1.0.4: Add Okteto pinger, which will wake the pod up, when it goes to sleep

hikka/main.py

@@ -56,7 +56,7 @@ from .translations.core import Translator
 
 from math import ceil
 
-__version__ = (1, 0, 3)
+__version__ = (1, 0, 4)
 try:
     from .web import core
 except ImportError:

hikka/modules/okteto.py

@@ -0,0 +1,85 @@
+# █ █ ▀ █▄▀ ▄▀█ █▀█ ▀    ▄▀█ ▀█▀ ▄▀█ █▀▄▀█ ▄▀█
+# █▀█ █ █ █ █▀█ █▀▄ █ ▄  █▀█  █  █▀█ █ ▀ █ █▀█
+#
+#              © Copyright 2022
+#
+#          https://t.me/hikariatama
+#
+# 🔒 Licensed under the GNU GPLv3
+# 🌐 https://www.gnu.org/licenses/agpl-3.0.html
+
+from .. import loader
+import logging
+import asyncio
+import os
+import time
+from telethon.tl.functions.messages import GetScheduledHistoryRequest
+
+logger = logging.getLogger(__name__)
+
+
+@loader.tds
+class OktetoMod(loader.Module):
+    """Stuff related to Hikka Okteto cloud installation"""
+
+    strings = {"name": "Okteto"}
+
+    async def client_ready(self, client, db) -> None:
+        if "OKTETO" not in os.environ:
+            raise loader.LoadError(
+                "This module can be loaded only if userbot is installed to ☁️ Okteto"
+            )
+
+        self._db = db
+        self._client = client
+        self._env_wait_interval = 10
+        self._overall_polling_interval = 60 * 60
+        self._plan = 3 * 24 * 60 * 60
+        self._messages_interval = 60 * 60
+        self._exception_timeout = 10
+        self._send_interval = 5
+        self._bot = "@WebpageBot"
+        self._task = asyncio.ensure_future(self._okteto_poller())
+
+    async def on_unload(self) -> None:
+        self._task.cancel()
+
+    async def _okteto_poller(self) -> None:
+        """Creates queue to Webpage bot to reset Okteto polling after app goes to sleep"""
+        while True:
+            try:
+                if "OKTETO_URI" not in os.environ:
+                    await asyncio.sleep(self._env_wait_interval)
+                    continue
+
+                uri = os.environ["OKTETO_URI"]
+                current_queue = (
+                    await self._client(
+                        GetScheduledHistoryRequest(
+                            peer=self._bot,
+                            hash=0,
+                        ),
+                    )
+                ).messages
+
+                try:
+                    last_date = max(
+                        time.mktime(m.date.timetuple()) for m in current_queue
+                    )
+                except ValueError:
+                    last_date = time.time()
+
+                while last_date < time.time() + self._plan:
+                    last_date += self._messages_interval
+                    await self._client.send_message(
+                        self._bot,
+                        uri,
+                        schedule=last_date,
+                    )
+                    logger.debug(f"Scheduled Okteto pinger to {last_date}")
+                    await asyncio.sleep(self._send_interval)
+
+                await asyncio.sleep(self._overall_polling_interval)
+            except Exception:
+                logger.exception("Caught exception on Okteto poller")
+                await asyncio.sleep(self._exception_timeout)

hikka/web/root.py

@@ -136,11 +136,11 @@ class Web:
         return web.Response()
 
     async def okteto(self, request):
-        if not self._check_session(request):
+        if "OKTETO_URI" in os.environ:
-            return web.Response(body="WAIT")
+            return web.Response(status=418)
 
         text = await request.text()
-        main.hikka.okteto_uri = text
+        os.environ["OKTETO_URI"] = text
         return web.Response(body="URI_SAVED")
 
     async def tg_code(self, request):

web-resources/static/okteto.js

@@ -11,3 +11,5 @@ function okteto() {
         }
     })
 }
+
+okteto()