fix: report module version for go install

Author: shadowfish07

internal/cli/root.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"runtime/debug"
 	"strings"
 	"time"
 
@@ -43,7 +44,7 @@ func newRootCommand(stdout, stderr io.Writer) *cobra.Command {
 	root := &cobra.Command{
 		Use:           "heybox",
 		Short:         "搜索小黑盒社区内容",
-		Version:       version,
+		Version:       buildVersion(),
 		SilenceUsage:  true,
 		SilenceErrors: true,
 	}
@@ -54,6 +55,24 @@ func newRootCommand(stdout, stderr io.Writer) *cobra.Command {
 	return root
 }
 
+func buildVersion() string {
+	info, ok := debug.ReadBuildInfo()
+	if !ok {
+		return version
+	}
+	return resolveVersion(version, info)
+}
+
+func resolveVersion(injected string, info *debug.BuildInfo) string {
+	if injected != "dev" {
+		return injected
+	}
+	if info != nil && info.Main.Version != "" && info.Main.Version != "(devel)" {
+		return strings.TrimPrefix(info.Main.Version, "v")
+	}
+	return injected
+}
+
 func newCompletionCommand(root *cobra.Command, stdout io.Writer) *cobra.Command {
 	return &cobra.Command{
 		Use:                   "completion [bash|zsh|fish|powershell]",

internal/cli/root_test.go

@@ -2,10 +2,33 @@ package cli
 
 import (
 	"bytes"
+	"runtime/debug"
 	"strings"
 	"testing"
 )
 
+func TestResolveVersion(t *testing.T) {
+	t.Parallel()
+	tests := []struct {
+		name     string
+		injected string
+		info     *debug.BuildInfo
+		want     string
+	}{
+		{name: "release injection wins", injected: "1.2.3", info: &debug.BuildInfo{Main: debug.Module{Version: "v9.9.9"}}, want: "1.2.3"},
+		{name: "go install module version", injected: "dev", info: &debug.BuildInfo{Main: debug.Module{Version: "v1.2.3"}}, want: "1.2.3"},
+		{name: "local build", injected: "dev", info: &debug.BuildInfo{Main: debug.Module{Version: "(devel)"}}, want: "dev"},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			t.Parallel()
+			if got := resolveVersion(test.injected, test.info); got != test.want {
+				t.Fatalf("resolveVersion() = %q, want %q", got, test.want)
+			}
+		})
+	}
+}
+
 func TestHelp(t *testing.T) {
 	t.Parallel()
 	var stdout, stderr bytes.Buffer