wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-14 15:32:35 +08:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Add ws event handling

Browse Source
This commit is contained in:
pythongosssss 2023-10-14 13:09:39 +01:00
parent 3c20922b33
commit 6a95dfc9bb
1 changed files with 25 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
web/extensions/core/groupNode.js
View File

@ -1,4 +1,5 @@
import { app } from "../../scripts/app.js"; import { app } from "../../scripts/app.js";
import { api } from "../../scripts/api.js";
import { getWidgetType } from "../../scripts/widgets.js"; import { getWidgetType } from "../../scripts/widgets.js";
const IS_GROUP_NODE = Symbol(); const IS_GROUP_NODE = Symbol();
@ -295,6 +296,28 @@ const ext = {
new ConvertToGroupAction().addOption(options, options.length); new ConvertToGroupAction().addOption(options, options.length);
return options; return options;
}; };
api.addEventListener("executing", ({ detail }) => {
if (detail) {
const node = app.graph.getNodeById(detail);
if (!node) {
const split = detail.split(":");
if (split.length === 2) {
api.dispatchEvent(new CustomEvent("executing", { detail: split[0] }));
}
}
}
});
api.addEventListener("executed", ({ detail }) => {
const node = app.graph.getNodeById(detail.node);
if (!node) {
const split = detail.node.split(":");
if (split.length === 2) {
api.dispatchEvent(new CustomEvent("executed", { detail: { ...detail, node: split[0] } }));
}
}
});
}, },
async beforeConfigureGraph(graphData) { async beforeConfigureGraph(graphData) {
const groupNodes = graphData?.extra?.groupNodes; const groupNodes = graphData?.extra?.groupNodes;
@ -399,7 +422,7 @@ const ext = {
} }
// Reconnect outputs // Reconnect outputs
for (let outputId = 0; outputId < node.outputs.length; outputId++) { for (let outputId = 0; outputId < node.outputs?.length; outputId++) {
const output = node.outputs[outputId]; const output = node.outputs[outputId];
if (!output.links) continue; if (!output.links) continue;
for (const l of output.links) { for (const l of output.links) {
@ -418,7 +441,7 @@ const ext = {
return getExtraMenuOptions?.apply(this, arguments); return getExtraMenuOptions?.apply(this, arguments);
}; };
node.updateLink = function (link, innerNodes) { node.updateLink = function (link) {
// Replace the group node reference with the internal node // Replace the group node reference with the internal node
link = { ...link }; link = { ...link };
const output = slots.outputs[link.origin_slot]; const output = slots.outputs[link.origin_slot];