From 743dc4879ab318ab3f483b9925b2ae00cea298a4 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <chenlei.hu@mail.utoronto.ca>
Date: Wed, 21 Aug 2024 10:31:07 -0400
Subject: [PATCH] Add setting to disable missing nodes dialog (#571)

* Add setting to disable missing nodes dialog

* nit

* nit
---
 src/scripts/app.ts         | 10 +++++++++-
 src/stores/settingStore.ts |  7 +++++++
 src/types/apiTypes.ts      |  1 +
 3 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/src/scripts/app.ts b/src/scripts/app.ts
index 4e2eb523c2..a6dfa1a4f6 100644
--- a/src/scripts/app.ts
+++ b/src/scripts/app.ts
@@ -2157,12 +2157,20 @@ export class ComfyApp {
   }
 
   showMissingNodesError(missingNodeTypes, hasAddedNodes = true) {
-    if (this.vueAppReady)
+    if (
+      !this.ui.settings.getSettingValue(
+        'Comfy.Workflow.ShowMissingNodesWarning'
+      )
+    )
+      return
+
+    if (this.vueAppReady) {
       showLoadWorkflowWarning({
         missingNodeTypes,
         hasAddedNodes,
         maximizable: true
       })
+    }
 
     this.logging.addEntry('Comfy.App', 'warn', {
       MissingNodes: missingNodeTypes
diff --git a/src/stores/settingStore.ts b/src/stores/settingStore.ts
index 6cedb01fff..f9abf76903 100644
--- a/src/stores/settingStore.ts
+++ b/src/stores/settingStore.ts
@@ -165,6 +165,13 @@ export const useSettingStore = defineStore('setting', {
         type: 'boolean',
         defaultValue: true
       })
+
+      app.ui.settings.addSetting({
+        id: 'Comfy.Workflow.ShowMissingNodesWarning',
+        name: 'Show missing nodes warning',
+        type: 'boolean',
+        defaultValue: true
+      })
     },
 
     set<K extends keyof Settings>(key: K, value: Settings[K]) {
diff --git a/src/types/apiTypes.ts b/src/types/apiTypes.ts
index 63d04b80c8..e31a4bac9b 100644
--- a/src/types/apiTypes.ts
+++ b/src/types/apiTypes.ts
@@ -402,6 +402,7 @@ const zSettings = z.record(z.any()).and(
       'Comfy.CustomColorPalettes': colorPalettesSchema,
       'Comfy.ConfirmClear': z.boolean(),
       'Comfy.DevMode': z.boolean(),
+      'Comfy.Workflow.ShowMissingNodesWarning': z.boolean(),
       'Comfy.DisableFloatRounding': z.boolean(),
       'Comfy.DisableSliders': z.boolean(),
       'Comfy.DOMClippingEnabled': z.boolean(),