VAULT 13901/inject version (#105)

Author: raymonstah

.github/workflows/build.yml

@@ -85,6 +85,7 @@ jobs:
         env:
           GOOS: ${{ matrix.goos }}
           GOARCH: ${{ matrix.goarch }}
+          VERSION: ${{ needs.get-product-version.outputs.product-version }}
         run: |
           make build
           pushd pkg

Makefile

@@ -2,12 +2,13 @@ GOOS?=linux
 GOARCH?=amd64
 CI_TEST_ARGS=
 TERRAFORM_ARGS=
-
+PKG=github.com/hashicorp/vault-lambda-extension/internal/config
+VERSION?=0.0.0-dev
 .PHONY: build zip lint test clean mod quick-start quick-start-destroy publish-layer-version
 
 build: clean
 	GOOS=$(GOOS) GOARCH=$(GOARCH) CGO_ENABLED=0 go build \
-		-ldflags '-s -w' \
+		-ldflags "-s -w -X '$(PKG).ExtensionVersion=$(VERSION)'" \
 		-a -o pkg/extensions/vault-lambda-extension \
 		.
 

internal/config/info.go

@@ -1,8 +1,12 @@
 package config
 
 const (
-	ExtensionName    = "vault-lambda-extension"
-	ExtensionVersion = "0.9.0"
-	VaultLogLevel    = "VAULT_LOG_LEVEL" // Optional, one of TRACE, DEBUG, INFO, WARN, ERROR, OFF
-	VaultRunMode     = "VAULT_RUN_MODE"
+	ExtensionName = "vault-lambda-extension"
+	VaultLogLevel = "VAULT_LOG_LEVEL" // Optional, one of TRACE, DEBUG, INFO, WARN, ERROR, OFF
+	VaultRunMode  = "VAULT_RUN_MODE"
+)
+
+var (
+	// ExtensionVersion should be a var type, so the go build tool can override and inject a custom version.
+	ExtensionVersion = "0.0.0-dev"
 )

main.go

@@ -33,6 +33,7 @@ func main() {
 		Level: hclog.LevelFromString(os.Getenv(config.VaultLogLevel)),
 	})
 
+	logger.Info(fmt.Sprintf("Starting Vault Lambda Extension %v", config.ExtensionVersion))
 	runMode := runmode.ModeDefault
 	if runModeEnv := os.Getenv(config.VaultRunMode); runModeEnv != "" {
 		runMode = runmode.ParseMode(runModeEnv)