Merge pull request #35 from Legonzaur/main

add websocket keepalive packets

src/process/onKill.ts

@@ -102,7 +102,10 @@ export default async function listenKills() {
             killEntry.max_distance = Math.max(killEntry.max_distance || 0, payload.distance)
         }
         deathEntry.deaths++
+        deathEntry.attacker_name = payload.victim_name
         deathWithWeaponEntry.deaths_with_weapon++
+        deathWithWeaponEntry.attacker_name = payload.victim_name
+        killEntry.attacker_name = payload.attacker_name
     })
     return pgClient
 }

src/server/server.ts

@@ -145,6 +145,7 @@ router.post(
   //body('servername').customSanitizer(e => e.replace(/[^a-z0-9]/gi, '')),
   validateErrors,
   async (req, res) => {
+    // Do we check the same thing twice ?????
     if (!req.headers.authorization) return res.sendStatus(403)
     const headers = req.headers.authorization.split(' ')
     if (headers[0].toLowerCase() != "bearer") return res.status(403).send("authorization must be token bearer")
@@ -233,12 +234,12 @@ router.post(
       .then((e) => {
         res.sendStatus(201)
         console.log(
-          `[${Date.now().toLocaleString()}] Kill submitted for server ${servername}, ${attacker_name} killed ${victim_name}`
+          `[${new Date().toLocaleString()}] Kill submitted for server ${servername}, ${attacker_name} killed ${victim_name}`
         )
       })
       .catch((e) => {
         res.sendStatus(500)
-        console.log({
+        console.error({
           killstat_version,
           servername,
           host,

src/websocket.ts

@@ -1,7 +1,7 @@
 import * as dotenv from 'dotenv'
 dotenv.config()
 import { Client } from 'pg'
-import { WebSocketServer } from 'ws'
+import { WebSocket, WebSocketServer } from 'ws'
 import { KillTable } from './db/model'
 
 const port = 3002
@@ -14,7 +14,7 @@ export const pgClient = new Client({
     password: process.env.POSTGRES_PASSWORD
 })
 
-wss.on('connection', function connection(ws: WebSocket, req) {
+wss.on('connection', function connection(ws: WebSocket & { isAlive: boolean }, req) {
     const ip =
         req?.headers['x-forwarded-for']?.toString().split(',')[0].trim() ||
         req.socket.remoteAddress
@@ -29,8 +29,30 @@ wss.on('connection', function connection(ws: WebSocket, req) {
             new Date().toLocaleString() + ',' + ip + ',close,' + wss.clients.size
         )
     }
+    ws.onmessage = function (msg) {
+        if (msg.data === "pong") {
+            return ws.isAlive = true
+        }
+        if (msg.data === "ping") {
+            return ws.send("pong")
+        }
+    }
+    ws.send("ping");
 })
 
+const interval = setInterval(function ping() {
+    wss.clients.forEach(function each(ws) {
+        if ((ws as WebSocket & { isAlive: boolean }).isAlive === false) return ws.terminate();
+
+        (ws as WebSocket & { isAlive: boolean }).isAlive = false;
+        ws.send("ping");
+    });
+}, 30000);
+
+wss.on('close', function close() {
+    clearInterval(interval);
+});
+
 export default new Promise(async (resolve, reject) => {
     await pgClient.connect()
     await pgClient.query('LISTEN new_kill')