Improve status messages for app bulk execute task

Author: nickwesselman

packages/app/src/cli/api/graphql/bulk-operations/generated/bulk-operation-run-mutation.ts

@@ -12,6 +12,7 @@ export type BulkOperationRunMutationMutationVariables = Types.Exact<{
 export type BulkOperationRunMutationMutation = {
   bulkOperationRunMutation?: {
     bulkOperation?: {
+      type: Types.BulkOperationType
       completedAt?: unknown | null
       createdAt: unknown
       errorCode?: Types.BulkOperationErrorCode | null
@@ -81,6 +82,7 @@ export const BulkOperationRunMutation = {
                   selectionSet: {
                     kind: 'SelectionSet',
                     selections: [
+                      {kind: 'Field', name: {kind: 'Name', value: 'type'}},
                       {kind: 'Field', name: {kind: 'Name', value: 'completedAt'}},
                       {kind: 'Field', name: {kind: 'Name', value: 'createdAt'}},
                       {kind: 'Field', name: {kind: 'Name', value: 'errorCode'}},

packages/app/src/cli/api/graphql/bulk-operations/generated/bulk-operation-run-query.ts

@@ -10,6 +10,7 @@ export type BulkOperationRunQueryMutationVariables = Types.Exact<{
 export type BulkOperationRunQueryMutation = {
   bulkOperationRunQuery?: {
     bulkOperation?: {
+      type: Types.BulkOperationType
       completedAt?: unknown | null
       createdAt: unknown
       errorCode?: Types.BulkOperationErrorCode | null
@@ -59,6 +60,7 @@ export const BulkOperationRunQuery = {
                   selectionSet: {
                     kind: 'SelectionSet',
                     selections: [
+                      {kind: 'Field', name: {kind: 'Name', value: 'type'}},
                       {kind: 'Field', name: {kind: 'Name', value: 'completedAt'}},
                       {kind: 'Field', name: {kind: 'Name', value: 'createdAt'}},
                       {kind: 'Field', name: {kind: 'Name', value: 'errorCode'}},

packages/app/src/cli/api/graphql/bulk-operations/generated/get-bulk-operation-by-id.ts

@@ -9,6 +9,7 @@ export type GetBulkOperationByIdQueryVariables = Types.Exact<{
 
 export type GetBulkOperationByIdQuery = {
   bulkOperation?: {
+    type: Types.BulkOperationType
     completedAt?: unknown | null
     createdAt: unknown
     errorCode?: Types.BulkOperationErrorCode | null
@@ -50,6 +51,7 @@ export const GetBulkOperationById = {
             selectionSet: {
               kind: 'SelectionSet',
               selections: [
+                {kind: 'Field', name: {kind: 'Name', value: 'type'}},
                 {kind: 'Field', name: {kind: 'Name', value: 'completedAt'}},
                 {kind: 'Field', name: {kind: 'Name', value: 'createdAt'}},
                 {kind: 'Field', name: {kind: 'Name', value: 'errorCode'}},

packages/app/src/cli/api/graphql/bulk-operations/generated/types.d.ts

@@ -188,6 +188,13 @@ export type BulkOperationStatus =
   /** The bulk operation is runnning. */
   | 'RUNNING'
 
+/** The valid values for the bulk operation's type. */
+export type BulkOperationType =
+  /** The bulk operation is a mutation. */
+  | 'MUTATION'
+  /** The bulk operation is a query. */
+  | 'QUERY'
+
 /** Possible error codes that can be returned by `BulkOperationUserError`. */
 export type BulkOperationUserErrorCode =
   /** The input value is invalid. */
@@ -203,6 +210,8 @@ export type BulkOperationsSortKeys =
   | 'COMPLETED_AT'
   /** Sort by the `created_at` value. */
   | 'CREATED_AT'
+  /** Sort by the `status` value. */
+  | 'STATUS'
 
 /**
  * The possible HTTP methods that can be used when sending a request to upload a file using information from a

packages/app/src/cli/api/graphql/bulk-operations/mutations/bulk-operation-run-mutation.graphql

@@ -9,6 +9,7 @@ mutation BulkOperationRunMutation(
     clientIdentifier: $clientIdentifier
   ) {
     bulkOperation {
+      type
       completedAt
       createdAt
       errorCode

packages/app/src/cli/api/graphql/bulk-operations/mutations/bulk-operation-run-query.graphql

@@ -3,6 +3,7 @@ mutation BulkOperationRunQuery($query: String!) {
     query: $query
   ) {
     bulkOperation {
+      type
       completedAt
       createdAt
       errorCode

packages/app/src/cli/api/graphql/bulk-operations/queries/get-bulk-operation-by-id.graphql

@@ -1,5 +1,6 @@
 query GetBulkOperationById($id: ID!) {
   bulkOperation(id: $id) {
+    type
     completedAt
     createdAt
     errorCode

packages/app/src/cli/services/bulk-operations/format-bulk-operation-status.test.ts

@@ -8,54 +8,62 @@ function createMockOperation(overrides: Partial<BulkOperation> = {}): BulkOperat
   return {
     id: 'gid://shopify/BulkOperation/123',
     status: 'CREATED',
+    type: 'QUERY',
     errorCode: null,
     createdAt: '2024-01-01T00:00:00Z',
     completedAt: null,
     objectCount: '0',
     url: null,
+    partialDataUrl: null,
     ...overrides,
   }
 }
 
 describe('formatBulkOperationStatus', () => {
-  test('formats RUNNING status with object count', () => {
-    const result = formatBulkOperationStatus(createMockOperation({status: 'RUNNING', objectCount: 42}))
-    expect(result.value).toContain('Bulk operation in progress...')
-    expect(result.value).toContain('(42 objects)')
+  test('formats RUNNING status for query with object count', () => {
+    const result = formatBulkOperationStatus(createMockOperation({status: 'RUNNING', type: 'QUERY', objectCount: '42'}))
+    expect(result.value).toContain('Bulk operation in progress')
+    expect(result.value).toContain('(42 objects read)')
+  })
+
+  test('formats RUNNING status for mutation with object count', () => {
+    const result = formatBulkOperationStatus(createMockOperation({status: 'RUNNING', type: 'MUTATION', objectCount: '42'}))
+    expect(result.value).toContain('Bulk operation in progress')
+    expect(result.value).toContain('(42 objects written)')
   })
 
   test('formats CREATED status', () => {
     const result = formatBulkOperationStatus(createMockOperation({status: 'CREATED'}))
-    expect(result.value).toBe('Starting...')
+    expect(result.value).toBe('Starting')
   })
 
   test('formats COMPLETED status', () => {
-    const result = formatBulkOperationStatus(createMockOperation({status: 'COMPLETED', objectCount: 100}))
+    const result = formatBulkOperationStatus(createMockOperation({status: 'COMPLETED', objectCount: '100'}))
     expect(result.value).toContain('Bulk operation succeeded:')
     expect(result.value).toContain('100 objects')
   })
 
   test('formats FAILED status with error code', () => {
     const result = formatBulkOperationStatus(
-      createMockOperation({status: 'FAILED', objectCount: 10, errorCode: 'ACCESS_DENIED'}),
+      createMockOperation({status: 'FAILED', objectCount: '10', errorCode: 'ACCESS_DENIED'}),
     )
     expect(result.value).toContain('Bulk operation failed.')
     expect(result.value).toContain('Error: ACCESS_DENIED')
   })
 
   test('formats FAILED status without error code', () => {
-    const result = formatBulkOperationStatus(createMockOperation({status: 'FAILED', objectCount: 10, errorCode: null}))
+    const result = formatBulkOperationStatus(createMockOperation({status: 'FAILED', objectCount: '10', errorCode: null}))
     expect(result.value).toContain('Bulk operation failed.')
     expect(result.value).toContain('Error: unknown')
   })
 
   test('formats CANCELING status', () => {
-    const result = formatBulkOperationStatus(createMockOperation({status: 'CANCELING', objectCount: 5}))
+    const result = formatBulkOperationStatus(createMockOperation({status: 'CANCELING', objectCount: '5'}))
     expect(result.value).toBe('Bulk operation canceling...')
   })
 
   test('formats CANCELED status', () => {
-    const result = formatBulkOperationStatus(createMockOperation({status: 'CANCELED', objectCount: 5}))
+    const result = formatBulkOperationStatus(createMockOperation({status: 'CANCELED', objectCount: '5'}))
     expect(result.value).toBe('Bulk operation canceled.')
   })
 

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

@@ -6,11 +6,11 @@ export function formatBulkOperationStatus(
 ): TokenizedString {
   switch (operation.status) {
     case 'RUNNING':
-      return outputContent`Bulk operation in progress... ${outputToken.gray(
-        `(${String(operation.objectCount)} objects)`,
+      return outputContent`Bulk operation in progress ${outputToken.gray(
+        `(${String(operation.objectCount)} objects ${operation.type === 'MUTATION' ? 'written' : 'read'})`,
       )}`
     case 'CREATED':
-      return outputContent`Starting...`
+      return outputContent`Starting`
     case 'COMPLETED':
       return outputContent`Bulk operation succeeded: ${outputToken.gray(`${String(operation.objectCount)} objects`)}`
     case 'FAILED':