MajoorWaldi-ComfyUI-Majoor-AssetsManager/js/utils/extractOutputFiles.js

export function extractOutputFiles(output) {
    const files = [];
    const seen = new Set();
    const MAX_ITEMS = 500;
    const MAX_DEPTH = 4;
    const MAX_SCAN_ITEMS = 2000;
    const SUPPORTED_PATH_EXT_RE =
        /\.(png|jpe?g|webp|gif|bmp|tiff?|avif|heic|heif|apng|hdr|svg|mp4|webm|mov|mkv|avi|m4v|mp3|wav|flac|ogg|glb|gltf|obj|fbx|ply|stl)$/i;
    const visited = typeof WeakSet !== "undefined" ? new WeakSet() : null;

    const buildKey = ({ type, subfolder, filename, path }) => {
        if (path) return `path|${String(path).trim().toLowerCase()}`;
        return `${type || ""}|${subfolder || ""}|${filename || ""}`.toLowerCase();
    };

    const pushItem = (item) => {
        if (!item) return;
        const rawPath =
            item.path || item.filepath || item.fullpath || item.fullPath || item.full_path;
        let filename = item.filename;
        if (!filename && rawPath) {
            const parts = String(rawPath).split(/[/\\\\]/);
            filename = parts[parts.length - 1];
        }
        if (!filename) return;
        filename = String(filename);
        if (!filename) return;
        const subfolder = item.subfolder || item.sub_folder || item.subFolder || "";
        const type = item.type || item.output_type || "output";
        const key = buildKey({ type, subfolder, filename, path: rawPath });
        if (seen.has(key)) return;
        seen.add(key);
        const entry = { filename, subfolder, type };
        if (rawPath) entry.path = rawPath;
        files.push(entry);
    };

    const pushPathString = (value) => {
        if (typeof value !== "string") return;
        const raw = value.trim().replace(/^['"]|['"]$/g, "");
        if (!raw || !SUPPORTED_PATH_EXT_RE.test(raw)) return;
        const parts = raw.split(/[/\\\\]/);
        const filename = parts[parts.length - 1];
        if (!filename) return;
        const key = buildKey({ type: "output", subfolder: "", filename, path: raw });
        if (seen.has(key)) return;
        seen.add(key);
        files.push({ filename, subfolder: "", type: "output", path: raw });
    };

    const addItems = (items) => {
        if (typeof items === "string") {
            pushPathString(items);
            return;
        }
        if (!Array.isArray(items)) return;
        let scanned = 0;
        for (const item of items) {
            if (files.length >= MAX_ITEMS || scanned >= MAX_SCAN_ITEMS) break;
            scanned += 1;
            if (item && typeof item === "object" && "filename" in item) {
                pushItem(item);
            } else if (typeof item === "string") {
                pushPathString(item);
            } else {
                walk(item, 1);
            }
        }
    };

    const walk = (node, depth) => {
        if (!node || depth > MAX_DEPTH || files.length >= MAX_ITEMS) return;
        if (typeof node !== "object") return;
        if (visited) {
            if (visited.has(node)) return;
            visited.add(node);
        }

        if (Array.isArray(node)) {
            let scanned = 0;
            for (const item of node) {
                if (files.length >= MAX_ITEMS || scanned >= MAX_SCAN_ITEMS) break;
                scanned += 1;
                if (item && typeof item === "object" && "filename" in item) {
                    pushItem(item);
                } else if (typeof item === "string") {
                    pushPathString(item);
                } else {
                    walk(item, depth + 1);
                }
            }
            return;
        }

        // Common ComfyUI output keys
        if (node.images || node.gifs || node.videos || node.meshes || node.mesh || node.audio) {
            addItems(node.images);
            addItems(node.gifs);
            addItems(node.videos);
            addItems(node.meshes);
            addItems(node.mesh);
            addItems(node.audio);
        }

        // Known wrappers
        if (node.output) walk(node.output, depth + 1);
        if (node.outputs) walk(node.outputs, depth + 1);
        if (node.result) walk(node.result, depth + 1);
        if (node.data) walk(node.data, depth + 1);

        // Shallow scan of object values for nested outputs
        if (depth < MAX_DEPTH) {
            for (const value of Object.values(node)) {
                if (files.length >= MAX_ITEMS) break;
                if (typeof value === "string") {
                    pushPathString(value);
                    continue;
                }
                if (!value || typeof value !== "object") continue;
                if (
                    value.images ||
                    value.gifs ||
                    value.videos ||
                    value.meshes ||
                    value.mesh ||
                    value.audio
                ) {
                    addItems(value.images);
                    addItems(value.gifs);
                    addItems(value.videos);
                    addItems(value.meshes);
                    addItems(value.mesh);
                    addItems(value.audio);
                    continue;
                }
                walk(value, depth + 1);
            }
        }
    };

    walk(output, 0);
    return files;
}