server.ts

import http from 'http';
import next from 'next';
import { parse } from 'url';
import { WebSocketManager } from './src/lib/ws/manager';
import { setWebSocketManager } from './src/lib/ws/broadcast';
import { seedDefaultsIfNeeded } from './src/lib/db/seed-defaults';

const dev = process.env.NODE_ENV !== 'production';
const port = parseInt(process.env.PORT ?? '3000', 10);

const app = next({ dev });
const handle = app.getRequestHandler();

app.prepare().then(() => {
  const handleUpgrade = app.getUpgradeHandler();
  seedDefaultsIfNeeded();

  const wsManager = new WebSocketManager();
  setWebSocketManager(wsManager);
  wsManager.start();

  const server = http.createServer((req, res) => {
    const parsedUrl = parse(req.url!, true);
    handle(req, res, parsedUrl);
  });

  server.on('upgrade', (req, socket, head) => {
    const { pathname } = parse(req.url!);
    if (pathname === '/ws') {
      wsManager.handleUpgrade(req, socket as import('net').Socket, head);
    } else {
      handleUpgrade(req, socket, head);
    }
  });

  server.listen(port, () => {
    console.log('> Ready on http://localhost:' + port);
  });

  const shutdown = () => {
    wsManager.stop();
    server.close(() => process.exit(0));
  };
  process.on('SIGINT', shutdown);
  process.on('SIGTERM', shutdown);
});