import { mkdtemp, rm, writeFile } from "node:fs/promises";
import { tmpdir } from "node:os";
import { join } from "node:path";

import {
  applyMigrations,
  generateEmojiData,
  seedTestUser,
  startDevServer,
  stopDevServer,
  closeContextLogs,
  printFailureContext,
  printLogTails,
  type E2eContext,
  type E2eContextFile,
} from "./harness";
import { startMockOidcProvider, type MockOidcServer } from "./oidc-mock";

export type { E2eContextFile } from "./harness";
export const E2E_CONTEXT_PATH_ENV = "BLAND_E2E_CONTEXT_PATH";

export default async function globalSetup(): Promise<() => Promise<void>> {
  const tempDir = await mkdtemp(join(tmpdir(), "bland-e2e-"));
  let ctx: E2eContext | null = null;
  let mockOidc: MockOidcServer | null = null;
  console.log(`[e2e] temp state: ${tempDir}`);

  try {
    await generateEmojiData();
    await applyMigrations(tempDir);
    await seedTestUser(tempDir);
    mockOidc = await startMockOidcProvider();
    console.log(`[e2e] mock OIDC provider issuer: ${mockOidc.issuer}`);
    ctx = await startDevServer(tempDir, mockOidc.issuer);

    const contextFile: E2eContextFile = {
      baseUrl: ctx.baseUrl,
      tempDir,
      oidcIssuer: mockOidc.issuer,
    };

    const contextPath = join(tempDir, ".e2e-context.json");
    await writeFile(contextPath, JSON.stringify(contextFile));

    process.env.BLAND_E2E_BASE_URL = ctx.baseUrl;
    process.env[E2E_CONTEXT_PATH_ENV] = contextPath;

    const captured = { ctx, mockOidc };
    return async () => {
      await teardown(captured.ctx, captured.mockOidc, tempDir);
    };
  } catch (error) {
    if (ctx) {
      await stopDevServer(ctx.serverProcess);
      await closeContextLogs(ctx);
    }
    if (mockOidc) {
      await mockOidc.close().catch(() => undefined);
    }

    if (process.env.BLAND_E2E_PRESERVE === "1") {
      if (ctx) {
        printFailureContext(ctx);
        await printLogTails(ctx);
      }
      console.log(`[e2e] temp state preserved after setup failure: ${tempDir}`);
    } else {
      if (ctx) await rm(ctx.devVarsPath, { force: true });
      await rm(tempDir, { recursive: true, force: true });
      console.log(`[e2e] temp state cleaned up after setup failure`);
    }

    throw error;
  }
}

async function teardown(ctx: E2eContext, mockOidc: MockOidcServer | null, tempDir: string): Promise<void> {
  await stopDevServer(ctx.serverProcess);
  await closeContextLogs(ctx);
  if (mockOidc) {
    await mockOidc.close().catch(() => undefined);
  }

  if (process.env.BLAND_E2E_PRESERVE === "1") {
    printFailureContext(ctx);
    await printLogTails(ctx);
    console.log(`[e2e] temp state preserved: ${tempDir}`);
  } else {
    await rm(ctx.devVarsPath, { force: true });
    await rm(tempDir, { recursive: true, force: true });
    console.log(`[e2e] temp state cleaned up`);
  }
}