From 0f1b72ce948e29b16f87771da4d50a6d050ce350 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <chenlei.hu@mail.utoronto.ca>
Date: Thu, 12 Sep 2024 17:00:13 +0900
Subject: [PATCH] Version (#140)

* Proper node width/height calculation

* 0.7.71
---
 package-lock.json     | 4 ++--
 package.json          | 2 +-
 public/litegraph.d.ts | 4 ++++
 src/litegraph.js      | 4 ++--
 4 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/package-lock.json b/package-lock.json
index f4508b7b2..17001a9bf 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "@comfyorg/litegraph",
-  "version": "0.7.70",
+  "version": "0.7.71",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "@comfyorg/litegraph",
-      "version": "0.7.70",
+      "version": "0.7.71",
       "license": "MIT",
       "devDependencies": {
         "@types/jest": "^29.5.12",
diff --git a/package.json b/package.json
index 9d05b0bac..c1b3251ba 100755
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@comfyorg/litegraph",
-  "version": "0.7.70",
+  "version": "0.7.71",
   "type": "module",
   "description": "A graph node editor similar to PD or UDK Blueprints. It works in an HTML5 Canvas and allows to export graphs to be included in applications.",
   "main": "./dist/litegraph.umd.js",
diff --git a/public/litegraph.d.ts b/public/litegraph.d.ts
index 845ed17d2..1800d4340 100644
--- a/public/litegraph.d.ts
+++ b/public/litegraph.d.ts
@@ -1126,6 +1126,10 @@ export declare class LGraphNode {
     /** Called by `LGraphCanvas.processContextMenu` */
     getMenuOptions?(graphCanvas: LGraphCanvas): ContextMenuItem[];
     getSlotMenuOptions?(slot: INodeSlot): ContextMenuItem[];
+
+    get width(): number;
+    get height(): number;
+    drawBadges?(ctx: CanvasRenderingContext2D, options: { gap?: number }): void;
 }
 
 export type LGraphNodeConstructor<T extends LGraphNode = LGraphNode> = {
diff --git a/src/litegraph.js b/src/litegraph.js
index 571a741e6..00ebb583c 100755
--- a/src/litegraph.js
+++ b/src/litegraph.js
@@ -4820,11 +4820,11 @@ const globalExport = {};
         }
 
         get width() {
-            return this.size[0];
+            return this.collapsed ? this._collapsed_width || LiteGraph.NODE_COLLAPSED_WIDTH : this.size[0];
         }
 
         get height() {
-            return this.size[1];
+            return this.collapsed ? LiteGraph.NODE_COLLAPSED_HEIGHT : this.size[1];
         }
 
         drawBadges(ctx, {gap = 2} = {}) {