Add custom nodes, Civitai loras (LFS), and vast.ai setup script
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
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>
This commit is contained in:
156
custom_nodes/rgthree-comfy/src_web/comfyui/menu_auto_nest.ts
Normal file
156
custom_nodes/rgthree-comfy/src_web/comfyui/menu_auto_nest.ts
Normal file
@@ -0,0 +1,156 @@
|
||||
import type {
|
||||
LGraphNode,
|
||||
ContextMenu,
|
||||
IContextMenuOptions,
|
||||
IContextMenuValue,
|
||||
} from "@comfyorg/frontend";
|
||||
|
||||
import {app} from "scripts/app.js";
|
||||
import {rgthree} from "./rgthree.js";
|
||||
import {SERVICE as CONFIG_SERVICE} from "./services/config_service.js";
|
||||
|
||||
const SPECIAL_ENTRIES = [/^(CHOOSE|NONE|DISABLE|OPEN)(\s|$)/i, /^\p{Extended_Pictographic}/gu];
|
||||
|
||||
/**
|
||||
* Handles a large, flat list of string values given ContextMenu and breaks it up into subfolder, if
|
||||
* they exist. This is experimental and initially built to work for CheckpointLoaderSimple.
|
||||
*/
|
||||
app.registerExtension({
|
||||
name: "rgthree.ContextMenuAutoNest",
|
||||
async setup() {
|
||||
const logger = rgthree.newLogSession("[ContextMenuAutoNest]");
|
||||
|
||||
const existingContextMenu = LiteGraph.ContextMenu;
|
||||
|
||||
// @ts-ignore: TypeScript doesn't like this override.
|
||||
LiteGraph.ContextMenu = function (
|
||||
values: IContextMenuValue[],
|
||||
options: IContextMenuOptions<string, {rgthree_doNotNest: boolean}>,
|
||||
) {
|
||||
const threshold = CONFIG_SERVICE.getConfigValue("features.menu_auto_nest.threshold", 20);
|
||||
const enabled = CONFIG_SERVICE.getConfigValue("features.menu_auto_nest.subdirs", false);
|
||||
|
||||
// If we're not enabled, or are incompatible, then just call out safely.
|
||||
let incompatible: string | boolean = !enabled || !!options?.extra?.rgthree_doNotNest;
|
||||
if (!incompatible) {
|
||||
if (values.length <= threshold) {
|
||||
incompatible = `Skipping context menu auto nesting b/c threshold is not met (${threshold})`;
|
||||
}
|
||||
// If there's a rgthree_originalCallback, then we're nested and don't need to check things
|
||||
// we only expect on the first nesting.
|
||||
if (!(options.parentMenu?.options as any)?.rgthree_originalCallback) {
|
||||
// On first context menu, we require a callback and a flat list of options as strings.
|
||||
if (!options?.callback) {
|
||||
incompatible = `Skipping context menu auto nesting b/c a callback was expected.`;
|
||||
} else if (values.some((i) => typeof i !== "string")) {
|
||||
incompatible = `Skipping context menu auto nesting b/c not all values were strings.`;
|
||||
}
|
||||
}
|
||||
}
|
||||
if (incompatible) {
|
||||
if (enabled) {
|
||||
const [n, v] = logger.infoParts(
|
||||
"Skipping context menu auto nesting for incompatible menu.",
|
||||
);
|
||||
console[n]?.(...v);
|
||||
}
|
||||
return existingContextMenu.apply(this as any, [...arguments] as any);
|
||||
}
|
||||
|
||||
const folders: {[key: string]: IContextMenuValue[]} = {};
|
||||
const specialOps: IContextMenuValue[] = [];
|
||||
const folderless: IContextMenuValue[] = [];
|
||||
for (const value of values) {
|
||||
if (!value) {
|
||||
folderless.push(value);
|
||||
continue;
|
||||
}
|
||||
const newValue = typeof value === "string" ? {content: value} : Object.assign({}, value);
|
||||
(newValue as any).rgthree_originalValue = (value as any).rgthree_originalValue || value;
|
||||
const valueContent = newValue.content || "";
|
||||
const splitBy = valueContent.indexOf("/") > -1 ? "/" : "\\";
|
||||
const valueSplit = valueContent.split(splitBy);
|
||||
if (valueSplit.length > 1) {
|
||||
const key = valueSplit.shift()!;
|
||||
newValue.content = valueSplit.join(splitBy);
|
||||
folders[key] = folders[key] || [];
|
||||
folders[key]!.push(newValue);
|
||||
} else if (SPECIAL_ENTRIES.some((r) => r.test(valueContent))) {
|
||||
specialOps.push(newValue);
|
||||
} else {
|
||||
folderless.push(newValue);
|
||||
}
|
||||
}
|
||||
|
||||
const foldersCount = Object.values(folders).length;
|
||||
if (foldersCount > 0) {
|
||||
// Propogate the original callback down through the options.
|
||||
(options as any).rgthree_originalCallback =
|
||||
(options as any).rgthree_originalCallback ||
|
||||
(options.parentMenu?.options as any)?.rgthree_originalCallback ||
|
||||
options.callback;
|
||||
const oldCallback = (options as any)?.rgthree_originalCallback;
|
||||
options.callback = undefined;
|
||||
const newCallback = (
|
||||
item: IContextMenuValue,
|
||||
options: IContextMenuOptions,
|
||||
event: MouseEvent,
|
||||
parentMenu: ContextMenu | undefined,
|
||||
node: LGraphNode,
|
||||
) => {
|
||||
oldCallback?.((item as any)?.rgthree_originalValue!, options, event, undefined, node);
|
||||
};
|
||||
const [n, v] = logger.infoParts(`Nested folders found (${foldersCount}).`);
|
||||
console[n]?.(...v);
|
||||
const newValues: IContextMenuValue[] = [];
|
||||
for (const [folderName, folderValues] of Object.entries(folders)) {
|
||||
newValues.push({
|
||||
content: `📁 ${folderName}`,
|
||||
has_submenu: true,
|
||||
callback: () => {
|
||||
/* no-op, use the item callback. */
|
||||
},
|
||||
submenu: {
|
||||
options: folderValues.map((value) => {
|
||||
value!.callback = newCallback;
|
||||
return value;
|
||||
}),
|
||||
},
|
||||
});
|
||||
}
|
||||
values = ([] as IContextMenuValue[]).concat(
|
||||
specialOps.map((f) => {
|
||||
if (typeof f === "string") {
|
||||
f = {content: f};
|
||||
}
|
||||
f!.callback = newCallback;
|
||||
return f;
|
||||
}),
|
||||
newValues,
|
||||
folderless.map((f) => {
|
||||
if (typeof f === "string") {
|
||||
f = {content: f};
|
||||
}
|
||||
f!.callback = newCallback;
|
||||
return f;
|
||||
}),
|
||||
);
|
||||
}
|
||||
if (options.scale == null) {
|
||||
options.scale = Math.max(app.canvas.ds?.scale || 1, 1);
|
||||
}
|
||||
|
||||
const oldCtrResponse = existingContextMenu.call(this as any, values, options as any);
|
||||
// For some reason, LiteGraph calls submenus with "this.constructor" which no longer allows
|
||||
// us to continue building deep nesting, as well as skips many other extensions (even
|
||||
// ComfyUI's core extensions like translations) from working on submenus. It also removes
|
||||
// search, etc. While this is a recent-ish issue, I can't seem to find the culpit as it looks
|
||||
// like old litegraph always did this. Perhaps changing it to a Class? Anyway, this fixes it;
|
||||
// Hopefully without issues.
|
||||
if ((oldCtrResponse as any)?.constructor) {
|
||||
(oldCtrResponse as any).constructor = LiteGraph.ContextMenu;
|
||||
}
|
||||
return this;
|
||||
};
|
||||
},
|
||||
});
|
||||
Reference in New Issue
Block a user