35 lines
1.4 KiB
JavaScript
35 lines
1.4 KiB
JavaScript
import { app } from "/scripts/app.js";
|
|
import { applyTextReplacements } from "/scripts/utils.js";
|
|
|
|
app.registerExtension({
|
|
name: "Comfy.Easy.SaveImageExtraOutput",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (["easy imageSave", "easy fullkSampler", "easy kSampler", "easy kSamplerTiled","easy kSamplerInpainting", "easy kSamplerDownscaleUnet", "easy kSamplerSDTurbo","easy detailerFix"].includes(nodeData.name)) {
|
|
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
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const r = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
|
|
|
|
const widget = this.widgets.find((w) => w.name === "filename_prefix" || w.name === 'save_prefix');
|
|
widget.serializeValue = () => {
|
|
return applyTextReplacements(app, widget.value);
|
|
};
|
|
|
|
return r;
|
|
};
|
|
} else {
|
|
// When any other node is created add a property to alias the node
|
|
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");
|
|
}
|
|
|
|
return r;
|
|
};
|
|
}
|
|
},
|
|
});
|