nodes/FaberVS-MultiModel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
FaberVS
2025-04-30 12:02:56 +03:00
committed by GitHub
parent 66cbac1b5e
commit aeaf1b1307
3 changed files with 176 additions and 222 deletions
Download Patch File Download Diff File Expand all files Collapse all files
js/ModelParamsNodeExt.js
+1 -41
View File
@@ -5,11 +5,8 @@ import { app } from "/scripts/app.js";
const TARGET_NODE_TYPE = "ModelParamsPipe"; // Цільовий тип вузла, який ми розширюємо
console.log("%c[ModelParamsNodeExt] JS файл завантажився! (v2-dynamic-inputs)", "color: white; background: #007acc; font-weight: bold; padding: 2px 8px; border-radius: 3px;");
// --- Loading js/ModelParamsNodeExt.js --- v2-dynamic-inputs
console.log(`--- Loading extension for ${TARGET_NODE_TYPE} ---`);
app.registerExtension({
name: "MultiModel.ModelParamsPipeNode.Notifier",
@@ -18,11 +15,8 @@ app.registerExtension({
// Додаємо базову ініціалізацію та прямий пошук вузлів
async setup() {
console.log(`[${TARGET_NODE_TYPE} Notifier] SETUP starting...`);
// Перевіряємо наявність сервісу одразу
if (app.HubNodesService) {
console.log(`[${TARGET_NODE_TYPE} Notifier] HubNodesService found during setup`);
} else {
console.error(`[${TARGET_NODE_TYPE} Notifier] HubNodesService NOT found during setup!`);
}
@@ -30,30 +24,23 @@ app.registerExtension({
// Зачекаємо, доки граф буде завантажено, і потім проскануємо його
const checkForGraph = () => {
if (app.graph) {
console.log(`[${TARGET_NODE_TYPE} Notifier] Graph found in setup`);
// Додаємо обробник зміни графу
if (app.graph.onNodeAdded && !app.graph._notifier_attached) {
const origNodeAdded = app.graph.onNodeAdded;
app.graph.onNodeAdded = function(node) {
console.log(`[${TARGET_NODE_TYPE} Notifier] Graph onNodeAdded called for node ID: ${node.id}, type: ${node.type}`);
if (origNodeAdded) {
origNodeAdded.call(this, node);
}
// Перевіряємо, чи це наш цільовий тип
if (node.type === TARGET_NODE_TYPE) {
console.log(`[${TARGET_NODE_TYPE} Notifier] Target node added to graph: ID ${node.id}`);
// Викликаємо наш обробник
const extension = app.extensions.find(ext => ext.name === "MultiModel.ModelParamsPipeNode.Notifier");
if (extension) {
extension.setupNodeMonitoring(node);
if (app.HubNodesService) {
console.log(`[${TARGET_NODE_TYPE} Notifier] Notifying service about new graph node: ID ${node.id}`);
app.HubNodesService.notifyHubAdded(node);
// Пробуємо різні методи оновлення інтерфейсу
@@ -70,22 +57,18 @@ app.registerExtension({
};
app.graph._notifier_attached = true;
console.log(`[${TARGET_NODE_TYPE} Notifier] Attached graph.onNodeAdded handler`);
}
// Зробимо сканування графу з невеликою затримкою для певності
setTimeout(() => {
if (!app.graph || !app.graph._nodes) {
console.log(`[${TARGET_NODE_TYPE} Notifier] No graph nodes found yet`);
return;
}
const targetNodes = app.graph._nodes.filter(node => node.type === TARGET_NODE_TYPE);
console.log(`[${TARGET_NODE_TYPE} Notifier] Initial graph scan found ${targetNodes.length} target nodes`);
const extension = app.extensions.find(ext => ext.name === "MultiModel.ModelParamsPipeNode.Notifier");
if (!extension) {
console.error(`[${TARGET_NODE_TYPE} Notifier] Extension not found in array!`);
return;
}
@@ -95,14 +78,11 @@ app.registerExtension({
// Якщо знайшли хоча б один вузол, спробуємо оновити інтерфейс
if (app.HubNodesService) {
console.log(`[${TARGET_NODE_TYPE} Notifier] Found existing node in scan: ID ${node.id}`);
}
}
// Якщо було знайдено вузли, форсуємо оновлення
if (targetNodes.length > 0 && app.HubNodesService) {
console.log(`[${TARGET_NODE_TYPE} Notifier] Forcing update after scan found ${targetNodes.length} nodes`);
if (typeof app.HubNodesService.forceUpdates === 'function') {
app.HubNodesService.forceUpdates();
} else if (typeof app.HubNodesService.updateWidgets === 'function') {
@@ -113,26 +93,20 @@ app.registerExtension({
}
}, 500);
} else {
console.log(`[${TARGET_NODE_TYPE} Notifier] Graph not available yet, will retry...`);
setTimeout(checkForGraph, 500);
}
};
// Запускаємо перевірку графу
checkForGraph();
console.log(`[${TARGET_NODE_TYPE} Notifier] SETUP completed`);
},
// Встановлення моніторингу title/видалення для вузла
setupNodeMonitoring(node, appInstance = app) {
if (!node) return false;
console.log(`[${TARGET_NODE_TYPE} Notifier] Setting up monitoring for node ID: ${node.id}`);
// --- Встановлюємо моніторинг title ---
if (!node.__title_monitored) {
console.log(`[${TARGET_NODE_TYPE} Notifier] Setting up title monitoring for node ID: ${node.id}`);
try {
const originalTitle = node.title;
@@ -148,11 +122,8 @@ app.registerExtension({
return this._monitored_title;
},
set: function(newTitle) {
console.log(`---> [${TARGET_NODE_TYPE} Notifier] title property CHANGED: "${this._monitored_title}" → "${newTitle}" for node ID: ${this.id}`);
// Якщо значення не змінилось, не виконуємо додаткові дії
if (this._monitored_title === newTitle) {
console.log(` Title not changed (same value), skipping notification.`);
return;
}
@@ -162,7 +133,6 @@ app.registerExtension({
// Повідомляємо сервіс про перейменування з невеликою затримкою
setTimeout(() => {
if (appInstance.HubNodesService) {
console.log(` Notifying service about rename for node ID: ${this.id}`);
appInstance.HubNodesService.notifyHubRenamed(this);
} else {
console.error(` HubNodesService not found! Cannot notify rename.`);
@@ -174,7 +144,6 @@ app.registerExtension({
});
node.__title_monitored = true;
console.log(`[${TARGET_NODE_TYPE} Notifier] Title property monitored for node ID: ${node.id}`);
} catch (error) {
console.error(`[${TARGET_NODE_TYPE} Notifier] Error setting up title monitoring:`, error);
}
@@ -186,11 +155,8 @@ app.registerExtension({
const originalOnRemoved = node.onRemoved;
node.onRemoved = function() {
console.log(`[${TARGET_NODE_TYPE} Notifier] Node removed: ID ${this.id}`);
// Повідомляємо сервіс про видалення
if (appInstance.HubNodesService) {
console.log(` Notifying service about removal for node ID: ${this.id}`);
appInstance.HubNodesService.notifyHubRemoved(this);
} else {
console.error(` HubNodesService not found! Cannot notify removal.`);
@@ -203,7 +169,6 @@ app.registerExtension({
};
node.onRemoved.__isNotifierOverridden = true;
console.log(`[${TARGET_NODE_TYPE} Notifier] onRemoved overridden for node ID: ${node.id}`);
} catch (error) {
console.error(`[${TARGET_NODE_TYPE} Notifier] Error overriding onRemoved:`, error);
}
@@ -214,24 +179,19 @@ app.registerExtension({
// Для логування проблемних випадків
async nodeCreated(node) {
console.log(`[${TARGET_NODE_TYPE} Notifier] nodeCreated called for node type: ${node ? node.type : 'undefined'}, ID: ${node ? node.id : 'undefined'}`);
},
// Для логування проблемних випадків
async onNodeAdded(node) {
console.log(`[${TARGET_NODE_TYPE} Notifier] onNodeAdded called for node type: ${node ? node.type : 'undefined'}, ID: ${node ? node.id : 'undefined'}`);
},
// Хук для обробки вузлів, що вже існують при завантаженні
async loadedGraphNode(node, appInstance) {
if (!node) return;
console.log(`[${TARGET_NODE_TYPE} Notifier] loadedGraphNode called for node type: ${node.type}, ID: ${node.id}`);
if (node.type === TARGET_NODE_TYPE) {
// Встановлюємо моніторинг для існуючого вузла
this.setupNodeMonitoring(node, appInstance);
}
}
});
console.log(`--- Extension for ${TARGET_NODE_TYPE} loaded ---`);
});