ComfyUI/web/extensions/core/saveImageExtraOutput.js

36 lines
1.3 KiB
JavaScript
Raw Normal View History

import { app } from "../../scripts/app.js";
import { applyTextReplacements } from "../../scripts/utils.js";
2023-03-29 21:09:11 +00:00
// Use widget values and dates in output filenames
app.registerExtension({
name: "Comfy.SaveImageExtraOutput",
async beforeRegisterNodeDef(nodeType, nodeData, app) {
if (nodeData.name === "SaveImage") {
const onNodeCreated = nodeType.prototype.onNodeCreated;
// When the SaveImage node is created we want to override the serialization of the output name widget to run our S&R
2023-03-29 21:09:11 +00:00
nodeType.prototype.onNodeCreated = function () {
const r = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
const widget = this.widgets.find((w) => w.name === "filename_prefix");
widget.serializeValue = () => {
return applyTextReplacements(app, widget.value);
2023-03-29 21:09:11 +00:00
};
return r;
};
} else {
// When any other node is created add a property to alias the node
2023-03-29 21:09:11 +00:00
const onNodeCreated = nodeType.prototype.onNodeCreated;
nodeType.prototype.onNodeCreated = function () {
const r = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
if (!this.properties || !("Node name for S&R" in this.properties)) {
this.addProperty("Node name for S&R", this.constructor.type, "string");
2023-03-29 21:09:11 +00:00
}
return r;
};
}
},
});