koa-websocket

import Koa from 'koa'; import route from 'koa-route'; import websockify from 'koa-websocket'; // Create a new Koa application and extend it with WebSocket capabilities const app = websockify(new Koa()); // Basic HTTP middleware (optional, but shows coexistence) app.use(async (ctx, next) => { console.log(`HTTP Request: ${ctx.method} ${ctx.path}`); await next(); }); // Global WebSocket middleware for all connections app.ws.use(async (ctx, next) => { console.log(`New WebSocket connection established for path: ${ctx.path}`); // You can perform authentication or other setup here await next(ctx); // Pass context to the next WebSocket middleware or route }); // WebSocket route handler using koa-route app.ws.use(route.all('/chat/:id', (ctx) => { const roomId = ctx.params.id; console.log(`Client connected to chat room: ${roomId}`); // Send a welcome message to the newly connected client ctx.websocket.send(`Welcome to chat room ${roomId}!`); // Listen for messages from this client ctx.websocket.on('message', (message) => { console.log(`Message from client in room ${roomId}: ${message}`); // Echo the message back to the sender ctx.websocket.send(`Echo (${roomId}): ${message}`); }); // Handle client disconnection ctx.websocket.on('close', () => { console.log(`Client disconnected from chat room: ${roomId}`); }); // Handle WebSocket errors ctx.websocket.on('error', (err) => { console.error(`WebSocket error in room ${roomId}:`, err.message); }); })); // Start the server on an available port, defaulting to 3000 const PORT = process.env.PORT || 3000; app.listen(PORT, () => { console.log(`Server listening on http://localhost:${PORT} and ws://localhost:${PORT}`); console.log('Try connecting with a WebSocket client to ws://localhost:3000/chat/general'); });