import * as fs from 'node:fs/promises';
import * as path from 'node:path';
import * as os from 'node:os';
import { atomicWriteFile } from './packages/core/src/utils/atomicFileWrite.js';

async function main() {
  const tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), 'atomic-e2e-'));
  console.log('Test dir:', tmpDir);
  let allPass = true;

  // Test 1: Write new file, no tmp residue
  const f1 = path.join(tmpDir, 'test1.txt');
  await atomicWriteFile(f1, 'hello atomic');
  const content1 = await fs.readFile(f1, 'utf-8');
  const files1 = await fs.readdir(tmpDir);
  const t1 = content1 === 'hello atomic' && files1.length === 1;
  console.log(t1 ? '✓' : '✗', 'Test 1 - Write new file, no tmp residue');
  if (!t1) allPass = false;

  // Test 2: Permission preservation
  const f2 = path.join(tmpDir, 'test2.txt');
  await fs.writeFile(f2, 'old content');
  await fs.chmod(f2, 0o600);
  await atomicWriteFile(f2, 'new content');
  const stat2 = await fs.stat(f2);
  const content2 = await fs.readFile(f2, 'utf-8');
  const t2 = content2 === 'new content' && (stat2.mode & 0o777) === 0o600;
  console.log(t2 ? '✓' : '✗', 'Test 2 - Permission preservation (mode:', (stat2.mode & 0o777).toString(8) + ')');
  if (!t2) allPass = false;

  // Test 3: Symlink preservation
  const realFile = path.join(tmpDir, 'real.txt');
  const linkFile = path.join(tmpDir, 'link.txt');
  await fs.writeFile(realFile, 'real data');
  await fs.symlink(realFile, linkFile);
  await atomicWriteFile(linkFile, 'updated data');
  const linkTarget = await fs.readlink(linkFile);
  const realContent = await fs.readFile(realFile, 'utf-8');
  const t3 = linkTarget === realFile && realContent === 'updated data';
  console.log(t3 ? '✓' : '✗', 'Test 3 - Symlink preserved, real file updated');
  if (!t3) allPass = false;

  // Test 4: Broken symlink write-through
  const brokenTarget = path.join(tmpDir, 'not-yet.txt');
  const brokenLink = path.join(tmpDir, 'broken.txt');
  await fs.symlink(brokenTarget, brokenLink);
  await atomicWriteFile(brokenLink, 'created via broken link');
  const brokenLinkTarget = await fs.readlink(brokenLink);
  const createdContent = await fs.readFile(brokenTarget, 'utf-8');
  const t4 = brokenLinkTarget === brokenTarget && createdContent === 'created via broken link';
  console.log(t4 ? '✓' : '✗', 'Test 4 - Broken symlink write-through');
  if (!t4) allPass = false;

  // Test 5: Multi-level symlink chain
  const chainReal = path.join(tmpDir, 'chain-real.txt');
  const chainA = path.join(tmpDir, 'chain-a.txt');
  const chainB = path.join(tmpDir, 'chain-b.txt');
  await fs.writeFile(chainReal, 'original');
  await fs.symlink(chainReal, chainA);
  await fs.symlink(chainA, chainB);
  await atomicWriteFile(chainB, 'chain updated');
  const chainContent = await fs.readFile(chainReal, 'utf-8');
  const chainATarget = await fs.readlink(chainA);
  const chainBTarget = await fs.readlink(chainB);
  const t5 = chainContent === 'chain updated' && chainATarget === chainReal && chainBTarget === chainA;
  console.log(t5 ? '✓' : '✗', 'Test 5 - Multi-level symlink chain');
  if (!t5) allPass = false;

  await fs.rm(tmpDir, { recursive: true, force: true });
  console.log('\n' + (allPass ? 'ALL PASS' : 'SOME FAILED'));
  process.exit(allPass ? 0 : 1);
}

main();