Update main.py

main.py

@@ -7,6 +7,7 @@ import urllib.parse
 import traceback
 import subprocess
 import shutil
+import time
 from flask import Flask
 from rubpy.bot import BotClient, filters
 
@@ -33,7 +34,6 @@ if not bot_token or not GITHUB_TOKEN:
 bot = BotClient(bot_token)
 BOT_GUID = None
 
-# قفل برای جلوگیری از تداخل کامیت‌های همزمان گیت
 git_lock = asyncio.Lock()
 
 # ==============================================================================
@@ -77,7 +77,7 @@ async def download_large_file(url, chat_id, message_id, client):
 # 🟢 تابع خرد کردن فایل برای WinRAR
 # ==============================================================================
 def split_file_for_winrar(file_path, dest_dir, base_filename):
-    chunk_size = 95 * 1024 * 1024  # محدودیت 95 مگابایتی برای گیت‌هاب
+    chunk_size = 95 * 1024 * 1024 
     part_num = 1
     
     os.makedirs(dest_dir, exist_ok=True)
@@ -96,7 +96,7 @@ def split_file_for_winrar(file_path, dest_dir, base_filename):
             part_num += 1
 
 # ==============================================================================
-# 🟢 پردازش و کامیت در شاخه مجزا (Orphan Branch) گیت‌هاب
+# 🟢 پردازش و کامیت در گیت‌هاب (با درج زمان در نام شاخه)
 # ==============================================================================
 async def process_and_commit_to_github(client, chat_id, message_id, url):
     file_path, filename, status_msg_id = await download_large_file(url, chat_id, message_id, client)
@@ -113,8 +113,9 @@ async def process_and_commit_to_github(client, chat_id, message_id, url):
             except: pass
 
         async with git_lock:
-            # استفاده از نام شاخه مجزا برای هر فایل
-            branch_name = f"DL-{uuid.uuid4().hex[:6]}"
+            # جاسازی Timestamp (زمان فعلی) در نام شاخه برای بررسی‌های آینده
+            current_timestamp = int(time.time())
+            branch_name = f"DL-{current_timestamp}-{uuid.uuid4().hex[:6]}"
             repo_url = f"https://oauth2:{GITHUB_TOKEN}@github.com/{GITHUB_REPO}.git"
             work_dir = "/app/git_workspace"
             
@@ -126,28 +127,27 @@ async def process_and_commit_to_github(client, chat_id, message_id, url):
             subprocess.run(["git", "config", "user.email", "bot@aialpha.local"], cwd=work_dir)
             subprocess.run(["git", "config", "user.name", "AI Alpha Bot"], cwd=work_dir)
             
-            # ایجاد یک شاخه کاملاً خالی (بدون تاریخچه و فایل‌های قبلی)
             subprocess.run(["git", "checkout", "--orphan", branch_name], cwd=work_dir, check=True)
             subprocess.run(["git", "rm", "-rf", "."], cwd=work_dir, stderr=subprocess.DEVNULL)
             
-            # خرد کردن فایل و قرار دادن در پوشه‌ای همنام فایل
             file_folder = os.path.join(work_dir, filename)
             split_file_for_winrar(file_path, file_folder, filename)
             
-            # کامیت و پوش کردن شاخه جدید به سرور
             subprocess.run(["git", "add", "."], cwd=work_dir, check=True)
             subprocess.run(["git", "commit", "-m", f"Add parts for {filename}"], cwd=work_dir, check=True)
             subprocess.run(["git", "push", "origin", branch_name], cwd=work_dir, check=True)
             
-            # لینک آرشیو مخصوص این شاخه (فقط حاوی همین فایل)
             zip_download_link = f"https://github.com/{GITHUB_REPO}/archive/refs/heads/{branch_name}.zip"
             
+            # پیغام زیبای اطلاع‌رسانی به کاربر
             success_text = (
-                f"🎉 فایل با موفقیت ذخیره شد!\n\n"
-                f"📁 نام: {filename}\n"
-                f"🔗 لینک دانلود مستقیم (بدون فایل‌های اضافه):\n{zip_download_link}\n\n"
-                f"💡 راهنما: فایل Zip بالا را دانلود و Extract کنید تا به پارت‌های اصلی برسید.\n\n"
-                f"🗑 **برای حذف این فایل از سرور و آزادسازی فضا، کافیست لینک بالا را مجدداً برای من ارسال کنید.**"
+                f"🎉 فایل شما با موفقیت آماده شد!\n\n"
+                f"📁 **نام فایل:** {filename}\n"
+                f"🔗 **لینک دانلود مستقیم (بدون مصرف ترافیک بین‌الملل):**\n{zip_download_link}\n\n"
+                f"💡 **راهنما:** فایل بالا را دانلود کرده و استخراج (Extract) کنید.\n\n"
+                f"⚠️ **توجه مهم:**\n"
+                f"جهت حفظ کیفیت خدمات و آزادسازی فضای سرور، این فایل **دقیقاً ۲ ساعت دیگر به صورت خودکار حذف خواهد شد.** ⏳\n"
+                f"لطفاً در این بازه زمانی نسبت به دانلود فایل خود اقدام نمایید."
             )
             
             if status_msg_id:
@@ -168,41 +168,53 @@ async def process_and_commit_to_github(client, chat_id, message_id, url):
             try: shutil.rmtree("/app/git_workspace")
             except: pass
 
+
 # ==============================================================================
-# 🟢 تابع حذف فایل از گیت‌هاب از طریق API
+# 🟢 سیستم هوشمند حذف خودکار فایل‌های منقضی شده (بیش از ۲ ساعت)
 # ==============================================================================
-async def delete_github_branch(client, chat_id, message_id, url):
-    try:
-        # استخراج نام شاخه از لینک ربات
-        branch_name = url.split("/archive/refs/heads/")[1].replace(".zip", "").strip()
-        
-        status_msg = await client.send_message(chat_id, "🗑 در حال پاکسازی فایل از سرور گیت‌هاب...", reply_to_message_id=message_id)
-        status_msg_id = getattr(status_msg, 'message_id', None)
-        if not status_msg_id and isinstance(status_msg, dict):
-            status_msg_id = status_msg.get('message_update', {}).get('message_id') or status_msg.get('message_id')
+async def cleanup_expired_branches():
+    """این تابع هر ۳۰ دقیقه ران می‌شود و شاخه‌های قدیمی‌تر از ۲ ساعت را مستقیم از گیت‌هاب پاک می‌کند."""
+    while True:
+        try:
+            api_url = f"https://api.github.com/repos/{GITHUB_REPO}/git/refs/heads"
+            headers = {
+                "Authorization": f"token {GITHUB_TOKEN}",
+                "Accept": "application/vnd.github.v3+json"
+            }
             
-        api_url = f"https://api.github.com/repos/{GITHUB_REPO}/git/refs/heads/{branch_name}"
-        headers = {
-            "Authorization": f"token {GITHUB_TOKEN}",
-            "Accept": "application/vnd.github.v3+json"
-        }
-        
-        # درخواست حذف شاخه (و فایل‌های داخلش)
-        async with aiohttp.ClientSession() as session:
-            async with session.delete(api_url, headers=headers) as resp:
-                if resp.status == 204:
-                    result_text = "✅ فایل و تمامی پارت‌های آن با موفقیت از ریپازیتوری حذف شدند و فضای سرور آزاد شد!"
-                elif resp.status == 422 or resp.status == 404:
-                    result_text = "❌ این فایل قبلاً حذف شده است یا وجود ندارد."
-                else:
-                    result_text = f"❌ خطا در حذف فایل. کد وضعیت: {resp.status}"
-                    
-        if status_msg_id:
-            try: await client.edit_message_text(chat_id, status_msg_id, result_text)
-            except: await client.send_message(chat_id, result_text)
+            async with aiohttp.ClientSession() as session:
+                async with session.get(api_url, headers=headers) as resp:
+                    if resp.status == 200:
+                        refs = await resp.json()
+                        current_time = int(time.time())
+                        
+                        for ref in refs:
+                            ref_name = ref.get("ref", "")
+                            # بررسی اینکه آیا شاخه مربوط به فایل‌های دانلودی ماست
+                            if "refs/heads/DL-" in ref_name:
+                                branch_name = ref_name.split("/")[-1]
+                                parts = branch_name.split("-")
+                                
+                                # استخراج Timestamp از نام شاخه
+                                if len(parts) >= 3 and parts[1].isdigit():
+                                    creation_time = int(parts[1])
+                                    
+                                    # اگر بیش از ۷۲۰۰ ثانیه (۲ ساعت) گذشته بود، حذف شود
+                                    if current_time - creation_time > 7200:
+                                        delete_url = f"https://api.github.com/repos/{GITHUB_REPO}/git/refs/heads/{branch_name}"
+                                        await session.delete(delete_url, headers=headers)
+                                        print(f"🗑 Branch {branch_name} automatically deleted (Expired).")
+        except Exception as e:
+            print(f"❌ Error in Auto-Cleanup: {e}")
             
-    except Exception as e:
-        await client.send_message(chat_id, f"❌ خطا در پردازش لینک حذف:\n{str(e)[:150]}")
+        # صبر برای ۳۰ دقیقه تا اجرای دور بعدی پاکسازی
+        await asyncio.sleep(1800)
+
+def run_cleanup_system():
+    """اجرای حلقه ایزوله برای تسک حذف خودکار در پس‌زمینه"""
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    loop.run_until_complete(cleanup_expired_branches())
 
 # ==============================================================================
 # 🟢 هندلر روبیکا
@@ -229,21 +241,20 @@ async def main_handler(client, update):
         if not user_text_str: return
 
         if user_text_str.lower() in ["/start", "سلام"]:
-            await client.send_message(chat_id, "👋 سلام! لینک مستقیم هر فایلی رو بفرست تا برات خرد کنم و بفرستم روی گیت‌هاب (قابل دانلود با اینترنت ملی).\nبرای حذف فایل‌های قبلی، کافیه لینک گیتهابی که بهت دادم رو برام بفرستی.")
+            await client.send_message(chat_id, "👋 سلام! لینک مستقیم هر فایلی رو بفرست تا برات خرد کنم و بفرستم روی گیت‌هاب (قابل دانلود با اینترنت ملی).\n\n⏳ **فایل‌ها جهت مدیریت سرور، دو ساعت بعد از ساخته‌شدن به صورت کاملاً خودکار حذف میشن.**")
             return
 
-        # اگر کاربر لینک گیت‌هابی که قبلا خود ربات ساخته را فرستاد (جهت حذف)
-        if f"github.com/{GITHUB_REPO}/archive/refs/heads/DL-" in user_text_str:
-            asyncio.create_task(delete_github_branch(client, chat_id, msg_id, user_text_str))
-            return
-
-        # اگر لینک دانلود معمولی بود
         if user_text_str.lower().startswith("http"):
             asyncio.create_task(process_and_commit_to_github(client, chat_id, msg_id, user_text_str))
             
     except Exception: traceback.print_exc()
 
 if __name__ == "__main__":
+    # اجرای Flask در یک Thread مجزا
     threading.Thread(target=run_flask, daemon=True).start()
-    print("🚀 ربات کلاستر گیت‌هاب روشن شد...")
+    
+    # اجرای سیستم پاکسازی خودکار در یک Thread مجزا
+    threading.Thread(target=run_cleanup_system, daemon=True).start()
+    
+    print("🚀 ربات کلاستر گیت‌هاب همراه با سیستم پاکسازی خودکار روشن شد...")
     bot.run()