minor bump

pre_run_hook added, it runs before all subcommands.

gametype command now calls maprestart command directly if the -f flag is passed.

README.md

@@ -10,6 +10,9 @@
 ## Table of Contents
 
 - [Installation](#installation)
+- [Configuration](#configuration)
+- [Use](#use)
+- [Shell Completion](#shell-completion)
 - [License](#license)
 
 ## Installation
@@ -60,7 +63,7 @@ export Q3RCON_CLI_PASSWORD="<rcon password>"
 Usage: q3rcon-cli [OPTIONS] COMMAND
 
 ┏━ Subcommands ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
-┃ fastrestart  Executes a fast restart of the server                                               ┃
+┃ fastrestart  Executes a fast restart of the map                                                  ┃
 ┃ 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                                          ┃
@@ -73,6 +76,7 @@ Usage: q3rcon-cli [OPTIONS] COMMAND
 
 ┏━ Options ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
 ┃ -i, --interactive   Whether to start in interactive mode (defaults to false)                     ┃
+┃ -v, --version       Show the version and exit                                                    ┃
 ┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
 
 ┏━ Connection options ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
@@ -82,6 +86,14 @@ Usage: q3rcon-cli [OPTIONS] COMMAND
 ┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
 ```
 
+## Shell Completion
+
+Shell completion scripts are available for *bash*, *zsh*, and *fish*.
+
+```console
+q3rcon-cli --install-autocomplete
+```
+
 ## Special Thanks
 
 - [lapetus-11](https://github.com/Iapetus-11) for writing the [aio-q3-rcon](https://github.com/Iapetus-11/aio-q3-rcon) package.

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.2.5'
+__version__ = '0.4.0'

src/q3rcon_cli/cli.py

@@ -1,9 +1,11 @@
 import clypi
 from aioq3rcon import Client, IncorrectPasswordError
 from clypi import Command, Spinner, arg
+from clypi import parsers as cp
 from typing_extensions import override
 
-from . import console
+from . import config, console
+from .__about__ import __version__
 from .commands import (
     Fastrestart,
     Gametype,
@@ -37,6 +39,7 @@ class Q3rconCli(Command):
         help='The host to connect to',
         env='Q3RCON_CLI_HOST',
         group='Connection',
+        parser=cp.Str(min=1),
     )
     port: int = arg(
         28960,
@@ -44,6 +47,7 @@ class Q3rconCli(Command):
         help='The port to connect to',
         env='Q3RCON_CLI_PORT',
         group='Connection',
+        parser=cp.Int(min=1, max=65535),
     )
     password: str = arg(
         '',
@@ -51,43 +55,64 @@ class Q3rconCli(Command):
         help='The password for authentication',
         env='Q3RCON_CLI_PASSWORD',
         group='Connection',
+        parser=cp.Str(min=8),
     )
     interactive: bool = arg(
         False,
         short='i',
         help='Whether to start in interactive mode (defaults to false)',
     )
+    version: bool = arg(
+        False,
+        short='v',
+        help='Show the version and exit',
+    )
+
+    @override
+    async def pre_run_hook(self):
+        if self.subcommand is not None and self.interactive:
+            console.err.print(
+                'Cannot use subcommands in interactive mode.',
+                style='yellow',
+            )
+            self.print_help()
+            raise SystemExit(1)
+
+        if self.subcommand:
+            (
+                self.subcommand.timeout,
+                self.subcommand.fragment_read_timeout,
+                self.subcommand.interpret,
+            ) = config.get(self.subcommand.prog().split()[0].lower())
 
     @override
     async def run(self):
+        if self.version:
+            print(f'q3rcon-cli version: {clypi.style(__version__, fg="green")}')
+            return
+
         if self.interactive:
             await self.run_interactive()
-        else:
+            return
+
+        if self.subcommand is None:
             await Status(self.host, self.port, self.password).run()
+            return
+
+        self.print_help()
 
     async def run_interactive(self):
         print(
             clypi.style('Entering interactive mode. Type', fg='blue'),
-            clypi.style("'Q'", fg='red'),
+            clypi.style("'Q'", fg='yellow'),
             clypi.style('to quit.', fg='blue'),
         )
-
-        DEFAULT_TIMEOUT = 2
-        DEFAULT_FRAGMENT_READ_TIMEOUT = 0.25
         while command := input(clypi.style('cmd: ', fg='green')):
             if command.lower() == 'q':
                 break
 
-            TIMEOUTS = {
-                'status': (2, 1, False),
-                'fast_restart': (3, 1, True),
-                'map_restart': (3, 1, True),
-                'map': (3, 1, True),
-                'map_rotate': (3, 1, True),
-            }
-            timeout, fragment_read_timeout, interpret = TIMEOUTS.get(
-                command.split()[0].lower(),
-                (DEFAULT_TIMEOUT, DEFAULT_FRAGMENT_READ_TIMEOUT, False),
+            timeout, fragment_read_timeout, interpret = config.get(
+                command.split()[0].lower()
             )
 
             async with Spinner(f"Sending command: '{command}'", suffix='...'):
@@ -95,20 +120,20 @@ class Q3rconCli(Command):
                     self.host,
                     self.port,
                     self.password,
-                    timeout=timeout or DEFAULT_TIMEOUT,
-                    fragment_read_timeout=fragment_read_timeout
-                    or DEFAULT_FRAGMENT_READ_TIMEOUT,
+                    timeout=timeout,
+                    fragment_read_timeout=fragment_read_timeout,
                 ) as client:
                     try:
-                        if response := await client.send_command(
+                        response = await client.send_command(
                             command, interpret=interpret
-                        ):
-                            console.out.print_response(response)
+                        )
                     except TimeoutError:
                         console.err.print(
                             f"Timeout waiting for response for command: '{command}'"
                         )
 
+            console.out.print_response(response)
+
 
 def main():
     try:

src/q3rcon_cli/commands/fastrestart.py

@@ -6,7 +6,7 @@ from q3rcon_cli import console
 
 
 class Fastrestart(Command):
-    """Executes a fast restart of the server."""
+    """Executes a fast restart of the map."""
 
     host: str = arg(inherited=True)
     port: int = arg(inherited=True)
@@ -15,7 +15,15 @@ class Fastrestart(Command):
     @override
     async def run(self):
         async with Spinner('Executing fast restart', suffix='...'):
-            async with Client(self.host, self.port, self.password) as client:
-                response = await client.send_command('fast_restart', interpret=True)
+            async with Client(
+                self.host,
+                self.port,
+                self.password,
+                timeout=self.timeout,
+                fragment_read_timeout=self.fragment_read_timeout,
+            ) as client:
+                response = await client.send_command(
+                    'fast_restart', interpret=self.interpret
+                )
 
         console.out.print_response(response)

src/q3rcon_cli/commands/gametype.py

@@ -4,6 +4,8 @@ from typing_extensions import override
 
 from q3rcon_cli import console
 
+from .maprestart import Maprestart
+
 
 class Gametype(Command):
     """Get or set the current gametype of the server."""
@@ -35,11 +37,7 @@ class Gametype(Command):
                 await client.send_command(f'g_gametype {self.new_gametype}')
 
         if self.force:
-            async with Spinner('Forcing gametype change', suffix='...'):
-                async with Client(self.host, self.port, self.password) as client:
-                    client.timeout = 3
-                    client.fragment_read_timeout = 1
-                    await client.send_command('map_restart')
+            await Maprestart(self.host, self.port, self.password).configure_and_run()
 
         console.out.print(
             f'Gametype changed successfully to {self.new_gametype}.', style='green'

src/q3rcon_cli/commands/map.py

@@ -27,9 +27,16 @@ class Map(Command):
 
         async with Spinner('Changing map', suffix='...'):
             async with Client(
-                self.host, self.port, self.password, fragment_read_timeout=1
+                self.host,
+                self.port,
+                self.password,
+                timeout=self.timeout,
+                fragment_read_timeout=self.fragment_read_timeout,
             ) as client:
-                await client.send_command(f'map mp_{self.new_map.removeprefix("mp_")}')
+                await client.send_command(
+                    f'map mp_{self.new_map.removeprefix("mp_")}',
+                    interpret=self.interpret,
+                )
 
         console.out.print(
             f'Map changed to {self.new_map.removeprefix("mp_")}', style='green'

src/q3rcon_cli/commands/maprestart.py

@@ -2,7 +2,7 @@ from aioq3rcon import Client
 from clypi import Command, Spinner, arg
 from typing_extensions import override
 
-from q3rcon_cli import console
+from q3rcon_cli import config, console
 
 
 class Maprestart(Command):
@@ -12,12 +12,32 @@ class Maprestart(Command):
     port: int = arg(inherited=True)
     password: str = arg(inherited=True)
 
+    async def configure_and_run(self):
+        """Configures the command with the appropriate configuration and runs it.
+
+
+        This method is used if we invoke the maprestart command from another command (e.g. gametype),
+        since the pre_run_hook is not called in that case.
+        """
+        (
+            self.timeout,
+            self.fragment_read_timeout,
+            self.interpret,
+        ) = config.get(self.prog().split()[0].lower())
+        await self.run()
+
     @override
     async def run(self):
         async with Spinner('Restarting map', suffix='...'):
             async with Client(
-                self.host, self.port, self.password, timeout=3, fragment_read_timeout=1
+                self.host,
+                self.port,
+                self.password,
+                timeout=self.timeout,
+                fragment_read_timeout=self.fragment_read_timeout,
             ) as client:
-                response = await client.send_command('map_restart', interpret=True)
+                response = await client.send_command(
+                    'map_restart', interpret=self.interpret
+                )
 
         console.out.print_response(response)

src/q3rcon_cli/commands/maprotate.py

@@ -19,9 +19,11 @@ class Maprotate(Command):
                 self.host,
                 self.port,
                 self.password,
-                timeout=3,
-                fragment_read_timeout=1,
+                timeout=self.timeout,
+                fragment_read_timeout=self.fragment_read_timeout,
             ) as client:
-                response = await client.send_command('map_rotate', interpret=True)
+                response = await client.send_command(
+                    'map_rotate', interpret=self.interpret
+                )
 
         console.out.print_response(response)

src/q3rcon_cli/commands/status.py

@@ -15,9 +15,7 @@ class Status(Command):
     @override
     async def run(self):
         async with Spinner('Fetching status', suffix='...'):
-            async with Client(
-                self.host, self.port, self.password, fragment_read_timeout=1
-            ) as client:
+            async with Client(self.host, self.port, self.password) as client:
                 response = await client.send_command('status')
 
         console.out.print_status(response)

src/q3rcon_cli/config.py

@@ -0,0 +1,35 @@
+from collections import UserDict
+
+
+class Config(UserDict):
+    DEFAULT_TIMEOUT: int = 2
+    DEFAULT_FRAGMENT_READ_TIMEOUT: float = 0.25
+
+    def __init__(self):
+        self.data = {
+            'status': (2, 0.25, False),
+            'fast_restart': (3, 1, True),
+            'map_restart': (3, 1, True),
+            'map': (3, 1, True),
+            'map_rotate': (3, 1, True),
+        }
+
+    def __getitem__(self, key):
+        return self.data.get(
+            key, (self.DEFAULT_TIMEOUT, self.DEFAULT_FRAGMENT_READ_TIMEOUT, False)
+        )
+
+
+_config = Config()
+
+
+def get(key: str):
+    match key:
+        case 'fast_restart' | 'fastrestart':
+            return _config['fast_restart']
+        case 'map_restart' | 'maprestart':
+            return _config['map_restart']
+        case 'map_rotate' | 'maprotate':
+            return _config['map_rotate']
+        case _:
+            return _config[key]

src/q3rcon_cli/console.py

@@ -50,9 +50,8 @@ class OutConsole(Console):
         return OutConsole.COLOUR_CODE_REGEX.sub('', s)
 
     def print_response(self, response: str):
-        response = self._remove_colour_codes(response).removeprefix('print\n')
-
-        cprint(f'\n{response}\n', fg=self.style)
+        if response := self._remove_colour_codes(response).removeprefix('print\n'):
+            cprint(response, fg=self.style)
 
     def print_status(self, response: str):
         _slots = []