import re

from rich.table import Table
from rich.text import Text

Renderable = Text | Table | str


class Writable:
    RE_COLOR_CODES = re.compile(r'\^[0-9]')
    RE_MAP_FROM_STATUS = re.compile(r'^map: (?P<mapname>mp_[a-z_]+)$')
    RE_PLAYER_FROM_STATUS = 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,
    )
    RE_CVAR = 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>.*?)["]$'
    )

    def __init__(self, tui):
        self._settings = tui._settings

    @staticmethod
    def remove_color_codes(s: str) -> str:
        return Writable.RE_COLOR_CODES.sub('', s)

    def parse(self, cmd, response: str, style=None) -> Renderable:
        response = self.remove_color_codes(response.removeprefix('print\n'))
        if self._settings.raw:
            return Text(response, style=style)

        if response in ['Bad rcon']:
            return Text('Incorrect RCON password', style='#c73d4b')

        match cmd:
            case 'status':
                return self.status_table(response)
            case _:
                if m := self.RE_CVAR.match(response):
                    return self.cvar_table(m)
                else:
                    return Text(response, style=style)

    def error(self, message: str) -> Text:
        return Text(message, style='#c73d4b')

    def status_table(self, status_response: str) -> Table | str:
        table = Table(show_header=True, header_style='bold #88c0d0')
        columns = [
            ('Slot', 'center'),
            ('Score', 'center'),
            ('Ping', 'center'),
            ('GUID', 'center'),
            ('Name', 'center'),
        ]
        for column, justify in columns:
            table.add_column(column, justify=justify)
        if not self._settings.min_status:
            table.add_column('Last', justify='center')
        if self._settings.min_status:
            table.add_column('IP', justify='center')
        else:
            table.add_column('IP:Port', justify='center')
        columns = [
            ('QPort', 'center'),
            ('Rate', 'center'),
        ]
        if not self._settings.min_status:
            for column, justify in columns:
                table.add_column(column, justify=justify)

        mapname = 'unable to parse map name'
        for line in status_response.splitlines():
            if m := self.RE_PLAYER_FROM_STATUS.match(line):
                name = m.group('name')
                if name == '':
                    name = '[no name]'
                row = [
                    m.group('slot'),
                    m.group('score'),
                    m.group('ping'),
                    m.group('guid'),
                    name,
                ]
                if self._settings.min_status:
                    row.append(m.group('ip'))
                else:
                    row.append(f'{m.group("ip")}:{m.group("port")}')
                    row.append(m.group('last'))
                    row.append(m.group('qport'))
                    row.append(m.group('rate'))
                table.add_row(*row)
            elif m := self.RE_MAP_FROM_STATUS.match(line):
                mapname = m.group('mapname')

        out = Text(f'Map: {mapname}\n', style='bold #88c0d0')
        if len(table.rows) == 0:
            return out.append('No players connected', style='#c73d4b')
        else:
            table.title = out
            return table

    def cvar_table(self, m: re.Match) -> Table:
        table = Table(show_header=True, header_style='bold #88c0d0')
        columns = [
            ('Name', 'center'),
            ('Value', 'center'),
            ('Default', 'center'),
            ('Info', 'center'),
        ]
        for column, justify in columns:
            table.add_column(column, justify=justify)

        table.add_row(
            m.group('name'),
            m.group('value'),
            m.group('default'),
            m.group('info'),
        )

        return table