From 00c63a19a19f848eea0b9741383889419c34fe37 Mon Sep 17 00:00:00 2001 From: Roman Date: Thu, 14 Dec 2023 18:11:11 +0200 Subject: [PATCH] add flag to customize database path --- main.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/main.go b/main.go index 0131208..9fd15ba 100644 --- a/main.go +++ b/main.go @@ -41,6 +41,7 @@ var ( flagSessionSecret string = util.RandomString(32) flagWgConfTemplate string flagBasePath string + flagDBPath string = "./db" ) const ( @@ -80,6 +81,7 @@ func init() { flag.StringVar(&flagSessionSecret, "session-secret", util.LookupEnvOrString("SESSION_SECRET", flagSessionSecret), "The key used to encrypt session cookies.") flag.StringVar(&flagWgConfTemplate, "wg-conf-template", util.LookupEnvOrString("WG_CONF_TEMPLATE", flagWgConfTemplate), "Path to custom wg.conf template.") flag.StringVar(&flagBasePath, "base-path", util.LookupEnvOrString("BASE_PATH", flagBasePath), "The base path of the URL") + flag.StringVar(&flagDBPath, "database-path", util.LookupEnvOrString("WG_DATABASE_PATH", flagDBPath), "The custom path to database.") flag.Parse() // update runtime config @@ -120,7 +122,7 @@ func init() { } func main() { - db, err := jsondb.New("./db") + db, err := jsondb.New(flagDBPath) if err != nil { panic(err) }