feat: add provider name and repo name to envs ls

cmoog · cmoog · commit 817285849421 · 2021-03-11T17:42:45.000-06:00
diff --git a/coder-sdk/env.go b/coder-sdk/env.go
@@ -26,8 +26,8 @@ type Environment struct {
 	CPUCores         float32          `json:"cpu_cores"          table:"CPUCores"`
 	MemoryGB         float32          `json:"memory_gb"          table:"MemoryGB"`
 	DiskGB           int              `json:"disk_gb"            table:"DiskGB"`
-	GPUs             int              `json:"gpus"               table:"GPUs"`
-	Updating         bool             `json:"updating"           table:"Updating"`
+	GPUs             int              `json:"gpus"               table:"-"`
+	Updating         bool             `json:"updating"           table:"-"`
 	LatestStat       EnvironmentStat  `json:"latest_stat"        table:"Status"`
 	RebuildMessages  []RebuildMessage `json:"rebuild_messages"   table:"-"`
 	CreatedAt        time.Time        `json:"created_at"         table:"-"`
diff --git a/coder-sdk/image.go b/coder-sdk/image.go
@@ -66,6 +66,15 @@ func (c *DefaultClient) ImportImage(ctx context.Context, req ImportImageReq) (*I
 	return &img, nil
 }
 
+// ImageByID returns an image entity, fetched by its ID.
+func (c *DefaultClient) ImageByID(ctx context.Context, id string) (*Image, error) {
+	var img Image
+	if err := c.requestBody(ctx, http.MethodGet, "/api/v0/images/"+id, nil, &img); err != nil {
+		return nil, err
+	}
+	return &img, nil
+}
+
 // OrganizationImages returns all of the images imported for orgID.
 func (c *DefaultClient) OrganizationImages(ctx context.Context, orgID string) ([]Image, error) {
 	var (
diff --git a/coder-sdk/interface.go b/coder-sdk/interface.go
@@ -136,6 +136,9 @@ type Client interface {
 	// ImportImage creates a new image and optionally a new registry.
 	ImportImage(ctx context.Context, req ImportImageReq) (*Image, error)
 
+	// ImageByID returns an image entity, fetched by its ID.
+	ImageByID(ctx context.Context, id string) (*Image, error)
+
 	// OrganizationImages returns all of the images imported for orgID.
 	OrganizationImages(ctx context.Context, orgID string) ([]Image, error)
 
diff --git a/coder-sdk/workspace_providers.go b/coder-sdk/workspace_providers.go
@@ -12,13 +12,13 @@ type WorkspaceProviders struct {
 
 // KubernetesProvider defines an entity capable of deploying and acting as an ingress for Coder environments.
 type KubernetesProvider struct {
-	ID                 string                  `json:"id" table:"-"`
-	Name               string                  `json:"name" table:"Name"`
-	Status             WorkspaceProviderStatus `json:"status" table:"Status"`
-	BuiltIn            bool                    `json:"built_in" table:"-"`
-	EnvproxyAccessURL  string                  `json:"envproxy_access_url" validate:"required" table:"Access URL"`
-	DevurlHost         string                  `json:"devurl_host" table:"Devurl Host"`
-	OrgWhitelist       []string                `json:"org_whitelist" table:"-"`
+	ID                 string                  `json:"id"                  table:"-"`
+	Name               string                  `json:"name"                table:"Name"`
+	Status             WorkspaceProviderStatus `json:"status"              table:"Status"`
+	BuiltIn            bool                    `json:"built_in"            table:"-"`
+	EnvproxyAccessURL  string                  `json:"envproxy_access_url" table:"Access URL" validate:"required"`
+	DevurlHost         string                  `json:"devurl_host"         table:"Devurl Host"`
+	OrgWhitelist       []string                `json:"org_whitelist"       table:"-"`
 	KubeProviderConfig `json:"config" table:"_"`
 }
 
diff --git a/internal/cmd/envs.go b/internal/cmd/envs.go
@@ -82,7 +82,11 @@ func lsEnvsCommand() *cobra.Command {
 
 			switch outputFmt {
 			case humanOutput:
-				err := tablewriter.WriteTable(cmd.OutOrStdout(), len(envs), func(i int) interface{} {
+				envs, err := coderutil.EnvsHumanTable(ctx, client, envs)
+				if err != nil {
+					return err
+				}
+				err = tablewriter.WriteTable(cmd.OutOrStdout(), len(envs), func(i int) interface{} {
 					return envs[i]
 				})
 				if err != nil {
diff --git a/internal/coderutil/env.go b/internal/coderutil/env.go
@@ -2,6 +2,7 @@ package coderutil
 
 import (
 	"context"
+	"fmt"
 	"net/url"
 
 	"golang.org/x/xerrors"
@@ -72,3 +73,57 @@ func DefaultWorkspaceProvider(ctx context.Context, c coder.Client) (*coder.Kuber
 	}
 	return nil, coder.ErrNotFound
 }
+
+type EnvTable struct {
+	Name     string  `table:"Name"`
+	Image    string  `table:"Image"`
+	CPU      float32 `table:"vCPU"`
+	MemoryGB float32 `table:"MemoryGB"`
+	DiskGB   int     `table:"DiskGB"`
+	Status   string  `table:"Status"`
+	Provider string  `table:"Provider"`
+	CVM      bool    `table:"CVM"`
+}
+
+// EnvsHumanTable performs the composition of each Environment with its associated ProviderName and ImageRepo.
+func EnvsHumanTable(ctx context.Context, client coder.Client, envs []coder.Environment) ([]EnvTable, error) {
+	imageMap := make(map[string]*coder.Image)
+	for _, e := range envs {
+		imageMap[e.ImageID] = nil
+	}
+	// TODO: make this concurrent
+	for id := range imageMap {
+		img, err := client.ImageByID(ctx, id)
+		if err != nil {
+			return nil, err
+		}
+		imageMap[id] = img
+	}
+
+	pooledEnvs := make([]EnvTable, 0, len(envs))
+	providers, err := client.WorkspaceProviders(ctx)
+	if err != nil {
+		return nil, err
+	}
+	providerMap := make(map[string]coder.KubernetesProvider, len(providers.Kubernetes))
+	for _, p := range providers.Kubernetes {
+		providerMap[p.ID] = p
+	}
+	for _, e := range envs {
+		envProvider, ok := providerMap[e.ResourcePoolID]
+		if !ok {
+			return nil, xerrors.Errorf("fetch env workspace provider: %w", coder.ErrNotFound)
+		}
+		pooledEnvs = append(pooledEnvs, EnvTable{
+			Name:     e.Name,
+			Image:    fmt.Sprintf("%s:%s", imageMap[e.ImageID].Repository, e.ImageTag),
+			CPU:      e.CPUCores,
+			MemoryGB: e.MemoryGB,
+			DiskGB:   e.DiskGB,
+			Status:   string(e.LatestStat.ContainerStatus),
+			Provider: envProvider.Name,
+			CVM:      e.UseContainerVM,
+		})
+	}
+	return pooledEnvs, nil
+}
