fix error when trying to run apikey sub-commands

cmd/apikey/apikey_current.go

@@ -15,6 +15,8 @@ var apikeyCurrentCmd = &cobra.Command{
 	Short:   "Set the current API key",
 	Args:    cobra.MinimumNArgs(1),
 	Run: func(cmd *cobra.Command, args []string) {
+		config.LoadConfig(config.GetConfigFilename())
+
 		index, err := apiKeyFind(args[0])
 		if err != nil {
 			utility.Error("Unable find the API key %s", err.Error())
@@ -26,6 +28,5 @@ var apikeyCurrentCmd = &cobra.Command{
 			config.SaveConfig()
 			fmt.Printf("Set the default API Key to be %s\n", utility.Green(index))
 		}
-
 	},
 }

cmd/apikey/apikey_list.go

@@ -20,6 +20,8 @@ If you wish to use a custom format, the available fields are:
 
 Example: civo apikey ls -o custom -f "Name: Key"`,
 	Run: func(cmd *cobra.Command, args []string) {
+		config.LoadConfig(config.GetConfigFilename())
+
 		keys := make([]string, 0, len(config.Current.APIKeys))
 		for k := range config.Current.APIKeys {
 			keys = append(keys, k)

cmd/apikey/apikey_remove.go

@@ -19,6 +19,8 @@ var apikeyRemoveCmd = &cobra.Command{
 	Args:    cobra.MinimumNArgs(1),
 	Example: "civo apikey remove NAME",
 	Run: func(cmd *cobra.Command, args []string) {
+		config.LoadConfig(config.GetConfigFilename())
+
 		index, err := apiKeyFind(args[0])
 		if err != nil {
 			utility.Error("Unable to find the API key %s", err.Error())
@@ -49,6 +51,5 @@ var apikeyRemoveCmd = &cobra.Command{
 		} else {
 			fmt.Println("Operation aborted.")
 		}
-
 	},
 }

cmd/apikey/apikey_save.go

@@ -110,6 +110,17 @@ var apikeySaveCmd = &cobra.Command{
 			apiKey = apiKeyEnv
 		}
 
+		if config.Current.APIKeys == nil {
+			filename := config.GetConfigFilename()
+			err := config.CheckConfigFile(filename)
+			if err != nil {
+				fmt.Println(err.Error())
+				os.Exit(1)
+			}
+
+			config.ProcessConfig(filename)
+		}
+
 		config.Current.APIKeys[name] = apiKey
 		if config.Current.Meta.DefaultRegion == "" {
 			client, err := civogo.NewClientWithURL(apiKey, config.Current.Meta.URL, "")

cmd/apikey/apikey_show.go

@@ -22,6 +22,8 @@ If you wish to use a custom format, the available fields are:
 * key
 `,
 	Run: func(cmd *cobra.Command, args []string) {
+		config.LoadConfig(config.GetConfigFilename())
+
 		keys := make([]string, 0, len(config.Current.APIKeys))
 		for k := range config.Current.APIKeys {
 			keys = append(keys, k)

cmd/root.go

@@ -44,6 +44,11 @@ as instances and Kubernetes clusters at Civo.com.`,
 			cmd.Help()
 		}
 	},
+	PersistentPreRun: func(cmd *cobra.Command, args []string) {
+		if cmd.Parent().Use != "apikey" {
+			config.ReadConfig()
+		}
+	},
 }
 
 // completionCmd represents the completion command
@@ -126,8 +131,6 @@ func Execute() {
 }
 
 func init() {
-	cobra.OnInitialize(config.ReadConfig)
-
 	rootCmd.AddCommand(completionCmd)
 	completionCmd.AddCommand(completionBashCmd)
 	completionCmd.AddCommand(completionZshCmd)

config/config.go

@@ -37,26 +37,44 @@ var Filename string
 
 // ReadConfig reads in config file and ENV variables if set.
 func ReadConfig() {
-	filename, found := os.LookupEnv("CIVO_CONFIG")
-	if found {
-		Filename = filename
+	filename := GetConfigFilename()
+	if filename != "" {
+		LoadConfig(filename)
+		ProcessConfig(filename)
+	} else {
+		fmt.Println("No configuration file found")
+		os.Exit(1)
 	}
+}
 
-	if Filename != "" {
-		loadConfig(Filename)
-	} else {
-		home, err := homedir.Dir()
-		if err != nil {
-			fmt.Println(err)
-			os.Exit(1)
-		}
-		loadConfig(fmt.Sprintf("%s/%s", home, ".civo.json"))
+// Function to retrieve the config filename from environment variable or default
+func GetConfigFilename() string {
+	if filename, found := os.LookupEnv("CIVO_CONFIG"); found {
+		return filename
+	}
+
+	homeDir := getHomeDir()
+	if homeDir != "" {
+		return fmt.Sprintf("%s/%s", homeDir, ".civo.json")
+	}
+
+	return ""
+}
+
+// Function to get the home directory of the current user
+func getHomeDir() string {
+	home, err := homedir.Dir()
+	if err != nil {
+		fmt.Println("Error retrieving home directory:", err)
+		os.Exit(1)
 	}
+
+	return home
 }
 
-func loadConfig(filename string) {
+func LoadConfig(filename string) {
 	var err error
-	err = checkConfigFile(filename)
+	err = CheckConfigFile(filename)
 	if err != nil {
 		fmt.Println(err.Error())
 		os.Exit(1)
@@ -87,6 +105,9 @@ func loadConfig(filename string) {
 			os.Exit(1)
 		}
 	}
+}
+
+func ProcessConfig(filename string) {
 	if Current.APIKeys == nil {
 		Current.APIKeys = map[string]string{}
 	}
@@ -97,12 +118,14 @@ func loadConfig(filename string) {
 	}
 
 	if Current.Meta.CurrentAPIKey != "" && Current.RegionToFeatures == nil {
-		Current.RegionToFeatures, err = regionsToFeature()
+		regionstoFeature, err := regionsToFeature()
 		if err != nil {
 			fmt.Printf("Error getting supported regions to feature %s \n", err)
 			os.Exit(1)
 		}
 
+		Current.RegionToFeatures = regionstoFeature
+
 		dataBytes, err := json.Marshal(Current)
 		if err != nil {
 			fmt.Printf("Error parsing JSON %s \n", err)
@@ -120,11 +143,13 @@ func loadConfig(filename string) {
 		Current.Meta.LatestReleaseCheck = time.Now()
 
 		if Current.Meta.CurrentAPIKey != "" {
-			Current.RegionToFeatures, err = regionsToFeature()
+			regionFeatures, err := regionsToFeature()
 			if err != nil {
 				fmt.Printf("Error getting supported regions to feature %s \n", err)
 				os.Exit(1)
 			}
+
+			Current.RegionToFeatures = regionFeatures
 		}
 
 		dataBytes, err := json.Marshal(Current)
@@ -140,7 +165,6 @@ func loadConfig(filename string) {
 		}
 		common.CheckVersionUpdate()
 	}
-
 }
 
 // SaveConfig saves the current configuration back out to a JSON file in
@@ -178,7 +202,7 @@ func SaveConfig() {
 
 }
 
-func checkConfigFile(filename string) error {
+func CheckConfigFile(filename string) error {
 	curr := Config{APIKeys: map[string]string{}}
 	curr.Meta = Metadata{
 		Admin:           false,
@@ -187,7 +211,8 @@ func checkConfigFile(filename string) error {
 		LastCmdExecuted: time.Now(),
 	}
 
-	if Current.Meta.CurrentAPIKey != "" {
+	currApiKey := Current.Meta.CurrentAPIKey
+	if currApiKey != "" {
 		var err error
 		curr.RegionToFeatures, err = regionsToFeature()
 		if err != nil {
@@ -229,6 +254,11 @@ func checkConfigFile(filename string) error {
 		os.Exit(1)
 	}
 
+	Current = curr
+	if currApiKey != "" {
+		Current.Meta.CurrentAPIKey = currApiKey
+	}
+
 	return nil
 }