97 lines
4.3 KiB
JavaScript
97 lines
4.3 KiB
JavaScript
import { app } from "/scripts/app.js";
|
|
|
|
app.registerExtension({
|
|
name: "PaintingCoder.TextCombiner",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
// 更新节点检查条件，只保留原始名称和新命名空间格式
|
|
if (nodeData.name === "TextCombiner" || nodeData.name === "PaintingCoder::TextCombiner") {
|
|
// 添加自定义样式
|
|
const style = document.createElement('style');
|
|
style.textContent = `
|
|
// 节点基本样式
|
|
.comfy-node[data-type="TextCombiner"] {
|
|
background-color: rgb(63, 77, 85) !important;
|
|
color: rgb(255, 255, 255) !important;
|
|
}
|
|
|
|
// 标题栏样式
|
|
.comfy-node[data-type="TextCombiner"] .title-box {
|
|
background-color: rgb(45, 55, 65) !important;
|
|
color: rgb(33, 37, 41) !important;
|
|
}
|
|
|
|
// 悬停效果
|
|
.comfy-node[data-type="TextCombiner"]:hover {
|
|
background-color: rgba(63, 77, 85, 0.9) !important;
|
|
}
|
|
`;
|
|
document.head.appendChild(style);
|
|
|
|
// 保存原始的 onNodeCreated 方法
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
|
|
// 重写 onNodeCreated 方法
|
|
nodeType.prototype.onNodeCreated = function() {
|
|
const result = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
|
|
// 初始化连接计数器
|
|
this.lastConnectedCount = 0;
|
|
// 添加第一个文本输入
|
|
this.addInput("text_1", "STRING");
|
|
|
|
// 设置节点颜色
|
|
if (this.constructor.nodeData?.name === "TextCombiner") {
|
|
this.bgcolor = "#2e414a";
|
|
this.color = "#132832";
|
|
if (this.title_box) {
|
|
this.title_box.style.backgroundColor = "rgb(45, 55, 65)";
|
|
this.title_box.style.color = "#212529";
|
|
}
|
|
}
|
|
|
|
return result;
|
|
};
|
|
|
|
// 处理连接变化的方法
|
|
nodeType.prototype.onConnectionsChange = function(connectionType, slot, isConnected) {
|
|
// 只处理输入连接的变化
|
|
if (connectionType === 1) {
|
|
// 计算当前已连接的文本输入数量
|
|
const connectedCount = this.inputs.reduce((count, input) =>
|
|
input.name.startsWith('text_') ? (count + (input.link ? 1 : 0)) : count, 0);
|
|
|
|
// 如果连接数量发生变化
|
|
if (connectedCount !== this.lastConnectedCount) {
|
|
this.lastConnectedCount = connectedCount;
|
|
const targetCount = Math.max(connectedCount + 1, 1); // 确保至少保留一个输入
|
|
|
|
// 获取所有text输入
|
|
const textInputs = this.inputs
|
|
.map((input, index) => ({ index, name: input.name }))
|
|
.filter(input => input.name.startsWith('text_'));
|
|
|
|
// 添加新的输入槽位
|
|
if (textInputs.length < targetCount) {
|
|
const newIndex = textInputs.length + 1;
|
|
this.addInput(`text_${newIndex}`, "STRING");
|
|
}
|
|
|
|
// 只在确实需要移除时才移除多余的输入槽位
|
|
if (textInputs.length > targetCount) {
|
|
// 从后向前检查未连接的输入
|
|
for (let i = textInputs.length - 1; i >= targetCount; i--) {
|
|
const input = this.inputs[textInputs[i].index];
|
|
if (!input.link) {
|
|
this.removeInput(textInputs[i].index);
|
|
}
|
|
}
|
|
}
|
|
|
|
// 更新节点大小和画布
|
|
this.setSize(this.computeSize());
|
|
app.graph.setDirtyCanvas(true);
|
|
}
|
|
}
|
|
};
|
|
}
|
|
}
|
|
}); |