mirror of
https://github.com/onyx-and-iris/q3rcon-tui.git
synced 2026-04-19 18:23:31 +00:00
onyx-and-iris
f59076f0a6
add host, port validation functions improve error handling for the ConfigScreen. Pydantic validations now occur on assignment as well as creation. minor version bump
137 lines
4.4 KiB
Python
137 lines
4.4 KiB
Python
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
|