From 93f81feb1576d690040be79d1eff44d670be740d Mon Sep 17 00:00:00 2001
From: Gero Posmyk-Leinemann <gero@gitpod.io>
Date: Thu, 12 Sep 2024 10:36:57 +0200
Subject: [PATCH] [gp] Introduce "gp user-info" showing userId and email
 (#20206)

* [gp] Introduce "gp user-info" showing userId and email

* [gp] gp user-info: Add --email
---
 components/gitpod-cli/cmd/user-info.go | 130 +++++++++++++++++++++++++
 1 file changed, 130 insertions(+)
 create mode 100644 components/gitpod-cli/cmd/user-info.go

diff --git a/components/gitpod-cli/cmd/user-info.go b/components/gitpod-cli/cmd/user-info.go
new file mode 100644
index 00000000000000..6d6fc915c4ee21
--- /dev/null
+++ b/components/gitpod-cli/cmd/user-info.go
@@ -0,0 +1,130 @@
+// Copyright (c) 2024 Gitpod GmbH. All rights reserved.
+// Licensed under the GNU Affero General Public License (AGPL).
+// See License.AGPL.txt in the project root for license information.
+
+package cmd
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"os"
+	"time"
+
+	"github.com/olekukonko/tablewriter"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"golang.org/x/xerrors"
+
+	"github.com/gitpod-io/gitpod/gitpod-cli/pkg/supervisor"
+	serverapi "github.com/gitpod-io/gitpod/gitpod-protocol"
+	supervisorapi "github.com/gitpod-io/gitpod/supervisor/api"
+)
+
+var userInfoCmdOpts struct {
+	// Json configures whether the command output is printed as JSON, to make it machine-readable.
+	Json bool
+
+	// EmailOnly returns only the email address of the user
+	EmailOnly bool
+}
+
+// infoCmd represents the info command.
+var userInfoCmd = &cobra.Command{
+	Use:   "user-info",
+	Short: "Display user info about the workspace owner, such as its ID, email, etc.",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
+		defer cancel()
+
+		client, err := connectToAPI(ctx)
+		if err != nil {
+			return err
+		}
+
+		user, err := client.GetLoggedInUser(ctx)
+		if err != nil {
+			return err
+		}
+
+		// determine which email to show: prefer SSO, with random as fallback
+		email := user.GetSSOEmail()
+		if email == "" {
+			email = user.GetRandomEmail()
+		}
+
+		data := &userInfoData{
+			UserId: user.ID,
+			Email:  email,
+		}
+
+		if userInfoCmdOpts.EmailOnly {
+			fmt.Println(data.Email)
+			return nil
+		}
+
+		if userInfoCmdOpts.Json {
+			content, _ := json.Marshal(data)
+			fmt.Println(string(content))
+			return nil
+		}
+		outputUserInfo(data)
+		return nil
+	},
+}
+
+type userInfoData struct {
+	UserId string `json:"user_id"`
+	Email  string `json:"email"`
+}
+
+func outputUserInfo(info *userInfoData) {
+	table := tablewriter.NewWriter(os.Stdout)
+	table.SetColWidth(50)
+	table.SetBorder(false)
+	table.SetColumnSeparator(":")
+	table.Append([]string{"User ID", info.UserId})
+	table.Append([]string{"Email", info.Email})
+	table.Render()
+}
+
+func connectToAPI(ctx context.Context) (*serverapi.APIoverJSONRPC, error) {
+	supervisorClient, err := supervisor.New(ctx)
+	if err != nil {
+		return nil, xerrors.Errorf("failed connecting to supervisor: %w", err)
+	}
+	defer supervisorClient.Close()
+
+	wsinfo, err := supervisorClient.Info.WorkspaceInfo(ctx, &supervisorapi.WorkspaceInfoRequest{})
+	if err != nil {
+		return nil, xerrors.Errorf("failed getting workspace info from supervisor: %w", err)
+	}
+
+	clientToken, err := supervisorClient.Token.GetToken(ctx, &supervisorapi.GetTokenRequest{
+		Host: wsinfo.GitpodApi.Host,
+		Kind: "gitpod",
+		Scope: []string{
+			"function:getLoggedInUser",
+		},
+	})
+	if err != nil {
+		return nil, xerrors.Errorf("failed getting token from supervisor: %w", err)
+	}
+
+	serverLog := log.NewEntry(log.StandardLogger())
+	client, err := serverapi.ConnectToServer(wsinfo.GitpodApi.Endpoint, serverapi.ConnectToServerOpts{
+		Token:   clientToken.Token,
+		Context: ctx,
+		Log:     serverLog,
+	})
+	if err != nil {
+		return nil, xerrors.Errorf("failed connecting to server: %w", err)
+	}
+	return client, nil
+}
+
+func init() {
+	userInfoCmd.Flags().BoolVarP(&userInfoCmdOpts.Json, "json", "j", false, "Output in JSON format")
+	userInfoCmd.Flags().BoolVar(&userInfoCmdOpts.EmailOnly, "email", false, "Only emit the email address of the user")
+	rootCmd.AddCommand(userInfoCmd)
+}