feat: add coder organizations switch to change org context

Emyrk · Emyrk · commit bb2bf189bcb8 · 2024-02-22T08:23:43.000-06:00
diff --git a/cli/organization.go b/cli/organization.go
@@ -1,7 +1,9 @@
 package cli
 
 import (
+	"errors"
 	"fmt"
+	"os"
 
 	"github.com/coder/coder/v2/cli/clibase"
 	"github.com/coder/coder/v2/cli/cliui"
@@ -14,7 +16,6 @@ func (r *RootCmd) organizations() *clibase.Cmd {
 		Use:         "organizations { current }",
 		Short:       "Organization related commands",
 		Aliases:     []string{"organization", "org", "orgs"},
-
 		Handler: func(inv *clibase.Invocation) error {
 			return inv.Command.HelpHandler(inv)
 		},
@@ -27,6 +28,45 @@ func (r *RootCmd) organizations() *clibase.Cmd {
 	return cmd
 }
 
+func (r *RootCmd) switchOrganization() *clibase.Cmd {
+	var (
+		client = new(codersdk.Client)
+	)
+
+	cmd := &clibase.Cmd{
+		Use:   "switch <organization name | ID>",
+		Short: "Switch the organization used by the cli. Pass an empty string to reset to the default organization.",
+		Long: "Switch the organization used by the cli. Pass an empty string to reset to the default organization.\n" + formatExamples(
+			example{
+				Description: "Remove the current organization and defer to the default.",
+				Command:     "coder organizations switch ''",
+			},
+			example{
+				Description: "Switch to a custom organization.",
+				Command:     "coder organizations switch my-org",
+			},
+		),
+		Middleware: clibase.Chain(
+			r.InitClient(client),
+		),
+		Options: clibase.OptionSet{},
+		Handler: func(inv *clibase.Invocation) error {
+			conf := r.createConfig()
+			// If the user passes an empty string, we want to remove the organization
+			if inv.Args[0] == "" {
+				err := conf.Organization().Delete()
+				if err != nil && !errors.Is(err, os.ErrNotExist) {
+					return fmt.Errorf("failed to unset organization: %w", err)
+				}
+			}
+
+			return nil
+		},
+	}
+
+	return cmd
+}
+
 func (r *RootCmd) currentOrganization() *clibase.Cmd {
 	var (
 		client    = new(codersdk.Client)
