import { createHash } from "node:crypto";
import { writeFile } from "node:fs/promises";
import { join } from "node:path";
import { expect } from "vitest";

import { keyedArchiveResultSchema } from "../../src/domain/apple/keyedArchive.js";
import { createTestTempDirectory } from "./temporaryDirectory.js";

/** Inert XML with producer-equivalent key spellings and unchanged ordinary data. */
export const keyedArchivePrototypeKeyFixture = async () => {
  const directory = await createTestTempDirectory("rea-keyed-xml-keys-");
  const path = join(directory, "archive.plist");
  const bytes = Buffer.from(`<?xml version="1.0" encoding="UTF-8"?>
<plist version="1.0"><dict>
  <key>$archiver</key><string>NSKeyedArchiver</string>
  <key>$version</key><integer>100000</integer>
  <key>$objects</key><array><string>$null</string><dict>
    <key >__proto__</key ><string>omitted outer value</string>
    <key>kept</key><string>a&#13;b&#x85;c&#x2028;d</string>
    <key>&#95;&#95;rea_prototype_key__</key><string>ordinary key</string>
    <key>nested</key><dict>
      <key><![CDATA[__proto__]]><!-- trailing comment --></key>
      <string>omitted nested value</string>
      <key>value</key><integer>7</integer>
    </dict>
  </dict></array>
  <key>$top</key><dict><key>root</key><dict>
    <key>CF$UID</key><integer>1</integer>
  </dict></dict>
</dict></plist>`);
  await writeFile(path, bytes);
  return { path, digest: createHash("sha256").update(bytes).digest("hex") };
};

/** Require original source identity, ordinary values and the exact omission count. */
export const expectKeyedArchivePrototypeKeys = (
  value: unknown,
  digest: string,
) => {
  const result = keyedArchiveResultSchema.parse(value);
  expect(result.archive_sha256).toBe(digest);
  expect(result.roots).toEqual({ root: { CF$UID: 1 } });
  expect(result.objects).toHaveLength(2);
  expect(result.objects[1]).toMatchObject({
    id: 1,
    kind: "dictionary",
    value: {
      kept: "a\rb\u0085c\u2028d",
      __rea_prototype_key__: "ordinary key",
      nested: { value: 7 },
    },
  });
  expect(result.limitations).toContain(
    "2 dictionary entries keyed __proto__ were omitted because REA results cannot represent that key.",
  );
  expect(JSON.stringify(result)).not.toContain("omitted outer value");
  expect(JSON.stringify(result)).not.toContain("omitted nested value");
};