Initial prototype of resources command

cmoog · cmoog · commit 166ed1489b22 · 2020-10-16T22:22:31.000-05:00
diff --git a/coder-sdk/env.go b/coder-sdk/env.go
@@ -21,7 +21,7 @@ type Environment struct {
 	UserID           string           `json:"user_id"            tab:"-"`
 	LastBuiltAt      time.Time        `json:"last_built_at"      tab:"-"`
 	CPUCores         float32          `json:"cpu_cores"          tab:"CPUCores"`
-	MemoryGB         int              `json:"memory_gb"          tab:"MemoryGB"`
+	MemoryGB         float32          `json:"memory_gb"          tab:"MemoryGB"`
 	DiskGB           int              `json:"disk_gb"            tab:"DiskGB"`
 	GPUs             int              `json:"gpus"               tab:"GPUs"`
 	Updating         bool             `json:"updating"           tab:"Updating"`
@@ -93,6 +93,16 @@ func (c Client) CreateEnvironment(ctx context.Context, orgID string, req CreateE
 	return &env, nil
 }
 
+// ListEnvironments lists environments returned by the given filter.
+// TODO: add the filter options
+func (c Client) ListEnvironments(ctx context.Context) ([]Environment, error) { 
+	var envs []Environment
+	if err := c.requestBody(ctx, http.MethodGet, "/api/environments", nil, &envs); err != nil {
+		return nil, err
+	}
+	return envs, nil
+}
+
 // EnvironmentsByOrganization gets the list of environments owned by the given user.
 func (c Client) EnvironmentsByOrganization(ctx context.Context, userID, orgID string) ([]Environment, error) {
 	var envs []Environment
diff --git a/coder-sdk/org.go b/coder-sdk/org.go
@@ -3,15 +3,30 @@ package coder
 import (
 	"context"
 	"net/http"
+	"time"
 )
 
 // Org describes an Organization in Coder
 type Org struct {
 	ID      string `json:"id"`
 	Name    string `json:"name"`
-	Members []User `json:"members"`
+	Members []OrganizationUser `json:"members"`
 }
 
+type OrganizationUser struct {
+	User
+	OrganizationRoles []Role    `json:"organization_roles"`
+	RolesUpdatedAt    time.Time `json:"roles_updated_at"`
+}
+
+type Role string
+
+const (
+	RoleOrgMember Role = "organization-member"
+	RoleOrgAdmin Role = "organization-admin"
+	RoleOrgManager Role = "organization-manager"
+)
+
 // Orgs gets all Organizations
 func (c Client) Orgs(ctx context.Context) ([]Org, error) {
 	var orgs []Org
@@ -20,3 +35,11 @@ func (c Client) Orgs(ctx context.Context) ([]Org, error) {
 	}
 	return orgs, nil
 }
+
+func (c Client) OrgMembers(ctx context.Context, orgID string) ([]OrganizationUser, error) {
+	var members []OrganizationUser
+	if err := c.requestBody(ctx, http.MethodGet, "/api/orgs/"+orgID+"/members", nil, &members); err != nil {
+		return nil, err
+	}
+	return members, nil
+}
diff --git a/internal/cmd/cmd.go b/internal/cmd/cmd.go
@@ -24,6 +24,7 @@ func Make() *cobra.Command {
 		makeEnvsCommand(),
 		makeSyncCmd(),
 		makeURLCmd(),
+		makeResourceCmd(),
 		completionCmd,
 		genDocs(app),
 	)
diff --git a/internal/cmd/resourcemanager.go b/internal/cmd/resourcemanager.go
@@ -0,0 +1,106 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+	"text/tabwriter"
+
+	"cdr.dev/coder-cli/coder-sdk"
+	"github.com/spf13/cobra"
+)
+
+func makeResourceCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use: "resources",
+		Short: "manager Coder resources with platform-level context (users, organizations, environments)",
+	}
+	cmd.AddCommand(resourceTop)
+	return cmd
+}
+
+var resourceTop = &cobra.Command{
+	Use: "top",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		ctx := cmd.Context()
+
+		client, err := newClient()
+		if err != nil {
+			return err
+		}
+
+		envs, err := client.ListEnvironments(ctx)
+		if err != nil {
+			return err
+		}
+
+		userEnvs := make(map[string][]coder.Environment)
+		for _, e := range envs {
+			userEnvs[e.UserID] = append(userEnvs[e.UserID], e)
+		}
+
+		users, err := client.Users(ctx)
+		if err != nil {
+			return err
+		}
+
+		orgs := make(map[string]coder.Org)
+		orglist, err := client.Orgs(ctx)
+		if err != nil {
+			return err
+		}
+		for _, o := range orglist {
+			orgs[o.ID] = o
+		}
+
+		tabwriter := tabwriter.NewWriter(os.Stdout, 0, 0, 4, ' ', 0)
+		for _, u := range users {
+			_, _ = fmt.Fprintf(tabwriter, "%s\t(%s)\t%s", u.Name, u.Email, aggregateEnvResources(userEnvs[u.ID]))		
+			if len(userEnvs[u.ID]) >0 {
+				_, _ = fmt.Fprintf(tabwriter, "\f")
+			}
+			for _, env := range userEnvs[u.ID] {
+				_, _ = fmt.Fprintf(tabwriter, "\t")
+				_, _ = fmt.Fprintln(tabwriter, fmtEnvResources(env, orgs))
+			}
+			fmt.Fprint(tabwriter, "\n")
+		}
+		_ = tabwriter.Flush()
+
+		return nil
+	},
+}
+
+func resourcesFromEnv(env coder.Environment) resources {
+			return resources{
+				cpuAllocation: env.CPUCores,
+				cpuUtilization: env.LatestStat.CPUUsage,
+				memAllocation: env.MemoryGB,
+				memUtilization: env.LatestStat.MemoryUsage,
+			}
+}
+
+func fmtEnvResources(env coder.Environment, orgs map[string]coder.Org) string {
+			return fmt.Sprintf("%s\t%s\t[org: %s]", env.Name, resourcesFromEnv(env), orgs[env.OrganizationID].Name)
+}
+
+func aggregateEnvResources(envs []coder.Environment) (resources) {
+	var aggregate resources
+	for _, e := range envs {
+		aggregate.cpuAllocation += e.CPUCores
+		aggregate.cpuUtilization += e.LatestStat.CPUUsage
+		aggregate.memAllocation += e.MemoryGB
+		aggregate.memUtilization += e.LatestStat.MemoryUsage
+	}
+	return aggregate
+}
+
+type resources struct {
+	cpuAllocation  float32
+	cpuUtilization float32
+	memAllocation  float32
+	memUtilization float32
+}
+
+func (a resources) String() string {
+	return fmt.Sprintf("[cpu: alloc=%.1fvCPU, util=%.1f]\t[mem: alloc=%.1fGB, util=%.1f]", a.cpuAllocation, a.cpuUtilization, a.memAllocation, a.memUtilization)
+}

Original file line number	Diff line number	Diff line change
`@@ -24,6 +24,7 @@ func Make() *cobra.Command {`
`24`	`24`	`makeEnvsCommand(),`
`25`	`25`	`makeSyncCmd(),`
`26`	`26`	`makeURLCmd(),`
	`27`	`+ makeResourceCmd(),`
`27`	`28`	`completionCmd,`
`28`	`29`	`genDocs(app),`
`29`	`30`	`)`