use our new watchBulkOperation rendering when --watch is provided

Author: jordanverasamy

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

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

packages/app/src/cli/flags.ts

@@ -55,4 +55,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.test.ts

@@ -1,13 +1,15 @@
 import {executeBulkOperation} from './execute-bulk-operation.js'
 import {runBulkOperationQuery} from './run-query.js'
 import {runBulkOperationMutation} from './run-mutation.js'
+import {watchBulkOperation} from './watch-bulk-operation.js'
 import {AppLinkedInterface} from '../../models/app/app.js'
-import {renderSuccess, renderWarning} from '@shopify/cli-kit/node/ui'
+import {renderSuccess, renderInfo, renderWarning} from '@shopify/cli-kit/node/ui'
 import {ensureAuthenticatedAdmin} from '@shopify/cli-kit/node/session'
 import {describe, test, expect, vi, beforeEach} from 'vitest'
 
 vi.mock('./run-query.js')
 vi.mock('./run-mutation.js')
+vi.mock('./watch-bulk-operation.js')
 vi.mock('@shopify/cli-kit/node/ui')
 vi.mock('@shopify/cli-kit/node/session')
 
@@ -135,9 +137,12 @@ describe('executeBulkOperation', () => {
       query,
     })
 
-    expect(renderSuccess).toHaveBeenCalledWith({
-      headline: 'Bulk operation started successfully!',
-      body: 'Congrats!',
+    expect(renderInfo).toHaveBeenCalledWith({
+      customSections: expect.arrayContaining([
+        expect.objectContaining({
+          title: 'Bulk operation started!',
+        }),
+      ]),
     })
   })
 
@@ -165,4 +170,36 @@ describe('executeBulkOperation', () => {
 
     expect(renderSuccess).not.toHaveBeenCalled()
   })
+
+  test('executeBulkOperation waits until operation completes and prints final results when watch flag is true', async () => {
+    const query = '{ products { edges { node { id } } } }'
+    const mockResponse = {
+      bulkOperation: successfulBulkOperation,
+      userErrors: [],
+    }
+    const completedOperation = {
+      ...successfulBulkOperation,
+      status: 'COMPLETED',
+      url: 'https://example.com/download',
+      objectCount: '650',
+    }
+
+    vi.mocked(runBulkOperationQuery).mockResolvedValue(mockResponse as any)
+    vi.mocked(watchBulkOperation).mockResolvedValue(completedOperation as any)
+
+    await executeBulkOperation({
+      app: mockApp,
+      storeFqdn,
+      query,
+      watch: true,
+    })
+
+    expect(watchBulkOperation).toHaveBeenCalledWith(mockAdminSession, successfulBulkOperation.id)
+    expect(renderSuccess).toHaveBeenCalledWith(
+      expect.objectContaining({
+        headline: 'Bulk operation complete!',
+        body: expect.stringContaining('https://example.com/download'),
+      }),
+    )
+  })
 })

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

@@ -1,5 +1,6 @@
 import {runBulkOperationQuery} from './run-query.js'
 import {runBulkOperationMutation} from './run-mutation.js'
+import {watchBulkOperation, type BulkOperation} 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'
@@ -12,10 +13,11 @@ interface ExecuteBulkOperationInput {
   storeFqdn: string
   query: string
   variables?: string[]
+  watch?: boolean
 }
 
 export async function executeBulkOperation(input: ExecuteBulkOperationInput): Promise<void> {
-  const {app, storeFqdn, query, variables} = input
+  const {app, storeFqdn, query, variables, watch = false} = input
 
   renderInfo({
     headline: 'Starting bulk operation.',
@@ -49,11 +51,36 @@ export async function executeBulkOperation(input: ExecuteBulkOperationInput): Pr
     return
   }
 
-  const result = bulkOperationResponse?.bulkOperation
-  if (result) {
-    const infoSections = [
+  const createdOperation = bulkOperationResponse?.bulkOperation
+  if (createdOperation) {
+    if (watch) {
+      const finalOperation = await watchBulkOperation(adminSession, createdOperation.id)
+      renderBulkOperationComplete(finalOperation)
+    } else {
+      renderBulkOperationStarted(createdOperation)
+    }
+  }
+}
+
+function renderBulkOperationComplete(operation: BulkOperation): void {
+  const url = operation.url ?? ''
+
+  renderSuccess({
+    headline: 'Bulk operation complete!',
+    body: outputContent`ID: ${outputToken.cyan(operation.id)}\nStatus: ${outputToken.yellow(
+      operation.status,
+    )}\nObject count: ${outputToken.gray(String(operation.objectCount))}\n\nDownload results ${outputToken.link(
+      'here.',
+      url,
+    )}`.value,
+  })
+}
+
+function renderBulkOperationStarted(result: {id: string; status: string; createdAt: unknown}): void {
+  renderInfo({
+    customSections: [
       {
-        title: 'Bulk Operation Created',
+        title: 'Bulk operation started!',
         body: [
           {
             list: {
@@ -66,15 +93,8 @@ export async function executeBulkOperation(input: ExecuteBulkOperationInput): Pr
           },
         ],
       },
-    ]
-
-    renderInfo({customSections: infoSections})
-
-    renderSuccess({
-      headline: 'Bulk operation started successfully!',
-      body: 'Congrats!',
-    })
-  }
+    ],
+  })
 }
 
 function isMutation(graphqlOperation: string): boolean {

packages/cli/oclif.manifest.json

@@ -901,6 +901,13 @@
           "hidden": false,
           "name": "verbose",
           "type": "boolean"
+        },
+        "watch": {
+          "allowNo": false,
+          "description": "Wait for bulk operation results before exiting.",
+          "env": "SHOPIFY_FLAG_WATCH",
+          "name": "watch",
+          "type": "boolean"
         }
       },
       "hasDynamicHelp": false,