Update index.js

index.js

@@ -5,26 +5,144 @@ const io = require('socket.io')(http, {
     cors: { origin: "*", methods: ["GET", "POST"] }
 });
 
-app.get('/', (req, res) => res.send('sSs Relay Online.'));
+app.get('/', (req, res) => res.send('sSs Relay + Justice System Online.'));
+
+// --- STATE MEMORY ---
+let broadcaster = null;
+let currentVisual = "default"; // Default state
+const bannedIPs = new Map(); // IP -> Expiration Timestamp
+const mutedIPs = new Map();  // IP -> Expiration Timestamp
+const kickedIPs = new Map(); // IP -> Expiration Timestamp (10s)
+
+// --- HELPER: GET IP ---
+function getClientIP(socket) {
+    const header = socket.handshake.headers['x-forwarded-for'];
+    return header ? header.split(',')[0].trim() : socket.handshake.address;
+}
+
+// --- HELPER: CLEAN LISTS ---
+function cleanLists() {
+    const now = Date.now();
+    [bannedIPs, mutedIPs, kickedIPs].forEach(map => {
+        for (const [ip, time] of map.entries()) {
+            if (now > time) map.delete(ip);
+        }
+    });
+}
 
 io.on('connection', (socket) => {
-    console.log(`User connected: ${socket.id}`);
+    const ip = getClientIP(socket);
+    cleanLists(); // Cleanup old bans on every connect
+
+    // 1. GATEKEEPER (Bans & Kicks)
+    if (bannedIPs.has(ip)) {
+        const timeLeft = Math.ceil((bannedIPs.get(ip) - Date.now()) / 60000);
+        socket.emit('force-disconnect', `BANNED. Time remaining: ${timeLeft} mins.`);
+        socket.disconnect(true);
+        return;
+    }
+    if (kickedIPs.has(ip)) {
+        socket.emit('force-disconnect', `KICKED. Cooldown: 10s.`);
+        socket.disconnect(true);
+        return;
+    }
 
-    // 1. SIMPLE CHAT RELAY (Echo)
-    // We just take whatever comes in and blast it to everyone.
-    socket.on('chat-message', (msg) => {
-        io.emit('chat-message', msg);
+    console.log(`User joined: ${socket.id} [${ip}]`);
+
+    // 2. BROADCASTER LOGIC
+    socket.on('broadcaster', () => {
+        broadcaster = socket.id;
+        socket.broadcast.emit('broadcaster');
+        // Send lists to admin immediately
+        sendAdminData(socket);
+    });
+
+    // 3. WATCHER LOGIC
+    socket.on('watcher', () => {
+        if (broadcaster) socket.to(broadcaster).emit('watcher', socket.id);
+        // Send current visual state immediately to new user
+        socket.emit('visual-update', currentVisual);
     });
 
-    // 2. VIEWER COUNT (Optional but nice)
-    io.emit('viewer-count', io.engine.clientsCount);
+    // 4. CHAT RELAY (With Mute Check)
+    socket.on('chat-message', (payload) => {
+        cleanLists();
+        // payload: { name, text, isHost }
+        if (mutedIPs.has(ip) && !payload.isHost) {
+            // Notify user they are muted? Or silent fail? Let's notify.
+            const timeLeft = Math.ceil((mutedIPs.get(ip) - Date.now()) / 60000);
+            socket.emit('system-msg', `YOU ARE MUTED (${timeLeft}m remaining).`);
+            return;
+        }
+
+        // Add IP hash for Admin ID
+        io.emit('chat-message', { ...payload, socketId: socket.id, ip: ip });
+    });
 
-    socket.on('disconnect', () => {
-        io.emit('viewer-count', io.engine.clientsCount);
+    // 5. VISUAL CONTROLLER
+    socket.on('push-visual', (url) => {
+        if (socket.id !== broadcaster) return;
+        currentVisual = url;
+        io.emit('visual-update', url);
+    });
+
+    // 6. ADMIN JUSTICE SYSTEM
+    socket.on('admin-action', (cmd) => {
+        if (socket.id !== broadcaster) return;
+        
+        // cmd: { action: 'ban1', targetIp: '...' } OR { action: 'ban1', targetSocket: '...' }
+        let targetIP = cmd.targetIp;
+        
+        // If we only have socket ID (from chat click), find the socket to get IP
+        if (!targetIP && cmd.targetSocket) {
+            const targetSock = io.sockets.sockets.get(cmd.targetSocket);
+            if (targetSock) {
+                targetIP = getClientIP(targetSock);
+                // Execute immediate disconnects if needed
+                if (cmd.action.includes('ban') || cmd.action === 'kick') {
+                    targetSock.emit('force-disconnect', 'You have been removed by the Host.');
+                    targetSock.disconnect(true);
+                }
+            }
+        }
+
+        if (!targetIP) return; // Can't act without IP
+
+        const now = Date.now();
+        const min = 60000;
+        const hour = min * 60;
+
+        switch (cmd.action) {
+            case 'kick': kickedIPs.set(targetIP, now + 10000); break; // 10s
+            
+            case 'ban1': bannedIPs.set(targetIP, now + (1 * hour)); break;
+            case 'ban2': bannedIPs.set(targetIP, now + (24 * hour)); break;
+            case 'ban3': bannedIPs.set(targetIP, now + (720 * hour)); break;
+            
+            case 'mute1': mutedIPs.set(targetIP, now + (1 * min)); break;
+            case 'mute2': mutedIPs.set(targetIP, now + (10 * min)); break;
+            case 'mute3': mutedIPs.set(targetIP, now + (160 * min)); break;
+            
+            case 'unban': bannedIPs.delete(targetIP); kickedIPs.delete(targetIP); break;
+            case 'unmute': mutedIPs.delete(targetIP); break;
+        }
+
+        // Update Admin UI
+        sendAdminData(socket);
+    });
+
+    socket.on('request-admin-data', () => {
+        if (socket.id === broadcaster) sendAdminData(socket);
     });
 });
 
+function sendAdminData(socket) {
+    // Convert Maps to Arrays for sending
+    const bans = Array.from(bannedIPs.entries()).map(([ip, time]) => ({ ip, time }));
+    const mutes = Array.from(mutedIPs.entries()).map(([ip, time]) => ({ ip, time }));
+    socket.emit('admin-data', { bans, mutes });
+}
+
 http.listen(7860, () => console.log('Server running on 7860'));
 
-// Auto-Restart every 24h to keep memory fresh
 setTimeout(() => process.exit(0), 24 * 60 * 60 * 1000);