chore(cli): validate various names locally

ethanndickson · ethanndickson · commit 8a7b733a34f6 · 2024-09-04T08:18:07.000Z
diff --git a/cli/create.go b/cli/create.go
@@ -60,7 +60,11 @@ func (r *RootCmd) create() *serpent.Command {
 				workspaceName, err = cliui.Prompt(inv, cliui.PromptOptions{
 					Text: "Specify a name for your workspace:",
 					Validate: func(workspaceName string) error {
-						_, err = client.WorkspaceByOwnerAndName(inv.Context(), codersdk.Me, workspaceName, codersdk.WorkspaceOptions{})
+						err = codersdk.NameValid(workspaceName)
+						if err != nil {
+							return xerrors.Errorf("Workspace name %s", err)
+						}
+						_, err = client.WorkspaceByOwnerAndName(inv.Context(), workspaceOwner, workspaceName, codersdk.WorkspaceOptions{})
 						if err == nil {
 							return xerrors.Errorf("A workspace already exists named %q!", workspaceName)
 						}
@@ -71,7 +75,10 @@ func (r *RootCmd) create() *serpent.Command {
 					return err
 				}
 			}
-
+			err = codersdk.NameValid(workspaceName)
+			if err != nil {
+				return xerrors.Errorf("Workspace name %s", err)
+			}
 			_, err = client.WorkspaceByOwnerAndName(inv.Context(), workspaceOwner, workspaceName, codersdk.WorkspaceOptions{})
 			if err == nil {
 				return xerrors.Errorf("A workspace already exists named %q!", workspaceName)
diff --git a/cli/organizationmanage.go b/cli/organizationmanage.go
@@ -30,6 +30,11 @@ func (r *RootCmd) createOrganization() *serpent.Command {
 		Handler: func(inv *serpent.Invocation) error {
 			orgName := inv.Args[0]
 
+			err := codersdk.NameValid(orgName)
+			if err != nil {
+				return xerrors.Errorf("Organization name %w", err)
+			}
+
 			// 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.
@@ -38,7 +43,7 @@ func (r *RootCmd) createOrganization() *serpent.Command {
 				return xerrors.Errorf("organization %q already exists", orgName)
 			}
 
-			_, err := cliui.Prompt(inv, cliui.PromptOptions{
+			_, 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."),
diff --git a/cli/templatepush.go b/cli/templatepush.go
@@ -10,7 +10,6 @@ import (
 	"path/filepath"
 	"strings"
 	"time"
-	"unicode/utf8"
 
 	"github.com/briandowns/spinner"
 	"github.com/google/uuid"
@@ -57,8 +56,16 @@ func (r *RootCmd) templatePush() *serpent.Command {
 				return err
 			}
 
-			if utf8.RuneCountInString(name) > 32 {
-				return xerrors.Errorf("Template name must be no more than 32 characters")
+			err = codersdk.NameValid(name)
+			if err != nil {
+				return xerrors.Errorf("Template name %w", err)
+			}
+
+			if versionName != "" {
+				err = codersdk.TemplateVersionNameValid(versionName)
+				if err != nil {
+					return xerrors.Errorf("Template version name %w", err)
+				}
 			}
 
 			var createTemplate bool
diff --git a/cli/usercreate.go b/cli/usercreate.go
@@ -44,6 +44,13 @@ func (r *RootCmd) userCreate() *serpent.Command {
 			if username == "" {
 				username, err = cliui.Prompt(inv, cliui.PromptOptions{
 					Text: "Username:",
+					Validate: func(username string) error {
+						err = codersdk.NameValid(username)
+						if err != nil {
+							return xerrors.Errorf("Username %s", err)
+						}
+						return nil
+					},
 				})
 				if err != nil {
 					return err
@@ -144,7 +151,15 @@ Create a workspace  `+pretty.Sprint(cliui.DefaultStyles.Code, "coder create")+`!
 			Flag:          "username",
 			FlagShorthand: "u",
 			Description:   "Specifies a username for the new user.",
-			Value:         serpent.StringOf(&username),
+			Value: serpent.Validate(serpent.StringOf(&username), func(username *serpent.String) error {
+				if username.String() != "" {
+					err := codersdk.NameValid(username.String())
+					if err != nil {
+						return xerrors.Errorf("Username %s", err)
+					}
+				}
+				return nil
+			}),
 		},
 		{
 			Flag:          "full-name",
diff --git a/enterprise/cli/groupcreate.go b/enterprise/cli/groupcreate.go
@@ -35,6 +35,11 @@ func (r *RootCmd) groupCreate() *serpent.Command {
 				return xerrors.Errorf("current organization: %w", err)
 			}
 
+			err = codersdk.GroupNameValid(inv.Args[0])
+			if err != nil {
+				return xerrors.Errorf("Group name %w", err)
+			}
+
 			group, err := client.CreateGroup(ctx, org.ID, codersdk.CreateGroupRequest{
 				Name:        inv.Args[0],
 				DisplayName: displayName,
@@ -61,7 +66,15 @@ func (r *RootCmd) groupCreate() *serpent.Command {
 			Flag:        "display-name",
 			Description: `Optional human friendly name for the group.`,
 			Env:         "CODER_DISPLAY_NAME",
-			Value:       serpent.StringOf(&displayName),
+			Value: serpent.Validate(serpent.StringOf(&displayName), func(displayName *serpent.String) error {
+				if displayName.String() != "" {
+					err := codersdk.DisplayNameValid(displayName.String())
+					if err != nil {
+						return xerrors.Errorf("Group display name %w", err)
+					}
+				}
+				return nil
+			}),
 		},
 	}
 	orgContext.AttachOptions(cmd)
