from dpnet.netserver import Netserver
from giants.masterserver import MasterServer
import asyncio

## stats
import pymysql


class Server:
    def __init__(self, **kwargs):
        self.listen_ip = kwargs.get("ip", "0.0.0.0")
        self.listen_port = kwargs.get("port", 19711)
        self.register_with_ms = kwargs.get("register", True)
        ## stats
        self.db = self.db_connect()

    @classmethod
    def db_connect(cls):
        return pymysql.connect(host='192.168.40.2', user="gck", passwd="gck", db="gck", autocommit=True)

    def db_execute(self, sql, args=None):
        try:
            with self.db.cursor() as cursor:
                cursor.execute(sql, args)
        except:
            self.db = self.db_connect()
            return self.db_execute(sql, args)

    def db_query(self, sql, args=None):
        try:
            with self.db.cursor() as cursor:
                cursor.execute(sql, args)
                t = cursor.fetchall()
                return t
        except:
            self.db = self.db_connect()
            return self.db_query(sql, args)

    async def on_connection_received(self, ip, port):
        version = 1.5
        map_name = "Players last 24h/last 48h"
        server_name = "\x03!! V1.5 IS OUT !! \x04 Download on Discord: \x06https://discord.gg/Avj4azU\x01"

        self.db_execute("INSERT INTO multiplayer_stats(ip, date) VALUES (%s, NOW())", (ip,))
        stats_24 = self.db_query("SELECT COUNT(DISTINCT ip) FROM multiplayer_stats WHERE date + INTERVAL 1 DAY > NOW() AND ip not in ('192.168.40.1')")[0][0]
        stats_48 = self.db_query("SELECT COUNT(DISTINCT ip) FROM multiplayer_stats WHERE date + INTERVAL 2 DAY > NOW() AND ip not in ('192.168.40.1')")[0][0]

        player_count = stats_24
        max_player_count = stats_48

        return version, server_name, player_count, max_player_count, map_name


if __name__ == '__main__':
    server = Server(port=19110)
    if server.register_with_ms:
        ms = MasterServer(server)
        register_thread = ms.register_and_run()

    loop = asyncio.get_event_loop()
    listen = loop.create_datagram_endpoint(lambda: Netserver(server), local_addr=(server.listen_ip, server.listen_port))
    transport, protocol = loop.run_until_complete(listen)
    try:
        loop.run_forever()
    except KeyboardInterrupt:
        pass
    transport.close()
    loop.close()