wire up new watch UI to execute command

Author: jordanverasamy

packages/app/src/cli/commands/app/execute.ts

@@ -38,6 +38,7 @@ export default class Execute extends AppLinkedCommand {
       app: appContextResult.app,
       storeFqdn: store.shopDomain,
       query: flags.query,
+      watch: flags.watch,
     })
 
     return {app: appContextResult.app}

packages/app/src/cli/flags.ts

@@ -48,4 +48,9 @@ export const bulkOperationFlags = {
     env: 'SHOPIFY_FLAG_STORE',
     parse: async (input) => normalizeStoreFqdn(input),
   }),
+  watch: Flags.boolean({
+    description: 'Wait for bulk operation results before exiting.',
+    env: 'SHOPIFY_FLAG_WATCH',
+    default: false,
+  }),
 }

packages/app/src/cli/services/bulk-operations/execute-bulk-operation.ts

@@ -1,4 +1,5 @@
 import {runBulkOperationQuery} from './run-query.js'
+import {watchBulkOperation} from './watch-bulk-operation.js'
 import {AppLinkedInterface} from '../../models/app/app.js'
 import {renderSuccess, renderInfo, renderWarning} from '@shopify/cli-kit/node/ui'
 import {outputContent, outputToken} from '@shopify/cli-kit/node/output'
@@ -7,10 +8,11 @@ interface ExecuteBulkOperationInput {
   app: AppLinkedInterface
   storeFqdn: string
   query: string
+  watch?: boolean
 }
 
 export async function executeBulkOperation(input: ExecuteBulkOperationInput): Promise<void> {
-  const {app, storeFqdn, query} = input
+  const {app, storeFqdn, query, watch = false} = input
 
   renderInfo({
     headline: 'Starting bulk operation.',
@@ -57,9 +59,65 @@ export async function executeBulkOperation(input: ExecuteBulkOperationInput): Pr
 
     renderInfo({customSections: infoSections})
 
-    renderSuccess({
-      headline: 'Bulk operation started successfully!',
-      body: 'Congrats!',
-    })
+    if (watch) {
+      const finalOperation = await watchBulkOperation({
+        id: result.id,
+        storeFqdn,
+      })
+
+      if (!finalOperation) {
+        renderWarning({
+          headline: 'Operation not found.',
+          body: 'The bulk operation could not be found',
+        })
+        return
+      }
+
+      if (!('status' in finalOperation) || !('objectCount' in finalOperation)) {
+        renderWarning({
+          headline: 'Operation incomplete.',
+          body: 'The bulk operation did not return complete data',
+        })
+        return
+      }
+
+      const count = finalOperation.objectCount
+      switch (finalOperation.status) {
+        case 'COMPLETED':
+          renderSuccess({
+            headline: 'Query complete!',
+            body: `${count} objects processed`,
+          })
+          break
+        case 'FAILED':
+          renderWarning({
+            headline: 'Query failed.',
+            body: `${count} objects processed before failure`,
+          })
+          break
+        case 'CANCELED':
+          renderWarning({
+            headline: 'Query canceled.',
+            body: `${count} objects processed before cancellation`,
+          })
+          break
+        case 'EXPIRED':
+          renderWarning({
+            headline: 'Query expired.',
+            body: `${count} objects processed before expiration`,
+          })
+          break
+        default:
+          renderWarning({
+            headline: `Query ${finalOperation.status.toLowerCase()}.`,
+            body: `${count} objects processed`,
+          })
+      }
+    } else {
+      renderSuccess({
+        headline: 'Bulk operation started successfully!',
+        body: 'Congrats!',
+      })
+    }
   }
 }