feat: implement cordon/uncordon workspace provider (#302)

Faris Huskovic · web-flow · commit a58a2e7f4588 · 2021-03-25T14:26:26.000-05:00
* feat: implement cordon/uncordon workspace provider
diff --git a/Makefile b/Makefile
@@ -1,6 +1,6 @@
 # Makefile for Coder CLI 
 
-.PHONY: clean build build/macos build/windows build/linux fmt lint gendocs test/go
+.PHONY: clean build build/macos build/windows build/linux fmt lint gendocs test/go dev
 
 PROJECT_ROOT := $(shell git rev-parse --show-toplevel)
 MAKE_ROOT := $(shell pwd)
@@ -42,3 +42,10 @@ test/coverage:
 		$$(go list ./... | grep -v pkg/tcli | grep -v ci/integration)
 
 	goveralls -coverprofile=coverage -service=github
+
+dev: build/linux
+	@echo "removing project root binary if exists"
+	-rm ./coder
+	@echo "untarring..."
+	@tar -xzf ./ci/bin/coder-cli-linux-amd64.tar.gz
+	@echo "new dev binary ready"
diff --git a/coder b/coder
diff --git a/coder-sdk/interface.go b/coder-sdk/interface.go
@@ -225,4 +225,11 @@ type Client interface {
 
 	// BaseURL returns the BaseURL configured for this Client.
 	BaseURL() url.URL
+
+	// CordonWorkspaceProvider prevents the provider from having any more workspaces placed on it.
+	CordonWorkspaceProvider(ctx context.Context, id, reason string) error
+
+	// UnCordonWorkspaceProvider changes an existing cordoned providers status to 'Ready';
+	// allowing it to continue creating new workspaces and provisioning resources for them.
+	UnCordonWorkspaceProvider(ctx context.Context, id string) error
 }
diff --git a/coder-sdk/workspace_providers.go b/coder-sdk/workspace_providers.go
@@ -98,3 +98,28 @@ func (c *DefaultClient) CreateWorkspaceProvider(ctx context.Context, req CreateW
 func (c *DefaultClient) DeleteWorkspaceProviderByID(ctx context.Context, id string) error {
 	return c.requestBody(ctx, http.MethodDelete, "/api/private/resource-pools/"+id, nil, nil)
 }
+
+// CordoneWorkspaceProviderReq defines the request parameters for creating a new workspace provider entity.
+type CordoneWorkspaceProviderReq struct {
+	Reason string `json:"reason"`
+}
+
+// CordonWorkspaceProvider prevents the provider from having any more workspaces placed on it.
+func (c *DefaultClient) CordonWorkspaceProvider(ctx context.Context, id, reason string) error {
+	req := CordoneWorkspaceProviderReq{Reason: reason}
+	err := c.requestBody(ctx, http.MethodPost, "/api/private/resource-pools/"+id+"/cordon", req, nil)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+// UnCordonWorkspaceProvider changes an existing cordoned providers status to 'Ready';
+// allowing it to continue creating new workspaces and provisioning resources for them.
+func (c *DefaultClient) UnCordonWorkspaceProvider(ctx context.Context, id string) error {
+	err := c.requestBody(ctx, http.MethodPost, "/api/private/resource-pools/"+id+"/uncordon", nil, nil)
+	if err != nil {
+		return err
+	}
+	return nil
+}
diff --git a/internal/cmd/providers.go b/internal/cmd/providers.go
@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"net/url"
 
+	"cdr.dev/coder-cli/internal/coderutil"
 	"cdr.dev/coder-cli/internal/x/xcobra"
 
 	"github.com/spf13/cobra"
@@ -26,6 +27,8 @@ func providersCmd() *cobra.Command {
 		createProviderCmd(),
 		listProviderCmd(),
 		deleteProviderCmd(),
+		cordonProviderCmd(),
+		unCordonProviderCmd(),
 	)
 	return cmd
 }
@@ -215,3 +218,69 @@ coder providers rm my-workspace-provider`,
 	}
 	return cmd
 }
+
+func cordonProviderCmd() *cobra.Command {
+	var reason string
+
+	cmd := &cobra.Command{
+		Use:   "cordon [workspace_provider_name]",
+		Args:  xcobra.ExactArgs(1),
+		Short: "cordon a workspace provider.",
+		Long:  "Prevent an existing Coder workspace provider from supporting any additional workspaces.",
+		Example: `# cordon an existing workspace provider by name
+coder providers cordon my-workspace-provider --reason "limit cloud clost"`,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := cmd.Context()
+			client, err := newClient(ctx)
+			if err != nil {
+				return err
+			}
+
+			wpName := args[0]
+			provider, err := coderutil.ProviderByName(ctx, client, wpName)
+			if err != nil {
+				return err
+			}
+
+			if err := client.CordonWorkspaceProvider(ctx, provider.ID, reason); err != nil {
+				return err
+			}
+			clog.LogSuccess(fmt.Sprintf("provider %q successfully cordoned - you can no longer create workspaces on this provider without uncordoning first", wpName))
+			return nil
+		},
+	}
+	cmd.Flags().StringVar(&reason, "reason", "", "reason for cordoning the provider")
+	_ = cmd.MarkFlagRequired("reason")
+	return cmd
+}
+
+func unCordonProviderCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "uncordon [workspace_provider_name]",
+		Args:  xcobra.ExactArgs(1),
+		Short: "uncordon a workspace provider.",
+		Long:  "Set a currently cordoned provider as ready; enabling it to continue provisioning resources for new workspaces.",
+		Example: `# uncordon an existing workspace provider by name
+coder providers uncordon my-workspace-provider`,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := cmd.Context()
+			client, err := newClient(ctx)
+			if err != nil {
+				return err
+			}
+
+			wpName := args[0]
+			provider, err := coderutil.ProviderByName(ctx, client, wpName)
+			if err != nil {
+				return err
+			}
+
+			if err := client.UnCordonWorkspaceProvider(ctx, provider.ID); err != nil {
+				return err
+			}
+			clog.LogSuccess(fmt.Sprintf("provider %q successfully uncordoned - you can now create workspaces on this provider", wpName))
+			return nil
+		},
+	}
+	return cmd
+}
