CopyParameters -first version

mtojek · mtojek · commit 3979b4199974 · 2023-11-09T11:02:06.000+01:00
diff --git a/cli/create.go b/cli/create.go
@@ -28,6 +28,7 @@ func (r *RootCmd) create() *clibase.Cmd {
 
 		parameterFlags workspaceParameterFlags
 		autoUpdates    string
+		copyParameters string
 	)
 	client := new(codersdk.Client)
 	cmd := &clibase.Cmd{
@@ -76,6 +77,22 @@ func (r *RootCmd) create() *clibase.Cmd {
 				return xerrors.Errorf("A workspace already exists named %q!", workspaceName)
 			}
 
+			var sourceWorkspace codersdk.Workspace
+			if copyParameters != "" {
+				sourceWorkspaceOwner, sourceWorkspaceName, err := splitNamedWorkspace(copyParameters)
+				if err != nil {
+					return err
+				}
+
+				sourceWorkspace, err = client.WorkspaceByOwnerAndName(inv.Context(), sourceWorkspaceOwner, sourceWorkspaceName, codersdk.WorkspaceOptions{})
+				if err != nil {
+					return err
+				}
+
+				_, _ = fmt.Fprintf(inv.Stdout, "Coder will use the same template %q as the source workspace.\n", sourceWorkspace.TemplateName)
+				templateName = sourceWorkspace.TemplateName
+			}
+
 			var template codersdk.Template
 			if templateName == "" {
 				_, _ = fmt.Fprintln(inv.Stdout, pretty.Sprint(cliui.DefaultStyles.Wrap, "Select a template below to preview the provisioned infrastructure:"))
@@ -134,9 +151,17 @@ func (r *RootCmd) create() *clibase.Cmd {
 				schedSpec = ptr.Ref(sched.String())
 			}
 
-			cliRichParameters, err := asWorkspaceBuildParameters(parameterFlags.richParameters)
-			if err != nil {
-				return xerrors.Errorf("can't parse given parameter values: %w", err)
+			var buildParameters []codersdk.WorkspaceBuildParameter
+			if copyParameters != "" {
+				buildParameters, err = client.WorkspaceBuildParameters(inv.Context(), sourceWorkspace.ID)
+				if err != nil {
+					return err
+				}
+			} else {
+				buildParameters, err = asWorkspaceBuildParameters(parameterFlags.richParameters)
+				if err != nil {
+					return xerrors.Errorf("can't parse given parameter values: %w", err)
+				}
 			}
 
 			richParameters, err := prepWorkspaceBuild(inv, client, prepWorkspaceBuildArgs{
@@ -145,7 +170,7 @@ func (r *RootCmd) create() *clibase.Cmd {
 				NewWorkspaceName:  workspaceName,
 
 				RichParameterFile: parameterFlags.richParameterFile,
-				RichParameters:    cliRichParameters,
+				RichParameters:    buildParameters,
 			})
 			if err != nil {
 				return xerrors.Errorf("prepare build: %w", err)
@@ -217,6 +242,13 @@ func (r *RootCmd) create() *clibase.Cmd {
 			Default:     string(codersdk.AutomaticUpdatesNever),
 			Value:       clibase.StringOf(&autoUpdates),
 		},
+		clibase.Option{
+			Flag:        "copy-parameters",
+			Env:         "CODER_WORKSPACE_COPY_PARAMETERS",
+			Description: "Specify the source workspace name to copy parameters from.",
+			Default:     string(codersdk.AutomaticUpdatesNever),
+			Value:       clibase.StringOf(&copyParameters),
+		},
 		cliui.SkipPromptOption(),
 	)
 	cmd.Options = append(cmd.Options, parameterFlags.cliParameters()...)
diff --git a/cli/testdata/coder_create_--help.golden b/cli/testdata/coder_create_--help.golden
@@ -14,6 +14,9 @@ OPTIONS:
           Specify automatic updates setting for the workspace (accepts 'always'
           or 'never').
 
+      --copy-parameters string, $CODER_WORKSPACE_COPY_PARAMETERS (default: never)
+          Specify the source workspace name to copy parameters from.
+
       --parameter string-array, $CODER_RICH_PARAMETER
           Rich parameter value in the format "name=value".
 
