import * as dotenv from 'dotenv'
dotenv.config()
import { Client } from 'pg'
import { WebSocket, WebSocketServer } from 'ws'
import { KillTable } from './db/model'

const port = 3002
const wss = new WebSocketServer({ port })

export const pgClient = new Client({
    host: process.env.POSTGRES_HOST,
    database: process.env.POSTGRES_DATABASE,
    user: process.env.POSTGRES_USER,
    password: process.env.POSTGRES_PASSWORD
})

wss.on('connection', function connection(ws: WebSocket & { isAlive: boolean }, req) {
    const ip =
        req?.headers['x-forwarded-for']?.toString().split(',')[0].trim() ||
        req.socket.remoteAddress
    console.log(
        new Date().toLocaleString() + ',' + ip + ',connect,' + wss.clients.size
    )
    ws.onclose = function () {
        const ip =
            req?.headers['x-forwarded-for']?.toString().split(',')[0].trim() ||
            req.socket.remoteAddress
        console.log(
            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')

    pgClient.on('notification', (data) => {
        if (!data.payload) return
        const payload = JSON.parse(data.payload) as KillTable
        wss.clients.forEach(function (client) {
            if (client.readyState === client.OPEN) {
                client.send(
                    JSON.stringify({
                        attacker_id: payload.attacker_id,
                        attacker_name: payload.attacker_name,
                        cause_of_death: payload.cause_of_death,
                        victim_id: payload.victim_id,
                        victim_name: payload.victim_name,
                        attacker_current_weapon: payload.attacker_current_weapon,
                        victim_current_weapon: payload.victim_current_weapon,
                        distance: payload.distance,
                        game_mode: payload.game_mode,
                        servername: payload.servername,
                        map: payload.map,
                        host: payload.host
                    })
                )
            }
        })
    })
})