62 lines
2.0 KiB
JavaScript
62 lines
2.0 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
|
|
app.registerExtension({
|
|
name: "Craftopia.SmartPromptController",
|
|
|
|
async nodeCreated(node) {
|
|
if (node.comfyClass !== "SmartPromptController") return;
|
|
|
|
// Canvas-drawn result display — clearly an output, not an input field
|
|
const statusWidget = node.addWidget("button", "_spc_result", null, () => {}, { serialize: false });
|
|
statusWidget.serialize = false;
|
|
statusWidget._text = "";
|
|
|
|
statusWidget.draw = function (ctx, node, widgetWidth, y, height) {
|
|
const margin = 14;
|
|
const innerW = widgetWidth - margin * 2;
|
|
|
|
ctx.save();
|
|
|
|
ctx.beginPath();
|
|
if (ctx.roundRect) ctx.roundRect(margin, y + 2, innerW, height - 4, 4);
|
|
else ctx.rect(margin, y + 2, innerW, height - 4);
|
|
ctx.fillStyle = "#111";
|
|
ctx.fill();
|
|
ctx.strokeStyle = this._text ? "#3a3a3a" : "#222";
|
|
ctx.lineWidth = 1;
|
|
ctx.stroke();
|
|
|
|
ctx.font = "11px sans-serif";
|
|
ctx.textAlign = "center";
|
|
ctx.textBaseline = "middle";
|
|
ctx.fillStyle = this._text ? "#f28f41" : "#666";
|
|
ctx.fillText(
|
|
this._text || "— run to see result —",
|
|
widgetWidth / 2,
|
|
y + height / 2
|
|
);
|
|
|
|
ctx.restore();
|
|
};
|
|
|
|
// Intercept onExecuted: update our widget, hide ComfyUI's default textarea
|
|
const origOnExecuted = node.onExecuted;
|
|
node.onExecuted = function (output) {
|
|
origOnExecuted?.call(this, output);
|
|
|
|
if (output?.text?.[0]) {
|
|
statusWidget._text = output.text[0];
|
|
|
|
// Collapse the default customtext widget ComfyUI adds for ui.text
|
|
for (const w of node.widgets) {
|
|
if (w !== statusWidget && w.type === "customtext") {
|
|
w.computeSize = () => [0, -4];
|
|
}
|
|
}
|
|
}
|
|
|
|
node.setDirtyCanvas(true);
|
|
};
|
|
},
|
|
});
|