修复-1 被替换的问题

This commit is contained in:
2026-07-23 13:36:52 +08:00
parent f17aa7db8a
commit 151df9712e
5 changed files with 132 additions and 4 deletions
+61
View File
@@ -0,0 +1,61 @@
function getLink(graph, linkId) {
const links = graph?.links;
if (!links || linkId == null) return null;
return typeof links.get === "function" ? links.get(linkId) : links[linkId];
}
function collectUpstreamNodes(graph, targetNode) {
if (!targetNode) return new Set(graph?._nodes || []);
const found = new Set([targetNode]);
const pending = [targetNode];
while (pending.length) {
const current = pending.pop();
for (const input of current?.inputs || []) {
const link = getLink(graph, input?.link);
if (!link) continue;
const origin = graph?.getNodeById?.(link.origin_id);
if (!origin || found.has(origin)) continue;
found.add(origin);
pending.push(origin);
}
}
return found;
}
function safeWidgetValue(value) {
if (value == null || ["string", "number", "boolean"].includes(typeof value)) {
return value;
}
if (typeof value === "bigint") return String(value);
if (Array.isArray(value)) {
const result = value.map(safeWidgetValue);
return result.some((item) => item === undefined) ? undefined : result;
}
return undefined;
}
/**
* Build only the metadata subset needed by SmartSave's path preview.
* Reading widget.value directly is intentional: serializeValue and queue hooks
* may randomize seeds, upload files, or mutate third-party node state.
*/
export function buildReadOnlyPrompt(graph, targetNode = null) {
const included = collectUpstreamNodes(graph, targetNode);
const prompt = {};
for (const node of graph?._nodes || []) {
if (!included.has(node)) continue;
const inputs = {};
for (const widget of node?.widgets || []) {
if (!widget?.name || widget?.options?.serialize === false) continue;
const value = safeWidgetValue(widget.value);
if (value !== undefined) inputs[widget.name] = value;
}
prompt[String(node.id)] = {
class_type: node.comfyClass || node.type || "",
inputs,
};
}
return prompt;
}
+2 -2
View File
@@ -1,5 +1,6 @@
import { app } from "../../scripts/app.js";
import { api } from "../../scripts/api.js";
import { buildReadOnlyPrompt } from "./read_only_prompt.mjs";
const NODE_NAME = "SmartSaveImage";
const CSS_HREF = "extensions/SmartSaveImage/smart_save.css";
@@ -289,9 +290,8 @@ function buildPanel(node) {
statusLine.textContent = "正在计算…";
statusLine.className = "ssi-status";
try {
const graph = await app.graphToPrompt();
const payload = {
prompt: graph?.output || {},
prompt: buildReadOnlyPrompt(app.graph, node),
root_mode: widgetValue(node, "root_mode", "output"),
custom_root: widgetValue(node, "custom_root", ""),
folder_template: widgetValue(node, "folder_template", ""),