DemonGatanjieuandClaude Opus 5.5
48e273f5c7
A node put on the canvas or a workflow opened and never run read as "never used". The page now notes what each opened workflow holds (usage_notes.js, POST /anomalous/usage/opened, kept apart from runs in the run record) and the open tabs just before a check. Packs whose nodes went unused but which also add to the page (buttons, menus, monitors) get their own group: unused nodes do not say the pack is. The packs count covers only packs that work through nodes alone. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
66 lines
2.8 KiB
JavaScript
66 lines
2.8 KiB
JavaScript
/**
|
|
* Tells the run record (api/usage_ledger.py) which nodes and models a workflow holds when it is
|
|
* opened, and which the open tabs hold just before a Tidy check: a workflow opened, or a node put
|
|
* on the canvas, counts as use even when it is never run. One post per distinct workflow content
|
|
* and page load.
|
|
*/
|
|
|
|
import { app } from "../../../scripts/app.js";
|
|
|
|
const MODEL_FILE = /\.(safetensors|ckpt|pt|pth|bin|sft|gguf)$/i;
|
|
const posted = new Set();
|
|
|
|
/** { types, models } a workflow (ComfyUI's saved form) holds, its subgraphs' nodes included. */
|
|
export function workflowUses(workflow) {
|
|
const types = new Set();
|
|
const models = new Set();
|
|
const subgraphs = workflow?.definitions?.subgraphs || [];
|
|
const subgraphIds = new Set(subgraphs.map(graph => graph?.id));
|
|
for (const graph of [workflow, ...subgraphs]) {
|
|
for (const node of graph?.nodes || []) {
|
|
if (typeof node?.type === 'string' && node.type && !subgraphIds.has(node.type)) types.add(node.type);
|
|
const stack = [node?.widgets_values];
|
|
while (stack.length) {
|
|
const value = stack.pop();
|
|
if (typeof value === 'string') {
|
|
if (MODEL_FILE.test(value.trim())) models.add(value.replace(/\\/g, '/').trim().toLowerCase());
|
|
} else if (value && typeof value === 'object') {
|
|
stack.push(...Object.values(value));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return { types: [...types].sort(), models: [...models].sort() };
|
|
}
|
|
|
|
async function note(workflow) {
|
|
const uses = workflowUses(workflow);
|
|
const body = JSON.stringify(uses);
|
|
if ((!uses.types.length && !uses.models.length) || posted.has(body)) return;
|
|
posted.add(body);
|
|
try {
|
|
const response = await fetch('/anomalous/usage/opened', { method: 'POST', headers: { 'Content-Type': 'application/json' }, body });
|
|
if (!response.ok) posted.delete(body); // another device looking on: nothing is noted from there
|
|
} catch (error) {
|
|
posted.delete(body);
|
|
console.warn('[AMB] Could not note the opened workflow.', error);
|
|
}
|
|
}
|
|
|
|
/** Notes every workflow ComfyUI opens (the one restored at startup too). */
|
|
export function watchOpenedWorkflows() {
|
|
const load = app.loadGraphData;
|
|
if (typeof load !== 'function') return;
|
|
app.loadGraphData = async function (...args) {
|
|
const result = await load.apply(this, args);
|
|
note(app.graph?.serialize?.());
|
|
return result;
|
|
};
|
|
}
|
|
|
|
/** Notes what the open tabs hold now (the canvas as it is, nodes put on it included). */
|
|
export async function noteOpenWorkflows() {
|
|
const tabs = app.extensionManager?.workflow?.openWorkflows || [];
|
|
await Promise.all([app.graph?.serialize?.(), ...tabs.map(tab => tab?.activeState)].filter(Boolean).map(note));
|
|
}
|