67 lines
2.7 KiB
JavaScript
67 lines
2.7 KiB
JavaScript
import { app } from "../../../scripts/app.js";
|
|
import { ComfyWidgets } from "../../../scripts/widgets.js";
|
|
|
|
app.registerExtension({
|
|
name: "ComfyUI-String-Helper.ShowTranslateString",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name === "ShowTranslateString") {
|
|
// 创建显示文本的函数
|
|
function populate(texts) {
|
|
if (this.widgets) {
|
|
// 只删除带有特定前缀的 widgets
|
|
const widgetsToRemove = this.widgets.filter(w => w.name?.startsWith("translated_text_"));
|
|
for (const widget of widgetsToRemove) {
|
|
widget.onRemove?.();
|
|
const index = this.widgets.indexOf(widget);
|
|
if (index > -1) {
|
|
this.widgets.splice(index, 1);
|
|
}
|
|
}
|
|
}
|
|
|
|
// 确保 texts 是数组
|
|
const textArray = Array.isArray(texts) ? texts : [texts];
|
|
|
|
// 为每个文本创建一个新的 widget，使用特定的命名前缀
|
|
for (let i = 0; i < textArray.length; i++) {
|
|
const w = ComfyWidgets["STRING"](this, `translated_text_${i}`, ["STRING", { multiline: true }], app).widget;
|
|
w.inputEl.readOnly = true;
|
|
w.inputEl.style.opacity = 0.6;
|
|
w.value = textArray[i];
|
|
}
|
|
|
|
// 调整节点大小
|
|
requestAnimationFrame(() => {
|
|
const sz = this.computeSize();
|
|
if (sz[0] < this.size[0]) {
|
|
sz[0] = this.size[0];
|
|
}
|
|
if (sz[1] < this.size[1]) {
|
|
sz[1] = this.size[1];
|
|
}
|
|
this.onResize?.(sz);
|
|
app.graph.setDirtyCanvas(true, false);
|
|
});
|
|
}
|
|
|
|
// 处理执行结果
|
|
const onExecuted = nodeType.prototype.onExecuted;
|
|
nodeType.prototype.onExecuted = function (message) {
|
|
onExecuted?.apply(this, arguments);
|
|
if (message.translated_strings) {
|
|
populate.call(this, message.translated_strings);
|
|
}
|
|
};
|
|
|
|
// 处理配置
|
|
const onConfigure = nodeType.prototype.onConfigure;
|
|
nodeType.prototype.onConfigure = function () {
|
|
onConfigure?.apply(this, arguments);
|
|
if (this.widgets_values?.length) {
|
|
populate.call(this, this.widgets_values.slice(2));
|
|
}
|
|
};
|
|
}
|
|
}
|
|
});
|