import re

import clypi
from clypi import Boxes, cprint


class Console:
    def __init__(self, style: str = 'yellow'):
        self.style = style

    def print(self, message: str, style: str | None = None):
        cprint(message, fg=style or self.style)


class ErrorConsole(Console):
    def __init__(self):
        super().__init__(style='red')


class OutConsole(Console):
    COLOUR_CODE_REGEX = re.compile(r'\^[0-9]')
    STATUS_PLAYER_REGEX = re.compile(
        r'^\s*(?P<slot>[0-9]+)\s+'
        r'(?P<score>[0-9-]+)\s+'
        r'(?P<ping>[0-9]+)\s+'
        r'(?P<guid>[0-9a-f]+)\s+'
        r'(?P<name>.*?)\s+'
        r'(?P<last>[0-9]+?)\s*'
        r'(?P<ip>(?:(?:25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9]?[0-9])\.){3}'
        r'(?:25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9]?[0-9])):?'
        r'(?P<port>-?[0-9]{1,5})\s*'
        r'(?P<qport>-?[0-9]{1,5})\s+'
        r'(?P<rate>[0-9]+)$',
        re.IGNORECASE | re.VERBOSE,
    )
    STATUS_MAP_REGEX = re.compile(r'^map: (?P<mapname>mp_[a-z_]+)$')

    CVAR_REGEX = re.compile(
        r'^["](?P<name>[a-z_]+)["]\sis[:]\s'
        r'["](?P<value>.*?)["]\s'
        r'default[:]\s'
        r'["](?P<default>.*?)["]\s'
        r'info[:]\s'
        r'["](?P<info>.*?)["]$'
    )

    @staticmethod
    def _remove_colour_codes(s: str) -> str:
        """Remove Quake 3 colour codes from a string."""
        return OutConsole.COLOUR_CODE_REGEX.sub('', s)

    def print_response(self, response: str):
        if response := self._remove_colour_codes(response).removeprefix('print\n'):
            cprint(response, fg=self.style)

    def print_status(self, response: str):
        _slots = []
        _scores = []
        _pings = []
        _guids = []
        _names = []
        _ips = []

        lines = response.splitlines()
        for line in lines:
            if m := OutConsole.STATUS_PLAYER_REGEX.match(line):
                _slots.append(m.group('slot'))
                _scores.append(m.group('score'))
                _pings.append(m.group('ping'))
                _guids.append(m.group('guid'))
                _names.append(self._remove_colour_codes(m.group('name')))
                _ips.append(m.group('ip'))
            elif m := OutConsole.STATUS_MAP_REGEX.match(line):
                cprint(f'Current map: {m.group("mapname")}', fg=self.style)

        if not _slots:
            cprint('No players connected.', fg='red')
            return

        slots = clypi.boxed(
            _slots, title='Slot', width=10, align='center', style=Boxes.ROUNDED
        )
        scores = clypi.boxed(
            _scores, title='Score', width=10, align='center', style=Boxes.ROUNDED
        )
        pings = clypi.boxed(
            _pings, title='Ping', width=10, align='center', style=Boxes.ROUNDED
        )
        guids = clypi.boxed(
            _guids,
            title='GUID',
            width=len(max(_guids, key=len)) + 4,
            style=Boxes.ROUNDED,
        )
        names = clypi.boxed(
            _names,
            title='Name',
            width=len(max(_names, key=len)) + 4,
            style=Boxes.ROUNDED,
        )
        ips = clypi.boxed(
            _ips, title='IP', width=len(max(_ips, key=len)) + 4, style=Boxes.ROUNDED
        )
        print(f'{clypi.stack(slots, scores, pings, guids, names, ips, padding=0)}')

    def print_cvar(self, response: str):
        response = self._remove_colour_codes(response).removeprefix('print\n')

        if m := self.CVAR_REGEX.match(response):
            name = clypi.boxed(
                [m.group('name')],
                title='Name',
                width=max(len(m.group('name')) + 4, 15),
                style=Boxes.ROUNDED,
            )
            value = clypi.boxed(
                [m.group('value')],
                title='Value',
                width=max(len(m.group('value')) + 4, 15),
                style=Boxes.ROUNDED,
            )
            default = clypi.boxed(
                [m.group('default')],
                title='Default',
                width=max(len(m.group('default')) + 4, 15),
                style=Boxes.ROUNDED,
            )
            info = clypi.boxed(
                [m.group('info')],
                title='Info',
                width=max(len(m.group('info')) + 4, 15),
                style=Boxes.ROUNDED,
            )
            print(f'{clypi.stack(name, value, default, info, padding=0)}')


out = OutConsole()
err = ErrorConsole()