Add ability to create new environments and images

f0ssel · f0ssel · commit 74c6417de1ce · 2020-08-21T15:02:39.000-05:00
diff --git a/coder-sdk/env.go b/coder-sdk/env.go
@@ -2,7 +2,9 @@ package coder
 
 import (
 	"context"
+	"fmt"
 	"net/http"
+	"nhooyr.io/websocket/wsjson"
 	"time"
 
 	"cdr.dev/coder-cli/internal/x/xjson"
@@ -75,3 +77,127 @@ func (c Client) DialWsep(ctx context.Context, env *Environment) (*websocket.Conn
 	}
 	return conn, nil
 }
+
+// CreateEnvironmentRequest is used to configure a new environment
+type CreateEnvironmentRequest struct {
+	Name     string   `json:"name"`
+	ImageID  string   `json:"image_id"`
+	ImageTag string   `json:"image_tag"`
+	CPUCores float32  `json:"cpu_cores"`
+	MemoryGB int      `json:"memory_gb"`
+	DiskGB   int      `json:"disk_gb"`
+	GPUs     int      `json:"gpus"`
+	Services []string `json:"services"`
+}
+
+// CreateEnvironment sends a request to create an environment.
+func (c Client) CreateEnvironment(ctx context.Context, orgID string, req CreateEnvironmentRequest) (Environment, error) {
+	var env Environment
+	err := c.requestBody(
+		ctx,
+		http.MethodPost, "/api/orgs/"+orgID+"/environments",
+		req,
+		&env,
+	)
+	return env, err
+}
+
+type envUpdate struct {
+	Type string `json:"type"`
+}
+
+// WaitForEnvironmentReady watches the environment update websocket and waits for the "done" message type before returning.
+func (c Client) WaitForEnvironmentReady(ctx context.Context, envID string) error {
+	u := c.copyURL()
+	if c.BaseURL.Scheme == "https" {
+		u.Scheme = "wss"
+	} else {
+		u.Scheme = "ws"
+	}
+	u.Path = "/api/environments/" + envID + "/watch-update"
+
+	conn, resp, err := websocket.Dial(ctx, u.String(),
+		&websocket.DialOptions{
+			HTTPHeader: map[string][]string{
+				"Cookie": {"session_token=" + c.Token},
+			},
+		},
+	)
+	if err != nil {
+		if resp != nil {
+			return bodyError(resp)
+		}
+		return err
+	}
+
+	for {
+		m := envUpdate{}
+		err = wsjson.Read(ctx, conn, &m)
+		if err != nil {
+			return fmt.Errorf("read ws json msg: %w", err)
+		}
+		if m.Type == "done" {
+			break
+		}
+	}
+
+	return nil
+}
+
+type stats struct {
+	ContainerStatus string `json:"container_status"`
+	StatError       string `json:"stat_error"`
+	Time            string `json:"time"`
+}
+
+// WatchEnvironmentStats watches the environment update websocket for a given duration.
+func (c Client) WatchEnvironmentStats(ctx context.Context, envID string, duration time.Duration) error {
+	u := c.copyURL()
+	if c.BaseURL.Scheme == "https" {
+		u.Scheme = "wss"
+	} else {
+		u.Scheme = "ws"
+	}
+	u.Path = "/api/environments/" + envID + "/watch-stats"
+
+	conn, resp, err := websocket.Dial(ctx, u.String(),
+		&websocket.DialOptions{
+			HTTPHeader: map[string][]string{
+				"Cookie": {"session_token=" + c.Token},
+			},
+		},
+	)
+	if err != nil {
+		if resp != nil {
+			return bodyError(resp)
+		}
+		return err
+	}
+
+	statsCtx, statsCancel := context.WithTimeout(ctx, duration)
+	defer statsCancel()
+
+	for {
+		select {
+		case <-statsCtx.Done():
+			return nil
+		default:
+			m := stats{}
+			err = wsjson.Read(ctx, conn, &m)
+			if err != nil {
+				return fmt.Errorf("read ws json msg: %w", err)
+			}
+		}
+	}
+}
+
+// DeleteEnvironment deletes the environment.
+func (c Client) DeleteEnvironment(ctx context.Context, envID string) error {
+	err := c.requestBody(
+		ctx,
+		http.MethodDelete, "/api/environments/" + envID,
+		nil,
+		nil,
+	)
+	return err
+}
diff --git a/coder-sdk/image.go b/coder-sdk/image.go
@@ -0,0 +1,55 @@
+package coder
+
+import (
+	"context"
+	"net/http"
+)
+
+// Image describes a Coder Image
+type Image struct {
+	ID              string  `json:"id"`
+	OrganizationID  string  `json:"organization_id"`
+	Repository      string  `json:"repository"`
+	Description     string  `json:"description"`
+	URL             string  `json:"url"` // user-supplied URL for image
+	DefaultCPUCores float32 `json:"default_cpu_cores"`
+	DefaultMemoryGB int     `json:"default_memory_gb"`
+	DefaultDiskGB   int     `json:"default_disk_gb"`
+	Deprecated      bool    `json:"deprecated"`
+}
+
+// NewRegistryRequest describes a docker registry used in importing an image
+type NewRegistryRequest struct {
+	FriendlyName string `json:"friendly_name"`
+	Registry     string `json:"registry"`
+	Username     string `json:"username"`
+	Password     string `json:"password"`
+}
+
+// ImportImageRequest is used to import new images and registries into Coder
+type ImportImageRequest struct {
+	// RegistryID is used to import images to existing registries.
+	RegistryID      *string `json:"registry_id"`
+	// NewRegistry is used when adding a new registry.
+	NewRegistry     *NewRegistryRequest `json:"new_registry"`
+	// Repository refers to the image. For example: "codercom/ubuntu".
+	Repository      string  `json:"repository"`
+	Tag             string  `json:"tag"`
+	DefaultCPUCores float32 `json:"default_cpu_cores"`
+	DefaultMemoryGB int     `json:"default_memory_gb"`
+	DefaultDiskGB   int     `json:"default_disk_gb"`
+	Description     string  `json:"description"`
+	URL             string  `json:"url"`
+}
+
+// ImportImage creates a new image and optionally a new registry
+func (c Client) ImportImage(ctx context.Context, orgID string, req ImportImageRequest) (Image, error) {
+	var img Image
+	err := c.requestBody(
+		ctx,
+		http.MethodPost, "/api/orgs/"+orgID+"/images",
+		req,
+		&img,
+	)
+	return img, err
+}
