Use built version

bufbuild · Mar 15, 2024 · f865a19 · f865a19
1 parent 046e540
commit f865a19
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 6 deletions.
diff --git a/cmd/protoc-gen-bigquery/main.go b/cmd/protoc-gen-bigquery/main.go
@@ -21,5 +21,5 @@ import (
 )
 
 func main() {
-	protoplugin.Main(protoplugin.HandlerFunc(pluginbigquery.Handle), protoplugin.WithVersion(protoschema.Version))
+	protoplugin.Main(protoplugin.HandlerFunc(pluginbigquery.Handle), protoplugin.WithVersion(protoschema.Version()))
 }
diff --git a/cmd/protoc-gen-jsonschema/main.go b/cmd/protoc-gen-jsonschema/main.go
@@ -21,5 +21,5 @@ import (
 )
 
 func main() {
-	protoplugin.Main(protoplugin.HandlerFunc(pluginjsonschema.Handle), protoplugin.WithVersion(protoschema.Version))
+	protoplugin.Main(protoplugin.HandlerFunc(pluginjsonschema.Handle), protoplugin.WithVersion(protoschema.Version()))
 }
diff --git a/cmd/protoc-gen-pubsub/main.go b/cmd/protoc-gen-pubsub/main.go
@@ -21,5 +21,5 @@ import (
 )
 
 func main() {
-	protoplugin.Main(protoplugin.HandlerFunc(pluginpubsub.Handle), protoplugin.WithVersion(protoschema.Version))
+	protoplugin.Main(protoplugin.HandlerFunc(pluginpubsub.Handle), protoplugin.WithVersion(protoschema.Version()))
 }
diff --git a/internal/protoschema/protoschema.go b/internal/protoschema/protoschema.go
@@ -14,7 +14,15 @@
 
 package protoschema
 
-const (
-	// Version is the current version of protoschema which is shared by all plugins.
-	Version = "0.2.0-dev"
+import (
+	"runtime/debug"
+	"strings"
 )
+
+func Version() string {
+	buildInfo, ok := debug.ReadBuildInfo()
+	if ok && buildInfo != nil && buildInfo.Main.Version != "" {
+		return strings.TrimSpace(buildInfo.Main.Version)
+	}
+	return "devel"
+}