feat: add feature flag (#344)

userbotindo · Jul 12, 2023 · 7a08b4a · 7a08b4a
1 parent 4d027b4
commit 7a08b4a
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 9 deletions.
diff --git a/anjani/core/telegram_bot.py b/anjani/core/telegram_bot.py
@@ -183,9 +183,10 @@ async def start(self: "Anjani") -> None:
 
         self.log.info("Bot is ready")
 
-        self.log.info("Catching up on missed events")
-        await self.dispatch_missed_events()
-        self.log.info("Finished catching up")
+        if not self.config.is_flag_active("disable_catchup"):
+            self.log.info("Catching up on missed events")
+            await self.dispatch_missed_events()
+            self.log.info("Finished catching up")
 
         # Dispatch final late start event
         await self.dispatch_event("started")

diff --git a/anjani/plugins/main.py b/anjani/plugins/main.py
@@ -100,16 +100,21 @@ async def on_stop(self) -> None:
             if not await file.exists():
                 return
 
-            data = await self.bot.client.invoke(GetState())
+            state = {}
+            if not self.bot.config.is_flag_active("disable_catchup"):
+                data = await self.bot.client.invoke(GetState())
+                state = {
+                    "date": data.date,
+                    "pts": data.pts,
+                    "qts": data.qts,
+                    "seq": data.seq,
+                }
             await self.db.update_one(
                 {"_id": sha256(self.bot.config.BOT_TOKEN.encode()).hexdigest()},
                 {
                     "$set": {
                         "session": Binary(await file.read_bytes()),
-                        "date": data.date,
-                        "pts": data.pts,
-                        "qts": data.qts,
-                        "seq": data.seq,
+                        **state,
                     }
                 },
                 upsert=True,

diff --git a/anjani/util/config.py b/anjani/util/config.py
@@ -23,6 +23,7 @@ class Config:
 
     LOGIN_URL: Optional[str]
     PLUGIN_FLAG: list[str]
+    FEATURE_FLAG: list[str]
 
     IS_CI: bool
 
@@ -45,7 +46,12 @@ def __init__(self) -> None:
         self.SW_API = getenv("SW_API")
 
         self.LOGIN_URL = getenv("LOGIN_URL")
-        self.PLUGIN_FLAG = list(filter(None, [i.strip() for i in getenv("PLUGIN_FLAG", "").split(";")]))
+        self.PLUGIN_FLAG = list(
+            filter(None, [i.strip() for i in getenv("PLUGIN_FLAG", "").split(";")])
+        )
+        self.FEATURE_FLAG = list(
+            filter(None, [i.strip() for i in getenv("FEATURE_FLAG", "").split(";")])
+        )
 
         self.IS_CI = getenv("IS_CI", "false").lower() == "true"
 
@@ -65,3 +71,6 @@ def __init__(self) -> None:
 
     def is_plugin_disabled(self, name: str) -> bool:
         return f'disable_{name.lower().replace(" ", "_")}_plugin' in self.PLUGIN_FLAG
+
+    def is_flag_active(self, name: str) -> bool:
+        return name in self.FEATURE_FLAG