diff --git a/sevenkeys/main.go b/sevenkeys/main.go index 2d001cf..30091bc 100644 --- a/sevenkeys/main.go +++ b/sevenkeys/main.go @@ -1,13 +1,20 @@ package main import ( + "os" "sevenkeys/cli" "sevenkeys/database" "sevenkeys/figlet" ) func main() { - db := database.GetDatabaseFromConfig("config.json") + var profile string + if len(os.Args) < 2 { + profile = "production" + } else { + profile = os.Args[1] + } + db := database.GetDatabaseFromConfig("config." + profile + ".json") figlet.ReadFigletFonts() cli.ShowSplashScreen()