import { describe, expect, it } from "vitest";

import { collectDeliveredMediaUrls } from "./delivery-evidence.js";

describe("collectDeliveredMediaUrls attachment recursion", () => {

it("collects media URLs across nested attachments", () => {

const urls = collectDeliveredMediaUrls({

payloads: [

{

url: "https://example.com/root.png",

attachments: [

{ mediaUrl: "https://example.com/child.png" },

{ attachments: [{ filePath: "/tmp/grandchild.jpg" }] },

],

},

],

});

expect(urls.toSorted()).toEqual([

"/tmp/grandchild.jpg",

"https://example.com/child.png",

"https://example.com/root.png",

]);

});

it("does not overflow the stack on a self-referential attachments cycle", () => {

// Payloads arrive as in-process `unknown` objects; a malformed self-referential

// attachments chain previously recursed until the stack overflowed.

const cyclic: Record<string, unknown> = { url: "https://example.com/loop.png" };

cyclic.attachments = [cyclic];

let urls: string[] = [];

expect(() => {

urls = collectDeliveredMediaUrls({ payloads: [cyclic] });

}).not.toThrow();

expect(urls).toEqual(["https://example.com/loop.png"]);

});

it("does not overflow on a mutual attachments cycle", () => {

const a: Record<string, unknown> = { mediaUrl: "https://example.com/a.png" };

const b: Record<string, unknown> = { mediaUrl: "https://example.com/b.png" };

a.attachments = [b];

b.attachments = [a];

const urls = collectDeliveredMediaUrls({ payloads: [a] });

expect(urls.toSorted()).toEqual(["https://example.com/a.png", "https://example.com/b.png"]);

});

});

}

}

function collectMediaUrlsFromRecord(record: Record<string, unknown>, output: Set<string>) {

function collectMediaUrlsFromRecord(

record: Record<string, unknown>,

output: Set<string>,

// Payloads arrive as in-process `unknown` objects, so a malformed

// self-referential `attachments` chain would recurse until the stack

// overflows. Track visited records to bound the descent, matching

// redactStringsDeep in embedded-agent-subscribe.tools.ts.

seen = new WeakSet<object>(),

) {

if (seen.has(record)) {

return;

}

seen.add(record);

collectStringValues(record.mediaUrl, output);

collectStringValues(record.mediaUrls, output);

collectStringValues(record.path, output);

if (Array.isArray(attachments)) {

for (const attachment of attachments) {

if (attachment && typeof attachment === "object" && !Array.isArray(attachment)) {

collectMediaUrlsFromRecord(attachment as Record<string, unknown>, output);

collectMediaUrlsFromRecord(attachment as Record<string, unknown>, output, seen);

}

}

}