import { afterEach, describe, expect, test } from "bun:test";
import { mkdir, mkdtemp, rm, writeFile } from "node:fs/promises";
import os from "node:os";
import path from "node:path";
import { resolveSsrPageEntries } from "./implicitRootLayout";

const tempRoots: string[] = [];

const makeTempRoot = async () => {
  const root = await mkdtemp(path.join(os.tmpdir(), "akan-implicit-root-layout-"));
  tempRoots.push(root);
  return root;
};

const write = async (filePath: string, content: string) => {
  await mkdir(path.dirname(filePath), { recursive: true });
  await writeFile(filePath, content);
};

afterEach(async () => {
  await Promise.all(tempRoots.splice(0).map((root) => rm(root, { recursive: true, force: true })));
});

describe("resolveSsrPageEntries", () => {
  test("inherits root layout static exports for grouped root boundaries", async () => {
    const appRoot = await makeTempRoot();
    const pageRoot = path.join(appRoot, "page");
    const rootLayoutPath = path.join(pageRoot, "_layout.tsx");
    const groupedLayoutPath = path.join(pageRoot, "(home)", "_layout.tsx");

    await write(path.join(appRoot, "env", "env.client.ts"), "export const env = {};\n");
    await write(
      rootLayoutPath,
      'export const theme = "dark";\nexport const fonts = [{ name: "pretendard" }];\nexport const metadata = { title: "Root" };\n',
    );
    await write(
      groupedLayoutPath,
      'export function generateMetadata() { return { title: "Home" }; }\nexport default function Layout({ children }) { return children; }\n',
    );

    const entries = await resolveSsrPageEntries({
      appCwdPath: appRoot,
      appName: "demo",
      pageKeys: ["./_layout.tsx", "./(home)/_layout.tsx", "./(home)/_index.tsx"],
    });

    const groupedRoot = entries.find((entry) => entry.key === "./(home)/__root_layout.tsx");
    expect(groupedRoot).toBeDefined();
    expect(groupedRoot?.seedAbsPaths).toContain(rootLayoutPath);
    expect(groupedRoot?.seedAbsPaths).toContain(groupedLayoutPath);

    const generatedSource = await Bun.file(groupedRoot?.moduleAbsPath ?? "").text();
    expect(generatedSource).toContain('import * as inheritedLayout from "../../../page/_layout.tsx";');
    expect(generatedSource).not.toContain("<System.Provider");
    expect(generatedSource).toContain("export async function generateMetadata(props: PageProps)");
    expect(generatedSource).toContain("if (userLayout.generateMetadata) return userLayout.generateMetadata(props);");
    expect(generatedSource).toContain("if (userLayout.metadata !== undefined) return userLayout.metadata;");
    expect(generatedSource).toContain(
      "if (inheritedLayout.generateMetadata) return inheritedLayout.generateMetadata(props);",
    );
    expect(generatedSource).toContain("return inheritedLayout.metadata;");
    expect(generatedSource).not.toContain("Object.keys(userLayout.metadata)");
    expect(generatedSource).not.toContain("Object.keys(inheritedLayout.metadata)");
    expect(generatedSource).toContain("export const NotFound = userLayout.NotFound ?? inheritedLayout.NotFound;");
    expect(generatedSource).toContain("export const Error = userLayout.Error ?? inheritedLayout.Error;");
    expect(generatedSource).toContain("export const pageConfig = userLayout.pageConfig ?? inheritedLayout.pageConfig;");
    expect(generatedSource).toContain(
      "<UserLayout params={params} searchParams={searchParams}>{children}</UserLayout>",
    );
  });

  test("keeps system provider for grouped root boundaries without an ancestor root", async () => {
    const appRoot = await makeTempRoot();
    const pageRoot = path.join(appRoot, "page");
    const groupedLayoutPath = path.join(pageRoot, "(home)", "_layout.tsx");

    await write(path.join(appRoot, "env", "env.client.ts"), "export const env = {};\n");
    await write(groupedLayoutPath, 'export const theme = "dark";\n');

    const entries = await resolveSsrPageEntries({
      appCwdPath: appRoot,
      appName: "demo",
      pageKeys: ["./(home)/_layout.tsx", "./(home)/_index.tsx"],
    });

    const groupedRoot = entries.find((entry) => entry.key === "./(home)/__root_layout.tsx");
    expect(groupedRoot).toBeDefined();

    const generatedSource = await Bun.file(groupedRoot?.moduleAbsPath ?? "").text();
    expect(generatedSource).toContain("<System.Provider");
    expect(generatedSource).toContain("theme={userLayout.theme ?? inheritedLayout.theme}");
    expect(generatedSource).toContain('import { allDictionary } from "../dict/useDict.ts";');
    expect(generatedSource).toContain(
      'allDictionary={process.env.AKAN_PUBLIC_RENDER_ENV === "ssr" ? allDictionary : undefined}',
    );
    expect(generatedSource).not.toContain("getAllDictionary");
    expect(generatedSource).not.toContain("// export default function GeneratedLayout");

    const generatedDictMacro = await Bun.file(path.join(appRoot, ".akan", "generated", "dict", "useDict.ts")).text();
    expect(generatedDictMacro).toContain(
      'import { getAllDictionary } from "@apps/demo/lib/dict" with { type: "macro" };',
    );
    expect(generatedDictMacro).toContain("export const allDictionary = getAllDictionary();");
  });
});