Add show version feature

.goreleaser.yml

@@ -9,7 +9,7 @@ builds:
       - amd64
       - arm64
     ldflags:
-      - "-s -w"
+      - "-s -w -X main.version={{ .Version }} -X main.architecture={{ .Arch }}"
 archives:
   - format: tar.gz
     name_template: "{{ .ProjectName }}_{{ .Version }}_{{ .Os }}_{{ .Arch }}"

main.go

@@ -7,6 +7,22 @@ import (
 	"os"
 )
 
+var (
+	version      string
+	architecture string
+)
+
+func initFlags() {
+	flag.CommandLine = flag.NewFlagSet(os.Args[0], flag.ExitOnError)
+	flag.BoolVar(&versionFlag, "version", false, "Print version information")
+	flag.BoolVar(&vFlag, "v", false, "Print version information")
+}
+
+var (
+	versionFlag bool
+	vFlag       bool
+)
+
 func run() int {
 	if len(os.Args) < 2 {
 		printUsage()
@@ -66,5 +82,16 @@ func handlePush(configs []Config) int {
 }
 
 func main() {
+	initFlags()
+	flag.Parse()
+
+	if versionFlag || vFlag {
+		fmt.Printf("Version: %s\n", version)
+		fmt.Printf("Architecture: %s\n", architecture)
+		return
+	}
+
+	fmt.Printf("Version: %s\n", version)
+	fmt.Printf("Architecture: %s\n", architecture)
 	os.Exit(run())
 }

main_test.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"bytes"
+	"flag"
 	"io"
 	"os"
 	"testing"
@@ -12,6 +13,10 @@ var osExit = os.Exit
 
 // Remove remaining pull-related tests
 
+func resetFlags() {
+	flag.CommandLine = flag.NewFlagSet(os.Args[0], flag.ExitOnError)
+}
+
 func TestMain_NoCommand_ShowsHelp(t *testing.T) {
 	// Save the original os.Exit function and restore it after the test
 	originalExit := osExit
@@ -40,17 +45,48 @@ func TestMain_NoCommand_ShowsHelp(t *testing.T) {
 	}
 }
 
+func TestMain_VersionFlag(t *testing.T) {
+	resetFlags()
+	originalArgs := os.Args
+	defer func() { os.Args = originalArgs }()
+
+	os.Args = []string{"cmd", "--version"}
+	output := captureOutput(func() {
+		main()
+	})
+
+	if !bytes.Contains([]byte(output), []byte("Version: ")) || !bytes.Contains([]byte(output), []byte("Architecture: ")) {
+		t.Errorf("Expected version and architecture information to be printed, got: %s", output)
+	}
+}
+
+func TestMain_VFlag(t *testing.T) {
+	resetFlags()
+	originalArgs := os.Args
+	defer func() { os.Args = originalArgs }()
+
+	os.Args = []string{"cmd", "-v"}
+	output := captureOutput(func() {
+		main()
+	})
+
+	if !bytes.Contains([]byte(output), []byte("Version: ")) || !bytes.Contains([]byte(output), []byte("Architecture: ")) {
+		t.Errorf("Expected version and architecture information to be printed, got: %s", output)
+	}
+}
+
 // Helper function to capture output
 func captureOutput(f func()) string {
 	r, w, _ := os.Pipe()
+	originalStdout := os.Stdout
 	os.Stdout = w
 
 	f()
 
 	w.Close()
 	var buf bytes.Buffer
 	io.Copy(&buf, r)
-	os.Stdout = os.Stdout
+	os.Stdout = originalStdout
 
 	return buf.String()
 }