From 86ecc194ac990342429f24ec3dc1a349b877b5d8 Mon Sep 17 00:00:00 2001
From: "dr.lt.data" <Dr.Lt.Data@gmail.com>
Date: Fri, 26 Jan 2024 13:31:01 +0900
Subject: [PATCH] fix: badge drawer is overriding node.onDrawForeground

---
 __init__.py           | 2 +-
 js/comfyui-manager.js | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 48d8e29b..20cd4f69 100644
--- a/__init__.py
+++ b/__init__.py
@@ -29,7 +29,7 @@ except:
     print(f"[WARN] ComfyUI-Manager: Your ComfyUI version is outdated. Please update to the latest version.")
 
 
-version = [2, 3, 1]
+version = [2, 3, 2]
 version_str = f"V{version[0]}.{version[1]}" + (f'.{version[2]}' if len(version) > 2 else '')
 print(f"### Loading: ComfyUI-Manager ({version_str})")
 
diff --git a/js/comfyui-manager.js b/js/comfyui-manager.js
index 2452dbea..341f2a09 100644
--- a/js/comfyui-manager.js
+++ b/js/comfyui-manager.js
@@ -1271,7 +1271,10 @@ app.registerExtension({
 	async nodeCreated(node, app) {
 		if(!node.badge_enabled) {
 			node.getNickname = function () { return getNickname(node, node.comfyClass.trim()) };
-			const orig = node.__proto__.onDrawForeground;
+			let orig = node.onDrawForeground;
+			if(!orig)
+				orig = node.__proto__.onDrawForeground;
+
 			node.onDrawForeground = function (ctx) {
 				drawBadge(node, orig, arguments)
 			};