From b0bcb0e062e9d34fe84ca1419448b19e46769978 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <chenlei.hu@mail.utoronto.ca>
Date: Fri, 9 Aug 2024 13:18:51 -0400
Subject: [PATCH] Remove unnecessary compatibility layer (#66)

---
 src/litegraph.js    | 12 ++----------
 test/LGraph.test.ts | 29 +++++++----------------------
 2 files changed, 9 insertions(+), 32 deletions(-)

diff --git a/src/litegraph.js b/src/litegraph.js
index 42f972baf..6598e3d94 100755
--- a/src/litegraph.js
+++ b/src/litegraph.js
@@ -2226,14 +2226,7 @@ const globalExport = {};
             //TODO
         }
     }
-    // Backward Compatibility for extending the functionality via prototype.
-    function LegacyLGraph(...args) {
-        return new LGraph(...args);
-    }
-    LegacyLGraph.prototype = LGraph.prototype;
-
-    globalThis.LGraph = LGraph;
-    globalThis.LegacyLGraph = LiteGraph.LGraph = LegacyLGraph;
+    globalThis.LGraph = LiteGraph.LGraph = LGraph;
 
     //this is the class in charge of storing link information
     function LLink(id, type, origin_id, origin_slot, target_id, target_slot) {
@@ -14455,8 +14448,7 @@ LGraphNode.prototype.executeAction = function(action)
 })(globalExport)
 
 export const LiteGraph = globalExport.LiteGraph;
-export const LGraphES6 = globalExport.LGraph;
-export const LGraph = globalExport.LegacyLGraph;
+export const LGraph = globalExport.LGraph;
 export const LLink = globalExport.LLink;
 export const LGraphNode = globalExport.LGraphNode;
 export const LGraphGroup = globalExport.LGraphGroup;
diff --git a/test/LGraph.test.ts b/test/LGraph.test.ts
index 5977b4aca..ff924dcff 100644
--- a/test/LGraph.test.ts
+++ b/test/LGraph.test.ts
@@ -1,39 +1,24 @@
 import {
-  LGraphES6 as LGraph,
-  LGraph as LegacyLGraph,
+  LGraph,
   LiteGraph,
 } from "../dist/litegraph.es.js";
 
 describe("LegacyLGraph Compatibility Layer", () => {
-  test("LegacyLGraph can be instantiated", () => {
-    const graph = new LegacyLGraph({extra: "TestGraph"});
+  test("LGraph can be instantiated", () => {
+    const graph = new LGraph({extra: "TestGraph"});
     expect(graph).toBeInstanceOf(LGraph);
-    expect(graph).toBeInstanceOf(LegacyLGraph);
     expect(graph.extra).toBe("TestGraph");
   });
 
-  test("LegacyLGraph can be extended via prototype", () => {
-    LegacyLGraph.prototype.newMethod = function () {
+  test("LGraph can be extended via prototype", () => {
+    const graph = new LGraph();
+    LGraph.prototype.newMethod = function () {
       return "New method added via prototype";
     };
-
-    const graph = new LegacyLGraph();
     expect(graph.newMethod()).toBe("New method added via prototype");
   });
 
-  test("Extensions to LegacyLGraph affect LGraph instances", () => {
-    LegacyLGraph.prototype.anotherMethod = function () {
-      return "Another method";
-    };
-
-    const legacyGraph = new LegacyLGraph();
-    const normalGraph = new LGraph();
-
-    expect(legacyGraph.anotherMethod()).toBe("Another method");
-    expect(normalGraph.anotherMethod()).toBe("Another method");
-  });
-
   test("LegacyLGraph is correctly assigned to LiteGraph", () => {
-    expect(LiteGraph.LGraph).toBe(LegacyLGraph);
+    expect(LiteGraph.LGraph).toBe(LGraph);
   });
 });