Adding version command

.goreleaser.yaml

@@ -13,10 +13,9 @@ builds:
     goarch:
       - amd64
       - arm64
-    # commented these out so i work on implementing them with airlock
-    # ldflags: &build-ldflags |
-    #   -X github.com/massdriver-cloud/mass/pkg/version.version={{.Version}}
-    #   -X github.com/massdriver-cloud/mass/pkg/version.gitSHA={{.FullCommit}}
+    ldflags: &build-ldflags |
+      -X github.com/massdriver-cloud/airlock/pkg/version.version={{.Version}}
+      -X github.com/massdriver-cloud/airlock/pkg/version.gitSHA={{.FullCommit}}
 
   - id: darwin-build
     binary: airlock

Makefile

@@ -1,4 +1,6 @@
 INSTALL_PATH ?= /usr/local/bin
+GIT_SHA := $(shell git log -1 --pretty=format:"%H")
+LD_FLAGS := "-X github.com/massdriver-cloud/airlock/pkg/version.version=dev -X github.com/massdriver-cloud/airlock/pkg/version.gitSHA=local-dev-${GIT_SHA}"
 
 MKFILE_PATH := $(abspath $(lastword $(MAKEFILE_LIST)))
 MKFILE_DIR := $(dir $(MKFILE_PATH))
@@ -19,7 +21,7 @@ generate: ${API_DIR}/zz_generated.go
 
 .PHONY: test
 test:
-	go test ./... -cover
+	go clean -testcache && go test ./... -cover
 
 bin:
 	mkdir bin
@@ -30,11 +32,11 @@ lint:
 
 .PHONY: build.macos
 build.macos: bin
-	GOOS=darwin GOARCH=arm64 go build -o bin/airlock-darwin-arm64
+	GOOS=darwin GOARCH=arm64 go build -o bin/airlock-darwin-arm64 -ldflags=${LD_FLAGS}
 
 .PHONY: build.linux
 build.linux: bin
-	GOOS=linux GOARCH=amd64 go build -o bin/airlock-linux-amd64
+	GOOS=linux GOARCH=amd64 go build -o bin/airlock-linux-amd64 -ldflags=${LD_FLAGS}
 
 .PHONY: install.macos
 install.macos: build.macos

cmd/root.go

@@ -36,6 +36,7 @@ func Execute() {
 	rootCmd.AddCommand(NewCmdHelm())
 	rootCmd.AddCommand(NewCmdOpenTofu())
 	rootCmd.AddCommand(NewCmdValidate())
+	rootCmd.AddCommand(NewCmdVersion())
 	if err := rootCmd.Execute(); err != nil {
 		os.Exit(1)
 	}

cmd/version.go

@@ -0,0 +1,33 @@
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/massdriver-cloud/airlock/pkg/prettylogs"
+	"github.com/massdriver-cloud/airlock/pkg/version"
+	"github.com/spf13/cobra"
+)
+
+func NewCmdVersion() *cobra.Command {
+	versionCmd := &cobra.Command{
+		Use:     "version",
+		Aliases: []string{"v"},
+		Short:   "Version of Airlock",
+		Run:     runVersion,
+	}
+	return versionCmd
+}
+
+func runVersion(cmd *cobra.Command, args []string) {
+	latestVersion, err := version.GetLatestVersion()
+	if err != nil {
+		fmt.Errorf("could not check for newer version, skipping.\nurl: %s\nerror: %w\n", version.LatestReleaseURL, err)
+	}
+
+	isOld, _ := version.CheckForNewerVersionAvailable(latestVersion)
+	if isOld {
+		fmt.Printf("A newer version of Airlock is available, you can download it here: %v\n", version.LatestReleaseURL)
+	}
+	airlockVersion := prettylogs.Green(version.AirlockVersion())
+	fmt.Printf("Airlock version: %v (git SHA: %v)\n", airlockVersion, version.AirlockGitSHA())
+}

go.mod

@@ -7,13 +7,15 @@ toolchain go1.23.2
 require (
 	github.com/Checkmarx/kics/v2 v2.1.3
 	github.com/charmbracelet/glamour v0.8.0
+	github.com/charmbracelet/lipgloss v0.13.0
 	github.com/hashicorp/hcl/v2 v2.22.0
 	github.com/massdriver-cloud/terraform-config-inspect v0.0.1
 	github.com/spf13/cobra v1.8.1
 	github.com/stretchr/testify v1.9.0
 	github.com/wk8/go-ordered-map/v2 v2.1.8
 	github.com/xeipuuv/gojsonschema v1.2.0
 	github.com/zclconf/go-cty v1.15.0
+	golang.org/x/mod v0.21.0
 	gopkg.in/yaml.v3 v3.0.1
 )
 
@@ -26,7 +28,6 @@ require (
 	github.com/aymerick/douceur v0.2.0 // indirect
 	github.com/bahlo/generic-list-go v0.2.0 // indirect
 	github.com/buger/jsonparser v1.1.1 // indirect
-	github.com/charmbracelet/lipgloss v0.13.0 // indirect
 	github.com/charmbracelet/x/ansi v0.3.2 // indirect
 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
 	github.com/dlclark/regexp2 v1.11.4 // indirect
@@ -55,7 +56,6 @@ require (
 	github.com/yuin/goldmark-emoji v1.0.3 // indirect
 	golang.org/x/crypto v0.28.0 // indirect
 	golang.org/x/exp v0.0.0-20240909161429-701f63a606c0 // indirect
-	golang.org/x/mod v0.21.0 // indirect
 	golang.org/x/net v0.30.0 // indirect
 	golang.org/x/sync v0.8.0 // indirect
 	golang.org/x/sys v0.26.0 // indirect

pkg/prettylogs/main.go

@@ -0,0 +1,15 @@
+package prettylogs
+
+import "github.com/charmbracelet/lipgloss"
+
+func Underline(word string) lipgloss.Style {
+	return lipgloss.NewStyle().SetString(word).Underline(true).Foreground(lipgloss.Color("#7D56f4"))
+}
+
+func Green(word string) lipgloss.Style {
+	return lipgloss.NewStyle().SetString(word).Foreground(lipgloss.Color("#00FF00"))
+}
+
+func Orange(word string) lipgloss.Style {
+	return lipgloss.NewStyle().SetString(word).Foreground(lipgloss.Color("#FFA500"))
+}

pkg/version/version.go

@@ -0,0 +1,66 @@
+package version
+
+import (
+	"context"
+	"net/http"
+	"strings"
+
+	"golang.org/x/mod/semver"
+)
+
+const LatestReleaseURL = "https://github.com/massdriver-cloud/airlock/releases/latest"
+
+var (
+	version = "unknown"
+	gitSHA  = "unknown"
+)
+
+func AirlockVersion() string {
+	return version
+}
+
+func AirlockGitSHA() string {
+	return gitSHA
+}
+
+func SetVersion(setVersion string) {
+	version = setVersion
+}
+
+func GetLatestVersion() (string, error) {
+	ctx := context.Background()
+	req, reqErr := http.NewRequestWithContext(ctx, http.MethodGet, LatestReleaseURL, nil)
+	if reqErr != nil {
+		return "", reqErr
+	}
+
+	resp, respErr := http.DefaultClient.Do(req)
+	if respErr != nil {
+		return "", respErr
+	}
+	defer resp.Body.Close()
+
+	redirectURL := resp.Request.URL.String()
+	parts := strings.Split(redirectURL, "/")
+	latestVersion := parts[len(parts)-1]
+	return latestVersion, nil
+}
+
+func CheckForNewerVersionAvailable(latestVersion string) (bool, string) {
+	currentVersion := version
+
+	// semver requires "v" for version (e.g., v1.0.0 not 1.0.0). Adds "v" if missing
+	if !strings.HasPrefix(currentVersion, "v") {
+		currentVersion = "v" + currentVersion
+	}
+
+	if !strings.HasPrefix(latestVersion, "v") {
+		latestVersion = "v" + latestVersion
+	}
+
+	if semver.Compare(currentVersion, latestVersion) < 0 {
+		return true, latestVersion
+	}
+
+	return false, latestVersion
+}

pkg/version/version_test.go

@@ -0,0 +1,48 @@
+package version_test
+
+import (
+	"testing"
+
+	"github.com/massdriver-cloud/airlock/pkg/version"
+)
+
+func TestCheckForNewerVersionAvailable(t *testing.T) {
+	tests := []struct {
+		name      string
+		current   string
+		latest    string
+		wantIsOld bool
+	}{
+		{
+			name:      "unknown version should be considered old",
+			current:   "unknown",
+			latest:    "v1.2.0",
+			wantIsOld: true,
+		},
+		{
+			name:      "current version is the latest version",
+			current:   "1.2.0",
+			latest:    "v1.2.0",
+			wantIsOld: false,
+		},
+		{
+			name:      "current version is older than the latest version",
+			current:   "1.0.0",
+			latest:    "v1.2.0",
+			wantIsOld: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			version.SetVersion(tt.current)
+
+			got, latestVersion := version.CheckForNewerVersionAvailable(tt.latest)
+			if got != tt.wantIsOld {
+				t.Errorf("CheckForNewerVersionAvailable() got = %v, want %v", got, tt.wantIsOld)
+			}
+			if latestVersion != tt.latest {
+				t.Errorf("CheckForNewerVersionAvailable() latestVersion = %v, want %v", latestVersion, tt.latest)
+			}
+		})
+	}
+}