feat: skip workspace removal prompt when target has none (#1271)

Signed-off-by: Ivan Dagelic <[email protected]>

pkg/cmd/target/remove.go

@@ -74,27 +74,50 @@ var targetRemoveCmd = &cobra.Command{
 				return err
 			}
 		} else {
-			form := huh.NewForm(
-				huh.NewGroup(
-					huh.NewConfirm().
-						Title(fmt.Sprintf("Delete all workspaces within %s?", selectedTargetName)).
-						Description("You might not be able to easily remove these workspaces later.").
-						Value(&yesFlag),
-				),
-			).WithTheme(views.GetCustomTheme())
-
-			err := form.Run()
+			var targetWorkspaceCount int
+
+			workspaceList, res, err := apiClient.WorkspaceAPI.ListWorkspaces(ctx).Execute()
 			if err != nil {
-				return err
+				return apiclient_util.HandleErrorResponse(res, err)
 			}
 
-			if yesFlag {
-				err := RemoveTargetWorkspaces(ctx, apiClient, selectedTargetName)
+			for _, workspace := range workspaceList {
+				if workspace.Target == selectedTargetName {
+					targetWorkspaceCount++
+				}
+			}
+
+			if targetWorkspaceCount > 0 {
+				title := fmt.Sprintf("Delete %d workspaces within %s?", targetWorkspaceCount, selectedTargetName)
+				description := "You might not be able to easily remove these workspaces later."
+
+				if targetWorkspaceCount == 1 {
+					title = fmt.Sprintf("Delete %d workspace within %s?", targetWorkspaceCount, selectedTargetName)
+					description = "You might not be able to easily remove this workspace later."
+				}
+
+				form := huh.NewForm(
+					huh.NewGroup(
+						huh.NewConfirm().
+							Title(title).
+							Description(description).
+							Value(&yesFlag),
+					),
+				).WithTheme(views.GetCustomTheme())
+
+				err := form.Run()
 				if err != nil {
 					return err
 				}
-			} else {
-				fmt.Println("Proceeding with target removal without deleting workspaces.")
+
+				if yesFlag {
+					err := RemoveTargetWorkspaces(ctx, apiClient, selectedTargetName)
+					if err != nil {
+						return err
+					}
+				} else {
+					fmt.Println("Proceeding with target removal without deleting workspaces.")
+				}
 			}
 		}