bump aio-q3-rcon dep version

patch bump

.gitignore

@@ -210,3 +210,5 @@ marimo/_lsp/
 __marimo__/
 
 # End of ignr
+
+test-*.py

.pre-commit-config.yaml

@@ -0,0 +1,15 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v2.3.0
+    hooks:
+      - id: check-yaml
+      - id: end-of-file-fixer
+      - id: trailing-whitespace
+
+  - repo: local
+    hooks:
+      - id: update-readme
+        name: update-readme
+        entry: python tools/update_readme_help.py
+        language: python
+        pass_filenames: false

README.md

@@ -49,8 +49,8 @@ example .envrc:
 ```env
 #!/usr/bin/env bash
 
-export Q3RCON_CLI_HOST="localhost"
-export Q3RCON_CLI_PORT="28960"
+export Q3RCON_CLI_HOST=localhost
+export Q3RCON_CLI_PORT=28960
 export Q3RCON_CLI_PASSWORD="<rcon password>"
 ```
 
@@ -59,26 +59,27 @@ export Q3RCON_CLI_PASSWORD="<rcon password>"
 ```console
 Usage: q3rcon-cli [OPTIONS] COMMAND
 
-┏━ Subcommands ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
-┃ status       Prints the status of the server                                         ┃
-┃ mapname      Prints the current map name of the server                               ┃
-┃ maprotate    Rotates the map to the next one in the map rotation list                ┃
+┏━ Subcommands ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
 ┃ fastrestart  Executes a fast restart of the server                                               ┃
 ┃ gametype     Get or set the current gametype of the server                                       ┃
 ┃ hostname     Get or set the current hostname of the server                                       ┃
 ┃ map          Get the current map or change to a new one                                          ┃
+┃ mapname      Prints the current map name of the server                                           ┃
+┃ maprestart   Restarts the current map                                                            ┃
+┃ maprotate    Rotates the map to the next one in the map rotation list                            ┃
 ┃ plugins      Prints the currently loaded plugins of the server                                   ┃
-┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
+┃ status       Prints the status of the server                                                     ┃
+┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
 
-┏━ Options ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
+┏━ Options ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
 ┃ -i, --interactive   Whether to start in interactive mode (defaults to false)                     ┃
-┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
+┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
 
-┏━ Connection options ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
+┏━ Connection options ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
 ┃ -h, --host <HOST>          The host to connect to                                                ┃
 ┃ -p, --port <PORT>          The port to connect to                                                ┃
 ┃ -P, --password <PASSWORD>  The password for authentication                                       ┃
-┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
+┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
 ```
 
 ## Special Thanks

pyproject.toml

@@ -20,7 +20,7 @@ classifiers = [
   "Programming Language :: Python :: Implementation :: CPython",
   "Programming Language :: Python :: Implementation :: PyPy",
 ]
-dependencies = ["aio-q3-rcon>=1.0.0", "clypi>=1.8.2"]
+dependencies = ["aio-q3-rcon>=1.0.1", "clypi>=1.8.2"]
 
 [project.scripts]
 q3rcon-cli = "q3rcon_cli.cli:main"
@@ -33,6 +33,9 @@ Source = "https://github.com/onyx-and-iris/q3rcon-cli"
 [tool.hatch.version]
 path = "src/q3rcon_cli/__about__.py"
 
+[tool.hatch.envs.default]
+workspace.members = [{ path = "../aio-q3-rcon" }]
+
 [tool.hatch.envs.types]
 extra-dependencies = ["mypy>=1.0.0"]
 [tool.hatch.envs.types.scripts]

src/q3rcon_cli/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2026-present onyx-and-iris <code@onyxandiris.online>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.1.0"
+__version__ = '0.2.3'

src/q3rcon_cli/cli.py

@@ -1,22 +1,31 @@
 import clypi
-from aioq3rcon import Client
+from aioq3rcon import Client, IncorrectPasswordError
 from clypi import Command, arg
 from typing_extensions import override
 
+from . import console
 from .commands import (
     Fastrestart,
     Gametype,
     Hostname,
     Map,
     Mapname,
+    Maprestart,
     Maprotate,
     Plugins,
     Status,
 )
-from .console import Console
 
 Subcommands = (
-    Status | Mapname | Maprotate | Fastrestart | Gametype | Hostname | Map | Plugins
+    Fastrestart
+    | Gametype
+    | Hostname
+    | Map
+    | Mapname
+    | Maprestart
+    | Maprotate
+    | Plugins
+    | Status
 )
 
 
@@ -54,7 +63,7 @@ class Q3rconCli(Command):
         if self.interactive:
             await self.run_interactive()
         else:
-            await Status.run(self)
+            await Status(self.host, self.port, self.password).run()
 
     async def run_interactive(self):
         print(
@@ -87,16 +96,20 @@ class Q3rconCli(Command):
             ) as client:
                 try:
                     if response := await client.send_command(command):
-                        Console.print_response(response)
+                        console.out.print_response(response)
                 except TimeoutError:
-                    print(
-                        clypi.style(
-                            'Timeout waiting for response for command:', fg='red'
-                        ),
-                        clypi.style(command, fg='yellow'),
+                    console.err.print(
+                        f"Timeout waiting for response for command: '{command}'"
                     )
 
 
 def main():
+    try:
         cli = Q3rconCli().parse()
         cli.start()
+    except IncorrectPasswordError:
+        console.err.print('Incorrect password provided.')
+    except TimeoutError:
+        console.err.print(
+            f"Timeout waiting for response for command: '{type(cli.subcommand).__name__.lower()}'"
+        )

src/q3rcon_cli/commands/__init__.py

@@ -3,6 +3,7 @@ from .gametype import Gametype
 from .hostname import Hostname
 from .map import Map
 from .mapname import Mapname
+from .maprestart import Maprestart
 from .maprotate import Maprotate
 from .plugins import Plugins
 from .status import Status
@@ -16,4 +17,5 @@ __all__ = [
     'Hostname',
     'Map',
     'Plugins',
+    'Maprestart',
 ]

src/q3rcon_cli/commands/fastrestart.py

@@ -2,7 +2,7 @@ from aioq3rcon import Client
 from clypi import Command, arg
 from typing_extensions import override
 
-from q3rcon_cli.console import Console
+from q3rcon_cli import console
 
 
 class Fastrestart(Command):
@@ -16,4 +16,4 @@ class Fastrestart(Command):
     async def run(self):
         async with Client(self.host, self.port, self.password) as client:
             if response := await client.send_command('fast_restart'):
-                Console.print_response(response)
+                console.out.print_response(response)

src/q3rcon_cli/commands/gametype.py

@@ -2,7 +2,7 @@ from aioq3rcon import Client
 from clypi import Command, Positional, Spinner, arg
 from typing_extensions import override
 
-from q3rcon_cli.console import Console
+from q3rcon_cli import console
 
 
 class Gametype(Command):
@@ -26,7 +26,7 @@ class Gametype(Command):
         if not Gametype.new_gametype:
             async with Client(self.host, self.port, self.password) as client:
                 if response := await client.send_command('g_gametype'):
-                    Console.print_cvar(response)
+                    console.out.print_cvar(response)
             return
 
         async with Client(self.host, self.port, self.password) as client:
@@ -40,4 +40,4 @@ class Gametype(Command):
 
                 client.fragment_read_timeout = DEFAULT_FRAGMENT_READ_TIMEOUT
                 if response := await client.send_command('g_gametype'):
-                    Console.print_cvar(response)
+                    console.out.print_cvar(response)

src/q3rcon_cli/commands/hostname.py

@@ -2,7 +2,7 @@ from aioq3rcon import Client
 from clypi import Command, Positional, arg
 from typing_extensions import override
 
-from q3rcon_cli.console import Console
+from q3rcon_cli import console
 
 
 class Hostname(Command):
@@ -21,10 +21,10 @@ class Hostname(Command):
         if not self.new_hostname:
             async with Client(self.host, self.port, self.password) as client:
                 if response := await client.send_command('sv_hostname'):
-                    Console.print_cvar(response)
+                    console.out.print_cvar(response)
             return
 
         async with Client(self.host, self.port, self.password) as client:
             await client.send_command(f'sv_hostname {self.new_hostname}')
             if response := await client.send_command('sv_hostname'):
-                Console.print_cvar(response)
+                console.out.print_cvar(response)

src/q3rcon_cli/commands/map.py

@@ -2,7 +2,7 @@ from aioq3rcon import Client
 from clypi import Command, Positional, Spinner, arg
 from typing_extensions import override
 
-from q3rcon_cli.console import Console
+from q3rcon_cli import console
 
 
 class Map(Command):
@@ -21,7 +21,7 @@ class Map(Command):
         if not self.new_map:
             async with Client(self.host, self.port, self.password) as client:
                 if response := await client.send_command('mapname'):
-                    Console.print_cvar(response)
+                    console.out.print_cvar(response)
             return
 
         async with Spinner('Changing map...'):
@@ -30,4 +30,4 @@ class Map(Command):
             ) as client:
                 await client.send_command(f'map mp_{self.new_map.removeprefix("mp_")}')
                 if response := await client.send_command('mapname'):
-                    Console.print_cvar(response)
+                    console.out.print_cvar(response)

src/q3rcon_cli/commands/mapname.py

@@ -2,7 +2,7 @@ from aioq3rcon import Client
 from clypi import Command, arg
 from typing_extensions import override
 
-from q3rcon_cli.console import Console
+from q3rcon_cli import console
 
 
 class Mapname(Command):
@@ -16,4 +16,4 @@ class Mapname(Command):
     async def run(self):
         async with Client(self.host, self.port, self.password) as client:
             if response := await client.send_command('mapname'):
-                Console.print_cvar(response)
+                console.out.print_cvar(response)

src/q3rcon_cli/commands/maprestart.py

@@ -0,0 +1,22 @@
+from aioq3rcon import Client
+from clypi import Command, Spinner, arg
+from typing_extensions import override
+
+from q3rcon_cli import console
+
+
+class Maprestart(Command):
+    """Restarts the current map."""
+
+    host: str = arg(inherited=True)
+    port: int = arg(inherited=True)
+    password: str = arg(inherited=True)
+
+    @override
+    async def run(self):
+        async with Spinner('Restarting map...'):
+            async with Client(
+                self.host, self.port, self.password, fragment_read_timeout=1
+            ) as client:
+                if response := await client.send_command('map_restart'):
+                    console.out.print_response(response)

src/q3rcon_cli/commands/maprotate.py

@@ -2,7 +2,7 @@ from aioq3rcon import Client
 from clypi import Command, Spinner, arg
 from typing_extensions import override
 
-from q3rcon_cli.console import Console
+from q3rcon_cli import console
 
 
 class Maprotate(Command):
@@ -19,4 +19,4 @@ class Maprotate(Command):
                 self.host, self.port, self.password, fragment_read_timeout=1
             ) as client:
                 if response := await client.send_command('map_rotate'):
-                    Console.print_response(response)
+                    console.out.print_response(response)

src/q3rcon_cli/commands/plugins.py

@@ -2,7 +2,7 @@ from aioq3rcon import Client
 from clypi import Command, arg
 from typing_extensions import override
 
-from q3rcon_cli.console import Console
+from q3rcon_cli import console
 
 
 class Plugins(Command):
@@ -16,4 +16,4 @@ class Plugins(Command):
     async def run(self):
         async with Client(self.host, self.port, self.password) as client:
             if response := await client.send_command('plugins'):
-                Console.print_response(response)
+                console.out.print_response(response)

src/q3rcon_cli/commands/status.py

@@ -2,7 +2,7 @@ from aioq3rcon import Client
 from clypi import Command, Spinner, arg
 from typing_extensions import override
 
-from q3rcon_cli.console import Console
+from q3rcon_cli import console
 
 
 class Status(Command):
@@ -19,4 +19,4 @@ class Status(Command):
                 self.host, self.port, self.password, fragment_read_timeout=0.5
             ) as client:
                 if response := await client.send_command('status'):
-                    Console.print_status(response)
+                    console.out.print_status(response)

src/q3rcon_cli/console.py

@@ -5,6 +5,19 @@ from clypi import 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+'
@@ -30,18 +43,16 @@ class Console:
     )
 
     @staticmethod
-    def remove_colour_codes(s: str) -> str:
+    def _remove_colour_codes(s: str) -> str:
         """Remove Quake 3 colour codes from a string."""
-        return Console.COLOUR_CODE_REGEX.sub('', s)
+        return OutConsole.COLOUR_CODE_REGEX.sub('', s)
 
-    @staticmethod
-    def print_response(response: str):
-        response = Console.remove_colour_codes(response).removeprefix('print\n')
+    def print_response(self, response: str):
+        response = self._remove_colour_codes(response).removeprefix('print\n')
 
-        cprint(f'\n{response}\n', fg='yellow')
+        cprint(f'\n{response}\n', fg=self.style)
 
-    @staticmethod
-    def print_status(response: str):
+    def print_status(self, response: str):
         _slots = []
         _scores = []
         _pings = []
@@ -51,45 +62,48 @@ class Console:
 
         lines = response.splitlines()
         for line in lines:
-            if m := Console.STATUS_PLAYER_REGEX.match(line):
+            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(m.group('name'))
+                _names.append(self._remove_colour_codes(m.group('name')))
                 _ips.append(m.group('ip'))
 
         if not _slots:
-            cprint('\nNo players connected.\n', fg='yellow')
+            cprint('\nNo players connected.\n', fg=self.style)
             return
 
-        slots = clypi.boxed(_slots, title='Slot', width=15)
-        scores = clypi.boxed(_scores, title='Score', width=15)
-        pings = clypi.boxed(_pings, title='Ping', width=15)
-        guids = clypi.boxed(_guids, title='GUID', width=40)
-        names = clypi.boxed(_names, title='Name', width=30)
-        ips = clypi.boxed(_ips, title='IP', width=30)
+        slots = clypi.boxed(_slots, title='Slot', width=10, align='center')
+        scores = clypi.boxed(_scores, title='Score', width=10, align='center')
+        pings = clypi.boxed(_pings, title='Ping', width=10, align='center')
+        guids = clypi.boxed(_guids, title='GUID', width=len(max(_guids, key=len)) + 4)
+        names = clypi.boxed(_names, title='Name', width=len(max(_names, key=len)) + 4)
+        ips = clypi.boxed(_ips, title='IP', width=len(max(_ips, key=len)) + 4)
         print(f'\n{clypi.stack(slots, scores, pings, guids, names, ips, padding=0)}')
 
-    @staticmethod
-    def print_cvar(response: str):
-        response = Console.remove_colour_codes(response).removeprefix('print\n')
+    def print_cvar(self, response: str):
+        response = self._remove_colour_codes(response).removeprefix('print\n')
 
-        if m := Console.CVAR_REGEX.match(response):
+        if m := self.CVAR_REGEX.match(response):
             name = clypi.boxed(
-                [m.group('name')], title='Name', width=max(len(m.group('name')) + 4, 30)
+                [m.group('name')], title='Name', width=max(len(m.group('name')) + 4, 15)
             )
             value = clypi.boxed(
                 [m.group('value')],
                 title='Value',
-                width=max(len(m.group('value')) + 4, 30),
+                width=max(len(m.group('value')) + 4, 15),
             )
             default = clypi.boxed(
                 [m.group('default')],
                 title='Default',
-                width=max(len(m.group('default')) + 4, 30),
+                width=max(len(m.group('default')) + 4, 15),
             )
             info = clypi.boxed(
-                [m.group('info')], title='Info', width=max(len(m.group('info')) + 4, 30)
+                [m.group('info')], title='Info', width=max(len(m.group('info')) + 4, 15)
             )
             print(f'\n{clypi.stack(name, value, default, info, padding=0)}')
+
+
+out = OutConsole()
+err = ErrorConsole()

tools/update_readme_help.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python3
+import re
+import subprocess
+from pathlib import Path
+
+REPO_ROOT = Path(__file__).parent.parent
+README_PATH = REPO_ROOT / 'README.md'
+HELP_CMD = ['hatch', 'run', 'q3rcon-cli', '--help']
+
+
+def get_help_output():
+    result = subprocess.run(HELP_CMD, capture_output=True, text=True, check=True)
+    output = result.stdout.strip()
+    ansi_escape = re.compile(r'\x1B\[[0-?]*[ -/]*[@-~]')
+    output = ansi_escape.sub('', output)
+    return output
+
+
+def update_readme(help_text):
+    readme = README_PATH.read_text()
+
+    pattern = re.compile(
+        r'```console\nUsage: q3rcon-cli \[OPTIONS\] COMMAND.*?```', re.DOTALL
+    )
+    new_block = f'```console\n{help_text}\n```'
+    new_readme, n = pattern.subn(new_block, readme, count=1)
+    if n == 0:
+        raise RuntimeError(
+            'Could not find the q3rcon-cli usage console block to replace in README.md'
+        )
+    README_PATH.write_text(new_readme)
+    print('README.md updated with latest --help output.')
+
+
+def main():
+    help_text = get_help_output()
+    update_readme(help_text)
+
+
+if __name__ == '__main__':
+    main()