🎨 Refactor config file creation and prompting logic

cmd/init.go

@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"bufio"
 	"fmt"
 	"os"
 	"path/filepath"
@@ -25,13 +26,42 @@ in the user's home directory.`,
 			fmt.Println("Error getting home directory:", err)
 			return
 		}
-		_ = createConfigFile(homeDir)
+		configFilePath := filepath.Join(homeDir, ".lazycommit.yaml")
+		if !checkForExistingConfig(configFilePath) {
+			return
+		}
+		apiKey := promptForAPIKey()
+		_ = createConfigFile(configFilePath, apiKey)
 	},
 }
 
-func createConfigFile(homeDir string) string {
-	configFilePath := filepath.Join(homeDir, ".lazycommit.yaml")
-	err := os.WriteFile(configFilePath, []byte(configContent), 0644)
+func promptForAPIKey() string {
+	reader := bufio.NewReader(os.Stdin)
+	fmt.Print("Enter your OpenAI API Key: ")
+	apiKey, _ := reader.ReadString('\n')
+	return apiKey
+}
+
+func promptForOverwrite() bool {
+	reader := bufio.NewReader(os.Stdin)
+	fmt.Print("Configuration file already exists. Overwrite? (y/n): ")
+	overwrite, _ := reader.ReadString('\n')
+	return overwrite == "y\n"
+}
+
+func checkForExistingConfig(configFilePath string) bool {
+	if _, err := os.Stat(configFilePath); err == nil {
+		if !promptForOverwrite() {
+			fmt.Println("Configuration file not overwritten.")
+			return false
+		}
+	}
+	return true
+}
+
+func createConfigFile(configFilePath, apiKey string) string {
+	configWithAPIKey := configContent + "\nopenai_api_key: " + apiKey
+	err := os.WriteFile(configFilePath, []byte(configWithAPIKey), 0644)
 	if err != nil {
 		fmt.Println("Error writing config file:", err)
 		return ""