import { expect, tap } from '@push.rocks/tapbundle';
import * as net from 'net';
import { PortProxy } from '../ts/smartproxy.portproxy.js';

let testServer: net.Server;
let portProxy: PortProxy;
const TEST_SERVER_PORT = 4000;
const PROXY_PORT = 4001;
const TEST_DATA = 'Hello through port proxy!';

// Helper function to create a test TCP server
function createTestServer(port: number): Promise<net.Server> {
  return new Promise((resolve) => {
    const server = net.createServer((socket) => {
      socket.on('data', (data) => {
        // Echo the received data back
        socket.write(`Echo: ${data.toString()}`);
      });

      socket.on('error', (error) => {
        console.error('[Test Server] Socket error:', error);
      });
    });

    server.listen(port, () => {
      console.log(`[Test Server] Listening on port ${port}`);
      resolve(server);
    });
  });
}

// Helper function to create a test client connection
function createTestClient(port: number, data: string): Promise<string> {
  return new Promise((resolve, reject) => {
    const client = new net.Socket();
    let response = '';

    client.connect(port, 'localhost', () => {
      console.log('[Test Client] Connected to server');
      client.write(data);
    });

    client.on('data', (chunk) => {
      response += chunk.toString();
      client.end();
    });

    client.on('end', () => {
      resolve(response);
    });

    client.on('error', (error) => {
      reject(error);
    });
  });
}

// Setup test environment
tap.test('setup port proxy test environment', async () => {
  testServer = await createTestServer(TEST_SERVER_PORT);
  portProxy = new PortProxy(PROXY_PORT, TEST_SERVER_PORT);
});

tap.test('should start port proxy', async () => {
  await portProxy.start();
  expect(portProxy.netServer.listening).toBeTrue();
});

tap.test('should forward TCP connections and data', async () => {
  const response = await createTestClient(PROXY_PORT, TEST_DATA);
  expect(response).toEqual(`Echo: ${TEST_DATA}`);
});

tap.test('should handle multiple concurrent connections', async () => {
  const concurrentRequests = 5;
  const requests = Array(concurrentRequests).fill(null).map((_, i) => 
    createTestClient(PROXY_PORT, `${TEST_DATA} ${i + 1}`)
  );
  
  const responses = await Promise.all(requests);
  
  responses.forEach((response, i) => {
    expect(response).toEqual(`Echo: ${TEST_DATA} ${i + 1}`);
  });
});

tap.test('should handle connection timeouts', async () => {
  const client = new net.Socket();
  
  await new Promise<void>((resolve) => {
    client.connect(PROXY_PORT, 'localhost', () => {
      // Don't send any data, just wait for timeout
      client.on('close', () => {
        resolve();
      });
    });
  });
});

tap.test('should stop port proxy', async () => {
  await portProxy.stop();
  expect(portProxy.netServer.listening).toBeFalse();
});

// Cleanup
tap.test('cleanup port proxy test environment', async () => {
  await new Promise<void>((resolve) => testServer.close(() => resolve()));
});

process.on('exit', () => {
  if (testServer) {
    testServer.close();
  }
  if (portProxy && portProxy.netServer) {
    portProxy.stop();
  }
});

export default tap.start();