jostreff/wireguard-ui
1
0
Fork 0
mirror of https://github.com/ngoduykhanh/wireguard-ui.git synced 2025-05-23 00:15:19 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

consume variables to set username and password otherwise use defaults (#90)

Browse source
This commit is contained in:
Robin Horton 2021-08-24 12:41:52 -07:00 • committed by GitHub
parent 1711530dda
commit acbe5b48e9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
util/db.go
View file

@ -15,6 +15,8 @@ import (
)
const dbPath = "./db"
const username_env_var = "WGUI_USERNAME"
const password_env_var = "WGUI_PASSWORD"
const defaultUsername = "admin"
const defaultPassword = "admin"
const defaultServerAddress = "10.252.1.0/24"
@ -33,6 +35,13 @@ func DBConn() (*scribble.Driver, error) {
return db, nil
}
func getCredVar(key, fallback string) string {
if value, ok := os.LookupEnv(key); ok {
return value
}
return fallback
}
// InitDB to create the default database
func InitDB() error {
var clientPath string = path.Join(dbPath, "clients")
@ -112,8 +121,8 @@ func InitDB() error {
}
user := new(model.User)
user.Username = defaultUsername
user.Password = defaultPassword
user.Username = getCredVar(username_env_var, defaultUsername)
user.Password = getCredVar(password_env_var, defaultPassword)
db.Write("server", "users", user)
}