mirror of
https://github.com/onyx-and-iris/q3rcon-cli.git
synced 2026-03-24 02:39:18 +00:00
139 lines
4.3 KiB
Python
139 lines
4.3 KiB
Python
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()
|