Spaces:
Running
on
L40S
Running
on
L40S
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]; | |
} | |
}, | |
}); | |