Integration of spoilers [WIP]

modules/commands/cancel.py

@@ -3,11 +3,12 @@ from pyrogram import filters
 from pyrogram.types import Message
 from pyrogram.client import Client
 from modules.utils import should_quote, logWrite, locale
-from modules.database import col_tmp
+from modules.database import col_tmp, col_spoilers
 from modules import custom_filters
 
 @app.on_message((custom_filters.enabled_applications | custom_filters.enabled_sponsorships) & ~filters.scheduled & filters.command("cancel", prefixes=["/"]))
 async def command_cancel(app: Client, msg: Message):
     col_tmp.delete_many( {"user": msg.from_user.id} )
+    col_spoilers.delete_many( {"user": msg.from_user.id, "completed": False} )
     await msg.reply_text(locale("cancel", "message", locale=msg.from_user), quote=should_quote(msg))
     logWrite(f"Cancelling all ongoing tmp operations for {msg.from_user.id}")

modules/commands/spoiler.py

@@ -0,0 +1,39 @@
+from app import app
+from pyrogram import filters
+from pyrogram.types import Message, ForceReply
+from pyrogram.client import Client
+from classes.holo_user import HoloUser, UserInvalidError, UserNotFoundError
+from modules.utils import locale
+from modules.database import col_spoilers
+from modules import custom_filters
+
+# Spoiler command ==============================================================================================================
+@app.on_message(custom_filters.member & ~filters.scheduled & filters.private & filters.command(["spoiler"], prefixes=["/"]))
+async def cmd_spoiler(app: Client, msg: Message):
+
+    try:
+        holo_user = HoloUser(msg.from_user)
+    except (UserInvalidError, UserNotFoundError):
+        return
+
+    if holo_user.application_state()[0] != "fill" and holo_user.sponsorship_state()[0] != "fill":
+
+        if col_spoilers.find_one( {"user": msg.from_user.id, "completed": False} ) is None:
+
+            col_spoilers.insert_one(
+                {
+                    "user": msg.from_user.id,
+                    "completed": False,
+                    "description": None,
+                    "photo": None,
+                    "video": None,
+                    "animation": None,
+                    "text": None
+                }
+            )
+
+            await msg.reply_text(locale("spoiler_started", "message", locale=msg.from_user), reply_markup=ForceReply(placeholder=locale("spoiler_content", "force_reply", locale=msg.from_user)))
+
+        else:
+            await msg.reply_text(locale("spoiler_unfinished", "message", locale=msg.from_user))
+# ==============================================================================================================================

modules/commands/start.py

@@ -3,8 +3,10 @@ from pyrogram import filters
 from pyrogram.types import ReplyKeyboardMarkup, Message
 from pyrogram.client import Client
 from modules.utils import locale, logWrite
-from modules.database import col_users
+from modules.database import col_users, col_spoilers
 from modules import custom_filters
+from bson.objectid import ObjectId
+from bson.errors import InvalidId
 
 # Start command ================================================================================================================
 @app.on_message(custom_filters.enabled_applications & ~filters.scheduled & filters.private & filters.command(["start"], prefixes=["/"]))
@@ -26,4 +28,18 @@ async def cmd_start(app: Client, msg: Message):
 
         logWrite(f"User {msg.from_user.id} started bot interaction")
         await msg.reply_text(locale("start", "message", locale=msg.from_user), reply_markup=ReplyKeyboardMarkup(locale("welcome", "keyboard", locale=msg.from_user), resize_keyboard=True))
+
+    if len(msg.command) > 1:
+        try:
+            spoiler = col_spoilers.find_one( {"_id": ObjectId(msg.command[1])} )
+            if spoiler["photo"] is not None:
+                await msg.reply_photo(spoiler["photo"])
+            if spoiler["video"] is not None:
+                await msg.reply_video(spoiler["video"])
+            if spoiler["animation"] is not None:
+                await msg.reply_animation(spoiler["animation"])
+            if spoiler["text"] is not None:
+                await msg.reply_text(spoiler["text"])
+        except InvalidId:
+            await msg.reply_text(f"Got an invalid ID {msg.command[1]}")
 # ==============================================================================================================================