feat: implement prompt folder management and default data handling

- Add a `promptFolder` variable and configure it with a command line flag
- Add logic to initialize and create the `promptFolder` if it does not exist
- Add a new `promptCmd` command to handle loading default prompt data
- Implement functionality to load and save default prompt data templates
- Add a `GetRawData` function to retrieve raw data of templates by name

Signed-off-by: Bo-Yi Wu <[email protected]>

Author: appleboy

cmd/cmd.go

@@ -21,8 +21,9 @@ var rootCmd = &cobra.Command{
 
 // Used for flags.
 var (
-	cfgFile  string
-	replacer = strings.NewReplacer("-", "_", ".", "_")
+	cfgFile      string
+	promptFolder string
+	replacer     = strings.NewReplacer("-", "_", ".", "_")
 )
 
 const (
@@ -34,12 +35,14 @@ func init() {
 	cobra.OnInitialize(initConfig)
 
 	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.config/codegpt/.codegpt.yaml)")
+	rootCmd.PersistentFlags().StringVar(&promptFolder, "prompt_folder", "", "prompt folder (default is $HOME/.config/codegpt/prompt)")
 	rootCmd.AddCommand(versionCmd)
 	rootCmd.AddCommand(configCmd)
 	rootCmd.AddCommand(commitCmd)
 	rootCmd.AddCommand(hookCmd)
 	rootCmd.AddCommand(reviewCmd)
 	rootCmd.AddCommand(CompletionCmd)
+	rootCmd.AddCommand(promptCmd)
 
 	// hide completion command
 	rootCmd.CompletionOptions.HiddenDefaultCmd = true
@@ -75,6 +78,30 @@ func initConfig() {
 		}
 	}
 
+	if promptFolder != "" {
+		viper.Set("prompt_folder", promptFolder)
+		if file.IsFile(promptFolder) {
+			log.Fatalf("prompt folder %s is a file", promptFolder)
+		}
+		// create the prompt folder if it doesn't exist
+		if !file.IsDir(promptFolder) {
+			if err := os.MkdirAll(promptFolder, os.ModePerm); err != nil {
+				log.Fatal(err)
+			}
+		}
+	} else {
+		// Find home directory.
+		home, err := os.UserHomeDir()
+		cobra.CheckErr(err)
+		targetFolder := path.Join(home, ".config", "codegpt", "prompt")
+		if !file.IsDir(targetFolder) {
+			if err := os.MkdirAll(targetFolder, os.ModePerm); err != nil {
+				log.Fatal(err)
+			}
+		}
+		viper.Set("prompt_folder", targetFolder)
+	}
+
 	viper.AutomaticEnv()
 	viper.SetEnvKeyReplacer(replacer)
 

cmd/prompt.go

@@ -0,0 +1,58 @@
+package cmd
+
+import (
+	"os"
+	"path"
+
+	"github.com/appleboy/CodeGPT/prompt"
+	"github.com/erikgeiser/promptkit/confirmation"
+	"github.com/fatih/color"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+var loadPromptData bool
+
+func init() {
+	promptCmd.PersistentFlags().BoolVar(&loadPromptData, "load", false,
+		"load default prompt data")
+}
+
+var defaultPromptDataKeys = []string{
+	prompt.CodeReviewTemplate,
+	prompt.SummarizeFileDiffTemplate,
+	prompt.SummarizeTitleTemplate,
+	prompt.ConventionalCommitTemplate,
+}
+
+var promptCmd = &cobra.Command{
+	Use:   "prompt",
+	Short: "load default prompt data",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		if loadPromptData {
+			change, err := confirmation.New("Do you want to load default prompt data, will overwrite your data", confirmation.No).RunPrompt()
+			if err != nil || !change {
+				return err
+			}
+
+			folder := viper.GetString("prompt_folder")
+			for _, key := range defaultPromptDataKeys {
+				// load default prompt data
+				out, err := prompt.GetRawData(key)
+				if err != nil {
+					return err
+				}
+
+				// save out to file
+				target := path.Join(folder, key)
+				if err := os.WriteFile(target, out, 0o600); err != nil {
+					return err
+				}
+				color.Cyan("save %s to %s", key, target)
+			}
+
+			return nil
+		}
+		return nil
+	},
+}

prompt/prompt.go

@@ -28,3 +28,9 @@ func init() { //nolint:gochecknoinits
 		log.Fatal(err)
 	}
 }
+
+// GetRawData returns the raw data of the template with the given name.
+func GetRawData(name string) ([]byte, error) {
+	key := "templates/" + name
+	return templatesFS.ReadFile(key)
+}