From dbb9a9acd4fa9d55599faf4841d1bd31df3f3535 Mon Sep 17 00:00:00 2001
From: Austin Mroz <austin@comfy.org>
Date: Tue, 16 Sep 2025 12:25:00 -0500
Subject: [PATCH] Move proxyWidgets out of extensions dir

---
 src/extensions/core/index.ts                    | 1 -
 src/{extensions/core => scripts}/proxyWidget.ts | 2 +-
 src/scripts/widgets.ts                          | 1 +
 3 files changed, 2 insertions(+), 2 deletions(-)
 rename src/{extensions/core => scripts}/proxyWidget.ts (99%)

diff --git a/src/extensions/core/index.ts b/src/extensions/core/index.ts
index 00c925935..5354ef4e9 100644
--- a/src/extensions/core/index.ts
+++ b/src/extensions/core/index.ts
@@ -11,7 +11,6 @@ import './maskeditor'
 import './nodeTemplates'
 import './noteNode'
 import './previewAny'
-import './proxyWidget'
 import './rerouteNode'
 import './saveImageExtraOutput'
 import './saveMesh'
diff --git a/src/extensions/core/proxyWidget.ts b/src/scripts/proxyWidget.ts
similarity index 99%
rename from src/extensions/core/proxyWidget.ts
rename to src/scripts/proxyWidget.ts
index 3b04a3a2c..c5aad76a6 100644
--- a/src/extensions/core/proxyWidget.ts
+++ b/src/scripts/proxyWidget.ts
@@ -7,10 +7,10 @@ import { DOMWidgetImpl } from '@/scripts/domWidget'
 import { useDomWidgetStore } from '@/stores/domWidgetStore'
 import { useCanvasStore } from '@/stores/graphStore'
 
-const canvasStore = useCanvasStore()
 const originalConfigureAfterSlots =
   SubgraphNode.prototype._internalConfigureAfterSlots
 SubgraphNode.prototype._internalConfigureAfterSlots = function () {
+  const canvasStore = useCanvasStore()
   const subgraphNode = this
   //Must give value to proxyWidgets prior to injecting or it won't serialize
   subgraphNode.properties.proxyWidgets ??= '[]'
diff --git a/src/scripts/widgets.ts b/src/scripts/widgets.ts
index 357dc2bf1..50944381a 100644
--- a/src/scripts/widgets.ts
+++ b/src/scripts/widgets.ts
@@ -29,6 +29,7 @@ import { useSettingStore } from '@/stores/settingStore'
 import type { ComfyApp } from './app'
 import './domWidget'
 import './errorNodeWidgets'
+import './proxyWidget'
 
 export type ComfyWidgetConstructorV2 = (
   node: LGraphNode,