Code-MCP

import { Server, Socket } from "socket.io"; import { createServer } from "http"; import express from "express"; // ============================================ // Types // ============================================ interface User { id: string; username: string; socketId: string; rooms: Set<string>; status: "online" | "away" | "busy"; } interface Message { id: string; roomId: string; userId: string; username: string; content: string; timestamp: Date; type: "text" | "system" | "file"; } interface Room { id: string; name: string; createdBy: string; members: Set<string>; isPrivate: boolean; createdAt: Date; } // Server -> Client events interface ServerToClientEvents { "user:joined": (data: { userId: string; username: string; roomId: string; }) => void; "user:left": (data: { userId: string; username: string; roomId: string; }) => void; "user:status": (data: { userId: string; status: User["status"] }) => void; "message:new": (message: Message) => void; "message:typing": (data: { userId: string; username: string; roomId: string; }) => void; "room:created": (room: { id: string; name: string }) => void; "room:members": (data: { roomId: string; members: { id: string; username: string }[]; }) => void; error: (data: { message: string; code: string }) => void; } // Client -> Server events interface ClientToServerEvents { "user:login": ( data: { username: string }, callback: (response: { userId: string }) => void, ) => void; "user:status": (data: { status: User["status"] }) => void; "room:join": (data: { roomId: string }) => void; "room:leave": (data: { roomId: string }) => void; "room:create": ( data: { name: string; isPrivate?: boolean }, callback: (response: { roomId: string }) => void, ) => void; "message:send": (data: { roomId: string; content: string; type?: Message["type"]; }) => void; "message:typing": (data: { roomId: string }) => void; } // ============================================ // Data Store // ============================================ class DataStore { users = new Map<string, User>(); rooms = new Map<string, Room>(); messages = new Map<string, Message[]>(); // roomId -> messages private nextUserId = 1; private nextMessageId = 1; constructor() { // Create default room this.createRoom("general", "system", false); } createUser(username: string, socketId: string): User { const user: User = { id: `user-${this.nextUserId++}`, username, socketId, rooms: new Set(), status: "online", }; this.users.set(user.id, user); return user; } getUserBySocketId(socketId: string): User | undefined { return Array.from(this.users.values()).find((u) => u.socketId === socketId); } removeUser(socketId: string): User | undefined { const user = this.getUserBySocketId(socketId); if (user) { this.users.delete(user.id); } return user; } createRoom(name: string, createdBy: string, isPrivate: boolean): Room { const id = name.toLowerCase().replace(/\s+/g, "-"); const room: Room = { id, name, createdBy, members: new Set(), isPrivate, createdAt: new Date(), }; this.rooms.set(id, room); this.messages.set(id, []); return room; } addMessage( roomId: string, userId: string, username: string, content: string, type: Message["type"] = "text", ): Message { const message: Message = { id: `msg-${this.nextMessageId++}`, roomId, userId, username, content, timestamp: new Date(), type, }; const messages = this.messages.get(roomId) ?? []; messages.push(message); this.messages.set(roomId, messages); return message; } getRecentMessages(roomId: string, limit = 50): Message[] { const messages = this.messages.get(roomId) ?? []; return messages.slice(-limit); } getRoomMembers(roomId: string): User[] { const room = this.rooms.get(roomId); if (!room) return []; return Array.from(room.members) .map((id) => this.users.get(id)) .filter((u): u is User => u !== undefined); } } const store = new DataStore(); // ============================================ // Server Setup // ============================================ const app = express(); const httpServer = createServer(app); const io = new Server<ClientToServerEvents, ServerToClientEvents>(httpServer, { cors: { origin: process.env.CORS_ORIGIN || "*", methods: ["GET", "POST"], }, pingTimeout: 60000, pingInterval: 25000, }); // ============================================ // Middleware // ============================================ io.use((socket, next) => { // Add authentication here if needed const token = socket.handshake.auth.token; if (token) { // Verify token in production socket.data.authenticated = true; } next(); }); // ============================================ // Connection Handler // ============================================ io.on( "connection", (socket: Socket<ClientToServerEvents, ServerToClientEvents>) => { console.log(`Socket connected: ${socket.id}`); // User Login socket.on("user:login", ({ username }, callback) => { const user = store.createUser(username, socket.id); socket.data.userId = user.id; socket.data.username = username; // Auto-join general room const generalRoom = store.rooms.get("general"); if (generalRoom) { socket.join("general"); user.rooms.add("general"); generalRoom.members.add(user.id); // Notify room socket.to("general").emit("user:joined", { userId: user.id, username, roomId: "general", }); // Send system message const systemMsg = store.addMessage( "general", "system", "System", `${username} joined the chat`, "system", ); io.to("general").emit("message:new", systemMsg); } callback({ userId: user.id }); }); // Update Status socket.on("user:status", ({ status }) => { const user = store.getUserBySocketId(socket.id); if (!user) return; user.status = status; // Notify all rooms user is in user.rooms.forEach((roomId) => { socket.to(roomId).emit("user:status", { userId: user.id, status }); }); }); // Join Room socket.on("room:join", ({ roomId }) => { const user = store.getUserBySocketId(socket.id); const room = store.rooms.get(roomId); if (!user || !room) { socket.emit("error", { message: "Invalid room", code: "INVALID_ROOM" }); return; } socket.join(roomId); user.rooms.add(roomId); room.members.add(user.id); // Notify room socket.to(roomId).emit("user:joined", { userId: user.id, username: user.username, roomId, }); // Send room members const members = store .getRoomMembers(roomId) .map((m) => ({ id: m.id, username: m.username })); socket.emit("room:members", { roomId, members }); // Send recent messages const recentMessages = store.getRecentMessages(roomId); recentMessages.forEach((msg) => socket.emit("message:new", msg)); }); // Leave Room socket.on("room:leave", ({ roomId }) => { const user = store.getUserBySocketId(socket.id); const room = store.rooms.get(roomId); if (!user || !room) return; socket.leave(roomId); user.rooms.delete(roomId); room.members.delete(user.id); socket.to(roomId).emit("user:left", { userId: user.id, username: user.username, roomId, }); }); // Create Room socket.on("room:create", ({ name, isPrivate = false }, callback) => { const user = store.getUserBySocketId(socket.id); if (!user) { socket.emit("error", { message: "Not logged in", code: "UNAUTHORIZED", }); return; } const room = store.createRoom(name, user.id, isPrivate); // Auto-join creator socket.join(room.id); user.rooms.add(room.id); room.members.add(user.id); // Notify all (for public rooms) if (!isPrivate) { io.emit("room:created", { id: room.id, name: room.name }); } callback({ roomId: room.id }); }); // Send Message socket.on("message:send", ({ roomId, content, type = "text" }) => { const user = store.getUserBySocketId(socket.id); if (!user || !user.rooms.has(roomId)) { socket.emit("error", { message: "Not in room", code: "NOT_IN_ROOM" }); return; } const message = store.addMessage( roomId, user.id, user.username, content, type, ); io.to(roomId).emit("message:new", message); }); // Typing Indicator socket.on("message:typing", ({ roomId }) => { const user = store.getUserBySocketId(socket.id); if (!user || !user.rooms.has(roomId)) return; socket.to(roomId).emit("message:typing", { userId: user.id, username: user.username, roomId, }); }); // Disconnect socket.on("disconnect", (reason) => { console.log(`Socket disconnected: ${socket.id} - ${reason}`); const user = store.removeUser(socket.id); if (user) { user.rooms.forEach((roomId) => { const room = store.rooms.get(roomId); if (room) { room.members.delete(user.id); socket.to(roomId).emit("user:left", { userId: user.id, username: user.username, roomId, }); // System message const systemMsg = store.addMessage( roomId, "system", "System", `${user.username} left the chat`, "system", ); io.to(roomId).emit("message:new", systemMsg); } }); } }); }, ); // ============================================ // HTTP Endpoints // ============================================ app.get("/health", (_, res) => { res.json({ status: "healthy", connections: io.engine.clientsCount, rooms: store.rooms.size, users: store.users.size, }); }); app.get("/rooms", (_, res) => { const publicRooms = Array.from(store.rooms.values()) .filter((r) => !r.isPrivate) .map((r) => ({ id: r.id, name: r.name, members: r.members.size })); res.json(publicRooms); }); // ============================================ // Start Server // ============================================ const PORT = process.env.PORT || 3000; httpServer.listen(PORT, () => { console.log(`🔌 WebSocket Server: http://localhost:${PORT}`); }); export { io, store };