File size: 4,457 Bytes
0c85e62 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 | // SPDX-FileCopyrightText: 2025-2026 Kforge Labs <https://github.com/malkuthro/ComfyUI-Koolook>
// SPDX-License-Identifier: GPL-3.0-only
import { sortJsonValue } from "./workflows_store.js";
// Pure snapshot-comparison helpers for the Compare-mode sidebar (issue #181).
// No DOM, no globals, no I/O — every function takes plain data and returns
// plain data so it can be unit-tested by piping a scenario through Node. The
// Compare view renders the existing sidebar a second time and uses these
// results purely to tint comparison rows (green = only in comparison, red =
// same key, contents differ); it never mutates the working state.
// Partition two pick lists (arrays of string identifiers) against each other.
// Order follows the `working` list for `onlyWorking`/`shared` and the
// `comparison` list for `onlyComparison`. Membership is by set, so duplicates
// collapse to a single classification.
export function diffPicks(working, comparison) {
const workingSet = new Set(working);
const comparisonSet = new Set(comparison);
return {
onlyWorking: working.filter((id) => !comparisonSet.has(id)),
onlyComparison: comparison.filter((id) => !workingSet.has(id)),
shared: working.filter((id) => comparisonSet.has(id)),
};
}
// Flatten a workflows store ({ directories: { [name]: DirNode } }, the shape
// returned by getAllWorkflowsForExport / stored under snapshot.workflows) into
// a Map of full path ("Basics/txt2img", "Basics/Sub/wf") -> WorkflowEntry.
function collectWorkflows(node, prefix, out, skipArchived = false) {
const dirs = node && typeof node.directories === "object" && node.directories ? node.directories : {};
for (const dirName of Object.keys(dirs)) {
const child = dirs[dirName];
if (!child || typeof child !== "object") continue;
const childPrefix = `${prefix}${dirName}/`;
const wfs = child.workflows && typeof child.workflows === "object" ? child.workflows : {};
for (const wfName of Object.keys(wfs)) {
// Archived entries are old versions surfaced under the synthetic
// "Archive" folder. The diff skips them (skipArchived) so stale
// setups don't read as new/modified noise (#197); the copy-lookup
// path leaves them in so an archived version can still be copied.
if (skipArchived && wfs[wfName] && wfs[wfName].archived === true) continue;
out.set(`${childPrefix}${wfName}`, wfs[wfName]);
}
collectWorkflows(child, childPrefix, out, skipArchived);
}
return out;
}
// Two workflow graphs are equal when their canonical (key-order-normalized)
// JSON matches. Comparing the `graph` subtree alone ignores the sibling
// `savedAt` stamp, which is rewritten on every save and would otherwise make
// every re-saved workflow read as changed.
function graphsEqual(a, b) {
return JSON.stringify(sortJsonValue(a)) === JSON.stringify(sortJsonValue(b));
}
// Classify every workflow in the comparison store against the working store,
// keyed by full path:
// "new" -> present only in the comparison (green tint)
// "diff" -> present in both, graph differs ignoring savedAt (red tint)
// "same" -> present in both, identical graph (plain)
export function diffWorkflows(workingStore, comparisonStore) {
const working = collectWorkflows(workingStore, "", new Map(), true);
const comparison = collectWorkflows(comparisonStore, "", new Map(), true);
const status = {};
for (const [path, entry] of comparison) {
if (!working.has(path)) {
status[path] = "new";
} else {
status[path] = graphsEqual(entry.graph, working.get(path).graph) ? "same" : "diff";
}
}
return status;
}
// Resolve a single workflow entry in a raw store by full path
// ("Basics/txt2img", "Basics/Sub/wf"), or null if absent. Reuses the same
// flatten the diff uses, so lookups follow identical path semantics. Pure:
// reads only the passed store. The Compare-mode pull-in (#197) uses this to
// read a snapshot workflow's graph + tags so it can copy the entry into the
// live kit WITHOUT first applying the snapshot.
export function getWorkflowEntryFromStore(store, fullPath) {
if (!store || typeof store !== "object" || typeof fullPath !== "string") return null;
const all = collectWorkflows(store, "", new Map());
return all.has(fullPath) ? all.get(fullPath) : null;
}
|