50 lines
2.3 KiB
JavaScript
50 lines
2.3 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
|
|
app.registerExtension({
|
|
name: "TextDrivenWorkflows.ResolutionSelector", // 拡張機能の登録名を変更
|
|
async nodeCreated(node) {
|
|
if (node.comfyClass === "TextDrivenResolutionSelector") { // 新しいノードIDをターゲットにする
|
|
const listWidget = node.widgets.find(w => w.name === "resolutions_list");
|
|
const comboWidget = node.widgets.find(w => w.name === "selected_resolution");
|
|
|
|
if (listWidget && comboWidget) {
|
|
const updateComboOptions = (textValue) => {
|
|
if (!textValue) return;
|
|
|
|
// Split the text block into individual rows
|
|
const lines = textValue.split("\n")
|
|
.map(line => line.trim())
|
|
.filter(line => line.length > 0);
|
|
|
|
// ComfyUI renders whatever is inside options.values.
|
|
// We feed the full human-readable string directly to the UI.
|
|
comboWidget.options.values = lines;
|
|
|
|
if (lines.length > 0 && !lines.includes(comboWidget.value)) {
|
|
comboWidget.value = lines[0];
|
|
}
|
|
|
|
if (node.graph) node.setDirtyCanvas(true, true);
|
|
};
|
|
|
|
// ★核心：表示を汚さず、サーバー送信時のみデータを「システム用の値」に化けさせる
|
|
comboWidget.serializeValue = function () {
|
|
const currentValue = comboWidget.value;
|
|
if (!currentValue) return "";
|
|
|
|
// Split by space as instructed, and send only the first token (e.g., "768x1024") to Python
|
|
const parts = currentValue.split(" ");
|
|
return parts[0].toLowerCase();
|
|
};
|
|
|
|
const originalCallback = listWidget.callback;
|
|
listWidget.callback = function (value) {
|
|
updateComboOptions(value);
|
|
if (originalCallback) originalCallback.apply(this, arguments);
|
|
};
|
|
|
|
updateComboOptions(listWidget.value);
|
|
}
|
|
}
|
|
}
|
|
}); |