Merge pull request #11 from major-technology/1

Fix config missing

Author: jasonzbao

.gitignore

@@ -1,2 +1,3 @@
 vendor/
 cli
+major

cmd/root.go

@@ -1,8 +1,10 @@
 package cmd
 
 import (
+	"encoding/json"
 	"os"
 
+	"github.com/major-technology/cli/configs"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 )
@@ -39,14 +41,33 @@ func init() {
 }
 
 func initConfig() {
-	if cfgFile != "" {
-		viper.SetConfigFile(cfgFile)
+	// Use embedded config based on defaultConfig variable
+	var configData []byte
+	if defaultConfig == "configs/prod.json" {
+		configData = configs.ProdConfig
 	} else {
-		home, err := os.UserHomeDir()
+		configData = configs.LocalConfig
+	}
+
+	// Parse embedded config into viper
+	var config map[string]interface{}
+	if err := json.Unmarshal(configData, &config); err != nil {
 		cobra.CheckErr(err)
-		viper.AddConfigPath(home)
-		viper.SetConfigType("yaml")
-		viper.SetConfigName(".cli")
 	}
-	cobra.CheckErr(viper.ReadInConfig())
+
+	for key, value := range config {
+		viper.Set(key, value)
+	}
+
+	// Allow user to override with custom config file if specified
+	if cfgFile != "" && cfgFile != defaultConfig {
+		if _, err := os.Stat(cfgFile); err == nil {
+			viper.SetConfigFile(cfgFile)
+			// Don't exit on error, just use embedded defaults
+			_ = viper.MergeInConfig()
+		}
+	}
+
+	viper.SetEnvPrefix("MAJOR")
+	viper.AutomaticEnv()
 }

configs/configs.go

@@ -0,0 +1,11 @@
+package configs
+
+import (
+	_ "embed"
+)
+
+//go:embed prod.json
+var ProdConfig []byte
+
+//go:embed local.json
+var LocalConfig []byte