git push Merge remote-tracking branch 'template/dev' into dev

Author: Paillat-dev

src/config/models.py

@@ -40,13 +40,23 @@ def __bool__(self) -> bool:
         return self.enabled
 
 
+class RestConfig(BaseModel):
+    enabled: bool = False
+    health: bool = False
+    host: str = "0.0.0.0"  # noqa: S104
+    port: int = 6000
+
+    def __bool__(self) -> bool:
+        return self.enabled
+
+
 class BotConfig(BaseModel):
     token: str
     public_key: str | None = None
     prefix: PrefixConfig | str = PrefixConfig(prefix="!", enabled=False)
     slash: SlashConfig = SlashConfig(enabled=False)
     cache: CacheConfig = CacheConfig()
-    rest: bool = False
+    rest: RestConfig = RestConfig()
 
 
 class LoggingConfig(BaseModel):

src/start.py

@@ -16,7 +16,7 @@
 
 from src import custom, i18n
 from src.config import config
-from src.config.models import BotConfig, Extension
+from src.config.models import BotConfig, Extension, RestConfig
 from src.i18n.classes import ExtensionTranslation
 from src.log import logger, patch
 from src.utils import setup_func, unzip_extensions, validate_module
@@ -28,17 +28,18 @@
     FunctionlistType = list[tuple[Callable[..., Any], Extension]]
 
 
-async def start_bot(bot: custom.Bot, token: str, public_key: str | None = None) -> None:
+async def start_bot(bot: custom.Bot, token: str, rest_config: RestConfig, public_key: str | None = None) -> None:
     try:
         if isinstance(bot, custom.CustomRestBot):
             if not public_key:
                 raise TypeError("CustomRestBot requires a public key to start.")  # noqa: TRY301
             await bot.start(
                 token=token,
                 public_key=public_key,
+                health=rest_config.health,
                 uvicorn_options={
-                    "host": "0.0.0.0",  # noqa: S104
-                    "port": 6000,
+                    "host": rest_config.host,
+                    "port": rest_config.port,
                 },
             )
         else:
@@ -172,7 +173,7 @@ async def setup_and_start_bot(
         bot.prefixed_commands = {}
     if not config.slash:
         bot._pending_application_commands = []  # pyright: ignore[reportPrivateUsage]
-    await start_bot(bot, config.token, config.public_key)
+    await start_bot(bot, config.token, config.rest, config.public_key)
 
 
 async def setup_and_start_backend(