xinyun/ComfyUI-Manager
1
0
Fork 0
mirror of https://git.datalinker.icu/ltdrdata/ComfyUI-Manager synced 2025-12-09 22:24:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix the original missing menu options

Browse Source
This commit is contained in:
johnqiao 2023-11-24 19:06:40 -07:00
parent 355991ecc9
commit e32072ab34
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
js/comfyui-manager.js
View File

@ -766,10 +766,19 @@ app.registerExtension({
},
_addExtraNodeContextMenu(node, app) {
node.prototype.getExtraMenuOptions = function (_, options) {
if (isOutputNode(node)) {
const origGetExtraMenuOptions = node.prototype.getExtraMenuOptions;
node.prototype.getExtraMenuOptions = function (_, options) {
origGetExtraMenuOptions?.apply?.(this, arguments);
if (isOutputNode(node)) {
const { potential_outputs } = getPotentialOutputsAndOutputNodes([this]);
const hasOutput = potential_outputs.length > 0;
// Check if the previous menu option is `null`. If it's not,
// then we need to add a `null` as a separator.
if (options[options.length - 1] !== null) {
options.push(null);
}
options.push({
content: "🏞️ Share Output",
disabled: !hasOutput,
@ -793,6 +802,6 @@ app.registerExtension({
}
}, null);
}
}
}
},
});