import test from 'node:test';
import assert from 'node:assert/strict';
import http from 'node:http';
import net from 'node:net';
import crypto from 'node:crypto';
import { createProxy } from '../src/proxy.js';

const config = { baseHost: 'project.webklar.com' };

// Test-Ziel: HTTP-Echo + WebSocket-Echo per rohem Upgrade
function startTarget() {
  const seen = [];
  const srv = http.createServer((req, res) => {
    const chunks = [];
    req.on('data', (c) => chunks.push(c));
    req.on('end', () => {
      seen.push({ method: req.method, url: req.url, headers: req.headers });
      res.writeHead(201, { 'X-Test': 'ja' });
      res.end(`${req.method} ${req.url} ${Buffer.concat(chunks).toString()}`);
    });
  });
  srv.on('upgrade', (req, socket) => {
    const acc = crypto.createHash('sha1').update(`${req.headers['sec-websocket-key']}258EAFA5-E914-47DA-95CA-C5AB0DC85B11`).digest('base64');
    socket.write(`HTTP/1.1 101 Switching Protocols\r\nUpgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Accept: ${acc}\r\n\r\n`);
    socket.on('data', (d) => socket.write(d)); // Echo
    socket.on('end', () => socket.end());
    socket.on('error', () => {});
  });
  return new Promise((r) => srv.listen(0, '127.0.0.1', () => r({ srv, seen, port: srv.address().port })));
}

async function withProxy(devImpl, fn, opts = {}) {
  const touched = [];
  const dev = { touch: (s) => touched.push(s), ...devImpl };
  const proxy = createProxy({ config, dev, ...opts });
  await new Promise((r) => proxy.listen(0, '127.0.0.1', r));
  try { await fn({ port: proxy.address().port, touched }); } finally { proxy.closeAllConnections?.(); await new Promise((r) => proxy.close(r)); }
}

const get = (port, host, { method = 'GET', path = '/', body } = {}) => new Promise((resolve, reject) => {
  const req = http.request({ port, host: '127.0.0.1', method, path, headers: { Host: host } }, (res) => {
    const c = []; res.on('data', (d) => c.push(d)); res.on('end', () => resolve({ res, text: Buffer.concat(c).toString() }));
  });
  req.on('error', reject);
  req.end(body);
});

test('online: Durchreichen inkl. POST-Body, X-Forwarded-*, Host, touch', async () => {
  const t = await startTarget();
  try {
    await withProxy({ wake: async (s) => ({ state: 'online', target: `http://127.0.0.1:${t.port}` }) }, async ({ port, touched }) => {
      const { res, text } = await get(port, 'demo.project.webklar.com:443', { method: 'POST', path: '/a?b=1', body: 'hallo' });
      assert.equal(res.statusCode, 201);
      assert.equal(res.headers['x-test'], 'ja');
      assert.equal(text, 'POST /a?b=1 hallo');
      assert.deepEqual(touched, ['demo']);
      const h = t.seen[0].headers;
      assert.equal(h.host, 'demo.project.webklar.com:443');
      assert.equal(h['x-forwarded-host'], 'demo.project.webklar.com:443');
      assert.equal(h['x-forwarded-proto'], 'https');
      assert.ok(h['x-forwarded-for']);
    });
  } finally { t.srv.closeAllConnections?.(); await new Promise((r) => t.srv.close(r)); }
});

test('WebSocket-Upgrade wird durchgereicht (Echo)', async () => {
  const t = await startTarget();
  try {
    await withProxy({ wake: async () => ({ state: 'online', target: `http://127.0.0.1:${t.port}` }) }, async ({ port, touched }) => {
      const sock = net.connect(port, '127.0.0.1');
      sock.on('error', () => {});
      let buf = '';
      const got = new Promise((resolve) => sock.on('data', (d) => { buf += d.toString('latin1'); if (buf.includes('\r\n\r\n') && buf.endsWith('ping-pong')) resolve(); }));
      sock.write('GET /ws HTTP/1.1\r\nHost: demo.project.webklar.com\r\nUpgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\nSec-WebSocket-Version: 13\r\n\r\n');
      await new Promise((r) => { const i = setInterval(() => { if (buf.includes('\r\n\r\n')) { clearInterval(i); r(); } }, 5); });
      assert.match(buf, /^HTTP\/1\.1 101/);
      try {
        sock.write('ping-pong');
        await got;
        assert.ok(buf.endsWith('ping-pong'));
        assert.deepEqual(touched, ['demo']);
      } finally { sock.destroy(); }
    });
  } finally { t.srv.closeAllConnections?.(); await new Promise((r) => t.srv.close(r)); }
});

test('starting ⇒ 503 mit Text, Retry-After und Meta-Refresh', async () => {
  await withProxy({ wake: async () => ({ state: 'starting' }) }, async ({ port, touched }) => {
    const { res, text } = await get(port, 'demo.project.webklar.com');
    assert.equal(res.statusCode, 503);
    assert.equal(res.headers['retry-after'], '3');
    assert.match(text, /Preview startet …/);
    assert.match(text, /<meta http-equiv="refresh" content="3">/);
    assert.deepEqual(touched, []);
  });
});

test('failed ⇒ 503 mit Fehlerseite', async () => {
  await withProxy({ wake: async () => ({ state: 'failed' }) }, async ({ port }) => {
    const { res, text } = await get(port, 'demo.project.webklar.com');
    assert.equal(res.statusCode, 503);
    assert.match(text, /Preview konnte nicht starten/);
    assert.match(text, /Protokoll im WOMS/);
    assert.ok(!text.includes('http-equiv="refresh"'));
  });
});

test('wake unknown ⇒ 503 „nicht verfügbar“', async () => {
  await withProxy({ wake: async () => ({ state: 'unknown' }) }, async ({ port }) => {
    const { res, text } = await get(port, 'demo.project.webklar.com');
    assert.equal(res.statusCode, 503);
    assert.match(text, /Diese Preview ist gerade nicht verfügbar\./);
  });
});

test('Speicherschutz (wake wirft 503) ⇒ 503-Seite, kein Absturz', async () => {
  await withProxy({ wake: async () => { throw Object.assign(new Error('Server ausgelastet'), { status: 503 }); } }, async ({ port }) => {
    const { res, text } = await get(port, 'demo.project.webklar.com');
    assert.equal(res.statusCode, 503);
    assert.match(text, /ausgelastet/);
  });
});

test('fremder Host, Großschreibung, ungültiger sub, mehrstufig ⇒ 404 ohne wake', async () => {
  let woken = 0;
  await withProxy({ wake: async () => { woken++; return { state: 'starting' }; } }, async ({ port }) => {
    for (const h of ['evil.example.com', 'project.webklar.com', 'Demo.project.webklar.com', 'a_b.project.webklar.com', 'a.b.project.webklar.com', '.project.webklar.com', 'demo.project.webklar.com.evil.com', '']) {
      const { res } = await get(port, h);
      assert.equal(res.statusCode, 404, h);
    }
    assert.equal(woken, 0);
  });
});

test('Upstream-Fehler ⇒ 502 deutscher Text', async () => {
  const dead = net.createServer(); await new Promise((r) => dead.listen(0, '127.0.0.1', r));
  const port0 = dead.address().port; await new Promise((r) => dead.close(r));
  await withProxy({ wake: async () => ({ state: 'online', target: `http://127.0.0.1:${port0}` }) }, async ({ port }) => {
    const { res, text } = await get(port, 'demo.project.webklar.com');
    assert.equal(res.statusCode, 502);
    assert.match(text, /nicht erreichbar/);
  });
});