31 lines
1.0 KiB
JavaScript
31 lines
1.0 KiB
JavaScript
import { app, ComfyApp } from "../../scripts/app.js";
|
|
// import { api } from "../../scripts/api.js";
|
|
// import { ComfyWidgets } from "/scripts/widgets.js";
|
|
|
|
const BUTTON_NAME = "Open Mask Editor";
|
|
|
|
app.registerExtension({
|
|
name: "Comfy.D2.D2_LoadImage",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name !== "D2 Load Image") return;
|
|
|
|
const origOnNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const r = origOnNodeCreated ? origOnNodeCreated.apply(this) : undefined;
|
|
return r;
|
|
};
|
|
},
|
|
getCustomWidgets(app) {
|
|
return {
|
|
D2MASKEDITOR(node, inputName, inputData, app) {
|
|
const widget = node.addWidget("button", BUTTON_NAME, 0, () => {
|
|
ComfyApp.copyToClipspace(node);
|
|
ComfyApp.clipspace_return_node = node;
|
|
ComfyApp.open_maskeditor();
|
|
});
|
|
return widget;
|
|
},
|
|
};
|
|
},
|
|
});
|