Added DB command line flags

Flags relating to databases

Signed-off-by: Computroniks <mnickson@sidingsmedia.com>
This commit is contained in:
Computroniks 2022-03-07 20:56:39 +00:00
parent 4be3a65691
commit 45b5a1f0d5
No known key found for this signature in database
GPG key ID: AB524D88499439A1
2 changed files with 24 additions and 0 deletions

18
main.go
View file

@ -35,6 +35,12 @@ var (
flagEmailFrom string flagEmailFrom string
flagEmailFromName string = "WireGuard UI" flagEmailFromName string = "WireGuard UI"
flagSessionSecret string flagSessionSecret string
flagDBType string = "jsondb"
flagDBHost string = "localhost"
flagDBPort int = 3306
flagDBDatabase string = "wireguard-ui"
flagDBUsername string
flagDBPassword string
) )
const ( const (
@ -61,6 +67,12 @@ func init() {
flag.StringVar(&flagEmailFrom, "email-from", util.LookupEnvOrString("EMAIL_FROM_ADDRESS", flagEmailFrom), "'From' email address.") 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(&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.") flag.StringVar(&flagSessionSecret, "session-secret", util.LookupEnvOrString("SESSION_SECRET", flagSessionSecret), "The key used to encrypt session cookies.")
flag.StringVar(&flagDBType, "db-type", util.LookupEnvOrString("DB_TYPE", flagDBType), "Type of database to use. One of: `jsondb`|`mysql`.")
flag.StringVar(&flagDBHost, "db-host", util.LookupEnvOrString("DB_HOST", flagDBHost), "Database host")
flag.IntVar(&flagDBPort, "db-port", util.LookupEnvOrInt("DB_PORT", flagDBPort), "Database port")
flag.StringVar(&flagDBDatabase, "db-database", util.LookupEnvOrString("DB_DATABASE", flagDBDatabase), "Database name")
flag.StringVar(&flagDBUsername, "db-username", util.LookupEnvOrString("DB_USERNAME", flagDBUsername), "Database username")
flag.StringVar(&flagDBPassword, "db-password", util.LookupEnvOrString("DB_PASSWORD", flagDBPassword), "Database password")
flag.Parse() flag.Parse()
// update runtime config // update runtime config
@ -76,6 +88,12 @@ func init() {
util.EmailFrom = flagEmailFrom util.EmailFrom = flagEmailFrom
util.EmailFromName = flagEmailFromName util.EmailFromName = flagEmailFromName
util.SessionSecret = []byte(flagSessionSecret) util.SessionSecret = []byte(flagSessionSecret)
util.DBType = flagDBType
util.DBHost = flagDBHost
util.DBPort = flagDBPort
util.DBDatabase = flagDBDatabase
util.DBUsername = flagDBUsername
util.DBPassword = flagDBPassword
// print app information // print app information
fmt.Println("Wireguard UI") fmt.Println("Wireguard UI")

View file

@ -16,6 +16,12 @@ var (
EmailSubject string EmailSubject string
EmailContent string EmailContent string
SessionSecret []byte SessionSecret []byte
DBType string
DBHost string
DBPort int
DBDatabase string
DBUsername string
DBPassword string
) )
const ( const (