From c756c6b1dc1a9b38b9630ed1d5746306ef158af0 Mon Sep 17 00:00:00 2001
From: profitroll <vozhd.kk@gmail.com>
Date: Wed, 3 Jan 2024 22:37:04 +0100
Subject: [PATCH] Closes #33

---
 src/libbot/pyrogram/classes/client.py | 82 +++++++++++++++++++++------
 tests/data/locale/en.json             | 12 ++++
 tests/data/locale/uk.json             | 12 ++++
 3 files changed, 89 insertions(+), 17 deletions(-)

diff --git a/src/libbot/pyrogram/classes/client.py b/src/libbot/pyrogram/classes/client.py
index c232280..90422f6 100644
--- a/src/libbot/pyrogram/classes/client.py
+++ b/src/libbot/pyrogram/classes/client.py
@@ -103,7 +103,7 @@ class PyroClient(Client):
 
         self.bot_locale: BotLocale = BotLocale(
             default_locale=self.config["locale"],
-            locales_root=(Path("locale") if locales_root is None else locales_root)
+            locales_root=(Path("locale") if locales_root is None else locales_root),
         )
         self.default_locale: str = self.bot_locale.default
         self.locales: dict = self.bot_locale.locales
@@ -129,6 +129,53 @@ class PyroClient(Client):
             getpid(),
         )
 
+        # Register default bot's info
+        try:
+            await self.set_bot_info(
+                name=self._("name", "bot"),
+                about=self._("about", "bot"),
+                description=self._("description", "bot"),
+                lang_code="",
+            )
+            logger.info(
+                "Bot's info for the default locale %s has been updated",
+                self.default_locale,
+            )
+        except KeyError:
+            logger.warning(
+                "Default locale %s has incorrect keys or values in bot section",
+                self.default_locale,
+            )
+
+        # Register bot's info for each available locale
+        for locale_code in self.locales:
+            locale = self.locales[locale_code]
+
+            if "metadata" not in locale or ("codes" not in locale["metadata"]):
+                logger.warning(
+                    "Locale %s is missing metadata or metadata.codes key", locale_code
+                )
+                continue
+
+            for code in locale["metadata"]["codes"]:
+                try:
+                    await self.set_bot_info(
+                        name=locale["bot"]["name"],
+                        about=locale["bot"]["about"],
+                        description=locale["bot"]["description"],
+                        lang_code=code,
+                    )
+                    logger.info(
+                        "Bot's info for the locale %s has been updated",
+                        self.default_locale,
+                    )
+                except KeyError:
+                    logger.warning(
+                        "Locale %s has incorrect keys or values in bot section",
+                        locale_code,
+                    )
+
+        # Send a message to the bot's reports chat about the startup
         try:
             await self.send_message(
                 chat_id=self.owner
@@ -136,22 +183,23 @@ class PyroClient(Client):
                 else self.config["reports"]["chat_id"],
                 text=f"Bot started PID `{getpid()}`",
             )
-
-            if self.scheduler is None:
-                return
-
-            if register_commands:
-                self.scheduler.add_job(
-                    self.register_commands,
-                    trigger="date",
-                    run_date=datetime.now() + timedelta(seconds=5),
-                    kwargs={"command_sets": await self.collect_commands()},
-                )
-
-            self.scheduler.start()
         except BadRequest:
             logger.warning("Unable to send message to report chat.")
 
+        if self.scheduler is None:
+            return
+
+        # Schedule the task to register all commands
+        if register_commands:
+            self.scheduler.add_job(
+                self.register_commands,
+                trigger="date",
+                run_date=datetime.now() + timedelta(seconds=5),
+                kwargs={"command_sets": await self.collect_commands()},
+            )
+
+        self.scheduler.start()
+
     async def stop(self, exit_completely: bool = True) -> None:
         try:
             await self.send_message(
@@ -170,16 +218,16 @@ class PyroClient(Client):
         if exit_completely:
             try:
                 exit()
-            except SystemExit as exp:
+            except SystemExit as exc:
                 raise SystemExit(
                     "Bot has been shut down, this is not an application error!"
-                ) from exp
+                ) from exc
 
     async def collect_commands(self) -> Union[List[CommandSet], None]:
         """Gather list of the bot's commands
 
         ### Returns:
-            * `List[CommandSet]`: List of the commands' sets
+            * `List[CommandSet]`: List of the commands' sets.
         """
         command_sets = None
 
diff --git a/tests/data/locale/en.json b/tests/data/locale/en.json
index 1b44b7e..914a805 100644
--- a/tests/data/locale/en.json
+++ b/tests/data/locale/en.json
@@ -1,4 +1,16 @@
 {
+    "metadata": {
+        "flag": "🇬🇧",
+        "name": "English",
+        "codes": [
+            "en"
+        ]
+    },
+    "bot": {
+        "name": "",
+        "about": "", 
+        "description": ""
+    },
     "foo": "bar",
     "messages": {
         "example": "okay"
diff --git a/tests/data/locale/uk.json b/tests/data/locale/uk.json
index 34ff1fa..444b17b 100644
--- a/tests/data/locale/uk.json
+++ b/tests/data/locale/uk.json
@@ -1,4 +1,16 @@
 {
+    "metadata": {
+        "flag": "🇺🇦",
+        "name": "Українська",
+        "codes": [
+            "uk"
+        ]
+    },
+    "bot": {
+        "name": "",
+        "about": "", 
+        "description": ""
+    },
     "foo": "бар",
     "messages": {
         "example": "окей"