nodes/BlakeOne-ComfyUI-NodeReset
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add support for widget callback

Browse Source
This commit is contained in:
BlakeOne
2024-04-18 12:11:47 -07:00
committed by GitHub
parent b9146caf01
commit 3f48df0fe9
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
js/extension.js
+11 -3
View File
@@ -5,17 +5,25 @@ app.registerExtension({
setup() {
const defaultWidgets = {};
const orig = LGraphCanvas.prototype.getNodeMenuOptions;
LGraphCanvas.prototype.getNodeMenuOptions = function(node) {
LGraphCanvas.prototype.getNodeMenuOptions = function(node) {
const options = orig.call(this, node);
options.push(null, {
content: "Reset",
content: "Restore default values",
callback: () => {
if (!defaultWidgets[node.type]) {
defaultWidgets[node.type] = LiteGraph.createNode(node.type, node.name, node.options).widgets;
}
defaultWidgets[node.type].forEach(defaultWidget => {
const widget = node.widgets.find(widget => widget.name === defaultWidget.name);
if (widget) widget.value = defaultWidget.value;
if (widget) {
widget.value = defaultWidget.value;
if (widget.options && widget.options.property && node.properties[widget.options.property] !== undefined) {
node.setProperty(widget.options.property, widget.value);
}
if (widget.callback) {
widget.callback(widget.value, this, node);
}
}
});
node.setDirtyCanvas(true, false);
}