Init cs2 rcon panel

modules/middleware.js

@@ -0,0 +1,14 @@
+function is_authenticated(req, res, next) {
+    if (req.session.user) {
+        next();
+    } else {
+        const accept_header = req.headers['accept'];
+        if (accept_header && accept_header.includes('text/html')) {
+            res.redirect('/');
+        } else {
+            res.status(401).json({ status: 401, message: 'Unauthorized' });
+        }
+    }
+}
+
+module.exports = is_authenticated;

modules/rcon.js

@@ -0,0 +1,90 @@
+const Rcon = require('rcon-srcds').default;
+const { better_sqlite_client } = require('../db');
+
+
+class RconManager {
+    constructor() {
+        this.rcons = {};
+        this.details = {};
+        this.init();
+    }
+
+    async init() {
+        try {
+            const servers_query = better_sqlite_client.prepare(`
+                SELECT * FROM servers
+            `);
+            const servers = servers_query.all();
+            console.log('All servers in DB:', servers);
+            for (const server of servers) {
+                const server_id = server.id.toString();
+                if (server_id in this.rcons) continue;
+                await this.connect(server_id, server)
+            }
+        } catch (error) {
+            console.error('Error connecting to MongoDB:', error);
+        }
+    }
+
+    async connect(server_id, server) {
+        let rcon_connection = null;
+        rcon_connection = new Rcon({ host: server.serverIP, port: server.serverPort, timeout: 5000 });
+        console.log("CONNECTING RCON", server_id, server.serverIP, server.serverPort);
+
+        // Set a timeout for the authentication process
+        const authenticationTimeout = setTimeout(async () => {
+            console.error('RCON Authentication timed out', server_id);
+            try {
+                await this.disconnect_rcon(server_id); // Disconnect the RCON connection
+                console.log('Timed out, disconnected RCON', server_id);
+            } catch (error) {
+                console.error('Error disconnecting RCON', server_id, error);
+            }
+        }, 10000);
+    
+        try {
+            await rcon_connection.authenticate(server.rconPassword);
+            clearTimeout(authenticationTimeout);
+            console.log('RCON Authenticated', server_id, server.serverIP, server.serverPort);
+        } catch (error) {
+            clearTimeout(authenticationTimeout);
+            console.error('RCON Authentication failed', server_id, error);
+            // Handle the authentication error here as needed.
+        }
+        
+        this.rcons[server_id] = rcon_connection;
+        this.details[server_id] = {
+            host: server.serverIP,
+            port: server.serverPort,
+            rcon_password: server.rconPassword,
+            connected: rcon_connection.isConnected(),
+            authenticated: rcon_connection.isAuthenticated()
+        };
+        return;
+    }
+
+    async disconnect_rcon(server_id) {
+        console.log('starting disconnect', server_id)
+        if (!this.rcons[server_id].connected) {
+            return Promise.resolve();
+        }
+    
+        this.rcons[server_id].authenticated = false;
+        this.rcons[server_id].connected = false;
+    
+        return new Promise((resolve, reject) => {
+            this.rcons[server_id].connection.once('close', () => {
+                resolve();
+            });
+    
+            this.rcons[server_id].connection.once('error', (e) => {
+                console.error('Socket error during disconnect:', e);
+                resolve();
+            });
+    
+            this.rcons[server_id].connection.end(); // Close the socket gracefully
+        });
+    }
+}
+
+module.exports = new RconManager();