修复-1 被替换的问题

2026-07-23 13:36:52 +08:00
parent f17aa7db8a
commit 151df9712e
5 changed files with 132 additions and 4 deletions
@@ -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;
+}