from __future__ import annotations

import asyncio
import inspect
import json
import struct
import sys

# TODO: Get rid of this import * lol
from .exceptions import (
    PyPresenceException,
    InvalidArgument,
    InvalidPipe,
    PipeClosed,
    ConnectionTimeout,
    ResponseTimeout,
    DiscordNotFound,
    InvalidID,
    DiscordError,
    ServerError,
)
from .payloads import Payload
from .utils import get_ipc_path, get_event_loop


class BaseClient:

    def __init__(self, client_id: str, **kwargs):
        loop = kwargs.get("loop", None)
        handler = kwargs.get("handler", None)
        self.pipe = kwargs.get("pipe", None)
        self.isasync = kwargs.get("isasync", False)
        self.connection_timeout = kwargs.get("connection_timeout", 30)
        self.response_timeout = kwargs.get("response_timeout", 10)

        client_id = str(client_id)

        if loop is not None:
            self.update_event_loop(loop)
        else:
            self.update_event_loop(get_event_loop())

        self.sock_reader: asyncio.StreamReader | None = None
        self.sock_writer: asyncio.StreamWriter | None = None

        self.client_id = client_id

        if handler is not None:
            if not inspect.isfunction(handler):
                raise PyPresenceException("Error handler must be a function.")
            args = inspect.getfullargspec(handler).args
            if args[0] == "self":
                args = args[1:]
            if len(args) != 2:
                raise PyPresenceException(
                    "Error handler should only accept two arguments."
                )

            if self.isasync:
                if not inspect.iscoroutinefunction(handler):
                    raise InvalidArgument(
                        "Coroutine",
                        "Subroutine",
                        "You are running async mode - "
                        "your error handler should be awaitable.",
                    )
                err_handler = self._async_err_handle
            else:
                err_handler = self._err_handle

            self.loop.set_exception_handler(err_handler)
            self.handler = handler

        if getattr(self, "on_event", None):  # Tasty bad code ;^)
            self._events_on = True
        else:
            self._events_on = False

    def update_event_loop(self, loop):
        # noinspection PyAttributeOutsideInit
        self.loop = loop
        asyncio.set_event_loop(self.loop)

    def _err_handle(self, loop, context: dict):
        result = self.handler(context["exception"], context["future"])
        if inspect.iscoroutinefunction(self.handler):
            loop.run_until_complete(result)

    # noinspection PyUnusedLocal
    async def _async_err_handle(self, loop, context: dict):
        await self.handler(context["exception"], context["future"])

    async def read_output(self):
        try:
            preamble = await asyncio.wait_for(
                self.sock_reader.read(8), self.response_timeout
            )
            status_code, length = struct.unpack("<II", preamble[:8])
            data = await asyncio.wait_for(
                self.sock_reader.read(length), self.response_timeout
            )
        except (BrokenPipeError, struct.error):
            raise PipeClosed
        except asyncio.TimeoutError:
            raise ResponseTimeout
        payload = json.loads(data.decode("utf-8"))
        if payload["evt"] == "ERROR":
            raise ServerError(payload["data"]["message"])
        return payload

    def send_data(self, op: int, payload: dict | Payload):
        if isinstance(payload, Payload):
            payload = payload.data
        payload_string = json.dumps(payload)

        assert (
            self.sock_writer is not None
        ), "You must connect your client before sending events!"

        self.sock_writer.write(
            struct.pack("<II", op, len(payload_string)) + payload_string.encode("utf-8")
        )

    async def create_reader_writer(self, ipc_path):
        try:
            if sys.platform == "linux" or sys.platform == "darwin":
                self.sock_reader, self.sock_writer = await asyncio.wait_for(
                    asyncio.open_unix_connection(ipc_path), self.connection_timeout
                )
            elif sys.platform == "win32":
                self.sock_reader = asyncio.StreamReader(loop=self.loop)
                reader_protocol = asyncio.StreamReaderProtocol(
                    self.sock_reader, loop=self.loop
                )
                self.sock_writer, _ = await asyncio.wait_for(
                    self.loop.create_pipe_connection(lambda: reader_protocol, ipc_path),
                    self.connection_timeout,
                )
        except FileNotFoundError:
            raise InvalidPipe
        except asyncio.TimeoutError:
            raise ConnectionTimeout

    async def handshake(self):
        ipc_path = get_ipc_path(self.pipe)
        if not ipc_path:
            raise DiscordNotFound

        await self.create_reader_writer(ipc_path)

        self.send_data(0, {"v": 1, "client_id": self.client_id})
        preamble = await self.sock_reader.read(8)
        if len(preamble) < 8:
            raise InvalidPipe  # this sometimes happens for some reason, perhaps discord cannot always accept all the connections?
        code, length = struct.unpack("<ii", preamble)
        data = json.loads(await self.sock_reader.read(length))
        if "code" in data:
            if data["message"] == "Invalid Client ID":
                raise InvalidID
            raise DiscordError(data["code"], data["message"])
        if self._events_on:
            self.sock_reader.feed_data = self.on_event
