feat: added whomai cmd to coder cli

BRAVO68WEB · BRAVO68WEB · commit 6f5e10110f05 · 2024-07-08T13:56:46.000Z
diff --git a/cli/root.go b/cli/root.go
@@ -117,6 +117,7 @@ func (r *RootCmd) CoreSubcommands() []*serpent.Command {
 		r.stop(),
 		r.unfavorite(),
 		r.update(),
+		r.whoami(),
 
 		// Hidden
 		r.gitssh(),
diff --git a/cli/whoami.go b/cli/whoami.go
@@ -0,0 +1,38 @@
+package cli
+
+import (
+	"fmt"
+
+	"github.com/coder/coder/v2/cli/cliui"
+	"github.com/coder/coder/v2/codersdk"
+	"github.com/coder/pretty"
+	"github.com/coder/serpent"
+)
+
+func (r *RootCmd) whoami() *serpent.Command {
+	client := new(codersdk.Client)
+	cmd := &serpent.Command{
+		Annotations: workspaceCommand,
+		Use:         "whoami",
+		Short:       "Fetch authenticated user info for Coder deployment",
+		Middleware: serpent.Chain(
+			serpent.RequireNArgs(0),
+			r.InitClient(client),
+		),
+		Handler: func(inv *serpent.Invocation) error {
+			ctx := inv.Context()
+			// Fetch the user info
+			resp, err := client.User(ctx, codersdk.Me)
+			// Get Coder instance url
+			deployConfig, _ := client.DeploymentConfig(ctx)
+
+			if err != nil {
+				return err
+			}
+
+			_, _ = fmt.Fprintf(inv.Stdout, Caret+"Coder is running at %s, You're authenticated as %s !\n", pretty.Sprint(cliui.DefaultStyles.Keyword, deployConfig.Values.AccessURL.Scheme+"://"+deployConfig.Values.AccessURL.Host), pretty.Sprint(cliui.DefaultStyles.Keyword, resp.Username))
+			return err
+		},
+	}
+	return cmd
+}
