Merge pull request #102 from GauravJangra9988/os-credentials

Secure API storage

Author: DFanso

cmd/cli/createMsg.go

@@ -23,9 +23,9 @@ import (
 // CreateCommitMsg launches the interactive flow for reviewing, regenerating,
 // editing, and accepting AI-generated commit messages in the current repo.
 // If dryRun is true, it displays the prompt without making an API call.
-func CreateCommitMsg(dryRun bool, autoCommit bool) {
+func CreateCommitMsg(Store *store.StoreMethods, dryRun bool, autoCommit bool) {
 	// Validate COMMIT_LLM and required API keys
-	useLLM, err := store.DefaultLLMKey()
+	useLLM, err := Store.DefaultLLMKey()
 	if err != nil {
 		pterm.Error.Printf("No LLM configured. Run: commit llm setup\n")
 		os.Exit(1)

cmd/cli/llmSetup.go

@@ -9,9 +9,11 @@ import (
 	"github.com/manifoldco/promptui"
 )
 
+
+
 // SetupLLM walks the user through selecting an LLM provider and storing the
 // corresponding API key or endpoint configuration.
-func SetupLLM() error {
+func SetupLLM(Store *store.StoreMethods) error {
 
 	providers := types.GetSupportedProviderStrings()
 	prompt := promptui.Select{
@@ -60,7 +62,7 @@ func SetupLLM() error {
 		APIKey: apiKey,
 	}
 
-	err = store.Save(LLMConfig)
+	err = Store.Save(LLMConfig)
 	if err != nil {
 		return err
 	}
@@ -71,7 +73,7 @@ func SetupLLM() error {
 
 // UpdateLLM lets the user switch defaults, rotate API keys, or delete stored
 // LLM provider configurations.
-func UpdateLLM() error {
+func UpdateLLM(Store *store.StoreMethods) error {
 
 	SavedModels, err := store.ListSavedModels()
 	if err != nil {
@@ -88,7 +90,7 @@ func UpdateLLM() error {
 	options2 := []string{"Set Default", "Change URL", "Delete"} //different option for local model
 
 	for _, p := range SavedModels.LLMProviders {
-		models = append(models, p.LLM.String())
+		models = append(models, p.String())
 	}
 
 	prompt := promptui.Select{
@@ -146,7 +148,7 @@ func UpdateLLM() error {
 		if !valid {
 			return fmt.Errorf("invalid LLM provider: %s", model)
 		}
-		err = store.UpdateAPIKey(modelProvider, apiKey)
+		err = Store.UpdateAPIKey(modelProvider, apiKey)
 		if err != nil {
 			return err
 		}
@@ -160,7 +162,7 @@ func UpdateLLM() error {
 		if !valid {
 			return fmt.Errorf("invalid LLM provider: %s", model)
 		}
-		err := store.DeleteModel(modelProvider)
+		err := Store.DeleteModel(modelProvider)
 		if err != nil {
 			return err
 		}

cmd/cli/root.go

@@ -6,9 +6,18 @@ package cmd
 import (
 	"os"
 
+	"github.com/dfanso/commit-msg/cmd/cli/store"
 	"github.com/spf13/cobra"
 )
 
+//store instance
+var Store *store.StoreMethods
+
+//Initailize store
+func StoreInit(sm *store.StoreMethods){
+	Store = sm
+}
+
 // rootCmd represents the base command when called without any subcommands
 var rootCmd = &cobra.Command{
 	Use:   "commit",
@@ -47,15 +56,15 @@ var llmSetupCmd = &cobra.Command{
 	Use:   "setup",
 	Short: "Setup your LLM provider and API key",
 	RunE: func(cmd *cobra.Command, args []string) error {
-		return SetupLLM()
+		return SetupLLM(Store)
 	},
 }
 
 var llmUpdateCmd = &cobra.Command{
 	Use:   "update",
 	Short: "Update or Delete LLM Model",
 	RunE: func(cmd *cobra.Command, args []string) error {
-		return UpdateLLM()
+		return UpdateLLM(Store)
 	},
 }
 
@@ -72,7 +81,7 @@ var creatCommitMsg = &cobra.Command{
 		if err != nil {
 			return err
 		}
-		CreateCommitMsg(dryRun, autoCommit)
+		CreateCommitMsg(Store, dryRun, autoCommit)
 		return nil
 	},
 }
@@ -86,6 +95,7 @@ func init() {
 
 	// Cobra also supports local flags, which will only run
 	// when this action is called directly.
+	
 	rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
 
 	// Add --dry-run and --auto as persistent flags so they show in top-level help