Extend ensureDirectoryConfirmed to display multi env error

When a command is run with multiple environment flags
we cant prompt for confirmation. This commit allows
`ensureDirectoryConfirmed` to accept environment
and multiEnvironment params

When a multi environment command run seems
to be acting on an invalid directory an error is
display and we do not confirm the action

This matches the behavior of other instances where
multi environment command actions cant be confirmed

Author: dejmedus

packages/theme/src/cli/utilities/theme-ui.test.ts

@@ -1,6 +1,9 @@
-import {themeComponent, themesComponent} from './theme-ui.js'
+import {themeComponent, themesComponent, ensureDirectoryConfirmed} from './theme-ui.js'
 import {Theme} from '@shopify/cli-kit/node/themes/types'
-import {test, describe, expect} from 'vitest'
+import {renderConfirmationPrompt, renderError, renderWarning} from '@shopify/cli-kit/node/ui'
+import {test, describe, expect, vi} from 'vitest'
+
+vi.mock('@shopify/cli-kit/node/ui')
 
 describe('themeComponent', () => {
   test('returns the ui for a theme', async () => {
@@ -26,6 +29,46 @@ describe('themesComponent', () => {
   })
 })
 
+describe('ensureDirectoryConfirmed', () => {
+  test('should prompt for confirmation when force flag is false', async () => {
+    vi.stubGlobal('process', {...process, stdout: {...process.stdout, isTTY: true}})
+    vi.mocked(renderConfirmationPrompt).mockResolvedValue(true)
+
+    const confirmed = await ensureDirectoryConfirmed(false)
+
+    expect(renderWarning).toHaveBeenCalledWith({
+      body: "It doesn't seem like you're running this command in a theme directory.",
+    })
+    expect(renderConfirmationPrompt).toHaveBeenCalledWith({
+      message: 'Do you want to proceed?',
+    })
+    expect(confirmed).toBe(true)
+  })
+
+  test('should not prompt for confirmation when called in a non-interactive environment', async () => {
+    vi.stubGlobal('process', {...process, stdout: {...process.stdout, isTTY: false}})
+
+    const confirmed = await ensureDirectoryConfirmed(false)
+
+    expect(renderWarning).toHaveBeenCalledWith({
+      body: "It doesn't seem like you're running this command in a theme directory.",
+    })
+    expect(confirmed).toBe(true)
+  })
+
+  describe('during a multi environment command run', () => {
+    test('should not prompt for confirmation and display an error', async () => {
+      const confirmed = await ensureDirectoryConfirmed(false, undefined, 'Production', true)
+
+      expect(renderError).toHaveBeenCalledWith({
+        headline: 'Environment: Production',
+        body: "It doesn't seem like you're running this command in a theme directory.",
+      })
+      expect(confirmed).toBe(false)
+    })
+  })
+})
+
 function theme(id: number) {
   return {id, name: `theme ${id}`} as Theme
 }

packages/theme/src/cli/utilities/theme-ui.ts

@@ -1,7 +1,7 @@
 import {recordEvent} from '@shopify/cli-kit/node/analytics'
 import {Theme} from '@shopify/cli-kit/node/themes/types'
 import {LIVE_THEME_ROLE} from '@shopify/cli-kit/node/themes/utils'
-import {Task, renderConfirmationPrompt, renderTasks, renderWarning} from '@shopify/cli-kit/node/ui'
+import {Task, renderConfirmationPrompt, renderError, renderTasks, renderWarning} from '@shopify/cli-kit/node/ui'
 
 export function themeComponent(theme: Theme) {
   return [
@@ -21,11 +21,21 @@ export function themesComponent(themes: Theme[]) {
 export async function ensureDirectoryConfirmed(
   force: boolean,
   message = "It doesn't seem like you're running this command in a theme directory.",
+  environment?: string,
+  multiEnvironment?: boolean,
 ) {
   if (force) {
     return true
   }
 
+  if (multiEnvironment) {
+    renderError({
+      headline: environment ? `Environment: ${environment}` : '',
+      body: message,
+    })
+    return false
+  }
+
   renderWarning({body: message})
 
   if (!process.stdout.isTTY) {