Perplexica/src/websocket/websocketServer.ts

import { WebSocketServer } from 'ws';
import { handleConnection } from './connectionManager';
import http from 'http';
import { getPort } from '../config';
import logger from '../utils/logger';

export const initServer = (
  server: http.Server<typeof http.IncomingMessage, typeof http.ServerResponse>,
) => {
  const port = getPort();
  const wss = new WebSocketServer({ server });

  wss.on('connection', (ws) => {
    handleConnection(ws);
  });

  logger.info(`WebSocket server started on port ${port}`);
};