[backport cloud/1.41] fix: hide template selector after shared workflow accept (#9915)

Backport of #9913 to `cloud/1.41`

Automatically created by backport workflow.

Co-authored-by: Alexander Brown <drjkl@comfy.org>
Co-authored-by: Amp <amp@ampcode.com>

src/platform/workflow/sharing/composables/useSharedWorkflowUrlLoader.test.ts

@@ -78,6 +78,7 @@ vi.mock('vue-i18n', () => ({
 
 const mockShowLayoutDialog = vi.hoisted(() => vi.fn())
 const mockCloseDialog = vi.hoisted(() => vi.fn())
+const mockHideTemplateSelector = vi.hoisted(() => vi.fn())
 
 vi.mock('@/services/dialogService', () => ({
   useDialogService: () => ({
@@ -91,6 +92,12 @@ vi.mock('@/stores/dialogStore', () => ({
   })
 }))
 
+vi.mock('@/composables/useWorkflowTemplateSelectorDialog', () => ({
+  useWorkflowTemplateSelectorDialog: () => ({
+    hide: mockHideTemplateSelector
+  })
+}))
+
 function makePayload(
   overrides: Partial<SharedWorkflowPayload> = {}
 ): SharedWorkflowPayload {
@@ -173,6 +180,18 @@ describe('useSharedWorkflowUrlLoader', () => {
     )
   })
 
+  it('hides template selector when user confirms opening shared workflow', async () => {
+    mockQueryParams = { share: 'share-id-1' }
+    mockShowLayoutDialog.mockImplementation(() => {
+      resolveDialogWithConfirm(makePayload())
+    })
+
+    const { loadSharedWorkflowFromUrl } = useSharedWorkflowUrlLoader()
+    await loadSharedWorkflowFromUrl()
+
+    expect(mockHideTemplateSelector).toHaveBeenCalledTimes(1)
+  })
+
   it('does not load graph when user cancels dialog', async () => {
     mockQueryParams = { share: 'share-id-1' }
     mockShowLayoutDialog.mockImplementation(() => {
@@ -190,6 +209,18 @@ describe('useSharedWorkflowUrlLoader', () => {
     )
   })
 
+  it('does not hide template selector when user cancels shared workflow dialog', async () => {
+    mockQueryParams = { share: 'share-id-1' }
+    mockShowLayoutDialog.mockImplementation(() => {
+      resolveDialogWithCancel()
+    })
+
+    const { loadSharedWorkflowFromUrl } = useSharedWorkflowUrlLoader()
+    await loadSharedWorkflowFromUrl()
+
+    expect(mockHideTemplateSelector).not.toHaveBeenCalled()
+  })
+
   it('calls import when non-owned assets exist and user confirms', async () => {
     mockQueryParams = { share: 'share-id-1' }
     const payload = makePayload({
@@ -241,6 +272,18 @@ describe('useSharedWorkflowUrlLoader', () => {
     expect(mockImportPublishedAssets).not.toHaveBeenCalled()
   })
 
+  it('hides template selector when user chooses open-only', async () => {
+    mockQueryParams = { share: 'share-id-1' }
+    mockShowLayoutDialog.mockImplementation(() => {
+      resolveDialogWithOpenOnly(makePayload())
+    })
+
+    const { loadSharedWorkflowFromUrl } = useSharedWorkflowUrlLoader()
+    await loadSharedWorkflowFromUrl()
+
+    expect(mockHideTemplateSelector).toHaveBeenCalledTimes(1)
+  })
+
   it('shows toast on import failure and returns loaded-without-assets', async () => {
     mockQueryParams = { share: 'share-id-1' }
     const payload = makePayload({

src/platform/workflow/sharing/composables/useSharedWorkflowUrlLoader.ts

@@ -2,6 +2,7 @@ import { useToast } from 'primevue/usetoast'
 import { useI18n } from 'vue-i18n'
 import { useRoute, useRouter } from 'vue-router'
 
+import { useWorkflowTemplateSelectorDialog } from '@/composables/useWorkflowTemplateSelectorDialog'
 import OpenSharedWorkflowDialogContent from '@/platform/workflow/sharing/components/OpenSharedWorkflowDialogContent.vue'
 import type { SharedWorkflowPayload } from '@/platform/workflow/sharing/types/shareTypes'
 import {
@@ -35,6 +36,7 @@ export function useSharedWorkflowUrlLoader() {
   const workflowShareService = useWorkflowShareService()
   const dialogService = useDialogService()
   const dialogStore = useDialogStore()
+  const templateSelectorDialog = useWorkflowTemplateSelectorDialog()
   const SHARE_NAMESPACE = PRESERVED_QUERY_NAMESPACES.SHARE
 
   function isValidParameter(param: string): boolean {
@@ -133,6 +135,8 @@ export function useSharedWorkflowUrlLoader() {
       return 'cancelled'
     }
 
+    templateSelectorDialog.hide()
+
     const { payload } = result
     const workflowName = payload.name || t('openSharedWorkflow.dialogTitle')
     const nonOwnedAssets = payload.assets.filter((a) => !a.in_library)