Create the template if it does not exist

mtojek · mtojek · commit ceebc7580dab · 2023-07-12T14:44:56.000+02:00
diff --git a/cli/templatepush.go b/cli/templatepush.go
@@ -186,8 +186,11 @@ func (r *RootCmd) templatePush() *clibase.Cmd {
 				return err
 			}
 
+			var templateExists bool
 			template, err := client.TemplateByName(inv.Context(), organization.ID, name)
-			if err != nil {
+			if create && err == nil {
+				templateExists = true
+			} else if !create && err != nil {
 				return err
 			}
 
@@ -208,19 +211,24 @@ func (r *RootCmd) templatePush() *clibase.Cmd {
 				return err
 			}
 
-			job, err := createValidTemplateVersion(inv, createValidTemplateVersionArgs{
-				Name:            versionName,
+			args := createValidTemplateVersionArgs{
 				Message:         message,
 				Client:          client,
 				Organization:    organization,
 				Provisioner:     database.ProvisionerType(provisioner),
 				FileID:          resp.ID,
+				ProvisionerTags: tags,
 				VariablesFile:   variablesFile,
 				Variables:       variables,
-				Template:        &template,
-				ReuseParameters: !alwaysPrompt,
-				ProvisionerTags: tags,
-			})
+			}
+
+			if templateExists {
+				args.Name = versionName
+				args.Template = &template
+				args.ReuseParameters = !alwaysPrompt
+			}
+
+			job, err := createValidTemplateVersion(inv, args)
 			if err != nil {
 				return err
 			}
@@ -229,7 +237,19 @@ func (r *RootCmd) templatePush() *clibase.Cmd {
 				return xerrors.Errorf("job failed: %s", job.Job.Status)
 			}
 
-			if activate {
+			if !templateExists {
+				_, err = client.CreateTemplate(inv.Context(), organization.ID, codersdk.CreateTemplateRequest{
+					Name:      name,
+					VersionID: job.ID,
+				})
+				if err != nil {
+					return err
+				}
+
+				_, _ = fmt.Fprintln(inv.Stdout, "\n"+cliui.DefaultStyles.Wrap.Render(
+					"The "+cliui.DefaultStyles.Keyword.Render(name)+" template has been created at "+cliui.DefaultStyles.DateTimeStamp.Render(time.Now().Format(time.Stamp))+"! "+
+						"Developers can provision a workspace with this template using:")+"\n")
+			} else if activate {
 				err = client.UpdateActiveTemplateVersion(inv.Context(), template.ID, codersdk.UpdateActiveTemplateVersion{
 					ID: job.ID,
 				})
