refactor: use debug BuildInfo for version if available

pressly · Jul 26, 2023 · 9c73699 · 9c73699
1 parent 0774ee6
commit 9c73699
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cmd/goose/main.go b/cmd/goose/main.go
@@ -1,13 +1,15 @@
 package main
 
 import (
+	_ "embed"
 	"errors"
 	"flag"
 	"fmt"
 	"io/fs"
 	"log"
 	"os"
 	"path/filepath"
+	"runtime/debug"
 	"sort"
 	"strconv"
 	"strings"
@@ -48,7 +50,11 @@ func main() {
 	}
 
 	if *versionFlag {
-		fmt.Printf("goose version: %s\n", version)
+		buildInfo, ok := debug.ReadBuildInfo()
+		if ok && buildInfo != nil && buildInfo.Main.Version != "" {
+			version = buildInfo.Main.Version
+		}
+		fmt.Printf("goose version: %s\n", strings.TrimSpace(version))
 		return
 	}
 	if *verbose {