import { mkdtemp, rm } from "node:fs/promises";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { resetEnvCache } from "@autoscholar/core";
import { closeDb, type Db, getDb } from "@autoscholar/db";
import { runMigrations } from "@autoscholar/db/migrate";
import { FilesystemStorage } from "@autoscholar/storage";
import { sql } from "drizzle-orm";

export const TEST_DB_URL = process.env.TEST_DATABASE_URL ?? "postgresql://localhost:5432/autoscholar_test";

export async function setupTestEnv(): Promise<{ db: Db; storage: FilesystemStorage; root: string }> {
  const root = await mkdtemp(join(tmpdir(), "as-int-"));
  process.env.DATABASE_URL = TEST_DB_URL;
  process.env.ARTIFACT_ROOT = root;
  process.env.SESSION_SECRET = "x".repeat(48);
  process.env.ADMIN_BOOTSTRAP_PASSWORD = "admin123";
  process.env.WORKER_ENROLLMENT_SECRET = "enroll-secret-for-tests-0001";
  process.env.LEASE_TTL_SECONDS = "30";
  process.env.NODE_ENV = "test";
  process.env.STAGING_MODE = "1";
  resetEnvCache();
  await closeDb();
  const db = getDb(TEST_DB_URL);
  await runMigrations(TEST_DB_URL);
  await db.execute(
    sql`truncate table papers, paper_versions, research_runs, research_projects, project_instruction_versions, compute_workers, admin_users, sessions, audit_events, security_events, login_attempts, platform_settings, paper_sequences, topics, series, research_templates restart identity cascade`,
  );
  return { db, storage: new FilesystemStorage(root), root };
}

export async function teardownTestEnv(root: string) {
  await closeDb();
  await rm(root, { recursive: true, force: true });
}

export const GOOD_INSTRUCTION = `---\nmode: theoretical\n---\n# Objective\nProve a small bound.\n## Domain and mode\nMathematics, theoretical.\n## Contribution\nA clean proof.\n## Literature\narXiv search.\n## Evidence\nProof only.\n## Procedure\nDerive.\n## Robustness\nCounterexamples.\n## Reproducibility\nSympy check.\n## Manuscript\nAutoScholar class.\n## Review\nStrict.\n## Limitations and ethics\nNone.\n## Acceptance criteria\nProof complete.\n${"detail ".repeat(100)}`;