chore(cmd): Add version and commit information to build process.

- Modify LDFLAGS to include version and commit information
- Add VERSION and COMMIT variables in version.go
- Remove indirect dependency on github.com/fatih/color in go.mod

Author: appleboy

Makefile

@@ -2,14 +2,21 @@ GO ?= go
 EXECUTABLE := codegpt
 GOFILES := $(shell find . -type f -name "*.go")
 TAGS ?=
-LDFLAGS ?= -X 'main.Version=$(VERSION)'
+LDFLAGS ?= -X 'github.com/appleboy/CodeGPT/cmd.Version=$(VERSION)' -X 'github.com/appleboy/CodeGPT/cmd.Commit=$(COMMIT)'
 
 ifneq ($(shell uname), Darwin)
 	EXTLDFLAGS = -extldflags "-static" $(null)
 else
 	EXTLDFLAGS =
 endif
 
+ifneq ($(DRONE_TAG),)
+	VERSION ?= $(DRONE_TAG)
+else
+	VERSION ?= $(shell git describe --tags --always || git rev-parse --short HEAD)
+endif
+COMMIT ?= $(shell git rev-parse --short HEAD)
+
 build: $(EXECUTABLE)
 
 $(EXECUTABLE): $(GOFILES)

cmd/commit.go

@@ -1,7 +1,7 @@
 package cmd
 
 import (
-	"log"
+	"errors"
 	"os"
 	"strings"
 
@@ -27,17 +27,17 @@ func init() {
 var commitCmd = &cobra.Command{
 	Use:   "commit",
 	Short: "Auto generate commit message",
-	Run: func(cmd *cobra.Command, args []string) {
+	RunE: func(cmd *cobra.Command, args []string) error {
 		var message string
 
 		// check git command exist
 		if !util.IsCommandAvailable("git") {
-			log.Fatal("To use CodeGPT, you must have git on your PATH")
+			return errors.New("To use CodeGPT, you must have git on your PATH")
 		}
 
 		diff, err := git.Diff()
 		if err != nil {
-			log.Fatal(err)
+			return err
 		}
 
 		color.Green("Summarize the commit message use " + viper.GetString("openai.model") + " model")
@@ -48,7 +48,7 @@ var commitCmd = &cobra.Command{
 			viper.GetString("openai.org_id"),
 		)
 		if err != nil {
-			log.Fatal(err)
+			return err
 		}
 
 		// Get summarize comment from diff datas
@@ -59,13 +59,13 @@ var commitCmd = &cobra.Command{
 			},
 		)
 		if err != nil {
-			log.Fatal(err)
+			return err
 		}
 
 		color.Cyan("We are trying to summarize a git diff")
 		summarizeDiff, err := client.Completion(cmd.Context(), out)
 		if err != nil {
-			log.Fatal(err)
+			return err
 		}
 
 		out, err = util.GetTemplate(
@@ -75,13 +75,13 @@ var commitCmd = &cobra.Command{
 			},
 		)
 		if err != nil {
-			log.Fatal(err)
+			return err
 		}
 
 		color.Cyan("We are trying to summarize a title for pull request")
 		summarizeTitle, err := client.Completion(cmd.Context(), out)
 		if err != nil {
-			log.Fatal(err)
+			return err
 		}
 
 		if prompt.GetLanguage(viper.GetString("output.lang")) != prompt.DefaultLanguage {
@@ -94,23 +94,26 @@ var commitCmd = &cobra.Command{
 				},
 			)
 			if err != nil {
-				log.Fatal(err)
+				return err
 			}
 
 			color.Cyan("We are trying to translate a git commit message to " + prompt.GetLanguage(viper.GetString("output.lang")) + "language")
 			summarize, err := client.Completion(cmd.Context(), out)
 			if err != nil {
-				log.Fatal(err)
+				return err
 			}
 			message = summarize
 		} else {
 			message = strings.TrimSpace(summarizeTitle) + "\n\n" + strings.TrimSpace(summarizeDiff)
 		}
-
+		color.Yellow("================Commit Summary====================")
+		color.Yellow("\n" + message + "\n\n")
+		color.Yellow("==================================================")
 		color.Cyan("Write the commit message to " + viper.GetString("output.file") + " file")
 		err = os.WriteFile(viper.GetString("output.file"), []byte(message), 0o644)
 		if err != nil {
-			log.Fatal(err)
+			return err
 		}
+		return nil
 	},
 }

cmd/version.go

@@ -6,12 +6,15 @@ import (
 	"github.com/spf13/cobra"
 )
 
-var version string = "0.0.1"
+var (
+	Version string = ""
+	Commit  string = ""
+)
 
 var versionCmd = &cobra.Command{
 	Use:   "version",
 	Short: "Print the version number",
 	Run: func(cmd *cobra.Command, args []string) {
-		fmt.Println("version:", version)
+		fmt.Println("version:", Version, "commit:", Commit)
 	},
 }

go.mod

@@ -4,13 +4,13 @@ go 1.20
 
 require (
 	github.com/appleboy/com v0.1.6
+	github.com/fatih/color v1.14.1
 	github.com/sashabaranov/go-openai v1.4.1
 	github.com/spf13/cobra v1.6.1
 	github.com/spf13/viper v1.15.0
 )
 
 require (
-	github.com/fatih/color v1.14.1 // indirect
 	github.com/fsnotify/fsnotify v1.6.0 // indirect
 	github.com/hashicorp/hcl v1.0.0 // indirect
 	github.com/inconshreveable/mousetrap v1.0.1 // indirect