mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-08-21 11:23:30 +08:00
46 lines
1.4 KiB
JavaScript
46 lines
1.4 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
import { BaseNodeModeChanger } from "./base_node_mode_changer.js";
|
|
import { NodeTypesString } from "./constants.js";
|
|
const MODE_BYPASS = 4;
|
|
const MODE_ALWAYS = 0;
|
|
class BypasserNode extends BaseNodeModeChanger {
|
|
constructor(title = BypasserNode.title) {
|
|
super(title);
|
|
this.comfyClass = NodeTypesString.FAST_BYPASSER;
|
|
this.modeOn = MODE_ALWAYS;
|
|
this.modeOff = MODE_BYPASS;
|
|
this.onConstructed();
|
|
}
|
|
async handleAction(action) {
|
|
if (action === "Bypass all") {
|
|
for (const widget of this.widgets || []) {
|
|
this.forceWidgetOff(widget, true);
|
|
}
|
|
}
|
|
else if (action === "Enable all") {
|
|
for (const widget of this.widgets || []) {
|
|
this.forceWidgetOn(widget, true);
|
|
}
|
|
}
|
|
else if (action === "Toggle all") {
|
|
for (const widget of this.widgets || []) {
|
|
this.forceWidgetToggle(widget, true);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
BypasserNode.exposedActions = ["Bypass all", "Enable all", "Toggle all"];
|
|
BypasserNode.type = NodeTypesString.FAST_BYPASSER;
|
|
BypasserNode.title = NodeTypesString.FAST_BYPASSER;
|
|
app.registerExtension({
|
|
name: "rgthree.Bypasser",
|
|
registerCustomNodes() {
|
|
BypasserNode.setUp();
|
|
},
|
|
loadedGraphNode(node) {
|
|
if (node.type == BypasserNode.title) {
|
|
node._tempWidth = node.size[0];
|
|
}
|
|
},
|
|
});
|