From 7235a3dccf371e327453c45a839dd824f6e86794 Mon Sep 17 00:00:00 2001
From: filtered <176114999+webfiltered@users.noreply.github.com>
Date: Tue, 13 Aug 2024 01:16:21 +1000
Subject: [PATCH] Adds support for LiteGraph LiteGraph ES6 classes

This should continue to work unchanged on old versions of ComfyUI.

The changes just allow the set/get nodes to also work with the latest version of litegraph - conversion to proper classes is slowly going on.
---
 web/js/setgetnodes.js | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/web/js/setgetnodes.js b/web/js/setgetnodes.js
index 00ea0d8..bb2b19d 100644
--- a/web/js/setgetnodes.js
+++ b/web/js/setgetnodes.js
@@ -30,6 +30,7 @@ function setColorAndBgColor(type) {
 }
 let isAlertShown = false;
 let disablePrefix = app.ui.settings.getSettingValue("KJNodes.disablePrefix")
+const LGraphNode = LiteGraph.LGraphNode
 
 function showAlertWithThrottle(message, delay) {
     if (!isAlertShown) {
@@ -41,7 +42,7 @@ function showAlertWithThrottle(message, delay) {
 app.registerExtension({
 	name: "SetNode",
 	registerCustomNodes() {
-		class SetNode {
+		class SetNode extends LGraphNode {
 			defaultVisibility = true;
 			serialize_widgets = true;
 			drawConnection = false;
@@ -50,7 +51,8 @@ app.registerExtension({
 			canvas = app.canvas;
 			menuEntry = "Show connections";
 
-			constructor() {
+			constructor(title) {
+				super(title)
 				if (!this.properties) {
 					this.properties = {
 						"previousName": ""
@@ -370,7 +372,7 @@ app.registerExtension({
 app.registerExtension({
 	name: "GetNode",
 	registerCustomNodes() {
-		class GetNode {
+		class GetNode extends LGraphNode {
 
 			defaultVisibility = true;
 			serialize_widgets = true;
@@ -379,7 +381,8 @@ app.registerExtension({
 			currentSetter = null;
 			canvas = app.canvas;
 
-			constructor() {
+			constructor(title) {
+				super(title)
 				if (!this.properties) {
 					this.properties = {};
 				}