105 lines
3.0 KiB
JavaScript
105 lines
3.0 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
|
|
app.registerExtension({
|
|
name: "Shellagent.extension",
|
|
async setup() {
|
|
window.parent.postMessage({
|
|
type: 'loaded'
|
|
}, '*');
|
|
window.addEventListener('message', (event) => {
|
|
if (event.data.type === 'save') {
|
|
app.graphToPrompt().then(data => {
|
|
window.parent.postMessage({
|
|
prompt: data?.output || {},
|
|
workflow: data?.workflow || {},
|
|
type: 'save'
|
|
}, "*");
|
|
});
|
|
}
|
|
if (event.data.type === 'load') {
|
|
app.loadGraphData(event.data.data, true, false);
|
|
}
|
|
if (event.data.type === 'load_default') {
|
|
// 使用FileReader读取JSON文件
|
|
fetch('extensions/ComfyUI-ShellAgent-Plugin/shellagent_default.json')
|
|
.then(response => response.blob())
|
|
.then(blob => {
|
|
const reader = new FileReader();
|
|
reader.onload = function(e) {
|
|
const json = JSON.parse(e.target.result);
|
|
app.loadGraphData(json, true, false);
|
|
};
|
|
reader.readAsText(blob);
|
|
})
|
|
.catch(error => console.error('加载默认JSON文件时出错:', error));
|
|
}
|
|
});
|
|
},
|
|
});app.registerExtension({
|
|
name: "Comfy.ShellAgent.UploadImage",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name === "ShellAgentPluginInputImage") {
|
|
if (
|
|
nodeData?.input?.required?.default_value?.[1]?.image_upload === true
|
|
) {
|
|
nodeData.input.required.upload = [
|
|
"IMAGEUPLOAD",
|
|
{ widget: "default_value" },
|
|
];
|
|
}
|
|
}
|
|
|
|
if (nodeData.name.indexOf('ShellAgentPlugin') === -1) {
|
|
addMenuHandler(nodeType, function (_, options) {
|
|
if (this.widgets) {
|
|
let toInput = [];
|
|
|
|
for (const w of this.widgets) {
|
|
if (["customtext"].indexOf(w.type) > -1) {
|
|
toInput.push({
|
|
content: `Input Text -> ${w.name}`,
|
|
callback: () => {
|
|
this.convertWidgetToInput(w);
|
|
const node = addNode("ShellAgentPluginInputText", this, { before: true });
|
|
node.connect(0, this, 1);
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
if (toInput.length) {
|
|
options.unshift({
|
|
content: "Convert to ShellAgent",
|
|
submenu: {
|
|
options: toInput
|
|
}
|
|
})
|
|
}
|
|
}
|
|
})
|
|
}
|
|
},
|
|
});
|
|
|
|
function addMenuHandler(nodeType, cb) {
|
|
const getOpts = nodeType.prototype.getExtraMenuOptions;
|
|
nodeType.prototype.getExtraMenuOptions = function () {
|
|
const r = getOpts.apply(this, arguments);
|
|
cb.apply(this, arguments);
|
|
return r;
|
|
};
|
|
}
|
|
|
|
function addNode(name, nextTo, options) {
|
|
options = { select: true, shiftY: 0, before: false, ...(options || {}) };
|
|
const node = LiteGraph.createNode(name);
|
|
app.graph.add(node);
|
|
node.pos = [
|
|
options.before ? nextTo.pos[0] - node.size[0] - 30 : nextTo.pos[0] + nextTo.size[0] + 30,
|
|
nextTo.pos[1] + options.shiftY,
|
|
];
|
|
if (options.select) {
|
|
app.canvas.selectNode(node, false);
|
|
}
|
|
return node;
|
|
} |