2020-04-18 11:17:49 +02:00
package main
import (
2020-10-09 23:33:29 +02:00
"flag"
2020-04-23 13:01:40 +02:00
"fmt"
2020-10-09 23:33:29 +02:00
"net/http"
"time"
2020-05-21 10:51:24 +02:00
rice "github.com/GeertJohan/go.rice"
"github.com/labstack/echo/v4"
2021-08-08 19:55:59 +02:00
"github.com/ngoduykhanh/wireguard-ui/emailer"
2020-04-18 11:17:49 +02:00
"github.com/ngoduykhanh/wireguard-ui/handler"
"github.com/ngoduykhanh/wireguard-ui/router"
2021-08-28 15:10:30 +02:00
"github.com/ngoduykhanh/wireguard-ui/store/jsondb"
2020-04-23 13:01:40 +02:00
"github.com/ngoduykhanh/wireguard-ui/util"
2020-04-18 11:17:49 +02:00
)
2020-10-09 23:33:29 +02:00
var (
2021-08-29 20:26:12 +02:00
// command-line banner information
2020-10-09 23:33:29 +02:00
appVersion = "development"
gitCommit = "N/A"
gitRef = "N/A"
buildTime = fmt . Sprintf ( time . Now ( ) . UTC ( ) . Format ( "01-02-2006 15:04:05" ) )
2021-08-29 20:26:12 +02:00
// configuration variables
flagDisableLogin bool = false
flagBindAddress string = "0.0.0.0:5000"
flagSendgridApiKey string
flagEmailFrom string
flagEmailFromName string = "WireGuard UI"
flagSessionSecret string
2020-10-09 23:33:29 +02:00
)
2021-08-08 19:55:59 +02:00
const (
defaultEmailSubject = "Your wireguard configuration"
defaultEmailContent = ` Hi , < / br >
2021-11-09 11:01:42 +01:00
< p > In this email you can find your personal configuration for our wireguard server . < / p >
2021-08-08 19:55:59 +02:00
< p > Best < / p >
`
)
2020-10-09 23:33:29 +02:00
func init ( ) {
2021-08-29 20:26:12 +02:00
// command-line flags and env variables
2021-11-12 21:17:18 +01:00
flag . BoolVar ( & flagDisableLogin , "disable-login" , util . LookupEnvOrBool ( "DISABLE_LOGIN" , flagDisableLogin ) , "Disable authentication on the app. This is potentially dangerous." )
2021-08-29 20:26:12 +02:00
flag . StringVar ( & flagBindAddress , "bind-address" , util . LookupEnvOrString ( "BIND_ADDRESS" , flagBindAddress ) , "Address:Port to which the app will be bound." )
flag . StringVar ( & flagSendgridApiKey , "sendgrid-api-key" , util . LookupEnvOrString ( "SENDGRID_API_KEY" , flagSendgridApiKey ) , "Your sendgrid api key." )
flag . StringVar ( & flagEmailFrom , "email-from" , util . LookupEnvOrString ( "EMAIL_FROM_ADDRESS" , flagEmailFrom ) , "'From' email address." )
flag . StringVar ( & flagEmailFromName , "email-from-name" , util . LookupEnvOrString ( "EMAIL_FROM_NAME" , flagEmailFromName ) , "'From' email name." )
flag . StringVar ( & flagSessionSecret , "session-secret" , util . LookupEnvOrString ( "SESSION_SECRET" , flagSessionSecret ) , "The key used to encrypt session cookies." )
2020-10-09 23:33:29 +02:00
flag . Parse ( )
// update runtime config
2021-08-29 20:26:12 +02:00
util . DisableLogin = flagDisableLogin
util . BindAddress = flagBindAddress
util . SendgridApiKey = flagSendgridApiKey
util . EmailFrom = flagEmailFrom
util . EmailFromName = flagEmailFromName
util . SessionSecret = [ ] byte ( flagSessionSecret )
2020-06-01 08:03:10 +02:00
// print app information
fmt . Println ( "Wireguard UI" )
fmt . Println ( "App Version\t:" , appVersion )
fmt . Println ( "Git Commit\t:" , gitCommit )
fmt . Println ( "Git Ref\t\t:" , gitRef )
fmt . Println ( "Build Time\t:" , buildTime )
fmt . Println ( "Git Repo\t:" , "https://github.com/ngoduykhanh/wireguard-ui" )
2020-10-09 23:33:29 +02:00
fmt . Println ( "Authentication\t:" , ! util . DisableLogin )
2020-10-10 16:01:35 +02:00
fmt . Println ( "Bind address\t:" , util . BindAddress )
2021-08-29 20:26:12 +02:00
//fmt.Println("Sendgrid key\t:", util.SendgridApiKey)
fmt . Println ( "Email from\t:" , util . EmailFrom )
fmt . Println ( "Email from name\t:" , util . EmailFromName )
//fmt.Println("Session secret\t:", util.SessionSecret)
2020-06-01 10:24:11 +02:00
2020-10-09 23:33:29 +02:00
}
func main ( ) {
2021-08-28 15:10:30 +02:00
db , err := jsondb . New ( "./db" )
if err != nil {
panic ( err )
}
if err := db . Init ( ) ; err != nil {
panic ( err )
}
2020-10-09 23:33:29 +02:00
// set app extra data
extraData := make ( map [ string ] string )
extraData [ "appVersion" ] = appVersion
2020-04-23 13:01:40 +02:00
2020-05-21 10:51:24 +02:00
// create rice box for embedded template
tmplBox := rice . MustFindBox ( "templates" )
// rice file server for assets. "assets" is the folder where the files come from.
assetHandler := http . FileServer ( rice . MustFindBox ( "assets" ) . HTTPBox ( ) )
2020-04-23 13:01:40 +02:00
// register routes
2021-08-08 19:55:59 +02:00
app := router . New ( tmplBox , extraData , util . SessionSecret )
2020-04-18 11:17:49 +02:00
2021-08-28 15:10:30 +02:00
app . GET ( "/" , handler . WireGuardClients ( db ) , handler . ValidSession )
2020-10-09 23:33:29 +02:00
if ! util . DisableLogin {
app . GET ( "/login" , handler . LoginPage ( ) )
2021-08-28 15:10:30 +02:00
app . POST ( "/login" , handler . Login ( db ) )
2020-10-09 23:33:29 +02:00
}
2021-08-08 19:55:59 +02:00
sendmail := emailer . NewSendgridApiMail ( util . SendgridApiKey , util . EmailFromName , util . EmailFrom )
app . GET ( "/logout" , handler . Logout ( ) , handler . ValidSession )
2021-08-28 15:10:30 +02:00
app . POST ( "/new-client" , handler . NewClient ( db ) , handler . ValidSession )
app . POST ( "/update-client" , handler . UpdateClient ( db ) , handler . ValidSession )
app . POST ( "/email-client" , handler . EmailClient ( db , sendmail , defaultEmailSubject , defaultEmailContent ) , handler . ValidSession )
app . POST ( "/client/set-status" , handler . SetClientStatus ( db ) , handler . ValidSession )
app . POST ( "/remove-client" , handler . RemoveClient ( db ) , handler . ValidSession )
app . GET ( "/download" , handler . DownloadClient ( db ) , handler . ValidSession )
app . GET ( "/wg-server" , handler . WireGuardServer ( db ) , handler . ValidSession )
app . POST ( "wg-server/interfaces" , handler . WireGuardServerInterfaces ( db ) , handler . ValidSession )
app . POST ( "wg-server/keypair" , handler . WireGuardServerKeyPair ( db ) , handler . ValidSession )
app . GET ( "/global-settings" , handler . GlobalSettings ( db ) , handler . ValidSession )
app . POST ( "/global-settings" , handler . GlobalSettingSubmit ( db ) , handler . ValidSession )
app . GET ( "/status" , handler . Status ( db ) , handler . ValidSession )
app . GET ( "/api/clients" , handler . GetClients ( db ) , handler . ValidSession )
app . GET ( "/api/client/:id" , handler . GetClient ( db ) , handler . ValidSession )
2021-08-08 19:55:59 +02:00
app . GET ( "/api/machine-ips" , handler . MachineIPAddresses ( ) , handler . ValidSession )
2021-08-28 15:10:30 +02:00
app . GET ( "/api/suggest-client-ips" , handler . SuggestIPAllocation ( db ) , handler . ValidSession )
app . GET ( "/api/apply-wg-config" , handler . ApplyServerConfig ( db , tmplBox ) , handler . ValidSession )
2020-05-21 10:51:24 +02:00
// servers other static files
app . GET ( "/static/*" , echo . WrapHandler ( http . StripPrefix ( "/static/" , assetHandler ) ) )
2020-10-10 16:01:35 +02:00
app . Logger . Fatal ( app . Start ( util . BindAddress ) )
2020-04-18 11:17:49 +02:00
}