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>
43 lines
1.8 KiB
JavaScript
43 lines
1.8 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
import { addConnectionLayoutSupport } from "./utils.js";
|
|
import { PowerPrompt } from "./base_power_prompt.js";
|
|
import { NodeTypesString } from "./constants.js";
|
|
let nodeData = null;
|
|
app.registerExtension({
|
|
name: "rgthree.PowerPrompt",
|
|
async beforeRegisterNodeDef(nodeType, passedNodeData) {
|
|
if (passedNodeData.name.includes("Power Prompt") && passedNodeData.name.includes("rgthree")) {
|
|
nodeData = passedNodeData;
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
onNodeCreated ? onNodeCreated.apply(this, []) : undefined;
|
|
this.powerPrompt = new PowerPrompt(this, passedNodeData);
|
|
};
|
|
addConnectionLayoutSupport(nodeType, app, [
|
|
["Left", "Right"],
|
|
["Right", "Left"],
|
|
]);
|
|
}
|
|
},
|
|
async loadedGraphNode(node) {
|
|
if (node.type === NodeTypesString.POWER_PROMPT) {
|
|
setTimeout(() => {
|
|
if (node.outputs[0].type === "STRING") {
|
|
if (node.outputs[0].links) {
|
|
node.outputs[3].links = node.outputs[3].links || [];
|
|
for (const link of node.outputs[0].links) {
|
|
node.outputs[3].links.push(link);
|
|
(node.graph || app.graph).links[link].origin_slot = 3;
|
|
}
|
|
node.outputs[0].links = null;
|
|
}
|
|
node.outputs[0].type = nodeData.output[0];
|
|
node.outputs[0].name = nodeData.output_name[0] || node.outputs[0].type;
|
|
node.outputs[0].color_on = undefined;
|
|
node.outputs[0].color_off = undefined;
|
|
}
|
|
}, 50);
|
|
}
|
|
},
|
|
});
|