Allow publish command to work with multiple environments

This commit uses the command function to pass multiple environments to the publish command.

Author: EvilGenius13

.changeset/hot-fishes-develop.md

@@ -0,0 +1,5 @@
+---
+'@shopify/theme': patch
+---
+
+Allow publish command to be called with multiple environments

packages/cli/oclif.manifest.json

@@ -6338,6 +6338,11 @@
       "hiddenAliases": [
       ],
       "id": "theme:publish",
+      "multiEnvironmentsFlags": [
+        "store",
+        "password",
+        "theme"
+      ],
       "pluginAlias": "@shopify/cli",
       "pluginName": "@shopify/cli",
       "pluginType": "core",

packages/theme/src/cli/commands/theme/publish.ts

@@ -1,10 +1,12 @@
-import {ensureThemeStore} from '../../utilities/theme-store.js'
 import ThemeCommand from '../../utilities/theme-command.js'
 import {themeFlags} from '../../flags.js'
 import {publish, renderArgumentsWarning} from '../../services/publish.js'
 import {Flags} from '@oclif/core'
 import {globalFlags} from '@shopify/cli-kit/node/cli'
-import {ensureAuthenticatedThemes} from '@shopify/cli-kit/node/session'
+import {OutputFlags} from '@oclif/core/interfaces'
+import {AdminSession} from '@shopify/cli-kit/node/session'
+
+type PublishFlags = OutputFlags<typeof Publish.flags>
 
 export default class Publish extends ThemeCommand {
   static summary = 'Set a remote theme as the live theme.'
@@ -37,16 +39,18 @@ If you want to publish your local theme, then you need to run \`shopify theme pu
     }),
   }
 
-  async run(): Promise<void> {
-    const {flags, argv} = await this.parse(Publish)
-    const store = await ensureThemeStore(flags)
-    const adminSession = await ensureAuthenticatedThemes(store, flags.password)
-    const themeId = flags.theme || argv[0]
+  static multiEnvironmentsFlags = ['store', 'password', 'theme']
 
-    if (argv.length > 0 && typeof argv[0] === 'string') {
-      renderArgumentsWarning(argv[0])
-    }
+  async command(flags: PublishFlags, adminSession: AdminSession) {
+    // Deprecated use of passing the theme id as an argument
+    // ex: `shopify theme publish <themeid>`
+    const positionalArgValue =
+      this.argv && this.argv.length > 0 && typeof this.argv[0] === 'string' ? this.argv[0] : undefined
 
-    await publish(adminSession, themeId, flags)
+    if (!flags.theme && positionalArgValue) {
+      flags.theme = positionalArgValue
+      renderArgumentsWarning(positionalArgValue)
+    }
+    await publish(adminSession, flags)
   }
 }

packages/theme/src/cli/services/publish.test.ts

@@ -33,7 +33,7 @@ describe('publish', () => {
     vi.mocked(renderConfirmationPrompt).mockResolvedValue(true)
 
     // When
-    await publish(session, '1', options)
+    await publish(session, options)
 
     // Then
     expect(renderConfirmationPrompt).toBeCalledWith({
@@ -66,7 +66,7 @@ describe('publish', () => {
     vi.mocked(renderConfirmationPrompt).mockResolvedValue(false)
 
     // When
-    await publish(session, '1', options)
+    await publish(session, options)
 
     // Then
     expect(renderConfirmationPrompt).toBeCalledWith({
@@ -84,7 +84,7 @@ describe('publish', () => {
     vi.mocked(renderConfirmationPrompt).mockResolvedValue(false)
 
     // When
-    await publish(session, '1', {...options, force: true})
+    await publish(session, {...options, force: true})
 
     // Then
     expect(renderConfirmationPrompt).not.toBeCalled()

packages/theme/src/cli/services/publish.ts

@@ -18,33 +18,38 @@ export function renderArgumentsWarning(id: string) {
   })
 }
 
-export async function publish(adminSession: AdminSession, themeId: string | undefined, options: {force: boolean}) {
-  const theme = await findOrSelectTheme(adminSession, {
+interface PublishServiceOptions {
+  theme: string | undefined
+  force: boolean
+}
+
+export async function publish(adminSession: AdminSession, options: PublishServiceOptions) {
+  const themeToPublish = await findOrSelectTheme(adminSession, {
     header: 'Select a theme to publish',
     filter: {
       development: false,
       live: false,
-      theme: themeId,
+      theme: options.theme,
     },
   })
 
-  const previewUrl = themePreviewUrl({...theme, role: 'live'} as Theme, adminSession)
+  const previewUrl = themePreviewUrl({...themeToPublish, role: 'live'} as Theme, adminSession)
 
   if (!options.force) {
     const accept = await renderConfirmationPrompt({
-      message: `Do you want to make '${theme.name}' the new live theme on ${adminSession.storeFqdn}?`,
-      confirmationMessage: `Yes, make '${theme.name}' the new live theme`,
+      message: `Do you want to make '${themeToPublish.name}' the new live theme on ${adminSession.storeFqdn}?`,
+      confirmationMessage: `Yes, make '${themeToPublish.name}' the new live theme`,
       cancellationMessage: 'No, cancel publish',
     })
     if (!accept) return
   }
 
-  await themePublish(theme.id, adminSession)
+  await themePublish(themeToPublish.id, adminSession)
 
   renderSuccess({
     body: [
       'The theme',
-      ...themeComponent(theme),
+      ...themeComponent(themeToPublish),
       'is now live at',
       {
         link: {