Fix easy config

debug-ing · Dec 11, 2024 · 7688aee · 7688aee
1 parent 3008c46
commit 7688aee
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"log"
 
 	"github.com/debug-ing/radio-music/config"
@@ -11,7 +12,9 @@ import (
 )
 
 func main() {
-	config := config.LoadConfig()
+	configPath := flag.String("config", "config.toml", "config file")
+	flag.Parse()
+	config := config.LoadConfig(*configPath)
 	client := internal.NewClient()
 	go internal.StartStream(config.App.Folder, client)
 	r := gin.Default()

diff --git a/config/config.go b/config/config.go
@@ -17,9 +17,9 @@ type Config struct {
 var configPublic *Config
 var once sync.Once
 
-func LoadConfig() (config *Config) {
+func LoadConfig(address string) (config *Config) {
 	once.Do(func() {
-		viper.SetConfigFile("config/config.toml")
+		viper.SetConfigFile(address)
 		if err := viper.ReadInConfig(); err != nil {
 			panic(err)
 		}