chore(cmd): add git hook management feature.

- Add a new `hookCmd` command to install and uninstall git hooks
- Remove `log` library import from `cmd/config.go`
- Replace `log.Fatal` calls with `errors.New` returns in `cmd/config.go`
- Add a `hookPath` function to get the git hook path
- Add a new `prepare-commit-msg` file to the `hook/templates` directory.

cmd/cmd.go

@@ -30,6 +30,7 @@ func init() {
 	rootCmd.AddCommand(versionCmd)
 	rootCmd.AddCommand(configCmd)
 	rootCmd.AddCommand(commitCmd)
+	rootCmd.AddCommand(hookCmd)
 
 	// hide completion command
 	rootCmd.CompletionOptions.HiddenDefaultCmd = true
@@ -83,7 +84,6 @@ func initConfig() {
 
 func Execute(ctx context.Context) {
 	if _, err := rootCmd.ExecuteContextC(ctx); err != nil {
-		fmt.Fprintln(os.Stderr, err)
 		os.Exit(1)
 	}
 }

cmd/config.go

@@ -1,8 +1,8 @@
 package cmd
 
 import (
+	"errors"
 	"fmt"
-	"log"
 	"strings"
 
 	"github.com/appleboy/com/array"
@@ -27,19 +27,20 @@ var configCmd = &cobra.Command{
 	Use:   "config",
 	Short: "Add openai config (openai.api_key, openai.model ...)",
 	Args:  cobra.MinimumNArgs(3),
-	Run: func(cmd *cobra.Command, args []string) {
+	RunE: func(cmd *cobra.Command, args []string) error {
 		if args[0] != "set" {
-			log.Fatal("config set key value. ex: config set openai.api_key sk-...")
+			return errors.New("config set key value. ex: config set openai.api_key sk-...")
 		}
 
 		if !array.InSlice(args[1], availableKeys) {
-			log.Fatal("available key list:", strings.Join(availableKeys, ", "))
+			return errors.New("available key list: " + strings.Join(availableKeys, ", "))
 		}
 
 		viper.Set(args[1], args[2])
 		if err := viper.WriteConfig(); err != nil {
-			log.Fatal(err)
+			return err
 		}
 		fmt.Println("you can see the config file:", viper.ConfigFileUsed())
+		return nil
 	},
 }

cmd/hook.go

@@ -0,0 +1,29 @@
+package cmd
+
+import (
+	"errors"
+
+	"github.com/appleboy/CodeGPT/hook"
+
+	"github.com/spf13/cobra"
+)
+
+var hookCmd = &cobra.Command{
+	Use:   "hook",
+	Short: "install/uninstall git prepare-commit-msg hook",
+	Args:  cobra.MinimumNArgs(1),
+	RunE: func(cmd *cobra.Command, args []string) error {
+		if args[0] != "install" && args[0] != "uninstall" {
+			return errors.New("only support install or uninstall command")
+		}
+
+		switch args[0] {
+		case "install":
+			return hook.Install()
+		case "uninstall":
+			return hook.Uninstall()
+		}
+
+		return nil
+	},
+}

git/git.go

@@ -55,6 +55,19 @@ func diffFiles() *exec.Cmd {
 	)
 }
 
+func hookPath() *exec.Cmd {
+	args := []string{
+		"rev-parse",
+		"--git-path",
+		"hooks",
+	}
+
+	return exec.Command(
+		"git",
+		args...,
+	)
+}
+
 // Diff compares the differences between two sets of data.
 // It returns a string representing the differences and an error.
 // If there are no differences, it returns an empty string and an error.
@@ -74,3 +87,9 @@ func Diff() (string, error) {
 
 	return string(output), nil
 }
+
+// Hook to show git hook path
+func Hook() (string, error) {
+	output, err := hookPath().Output()
+	return string(output), err
+}

hook/templates/prepare-commit-msg

@@ -0,0 +1,3 @@
+#!/bin/sh
+
+codegpt commit --file $1