From 9b363cc9bd6ee086b8659300011d934ef76ae56a Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Tue, 28 Nov 2023 02:23:59 +0900
Subject: [PATCH] enhance: badge - hide built-in option is added

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

diff --git a/__init__.py b/__init__.py
index 9a9f0f6a..c9ca9bac 100644
--- a/__init__.py
+++ b/__init__.py
@@ -17,7 +17,7 @@ import http.client
 import re
 import signal
 
-version = "V1.5.1"
+version = "V1.5.2"
 print(f"### Loading: ComfyUI-Manager ({version})")
 
 
diff --git a/js/comfyui-manager.js b/js/comfyui-manager.js
index dabd011f..93c9235e 100644
--- a/js/comfyui-manager.js
+++ b/js/comfyui-manager.js
@@ -425,7 +425,9 @@ class ManagerMenuDialog extends ComfyDialog {
 		let badge_combo = document.createElement("select");
 		badge_combo.appendChild($el('option', { value: 'none', text: 'Badge: None' }, []));
 		badge_combo.appendChild($el('option', { value: 'nick', text: 'Badge: Nickname' }, []));
+		badge_combo.appendChild($el('option', { value: 'nick_hide', text: 'Badge: Nickname (hide built-in)' }, []));
 		badge_combo.appendChild($el('option', { value: 'id_nick', text: 'Badge: #ID Nickname' }, []));
+		badge_combo.appendChild($el('option', { value: 'id_nick_hide', text: 'Badge: #ID Nickname (hide built-in)' }, []));
 
 		api.fetchApi('/manager/badge_mode')
 			.then(response => response.text())
@@ -722,14 +724,19 @@ app.registerExtension({
 
 			if (!this.flags.collapsed && badge_mode != 'none' && nodeType.title_mode != LiteGraph.NO_TITLE) {
 				let text = "";
-				if (badge_mode == 'id_nick')
+				if (badge_mode.startsWith('id_nick'))
 					text = `#${this.id} `;
 
 				if (nicknames[nodeData.name.trim()]) {
 					let nick = nicknames[nodeData.name.trim()];
 
 					if (nick == 'ComfyUI') {
-						nick = "🦊"
+						if(badge_mode.endsWith('hide')) {
+							nick = "";
+						}
+						else {
+							nick = "🦊"
+						}
 					}
 
 					if (nick.length > 25) {
@@ -772,14 +779,19 @@ app.registerExtension({
 
 				if (!this.flags.collapsed && badge_mode != 'none') {
 					let text = "";
-					if (badge_mode == 'id_nick')
+					if (badge_mode.startsWith('id_nick'))
 						text = `#${this.id} `;
 
 					if (nicknames[node.type.trim()]) {
 						let nick = nicknames[node.type.trim()];
 
 						if (nick == 'ComfyUI') {
-							nick = "🦊"
+							if(badge_mode.endsWith('hide')) {
+								nick = "";
+							}
+							else {
+								nick = "🦊"
+							}
 						}
 
 						if (nick.length > 25) {