Dariusz L
514b7d5114
- Detect subgraph instance scope and distinguish top-level vs nested nodes - Aggregate inner resolved models for outer subgraph instances so menus show promoted and hidden models together - Patch both class prototypes and individual node instances when they define getExtraMenuOptions or getMenuOptions, preserving native menu hooks - Re-inject context menu after loadedGraphNode and queueMicrotask for subgraph-created nodes - Add tests for menu injection on overridden getMenuOptions and instance patching, plus subgraph scope and aggregation behavior
58 lines
2.1 KiB
JavaScript
58 lines
2.1 KiB
JavaScript
/**
|
|
* ComfyUI Model Resolver Extension - Frontend
|
|
*
|
|
* Provides a menu button and dialog interface for relinking missing models in workflows.
|
|
*/
|
|
|
|
import { app } from "../../../scripts/app.js";
|
|
import {
|
|
MODEL_RESOLVER_OPEN_COMMAND_ID,
|
|
MODEL_RESOLVER_OPEN_DEFAULT_KEYBINDING,
|
|
ModelResolver as ModelResolverClass,
|
|
} from "./resolver/model_resolver.js";
|
|
import { registerGlobalHelpers } from "./resolver/globals.js";
|
|
|
|
registerGlobalHelpers();
|
|
|
|
const modelResolver = new ModelResolverClass();
|
|
|
|
app.registerExtension({
|
|
name: "Model Resolver",
|
|
commands: [
|
|
{
|
|
id: MODEL_RESOLVER_OPEN_COMMAND_ID,
|
|
label: "Open Model Resolver",
|
|
function: () => modelResolver.activateResolverButton(),
|
|
},
|
|
],
|
|
keybindings: [
|
|
MODEL_RESOLVER_OPEN_DEFAULT_KEYBINDING,
|
|
],
|
|
setup: modelResolver.setup,
|
|
beforeRegisterNodeDef(nodeType, nodeData) {
|
|
modelResolver.configureWorkflowDependencyMarkerNodeType(nodeType, nodeData);
|
|
modelResolver.configureNodeContextMenu(nodeType);
|
|
},
|
|
nodeCreated(node) {
|
|
modelResolver.configureNodeContextMenu(node?.constructor);
|
|
modelResolver.configureNodeContextMenu(node);
|
|
queueMicrotask(() => modelResolver.configureNodeContextMenu(node));
|
|
modelResolver.configureWorkflowDependencyMarkerNode(node);
|
|
modelResolver.configureCustomNodeModelAdapter(node);
|
|
modelResolver.scheduleNodeContextMenuAnalysis();
|
|
},
|
|
loadedGraphNode(node) {
|
|
modelResolver.configureNodeContextMenu(node?.constructor);
|
|
modelResolver.configureNodeContextMenu(node);
|
|
queueMicrotask(() => modelResolver.configureNodeContextMenu(node));
|
|
modelResolver.configureWorkflowDependencyMarkerNode(node);
|
|
modelResolver.configureCustomNodeModelAdapter(node);
|
|
modelResolver.scheduleNodeContextMenuAnalysis();
|
|
},
|
|
afterConfigureGraph() {
|
|
modelResolver.configureWorkflowDependencyMarkerNodes();
|
|
modelResolver.checkAndOpenForMissingModels();
|
|
modelResolver.scheduleNodeContextMenuAnalysis(500);
|
|
},
|
|
});
|