|
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]; |
|
} |
|
}, |
|
}); |
|
|