import {WebSocketServer} from 'ws';

import Engine from '@/engine/engine.js';
import Server from '@/net/server/server.js';

const wss = new WebSocketServer({
  noServer: true,
});

function onUpgrade(request, socket, head) {
  const {pathname} = new URL(request.url, 'wss://base.url');
  if (pathname === '/ws') {
    wss.handleUpgrade(request, socket, head, function done(ws) {
      wss.emit('connection', ws, request);
    });
  }
  else {
    socket.destroy();
  }
}

export async function listen(server) {
  server.on('upgrade', onUpgrade);

  class SocketServer extends Server {
    transmit(ws, packed) { ws.send(packed); }
  }

  const engine = new Engine(SocketServer);

  await engine.load();
  engine.start();

  async function onConnect(ws) {
    ws.on('close', () => {
      engine.disconnectPlayer(ws);
    })
    ws.on('message', (packed) => {
      engine.server.accept(ws, new DataView(packed.buffer, packed.byteOffset, packed.length));
    });
    await engine.connectPlayer(ws);
  }

  wss.on('connection', onConnect);
}