ws.js

const PORT = 4000;
const server = new WebSocketServer({ port: PORT });

server.on('data', async (message, reply) => {
  if (!message) return;

  const data = JSON.parse(message);
  const { method, args = [] } = data;
  const handler = api[method];
  if (!handler) return reply({ error: 'Not Found' });

  try {
    const result = await handler(...args);
    reply(result);
  } catch (error) {
    console.error(error);
    reply({ error: 'Internal Server Error' });
  }
});

server.listen(() => {
  console.log(`WebSocket server listening on port ${PORT}`);
});