# Developed by KuiToi Dev
# File core.udp_server.py
# Written by: SantaSpeen
# Core version: 0.4.0
# Licence: FPA
# (c) kuitoi.su 2023
import asyncio

from core import utils


class UDPServer(asyncio.DatagramTransport):
    transport = None

    def __init__(self, core, host=None, port=None):
        super().__init__()
        self.log = utils.get_logger("UDPServer")
        self.loop = asyncio.get_event_loop()
        self.Core = core
        self.host = host
        self.port = port
        self.run = False
        # self.transport = transport

    def connection_made(self, transport):
        self.log.debug("set connection transport")
        # self.transport = self.transport()

    async def handle_datagram(self, data, addr):
        try:
            cid = data[0] - 1
            code = data[2:3].decode()

            client = self.Core.get_client(cid=cid)
            if client:
                if client._udp_sock != (self.transport, addr):
                    client._udp_sock = (self.transport, addr)
                    self.log.debug(f"Set UDP Sock for CID: {cid}")
                match code:
                    case "p":
                        self.log.debug(f"[{cid}] Send ping")
                        # TODO: Call event onSentPing
                        self.transport.sendto(b"p", addr)  # Send ping
                    case "Z":
                        # TODO: Call event onChangePosition
                        if client:
                            await client._send(data[2:], to_all=True, to_self=False, to_udp=True)
                    case _:
                        self.log.debug(f"[{cid}] Unknown code: {code}")
            else:
                self.log.debug(f"Client not found.")

        except Exception as e:
            self.log.error(f"Error handle_datagram: {e}")

    def datagram_received(self, *args, **kwargs):
        self.loop.create_task(self.handle_datagram(*args, **kwargs))

    def connection_lost(self, exc):
        if exc is not None and exc != KeyboardInterrupt:
            self.log.debug(f'Connection raised: {exc}')
        self.log.debug(f'Disconnected.')

    def error_received(self, exc):
        self.log.debug(f'error_received: {exc}')
        self.log.exception(exc)
        self.connection_lost(exc)
        self.transport.close()

    async def _start(self):
        self.log.debug("Starting UDP server.")
        try:
            while self.Core.run:
                await asyncio.sleep(0.2)

                d = UDPServer
                self.transport, p = await self.loop.create_datagram_endpoint(
                    lambda: d(self.Core),
                    local_addr=(self.host, self.port)
                )
                d.transport = self.transport

                if not self.run:
                    self.log.debug(f"UDP server started on {self.transport.get_extra_info('sockname')}")

                self.run = True
                while not self.transport.is_closing():
                    await asyncio.sleep(0.2)
                self.log.info("UDP сервер сдох 1")
        except OSError as e:
            self.log.error("Cannot bind port or other error")
            self.log.exception(e)
        except Exception as e:
            self.log.error(f"Error: {e}")
            self.log.exception(e)
        finally:
            self.run = False
            self.Core.run = False

    def _stop(self):
        self.log.debug("Stopping UDP server")
        self.transport.close()