jaidaken
f09734b0ee
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
66 lines
3.4 KiB
JavaScript
66 lines
3.4 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
import { rgthree } from "./rgthree.js";
|
|
import { getGroupNodes, getOutputNodes } from "./utils.js";
|
|
import { SERVICE as CONFIG_SERVICE } from "./services/config_service.js";
|
|
function showQueueNodesMenuIfOutputNodesAreSelected(existingOptions) {
|
|
if (CONFIG_SERVICE.getConfigValue("features.menu_queue_selected_nodes") === false) {
|
|
return;
|
|
}
|
|
const outputNodes = getOutputNodes(Object.values(app.canvas.selected_nodes));
|
|
const menuItem = {
|
|
content: `Queue Selected Output Nodes (rgthree) `,
|
|
className: "rgthree-contextmenu-item",
|
|
callback: () => {
|
|
rgthree.queueOutputNodes(outputNodes);
|
|
},
|
|
disabled: !outputNodes.length,
|
|
};
|
|
let idx = existingOptions.findIndex((o) => (o === null || o === void 0 ? void 0 : o.content) === "Outputs") + 1;
|
|
idx = idx || existingOptions.findIndex((o) => (o === null || o === void 0 ? void 0 : o.content) === "Align") + 1;
|
|
idx = idx || 3;
|
|
existingOptions.splice(idx, 0, menuItem);
|
|
}
|
|
function showQueueGroupNodesMenuIfGroupIsSelected(existingOptions) {
|
|
if (CONFIG_SERVICE.getConfigValue("features.menu_queue_selected_nodes") === false) {
|
|
return;
|
|
}
|
|
const group = rgthree.lastCanvasMouseEvent &&
|
|
(app.canvas.getCurrentGraph() || app.graph).getGroupOnPos(rgthree.lastCanvasMouseEvent.canvasX, rgthree.lastCanvasMouseEvent.canvasY);
|
|
const outputNodes = (group && getOutputNodes(getGroupNodes(group))) || null;
|
|
const menuItem = {
|
|
content: `Queue Group Output Nodes (rgthree) `,
|
|
className: "rgthree-contextmenu-item",
|
|
callback: () => {
|
|
outputNodes && rgthree.queueOutputNodes(outputNodes);
|
|
},
|
|
disabled: !(outputNodes === null || outputNodes === void 0 ? void 0 : outputNodes.length),
|
|
};
|
|
let idx = existingOptions.findIndex((o) => { var _a; return (_a = o === null || o === void 0 ? void 0 : o.content) === null || _a === void 0 ? void 0 : _a.startsWith("Queue Selected "); }) + 1;
|
|
idx = idx || existingOptions.findIndex((o) => (o === null || o === void 0 ? void 0 : o.content) === "Outputs") + 1;
|
|
idx = idx || existingOptions.findIndex((o) => (o === null || o === void 0 ? void 0 : o.content) === "Align") + 1;
|
|
idx = idx || 3;
|
|
existingOptions.splice(idx, 0, menuItem);
|
|
}
|
|
app.registerExtension({
|
|
name: "rgthree.QueueNode",
|
|
async beforeRegisterNodeDef(nodeType, nodeData) {
|
|
const getExtraMenuOptions = nodeType.prototype.getExtraMenuOptions;
|
|
nodeType.prototype.getExtraMenuOptions = function (canvas, options) {
|
|
var _a;
|
|
const extraOptions = (_a = getExtraMenuOptions === null || getExtraMenuOptions === void 0 ? void 0 : getExtraMenuOptions.call(this, canvas, options)) !== null && _a !== void 0 ? _a : [];
|
|
showQueueNodesMenuIfOutputNodesAreSelected(options);
|
|
showQueueGroupNodesMenuIfGroupIsSelected(options);
|
|
return extraOptions;
|
|
};
|
|
},
|
|
async setup() {
|
|
const getCanvasMenuOptions = LGraphCanvas.prototype.getCanvasMenuOptions;
|
|
LGraphCanvas.prototype.getCanvasMenuOptions = function (...args) {
|
|
const options = getCanvasMenuOptions.apply(this, [...args]);
|
|
showQueueNodesMenuIfOutputNodesAreSelected(options);
|
|
showQueueGroupNodesMenuIfGroupIsSelected(options);
|
|
return options;
|
|
};
|
|
},
|
|
});
|