From 73396784a8ea572cd9b766ec1a90a86a419523c0 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <huchenlei@proton.me>
Date: Wed, 4 Dec 2024 12:45:38 -0800
Subject: [PATCH] Keep shim of several legacy components (#1796)

---
 src/scripts/app.ts           | 12 +-----------
 src/scripts/ui/menu/index.ts |  3 +++
 2 files changed, 4 insertions(+), 11 deletions(-)

diff --git a/src/scripts/app.ts b/src/scripts/app.ts
index 38acb87955..99b4e6400a 100644
--- a/src/scripts/app.ts
+++ b/src/scripts/app.ts
@@ -15,14 +15,12 @@ import {
   getLatentMetadata
 } from './pnginfo'
 import { createImageHost, calculateImageGrid } from './ui/imagePreview'
-import { DraggableList } from './ui/draggableList'
-import { applyTextReplacements, addStylesheet } from './utils'
 import type { ComfyExtension, MissingNodeType } from '@/types/comfy'
 import {
   type ComfyWorkflowJSON,
   type NodeId,
   validateComfyWorkflow
-} from '../types/comfyWorkflow'
+} from '@/types/comfyWorkflow'
 import type { ComfyNodeDef } from '@/types/apiTypes'
 import { adjustColor, ColorAdjustOptions } from '@/utils/colorUtil'
 import { ComfyAppMenu } from './ui/menu/index'
@@ -114,14 +112,6 @@ export class ComfyApp {
   static open_maskeditor = null
   static clipspace_return_node = null
 
-  // Force vite to import utils.ts as part of index.
-  // Force import of DraggableList.
-  static utils = {
-    applyTextReplacements,
-    addStylesheet,
-    DraggableList
-  }
-
   vueAppReady: boolean
   ui: ComfyUI
   extensions: ComfyExtension[]
diff --git a/src/scripts/ui/menu/index.ts b/src/scripts/ui/menu/index.ts
index 4a78a3959b..5c90bb54a0 100644
--- a/src/scripts/ui/menu/index.ts
+++ b/src/scripts/ui/menu/index.ts
@@ -7,6 +7,9 @@ import './menu.css'
 export { ComfyButton } from '../components/button'
 export { ComfySplitButton } from '../components/splitButton'
 export { ComfyPopup } from '../components/popup'
+export { ComfyAsyncDialog } from '@/scripts/ui/components/asyncDialog'
+export { DraggableList } from '@/scripts/ui/draggableList'
+export { applyTextReplacements, addStylesheet } from '@/scripts/utils'
 
 export class ComfyAppMenu {
   app: ComfyApp