From 4aab7b5107d88fb8cb685d97f5b8dddb4d42ae1c Mon Sep 17 00:00:00 2001
From: Joseph Schorr <josephschorr@users.noreply.github.com>
Date: Wed, 27 Jul 2022 13:19:29 -0400
Subject: [PATCH] Have `zed version` print the service version as well

The version command now makes a ReadSchema call to the service, and prints out the version returned (if any)

Fixes #132
---
 cmd/zed/main.go    |  6 +++-
 cmd/zed/version.go | 84 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 89 insertions(+), 1 deletion(-)
 create mode 100644 cmd/zed/version.go

diff --git a/cmd/zed/main.go b/cmd/zed/main.go
index 17b40b32..e9aef516 100644
--- a/cmd/zed/main.go
+++ b/cmd/zed/main.go
@@ -89,9 +89,13 @@ func main() {
 	versionCmd := &cobra.Command{
 		Use:   "version",
 		Short: "display zed version information",
-		RunE:  cobrautil.VersionRunFunc("zed"),
+		RunE: cobrautil.CommandStack(
+			LogCmdFunc,
+			versionCmdFunc,
+		),
 	}
 	cobrautil.RegisterVersionFlags(versionCmd.Flags())
+	versionCmd.Flags().Bool("include-remote-version", true, "whether to display the version of Authzed or SpiceDB for the current context")
 	rootCmd.AddCommand(versionCmd)
 
 	// Register root-level aliases
diff --git a/cmd/zed/version.go b/cmd/zed/version.go
new file mode 100644
index 00000000..8c6f5395
--- /dev/null
+++ b/cmd/zed/version.go
@@ -0,0 +1,84 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/authzed/authzed-go/pkg/responsemeta"
+	v1 "github.com/authzed/authzed-go/proto/authzed/api/v1"
+	"github.com/authzed/authzed-go/v1"
+	"github.com/gookit/color"
+	"github.com/jzelinskie/cobrautil"
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/cobra"
+	"golang.org/x/term"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/metadata"
+
+	"github.com/authzed/zed/internal/storage"
+)
+
+func versionCmdFunc(cmd *cobra.Command, args []string) error {
+	if !term.IsTerminal(int(os.Stdout.Fd())) {
+		color.Disable()
+	}
+
+	includeRemoteVersion := cobrautil.MustGetBool(cmd, "include-remote-version")
+	hasContext := false
+	configStore, secretStore := defaultStorage()
+	if includeRemoteVersion {
+		_, err := storage.DefaultToken(
+			cobrautil.MustGetString(cmd, "endpoint"),
+			cobrautil.MustGetString(cmd, "token"),
+			configStore,
+			secretStore,
+		)
+		hasContext = err == nil
+	}
+
+	if hasContext && includeRemoteVersion {
+		green := color.FgGreen.Render
+		fmt.Print(green("client: "))
+	}
+
+	_, err := fmt.Println(cobrautil.UsageVersion("zed", cobrautil.MustGetBool(cmd, "include-deps")))
+	if err != nil {
+		return err
+	}
+
+	if hasContext && includeRemoteVersion {
+		token, err := storage.DefaultToken(
+			cobrautil.MustGetString(cmd, "endpoint"),
+			cobrautil.MustGetString(cmd, "token"),
+			configStore,
+			secretStore,
+		)
+		if err != nil {
+			return err
+		}
+		log.Trace().Interface("token", token).Send()
+
+		client, err := authzed.NewClient(token.Endpoint, dialOptsFromFlags(cmd, token)...)
+		if err != nil {
+			return err
+		}
+
+		// NOTE: we ignore the error here, as it may be due to a schema not existing, or
+		// the client being unable to connect, etc. We just treat all such cases as an unknown
+		// version.
+		var headerMD metadata.MD
+		_, _ = client.ReadSchema(context.Background(), &v1.ReadSchemaRequest{}, grpc.Header(&headerMD))
+		version := headerMD.Get(string(responsemeta.ServerVersion))
+
+		blue := color.FgLightBlue.Render
+		fmt.Print(blue("service: "))
+		if len(version) == 1 {
+			fmt.Println(version[0])
+		} else {
+			fmt.Println("(unknown)")
+		}
+	}
+
+	return nil
+}