Update main.py

main.py

@@ -16,12 +16,14 @@ logging.basicConfig(level=logging.INFO)
 
 WELCOME_TEXT = """
 Hey! {first_name}
+
 Saya adalah bot untuk mengunduh video PornoHub di telegram.
 
 - Jangan share video ke group (mending private bot aman)
 
 - Command: /hubsearch boobs
 - Command: /hubdl link xnxx
+- Command: /freemode
 
 Powered by @xtdevs
 """
@@ -68,6 +70,21 @@ def set_user_update_in_db(user_id: int, first_name: str, username: str):
         return None
     return response.json()
 
+def toggle_free_mode(user_id):
+    try:
+        response = requests.post(f"https://private-akeno.randydev.my.id/api/v2/akenohub/freemode?user_id={user_id}")
+        return response.json()
+    except requests.RequestException as e:
+        return {"message": f"Error toggling free mode: {str(e)}", "free_mode_enabled": False}
+
+def check_expiration(user_id):
+    try:
+        response = requests.get(f"https://private-akeno.randydev.my.id/api/v2/akenohub/check?user_id={user_id}")
+        response.raise_for_status()
+        return response.json()
+    except requests.RequestException as e:
+        return {"expired": True, "free_mode": False, "message": f"Error checking expiration: {str(e)}"}
+
 @client.on_message(
     filters.incoming
     & filters.private,
@@ -100,7 +117,6 @@ async def must_join_channel(bot: Client, msg: Message):
     except ChatAdminRequired:
         pass
 
-
 async def check_membership(channel_id, bot, msg):
     try:
         user_id = msg.from_user.id if msg.from_user else 0
@@ -179,8 +195,13 @@ async def searchhub(client: Client, message: Message):
         return await message.reply_text("please search for pornohub.")
     pro = await message.reply_text("Processing.....")
     if query in not_allowed:
-        return await pro.edit_text("I don't understand, can you help me?") 
+        return await pro.edit_text("I don't understand, can you help me?")
+    user_status = check_expiration(user_id)
+    if user_status["expired"] and not user_status["free_mode"]:
+        await pro.edit_text("Your session has expired, please renew or activate free mode.")
+        return
     try:
+        protect_content = not user_status["free_mode"]
         response = await api.x_search(query=query)
         if response is None:
             return await pro.edit_text("nothing found gay")
@@ -203,7 +224,7 @@ async def searchhub(client: Client, message: Message):
             caption=f"• Powered by {client.me.mention}",
             thumb=thumb,
             has_spoiler=True,
-            protect_content=True,
+            protect_content=protect_content,
             progress=progress,
             progress_args=(
                 pro,
@@ -230,7 +251,12 @@ async def porno_download(client: Client, message: Message):
     if not link.startswith("https://www.xnxx.com/"):
         return await message.reply_text("invalid link.")
     pro = await message.reply_text("Processing.....")
+    user_status = check_expiration(user_id)
+    if user_status["expired"] and not user_status["free_mode"]:
+        await pro.edit_text("Your session has expired, please renew or activate free mode.")
+        return
     try:
+        protect_content = not user_status["free_mode"]
         file_path, thumb, _ = await api.x_download(url=link, is_stream=True)
         upload_text = f"**⬆️ 𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 video ...**"
         await pro.edit_text(upload_text)
@@ -250,7 +276,7 @@ async def porno_download(client: Client, message: Message):
             caption=f"• Powered by {client.me.mention}",
             thumb=thumb,
             has_spoiler=True,
-            protect_content=True,
+            protect_content=protect_content,
             progress=progress,
             progress_args=(
                 pro,
@@ -264,4 +290,13 @@ async def porno_download(client: Client, message: Message):
     except Exception as e:
         await pro.edit_text(str(e))
 
-client.run()
+@client.on_message(filters.command("freemode"))
+async def freemode_handler(client, message):
+    user_id = message.from_user.id
+    free_mode_status = toggle_free_mode(user_id)
+    if free_mode_status.get("free_mode_enabled"):
+        await message.reply(f"Free mode activated for user {user_id}.")
+    elif "wait" in free_mode_status["message"]:
+        await message.reply(f"{free_mode_status['message']}")
+    else:
+        await message.reply(f"Free mode deactivated for user {user_id}.")