69 lines
2.9 KiB
JavaScript
69 lines
2.9 KiB
JavaScript
import { app } from "../../../scripts/app.js";
|
|
import { api } from "../../../scripts/api.js";
|
|
|
|
app.registerExtension({
|
|
name: "FlipStreamViewer.RunOnce",
|
|
async setup() {
|
|
api.addEventListener("FlipStreamViewer_run_once", async () => {
|
|
await api.interrupt();
|
|
await api.fetchApi("/queue", {
|
|
method: "POST",
|
|
body: JSON.stringify({ clear: true }),
|
|
});
|
|
await app.queuePrompt(0, 1);
|
|
});
|
|
}
|
|
});
|
|
|
|
app.registerExtension({
|
|
name: "FlipStreamViewer.FlipStreamChatJson",
|
|
async beforeRegisterNodeDef(t, d) {
|
|
const config = {
|
|
"FlipStreamGet": { in: 0, out: 2, range: 20, iname:'label', oname:'value', useo: true },
|
|
"FlipStreamChatJson": { in: 11, out: 3, range: 20, iname:'label', oname:'value', useo: true },
|
|
"FlipStreamTextConcat": { in: 2, out: 1, range: 20, iname:'text', oname:'', useo: false }
|
|
};
|
|
const CFG = config[d.name];
|
|
if (!CFG) return;
|
|
|
|
const sync = (n) => {
|
|
const w = n.widgets;
|
|
if (CFG.useo) {
|
|
while (n.outputs.length > w.length + CFG.out - CFG.in) {
|
|
if (n.outputs.length < CFG.out || n.outputs[n.outputs.length - 1]?.links?.length) break;
|
|
n.removeOutput(n.outputs.length - 1);
|
|
n.outputs_dirty = true;
|
|
}
|
|
}
|
|
if (w.length < CFG.in + CFG.range && (w.at(-1)?.value || n.inputs[w.length - 1]?.link !== null)) {
|
|
if (CFG.useo) {
|
|
n.addOutput(`${CFG.oname}${w.length - CFG.in}`, "*");
|
|
n.outputs_dirty = true;
|
|
}
|
|
const inputName = `${CFG.iname}${w.length - CFG.in}`;
|
|
n.addWidget("string", inputName, "", () => {}, { forceInput: true });
|
|
n.inputs_dirty = true;
|
|
}
|
|
while (w.length > CFG.in + 1 && (!w.at(-1).value && n.inputs[w.length - 1]?.link === null) && (!w.at(-2).value && n.inputs[w.length - 2]?.link === null)) {
|
|
if (n.outputs[w.length + CFG.out - CFG.in - 1]?.links?.length) break;
|
|
if (CFG.useo) {
|
|
n.removeOutput(w.length + CFG.out - CFG.in - 1);
|
|
n.outputs_dirty = true;
|
|
}
|
|
w.pop();
|
|
n.inputs_dirty = true;
|
|
}
|
|
n.setSize(n.computeSize());
|
|
n.setDirtyCanvas(true, true);
|
|
};
|
|
|
|
t.prototype.onNodeCreated = function() { sync(this); };
|
|
t.prototype.onWidgetChanged = function() { sync(this); };
|
|
t.prototype.onConnectionsChange = function(type, slot, connected, info) { requestAnimationFrame(() => sync(this)); };
|
|
|
|
t.prototype.onConfigure = function(c) {
|
|
c.widgets_values.forEach((v, i) => { if (this.widgets[i]) this.widgets[i].value = v; sync(this);});
|
|
};
|
|
}
|
|
});
|