rollback updates

rollback updates

hikka/langpacks/de.yml

@@ -348,6 +348,9 @@ updater:
   cancel: "🚫 Abbrechen"
   _cls_doc: "Aktualisiert den Userbot"
   lavhost_update: "<emoji document_id=5208661316947955396>✌️</emoji> <b>Dein {} wird aktualisiert...</b>"
+  invalid_args: "<emoji document_id=5210952531676504517>🚫</emoji> <b>Sie müssen die Anzahl der Commits angeben, die zurückgesetzt werden sollen!</b>"
+  rollback_too_far: "<emoji document_id=5210952531676504517>🚫</emoji> <b>Du rollst zu weit zurück!</b>"
+  rollback_confirm: "❓ <b>Sind Sie sicher, dass Sie {num} Commit(s) zurücksetzen möchten?</b>"
   _cmd_doc_restart: "Startet den Userbot neu"
   _cmd_doc_source: "Zeigt den Link zum Quellcode des Projekts"
   _cmd_doc_update: "Lädt Updates für den Userbot herunter"

hikka/langpacks/en.yml

@@ -370,6 +370,9 @@ updater:
   lavhost_update: "<emoji document_id=5469986291380657759>✌️</emoji> <b>Your {} is updating...</b>"
   full_success: "<emoji document_id=6028215594707324419>👍</emoji> <b>Userbot is fully loaded! {}</b>\n<i>Full restart took {}s</i>"
   secure_boot_complete: "<emoji document_id=5472308992514464048>🔐</emoji> <b>Secure boot completed! {}</b>\n<i>Restart took {}s</i>"
+  invalid_args: "<emoji document_id=5210952531676504517>🚫</emoji> <b>You must provide number of commits to rollback!</b>"
+  rollback_too_far: "<emoji document_id=5210952531676504517>🚫</emoji> <b>You're rolling back too far!</b>"
+  rollback_confirm: "❓ <b>Are you sure you want to rollback {num} commit(-s)?</b>"
   _cmd_doc_restart: "Restarts the userbot"
   _cmd_doc_source: "Links the source code of this project"
   _cmd_doc_update: "Downloads userbot updates"

hikka/langpacks/ru.yml

@@ -339,6 +339,9 @@ updater:
   cancel: "🚫 Отмена"
   _cls_doc: "Обновляет юзербот"
   lavhost_update: "<emoji document_id=5469986291380657759>✌️</emoji> <b>Твой {} обновляется...</b>"
+  invalid_args: "<emoji document_id=5210952531676504517>🚫</emoji> <b>Вы должны указать количество коммитов для отката!</b>"
+  rollback_too_far: "<emoji document_id=5210952531676504517>🚫</emoji> <b>Вы откатываетесь слишком далеко!</b>"
+  rollback_confirm: "❓ <b>Вы уверены, что хотите откатить {num} коммит(-а,-ов)?</b>"
   _cmd_doc_restart: "Перезагружает юзербот"
   _cmd_doc_source: "Показать ссылку на исходный код проекта"
   _cmd_doc_update: "Скачивает обновления юзербота"

hikka/langpacks/ua.yml

@@ -348,6 +348,9 @@ updater:
   cancel: "🚫 Скасувати"
   _cls_doc: "Оновлює юзербот"
   lavhost_update: "<emoji document_id=5469986291380657759>✌️</emoji> <b>Твій {} оновлюється...</b>"
+  invalid_args: "<emoji document_id=5210952531676504517>🚫</emoji> <b>Вы должны указать количество коммитов для отката!</b>"
+  rollback_too_far: "<emoji document_id=5210952531676504517>🚫</emoji> <b>Ви відкочуєтесь занадто далеко!</b>"
+  rollback_confirm: "❓ <b>Вы уверены, что хотите откатить {num} коммит(-а,-ов)?</b>"
   _cmd_doc_restart: "Перезавантажує юзербот"
   _cmd_doc_source: "Показати посилання на вихідний код проекту"
   _cmd_doc_update: "Завантажує оновлення юзербота"

hikka/modules/updater.py

@@ -430,3 +430,37 @@ class UpdaterMod(loader.Module):
             inline_message_id=ms,
             text=self.inline.sanitise_text(msg),
         )
+
+    @loader.command()
+    async def rollback(self, message: Message):
+        if not (args := utils.get_args_raw(message)).isdigit():
+            await utils.answer(message, self.strings('invalid_args'))
+            return
+        if int(args) > 10:
+            await utils.answer(message, self.strings('rollback_too_far'))
+            return
+        form = await self.inline.form(
+            message=message,
+            text=self.strings('rollback_confirm').format(num=args),
+            reply_markup=[
+                [
+                    {
+                        "text": "✅",
+                        "callback": self.rollback_confirm,
+                        "args": [args],
+                    }
+                ],
+                [
+                    {
+                        "text": "❌",
+                        "action": "close",
+                    }
+                ]
+            ]
+        )
+
+    async def rollback_confirm(self, call: InlineCall, number: int):
+        await utils.answer(call, self.strings('rollback_process').format(num=number))
+        await asyncio.create_subprocess_shell(f'git reset --hard HEAD~{number}', stdout=asyncio.subprocess.PIPE)
+        await self.restart_common(call)
+