populate templateNames from the interactive picker too

ketang · ketang · commit 5802c6b43fff · 2022-07-08T15:06:41.000-05:00
diff --git a/cli/templatedelete.go b/cli/templatedelete.go
@@ -33,6 +33,16 @@ func templateDelete() *cobra.Command {
 
 			if len(args) > 0 {
 				templateNames = args
+
+				for _, templateName := range templateNames {
+					template, err := client.TemplateByName(ctx, organization.ID, templateName)
+					if err != nil {
+						return xerrors.Errorf("get template by name: %w", err)
+					}
+
+					templates = append(templates, template)
+				}
+
 			} else {
 				allTemplates, err := client.TemplatesByOrganization(ctx, organization.ID)
 				if err != nil {
@@ -58,19 +68,11 @@ func templateDelete() *cobra.Command {
 				for _, template := range allTemplates {
 					if template.Name == selection {
 						templates = append(templates, template)
+						templateNames = append(templateNames, template.Name)
 					}
 				}
 			}
 
-			for _, templateName := range templateNames {
-				template, err := client.TemplateByName(ctx, organization.ID, templateName)
-				if err != nil {
-					return xerrors.Errorf("get template by name: %w", err)
-				}
-
-				templates = append(templates, template)
-			}
-
 			// Confirm deletion of the template.
 			_, err = cliui.Prompt(cmd, cliui.PromptOptions{
 				Text:      fmt.Sprintf("Delete these templates: %s?", cliui.Styles.Code.Render(strings.Join(templateNames, ", "))),
