feat: add envs create-from-repo subcommand

kylecarbs · kylecarbs · commit 921d24ded8fd · 2021-01-07T08:42:47.000Z
diff --git a/coder-sdk/env.go b/coder-sdk/env.go
@@ -92,6 +92,24 @@ func (c Client) CreateEnvironment(ctx context.Context, orgID string, req CreateE
 	return &env, nil
 }
 
+// CreateEnvironmentFromRepoRequest is used to configure a new environment from a repo.
+type CreateEnvironmentFromRepoRequest struct {
+	RepositoryURL string `json:"repository_url"`
+	// Optional. The default branch will be used if not provided.
+	Branch string `json:"branch"`
+	// Optional. The template name will be used if not provided.
+	Name string `json:"name"`
+}
+
+// CreateEnvironmentFromRepo sends a request to create an environment from a repository.
+func (c Client) CreateEnvironmentFromRepo(ctx context.Context, orgID string, req CreateEnvironmentFromRepoRequest) (*Environment, error) {
+	var env Environment
+	if err := c.requestBody(ctx, http.MethodPost, "/api/private/orgs/"+orgID+"/environments/from-repo", req, &env); err != nil {
+		return nil, err
+	}
+	return &env, nil
+}
+
 // Environments lists environments returned by the given filter.
 // TODO: add the filter options, explore performance issue.
 func (c Client) Environments(ctx context.Context) ([]Environment, error) {
diff --git a/internal/cmd/envs.go b/internal/cmd/envs.go
@@ -31,6 +31,7 @@ func envsCmd() *cobra.Command {
 		watchBuildLogCommand(),
 		rebuildEnvCommand(),
 		createEnvCmd(),
+		createEnvFromRepoCmd(),
 		editEnvCmd(),
 	)
 	return cmd
@@ -183,7 +184,6 @@ coder envs create my-new-powerful-env --cpu 12 --disk 100 --memory 16 --image ub
 			if multiOrgMember && org == "" {
 				return xerrors.New("org is required for multi-org members")
 			}
-
 			importedImg, err := findImg(ctx, client, findImgConf{
 				email:   coder.Me,
 				imgName: img,
@@ -250,6 +250,85 @@ coder envs create my-new-powerful-env --cpu 12 --disk 100 --memory 16 --image ub
 	return cmd
 }
 
+func createEnvFromRepoCmd() *cobra.Command {
+	var (
+		org    string
+		branch string
+		name   string
+		follow bool
+	)
+
+	cmd := &cobra.Command{
+		Use:    "create-from-repo [environment_name]",
+		Short:  "create a new environment from a git repository.",
+		Args:   cobra.ExactArgs(1),
+		Long:   "Create a new Coder environment from a Git repository.",
+		Hidden: true,
+		Example: `# create a new environment from git repository template
+coder envs create-from-repo github.com/cdr/m
+coder envs create-from-repo github.com/cdr/m --branch envs-as-code`,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := cmd.Context()
+
+			client, err := newClient(ctx)
+			if err != nil {
+				return err
+			}
+
+			multiOrgMember, err := isMultiOrgMember(ctx, client, coder.Me)
+			if err != nil {
+				return err
+			}
+
+			if multiOrgMember && org == "" {
+				return xerrors.New("org is required for multi-org members")
+			}
+
+			if org == "" {
+				// Definitely janky... but since we aren't importing
+				// an image we need to get the org name somehow.
+				orgs, err := client.Organizations(ctx)
+				if err != nil {
+					return err
+				}
+				org = orgs[0].Name
+			}
+
+			// ExactArgs(1) ensures our name value can't panic on an out of bounds.
+			createReq := &coder.CreateEnvironmentFromRepoRequest{
+				RepositoryURL: args[0],
+				Name:          name,
+				Branch:        branch,
+			}
+
+			env, err := client.CreateEnvironmentFromRepo(ctx, org, *createReq)
+			if err != nil {
+				return xerrors.Errorf("create environment: %w", err)
+			}
+
+			if follow {
+				clog.LogSuccess("creating environment...")
+				if err := trailBuildLogs(ctx, client, env.ID); err != nil {
+					return err
+				}
+				return nil
+			}
+
+			clog.LogSuccess("creating environment...",
+				clog.BlankLine,
+				clog.Tipf(`run "coder envs watch-build %s" to trail the build logs`, env.Name),
+			)
+			return nil
+		},
+	}
+	cmd.Flags().StringVarP(&org, "org", "o", "", "name of the organization the environment should be created under.")
+	cmd.Flags().StringVarP(&name, "name", "n", "", "name of the environment to create.")
+	cmd.Flags().StringVarP(&branch, "branch", "b", "", "name of the branch to create the environment from.")
+	cmd.Flags().BoolVar(&follow, "follow", false, "follow buildlog after initiating rebuild")
+	_ = cmd.MarkFlagRequired("image")
+	return cmd
+}
+
 func editEnvCmd() *cobra.Command {
 	var (
 		org    string
