read api keys from config file

Author: GauravJangra9988

cmd/cli/createMsg.go

@@ -0,0 +1,140 @@
+package cmd
+
+import (
+	"log"
+	"os"
+
+	"github.com/atotto/clipboard"
+	"github.com/dfanso/commit-msg/cmd/cli/store"
+	"github.com/dfanso/commit-msg/internal/chatgpt"
+	"github.com/dfanso/commit-msg/internal/claude"
+	"github.com/dfanso/commit-msg/internal/display"
+	"github.com/dfanso/commit-msg/internal/gemini"
+	"github.com/dfanso/commit-msg/internal/git"
+	"github.com/dfanso/commit-msg/internal/grok"
+	"github.com/dfanso/commit-msg/internal/stats"
+	"github.com/dfanso/commit-msg/pkg/types"
+	"github.com/pterm/pterm"
+)
+
+
+func CreateCommitMsg () {
+	
+    // Validate COMMIT_LLM and required API keys
+	useLLM,err := store.DefaultLLMKey()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	commitLLM := useLLM.LLM
+	apiKey := useLLM.APIKey
+
+
+		// Get current directory
+		currentDir, err := os.Getwd()
+		if err != nil {
+			log.Fatalf("Failed to get current directory: %v", err)
+		}
+
+		// Check if current directory is a git repository
+		if !git.IsRepository(currentDir) {
+			log.Fatalf("Current directory is not a Git repository: %s", currentDir)
+		}
+
+		// Create a minimal config for the API
+		config := &types.Config{
+			GrokAPI: "https://api.x.ai/v1/chat/completions",
+		}
+
+		// Create a repo config for the current directory
+		repoConfig := types.RepoConfig{
+			Path: currentDir,
+		}
+
+		// Get file statistics before fetching changes
+		fileStats, err := stats.GetFileStatistics(&repoConfig)
+		if err != nil {
+			log.Fatalf("Failed to get file statistics: %v", err)
+		}
+
+		// Display header
+		pterm.DefaultHeader.WithFullWidth().
+			WithBackgroundStyle(pterm.NewStyle(pterm.BgDarkGray)).
+			WithTextStyle(pterm.NewStyle(pterm.FgLightWhite)).
+			Println("🚀 Commit Message Generator")
+
+		pterm.Println()
+
+		// Display file statistics with icons
+		display.ShowFileStatistics(fileStats)
+
+		if fileStats.TotalFiles == 0 {
+			pterm.Warning.Println("No changes detected in the Git repository.")
+			return
+		}
+
+		// Get the changes
+		changes, err := git.GetChanges(&repoConfig)
+		if err != nil {
+			log.Fatalf("Failed to get Git changes: %v", err)
+		}
+
+		if len(changes) == 0 {
+			pterm.Warning.Println("No changes detected in the Git repository.")
+			return
+		}
+
+		pterm.Println()
+
+		// Show generating spinner
+		spinnerGenerating, err := pterm.DefaultSpinner.
+			WithSequence("⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏").
+			Start("🤖 Generating commit message...")
+		if err != nil {
+			log.Fatalf("Failed to start spinner: %v", err)
+		}
+
+		var commitMsg string
+
+		switch commitLLM {
+			
+		case "Gemini":
+			commitMsg, err = gemini.GenerateCommitMessage(config, changes, apiKey)
+
+		case "OpenAI":
+			commitMsg, err = chatgpt.GenerateCommitMessage(config, changes, apiKey)
+		
+		case "Claude":
+			commitMsg, err = claude.GenerateCommitMessage(config, changes, apiKey)
+
+		default:
+			commitMsg, err = grok.GenerateCommitMessage(config, changes, apiKey)
+		}
+
+		
+		if err != nil {
+			spinnerGenerating.Fail("Failed to generate commit message")
+			log.Fatalf("Error: %v", err)
+		}
+
+		spinnerGenerating.Success("✅ Commit message generated successfully!")
+
+		pterm.Println()
+
+		// Display the commit message in a styled panel
+		display.ShowCommitMessage(commitMsg)
+
+		// Copy to clipboard
+		err = clipboard.WriteAll(commitMsg)
+		if err != nil {
+			pterm.Warning.Printf("⚠️  Could not copy to clipboard: %v\n", err)
+		} else {
+			pterm.Success.Println("📋 Commit message copied to clipboard!")
+		}
+
+		pterm.Println()
+
+		// Display changes preview
+		display.ShowChangesPreview(fileStats)
+
+}

cmd/cli/llmSetup.go

@@ -0,0 +1,53 @@
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/dfanso/commit-msg/cmd/cli/store"
+	"github.com/manifoldco/promptui"
+)
+
+
+func SetupLLM() error {
+
+	providers := []string{"OpenAI", "Claude", "Gemini", "Grok"}
+	prompt := promptui.Select{
+		Label: "Select LLM",
+		Items: providers,
+	}
+
+	_, model, err := prompt.Run()
+	if err != nil {
+		return fmt.Errorf("prompt failed")
+	}
+
+	apiKeyPrompt := promptui.Prompt{
+		Label: "Enter API Key",
+		
+	}
+
+	apiKey, err := apiKeyPrompt.Run()
+	if err != nil {
+		return fmt.Errorf("invalid API Key")
+	}
+
+	LLMConfig := store.LLMProvider{
+		LLM:    model,
+		APIKey: apiKey,
+	}
+
+
+
+	err = store.Save(LLMConfig)
+	if err != nil {
+		return err
+	}
+
+	fmt.Println("LLM model added")
+	return nil
+}
+
+func UpdateLLM() error {
+	fmt.Println("Update LLM config")
+	return nil
+}

cmd/root.go cmd/cli/root.gocmd/root.go renamed to cmd/cli/root.go

@@ -1,6 +1,5 @@
 /*
 Copyright © 2025 NAME HERE <EMAIL ADDRESS>
-
 */
 package cmd
 
@@ -10,18 +9,11 @@ import (
 	"github.com/spf13/cobra"
 )
 
-
-
 // rootCmd represents the base command when called without any subcommands
 var rootCmd = &cobra.Command{
-	Use:   "commit-msg",
-	Short: "A brief description of your application",
-	Long: `A longer description that spans multiple lines and likely contains
-examples and usage of using your application. For example:
-
-Cobra is a CLI library for Go that empowers applications.
-This application is a tool to generate the needed files
-to quickly create a Cobra application.`,
+	Use:   "commit",
+	Short: "CLI tool to write commit message",
+	Long:  `Write a commit message with AI of your choice`,
 	// Uncomment the following line if your bare application
 	// has an action associated with it:
 	// Run: func(cmd *cobra.Command, args []string) { },
@@ -36,6 +28,36 @@ func Execute() {
 	}
 }
 
+var llmCmd = &cobra.Command{
+	Use:   "llm",
+	Short: "Manage LLM configuration",
+}
+
+var llmSetupCmd = &cobra.Command{
+	Use:   "setup",
+	Short: "Setup your LLM provider and API key",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		return SetupLLM()
+	},
+}
+
+var llmUpdateCmd = &cobra.Command{
+	Use:   "update",
+	Short: "Update or Delete LLM Model",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		return UpdateLLM()
+	},
+}
+
+var creatCommitMsg = &cobra.Command{
+	Use: ".",
+	Short: "Create Commit Message",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		CreateCommitMsg()
+		return nil
+	},
+}
+
 func init() {
 	// Here you will define your flags and configuration settings.
 	// Cobra supports persistent flags, which, if defined here,
@@ -46,6 +68,9 @@ 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")
+	rootCmd.AddCommand(llmCmd)
+	llmCmd.AddCommand(llmSetupCmd)
+	llmCmd.AddCommand(llmUpdateCmd)
+	rootCmd.AddCommand(creatCommitMsg)
 }
 
-