gitpod/components/local-app/cmd/workspace-list.go

// Copyright (c) 2023 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"
	"fmt"
	"time"

	"github.com/bufbuild/connect-go"
	v1 "github.com/gitpod-io/gitpod/components/public-api/go/experimental/v1"
	"github.com/gitpod-io/local-app/pkg/config"
	"github.com/gitpod-io/local-app/pkg/helper"
	"github.com/gitpod-io/local-app/pkg/prettyprint"
	"github.com/sagikazarmark/slog-shim"
	"github.com/spf13/cobra"
)

// workspaceListCmd lists all available workspaces
var workspaceListCmd = &cobra.Command{
	Use:     "list",
	Short:   "Lists workspaces",
	Aliases: []string{"ls"},
	RunE: func(cmd *cobra.Command, args []string) error {
		cmd.SilenceUsage = true

		ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
		defer cancel()

		gitpod, err := getGitpodClient(ctx)
		if err != nil {
			return err
		}

		cfg := config.FromContext(ctx)
		gpctx, err := cfg.GetActiveContext()
		if err != nil {
			return err
		}
		orgId := gpctx.OrganizationID

		workspaces, err := gitpod.Workspaces.ListWorkspaces(ctx, connect.NewRequest(&v1.ListWorkspacesRequest{
			OrganizationId: orgId,
		}))
		if err != nil {
			return err
		}

		result := make([]tabularWorkspace, 0, len(workspaces.Msg.GetResult()))
		for _, ws := range workspaces.Msg.GetResult() {
			r := newTabularWorkspace(ws)
			if r == nil {
				continue
			}
			if workspaceListOpts.RunningOnly && ws.Status.Instance.Status.Phase != v1.WorkspaceInstanceStatus_PHASE_RUNNING {
				continue
			}
			result = append(result, *r)
		}

		return WriteTabular(result, workspaceListOpts.Format, prettyprint.WriterFormatWide)
	},
}

func newTabularWorkspace(ws *v1.Workspace) *tabularWorkspace {
	if !helper.HasInstanceStatus(ws) {
		slog.Debug("workspace has no instance status - removing from output", "workspace", ws.WorkspaceId)
		return nil
	}

	var repo string
	wsDetails := ws.Context.GetDetails()
	switch d := wsDetails.(type) {
	case *v1.WorkspaceContext_Git_:
		repo = fmt.Sprintf("%s/%s", d.Git.Repository.Owner, d.Git.Repository.Name)
	case *v1.WorkspaceContext_Prebuild_:
		repo = fmt.Sprintf("%s/%s", d.Prebuild.OriginalContext.Repository.Owner, d.Prebuild.OriginalContext.Repository.Name)
	}
	var branch string
	if ws.Status.Instance.Status.GitStatus != nil {
		branch = ws.Status.Instance.Status.GitStatus.Branch
		if branch == "" || branch == "(detached)" {
			branch = ""
		}
	}
	return &tabularWorkspace{
		ID:         ws.WorkspaceId,
		Repository: repo,
		Branch:     branch,
		Status:     prettyprint.FormatWorkspacePhase(ws.Status.Instance.Status.Phase),
	}
}

type tabularWorkspace struct {
	ID         string `print:"id"`
	Repository string `print:"repository"`
	Branch     string `print:"branch"`
	Status     string `print:"status"`
}

var workspaceListOpts struct {
	Format      formatOpts
	RunningOnly bool
}

func init() {
	workspaceCmd.AddCommand(workspaceListCmd)
	addFormatFlags(workspaceListCmd, &workspaceListOpts.Format)
	workspaceListCmd.Flags().BoolVarP(&workspaceListOpts.RunningOnly, "running-only", "r", false, "Only list running workspaces")
}