add edit workflow app

Author: Weixuanf

web/scripts/api.js

@@ -353,6 +353,7 @@ class ComfyApi extends EventTarget {
 	 * @returns { Promise<string, unknown> } A dictionary of id -> value
 	 */
 	async getSettings() {
+    return {};
 		return (await this.fetchApi("/settings")).json();
 	}
 

web/scripts/app.js

@@ -54,7 +54,7 @@ export class ComfyApp {
 	static clipspace_return_node = null;
 
 	constructor() {
-		this.ui = new ComfyUI(this);
+		// this.ui = new ComfyUI(this);
 		// this.logging = new ComfyLogging(this);
 
 		/**
@@ -1454,7 +1454,7 @@ export class ComfyApp {
 	 */
 	async setup() {
 		// await this.#setUser();
-		// await this.ui.settings.load();
+		await this.ui.settings.load();
 		await this.#loadExtensions();
 
 		// Create and mount the LiteGraph in the DOM
@@ -2239,7 +2239,8 @@ export async function loadModuleBasedOnPath() {
     app = new ComfyViewNodePackageApp();
   } else {
     // For any other path, import app.js and perform setup
-    app = new ComfyApp()
+    const {ComfyEditWorkflowApp} = await import("./comfyspace_editWorkflowApp.js");
+    app = new ComfyEditWorkflowApp()
   }
 }
 

web/scripts/comfyspace_editWorkflowApp.js

@@ -0,0 +1,43 @@
+
+import { ComfyApp } from "./app.js";
+import {serverNodeDefs} from '../../serverNodeDefs.js'
+import { ComfyUI, $el } from "./ui.js";
+
+const COMFYUI_CORE_EXTENSIONS = [
+  "/extensions/core/clipspace.js",
+  "/extensions/core/colorPalette.js",
+  "/extensions/core/contextMenuFilter.js",
+  "/extensions/core/dynamicPrompts.js",
+  "/extensions/core/editAttention.js",
+  "/extensions/core/groupNode.js",
+  "/extensions/core/groupNodeManage.js",
+  "/extensions/core/groupOptions.js",
+  "/extensions/core/invertMenuScrolling.js",
+  "/extensions/core/keybinds.js",
+  "/extensions/core/linkRenderMode.js",
+  "/extensions/core/maskeditor.js",
+  "/extensions/core/nodeTemplates.js",
+  "/extensions/core/noteNode.js",
+  "/extensions/core/rerouteNode.js",
+  "/extensions/core/saveImageExtraOutput.js",
+  "/extensions/core/slotDefaults.js",
+  "/extensions/core/snapToGrid.js",
+  "/extensions/core/undoRedo.js",
+  "/extensions/core/uploadImage.js",
+  "/extensions/core/widgetInputs.js",
+  "/extensions/dp.js",
+]
+
+export class ComfyEditWorkflowApp extends ComfyApp {
+  extensionFilesPath = COMFYUI_CORE_EXTENSIONS;
+  /** comfyspace cloud @type {string[]} */
+  nodeDefs = serverNodeDefs;
+
+  constructor() {
+    super();  
+		this.ui = new ComfyUI(this);
+  }
+  async setup() {
+    await super.setup();
+  }
+}

web/scripts/comfyspace_viewNodePackageApp.js

@@ -1,5 +1,7 @@
 
 import { ComfyApp } from "./app.js";
+import { ComfySettingsDialog } from "./ui/settings.js";
+
 const COMFYUI_CORE_EXTENSIONS = [
   // "/extensions/core/clipspace.js",
   "/extensions/core/colorPalette.js",
@@ -23,7 +25,15 @@ const COMFYUI_CORE_EXTENSIONS = [
   "/extensions/core/uploadImage.js",
   "/extensions/core/widgetInputs.js",
   "/extensions/dp.js",
-]
+];
+
+class CustomComfyUI {
+	constructor(app) {
+		this.app = app;
+		this.settings = new ComfySettingsDialog(app);
+  }
+}
+
 export class ComfyViewNodePackageApp extends ComfyApp {
   /** @type {{nodeDefs:string,jsFilePaths:string}} */
   nodePackage = null;
@@ -64,6 +74,7 @@ export class ComfyViewNodePackageApp extends ComfyApp {
 
   }
   async setup() {
+    this.ui = new CustomComfyUI(this);
     // to disable mousewheel zooming
     // LGraphCanvas.prototype.processMouseWheel =()=>{}
     this.setupMouseWheel();

web/scripts/comfyspace_viewWorkflowApp.js

@@ -1,5 +1,7 @@
 
 import { ComfyApp } from "./app.js";
+import {serverNodeDefs} from '../../serverNodeDefs.js'
+import { ComfySettingsDialog } from "./ui/settings.js";
 
 const COMFYUI_CORE_EXTENSIONS = [
   // "/extensions/core/clipspace.js",
@@ -25,12 +27,19 @@ const COMFYUI_CORE_EXTENSIONS = [
   "/extensions/core/widgetInputs.js",
   "/extensions/dp.js",
 ]
-
+class CustomComfyUI {
+	constructor(app) {
+		this.app = app;
+		this.settings = new ComfySettingsDialog(app);
+  }
+}
 export class ComfyViewWorkflowApp extends ComfyApp {
   #workflow = null
 
   async setup() {
+    this.ui = new CustomComfyUI(this);
     this.extensionFilesPath = COMFYUI_CORE_EXTENSIONS;
+    this.nodeDefs = serverNodeDefs;
 
     const queryParams = new URLSearchParams(window.location.search);
     const workflowVersionID = queryParams.get('workflowVersionID');
@@ -43,9 +52,15 @@ export class ComfyViewWorkflowApp extends ComfyApp {
         return response.json();
       })
       .then((data) => {
-        console.log("getCloudflowVersion data", data);
         const workflowVer = data.data;
-        this.nodeDefs = JSON.parse(workflowVer.nodeDefs);
+
+        if (workflowVer?.nodeDefs != null) {
+          const defs = JSON.parse(workflowVer.nodeDefs);
+          this.nodeDefs = {
+            ...this.nodeDefs,
+            ...defs,
+          }
+        }
         this.#workflow = JSON.parse(workflowVer.json);
       })
       .catch((error) => {

web/scripts/ui.js

@@ -294,7 +294,7 @@ export class ComfyUI {
 		this.lastQueueSize = 0;
 		this.queue = new ComfyList("Queue");
 		this.history = new ComfyList("History", "history", true);
-		return;
+		
 		api.addEventListener("status", () => {
 			this.queue.update();
 			this.history.update();