basic versioning

thushan · thushan · commit af9868fce7ce · 2025-10-27T11:48:00.000+11:00
diff --git a/.goreleaser.yml b/.goreleaser.yml
@@ -9,8 +9,12 @@ builds:
       - -pgo=auto
     env:
       - CGO_ENABLED=0
-    ldflags: |
-      -s -w -extldflags '-static'
+    ldflags:
+      - -s -w -extldflags '-static'
+      - -X main.Version={{.Version}}
+      - -X main.Commit={{.Commit}}
+      - -X main.Date={{.Date}}
+      - -X main.Timestamp={{.Timestamp}}
     goos:
       - linux
       - darwin
diff --git a/cmd/betteralign/main.go b/cmd/betteralign/main.go
@@ -7,6 +7,9 @@
 package main
 
 import (
+	"fmt"
+	"os"
+
 	"github.com/KimMachineGun/automemlimit/memlimit"
 	"github.com/dkorunic/betteralign"
 	"go.uber.org/automaxprocs/maxprocs"
@@ -29,5 +32,15 @@ func main() {
 	undo, _ := maxprocs.Set()
 	defer undo()
 
+	args := os.Args[1:]
+
+	// check for version flags
+	for _, arg := range args {
+		if arg == "-V" || arg == "--version" {
+			fmt.Println(getVersionString())
+			os.Exit(0)
+		}
+	}
+
 	singlechecker.Main(betteralign.Analyzer)
 }
diff --git a/cmd/betteralign/version.go b/cmd/betteralign/version.go
@@ -0,0 +1,43 @@
+package main
+
+import "fmt"
+
+const (
+	defaultUnknownInfo = "unknown"
+)
+
+var (
+	Version   = "dev"
+	Commit    = defaultUnknownInfo
+	Date      = defaultUnknownInfo
+	Timestamp = defaultUnknownInfo
+)
+
+// getVersionString returns version info like:
+// "betteralign version X.Y.X (commit short-SHA) built on YYYY-MM-DD"
+func getVersionString() string {
+	version := fmt.Sprintf("betteralign version %s", Version)
+
+	if Commit != defaultUnknownInfo {
+		shortCommit := Commit
+		if len(Commit) > 7 {
+			shortCommit = Commit[:7]
+		}
+		version += fmt.Sprintf(" (commit %s)", shortCommit)
+	}
+
+	builtDate := Date
+	if builtDate == defaultUnknownInfo && Timestamp != defaultUnknownInfo {
+		builtDate = Timestamp
+	}
+
+	if builtDate != defaultUnknownInfo {
+		if len(builtDate) >= 10 {
+			version += fmt.Sprintf(" built on %s", builtDate[:10])
+		} else {
+			version += fmt.Sprintf(" built on %s", builtDate)
+		}
+	}
+
+	return version
+}
