From 19baca55da446239c73869323a73c0d6e5bc4df5 Mon Sep 17 00:00:00 2001
From: Steven Masley <Emyrk@users.noreply.github.com>
Date: Tue, 27 Feb 2024 10:13:08 -0600
Subject: [PATCH] feat: implement create org commands from cli (#12308)

* feat: implement create org commands from cli
---
 cli/organization.go       |  1 +
 cli/organizationmanage.go | 65 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 66 insertions(+)
 create mode 100644 cli/organizationmanage.go

diff --git a/cli/organization.go b/cli/organization.go
index f014cc40ee..74b71464d7 100644
--- a/cli/organization.go
+++ b/cli/organization.go
@@ -27,6 +27,7 @@ func (r *RootCmd) organizations() *clibase.Cmd {
 		Children: []*clibase.Cmd{
 			r.currentOrganization(),
 			r.switchOrganization(),
+			r.createOrganization(),
 		},
 	}
 
diff --git a/cli/organizationmanage.go b/cli/organizationmanage.go
new file mode 100644
index 0000000000..577505a0c4
--- /dev/null
+++ b/cli/organizationmanage.go
@@ -0,0 +1,65 @@
+package cli
+
+import (
+	"fmt"
+
+	"github.com/google/uuid"
+
+	"github.com/coder/coder/v2/cli/clibase"
+	"github.com/coder/coder/v2/cli/cliui"
+	"github.com/coder/coder/v2/codersdk"
+	"github.com/coder/pretty"
+)
+
+func (r *RootCmd) createOrganization() *clibase.Cmd {
+	client := new(codersdk.Client)
+
+	cmd := &clibase.Cmd{
+		Use:   "create <organization name>",
+		Short: "Create a new organization.",
+		// This action is currently irreversible, so it's hidden until we have a way to delete organizations.
+		Hidden: true,
+		Middleware: clibase.Chain(
+			r.InitClient(client),
+			clibase.RequireNArgs(1),
+		),
+		Options: clibase.OptionSet{
+			cliui.SkipPromptOption(),
+		},
+		Handler: func(inv *clibase.Invocation) error {
+			orgName := inv.Args[0]
+
+			// This check is not perfect since not all users can read all organizations.
+			// So ignore the error and if the org already exists, prevent the user
+			// from creating it.
+			existing, _ := client.OrganizationByName(inv.Context(), orgName)
+			if existing.ID != uuid.Nil {
+				return fmt.Errorf("organization %q already exists", orgName)
+			}
+
+			_, err := cliui.Prompt(inv, cliui.PromptOptions{
+				Text: fmt.Sprintf("Are you sure you want to create an organization with the name %s?\n%s",
+					pretty.Sprint(cliui.DefaultStyles.Code, orgName),
+					pretty.Sprint(cliui.BoldFmt(), "This action is irreversible."),
+				),
+				IsConfirm: true,
+				Default:   cliui.ConfirmNo,
+			})
+			if err != nil {
+				return err
+			}
+
+			organization, err := client.CreateOrganization(inv.Context(), codersdk.CreateOrganizationRequest{
+				Name: orgName,
+			})
+			if err != nil {
+				return fmt.Errorf("failed to create organization: %w", err)
+			}
+
+			_, _ = fmt.Fprintf(inv.Stdout, "Organization %s (%s) created.\n", organization.Name, organization.ID)
+			return nil
+		},
+	}
+
+	return cmd
+}