Update main.py

main.py

@@ -1,73 +1,149 @@
-import os, asyncio, aiohttp, threading, uuid, base64, time, builtins
+import os
+import asyncio
+import aiohttp
+import threading
+import uuid
+import urllib.parse
+import traceback
+import base64
 from flask import Flask
+
+# نصب کتابخانه rubpy==6.9.9 الزامی است
 from rubpy import Client, filters
 
+# ==============================================================================
+# 🟢 تنظیمات وب‌سرور برای زنده نگه داشتن اسپیس
+# ==============================================================================
 app = Flask(__name__)
-@app.route('/')
-def home(): return "🚀 ربات در حال اجراست. بخش Logs را چک کنید."
 
-def run_flask(): app.run(host="0.0.0.0", port=7860)
+@app.route('/')
+def home():
+    return "🚀 سلف‌بات آپلودر ۲ گیگابایتی فعال است! لینک‌ها را در پیام‌های ذخیره شده بفرستید."
 
-# ترفند دور زدن ورودی کیبورد
-def mock_input(prompt):
-    p = prompt.lower()
-    if "phone" in p or "شماره" in p:
-        return os.environ.get("phone", "989120132730")
-    if "code" in p or "کد" in p:
-        # اگر کد در سکرت‌ها نبود، اینجا گیر می‌کنه تا تو بری سکرت رو آپدیت کنی
-        while True:
-            c = os.environ.get("code", "")
-            if len(c) > 4: return c
-            time.sleep(2)
-    return "y"
+def run_flask():
+    # پورت ۷۸۶۰ مخصوص هاگینگ فیس است
+    app.run(host="0.0.0.0", port=7860, threaded=True)
 
-builtins.input = mock_input
+# ==============================================================================
+# 🟢 استخراج فایل سشن از متغیر SESSION_STRING
+# ==============================================================================
+session_string = os.environ.get("SESSION_STRING", "")
+session_file = "my_userbot.rp"
 
-async def start_bot():
-    session_file = "my_userbot.sqlite"
-    
-    # اول چک می‌کنیم سشن نهایی داریم یا نه
-    s_string = os.environ.get("SESSION_STRING", "")
-    if s_string:
+if session_string:
+    try:
         with open(session_file, "wb") as f:
-            f.write(base64.b64decode(s_string))
-        print("✅ سشن نهایی بارگذاری شد.")
-    
-    bot = Client("my_userbot")
+            f.write(base64.b64decode(session_string))
+        print("✅ فایل سشن با موفقیت از متغیر مخفی بازسازی شد.")
+    except Exception as e:
+        print(f"❌ خطا در بازسازی سشن: {e}")
+else:
+    print("⚠️ ارور: SESSION_STRING یافت نشد! ربات نمی‌تواند آنلاین شود.")
+
+# ساخت کلاینت سلف‌بات (اکانت شخصی)
+bot = Client("my_userbot")
+
+# ==============================================================================
+# 🟢 تابع دانلود فایل از لینک
+# ==============================================================================
+async def download_file(url, chat_id, message_id, client):
+    # اصلاح لینک‌های دراپ‌باکس برای دانلود مستقیم
+    if "dropbox.com" in url:
+        url = url.replace("dl=0", "dl=1") if "dl=0" in url else (url + "&dl=1" if "?" in url else url + "?dl=1")
+
+    temp_dir = "/app/downloads"
+    os.makedirs(temp_dir, exist_ok=True)
     
-    # شروع فرآیند لاگین (اگر سشن نباشد شماره و کد را از سکرت‌ها می‌گیرد)
-    await bot.start()
+    # استخراج نام فایل از لینک یا ساخت نام تصادفی
+    parsed_url = urllib.parse.urlparse(url)
+    filename = os.path.basename(parsed_url.path) or f"file_{uuid.uuid4().hex[:6]}.dat"
+    file_path = os.path.join(temp_dir, f"{uuid.uuid4().hex[:4]}_{filename}")
     
-    # اگر تازه لاگین شده، کد سشن رو چاپ می‌کنه
-    if not s_string:
-        with open(session_file, "rb") as f:
-            encoded = base64.b64encode(f.read()).decode('utf-8')
-            print("\n" + "⭐"*20)
-            print("✅ لاگین موفق! این کد SESSION_STRING شماست:\n")
-            print(encoded)
-            print("\n" + "⭐"*20)
-
-    @bot.on_message_updates(filters.private)
-    async def main_handler(client, update):
-        msg = update.message if hasattr(update, 'message') else update
-        if msg.text and msg.text.startswith("http"):
-            asyncio.create_task(process_upload(client, msg.object_guid, msg.message_id, msg.text))
+    status_msg_id = None
+    try:
+        # ارسال پیام وضعیت
+        status_msg = await client.send_message(chat_id, "📥 شروع دانلود از سرور اصلی...\n(حجم مجاز: تا 2 گیگابایت)", reply_to_message_id=message_id)
+        status_msg_id = status_msg.message_id
+    except: pass
 
-async def process_upload(client, chat_id, message_id, url):
-    path = f"dl_{uuid.uuid4().hex[:6]}.dat"
     try:
-        s = await client.send_message(chat_id, "⏳ در حال دانلود و آپلود (تا ۲ گیگ)...")
-        async with aiohttp.ClientSession() as sess:
-            async with sess.get(url, timeout=0) as r:
-                with open(path, 'wb') as f:
-                    async for c in r.content.iter_chunked(2*1024*1024): f.write(c)
-        await client.send_document(chat_id, path, caption=f"✅ فایل آماده شد:\n{url}")
-        await client.delete_messages(chat_id, [s.message_id])
+        # استفاده از aiohttp برای دانلود تکه‌ای (Chunked) جهت مدیریت رم
+        async with aiohttp.ClientSession() as session:
+            async with session.get(url, timeout=None) as response:
+                if response.status != 200:
+                    if status_msg_id: await client.edit_message_text(chat_id, status_msg_id, f"❌ خطا در دانلود! کد وضعیت: {response.status}")
+                    return None, status_msg_id
+                
+                with open(file_path, 'wb') as f:
+                    async for chunk in response.content.iter_chunked(1024 * 1024 * 2): # تکه‌های ۲ مگابایتی
+                        if chunk:
+                            f.write(chunk)
+                            
+        if status_msg_id: await client.edit_message_text(chat_id, status_msg_id, "✅ دانلود کامل شد. در حال آپلود در روبیکا...")
+        return file_path, status_msg_id
     except Exception as e:
-        await client.send_message(chat_id, f"❌ خطا: {str(e)[:100]}")
-    finally:
-        if os.path.exists(path): os.remove(path)
+        if status_msg_id: await client.edit_message_text(chat_id, status_msg_id, f"❌ خطای شبکه در دانلود: {str(e)[:100]}")
+        return None, status_msg_id
+
+# ==============================================================================
+# 🟢 پردازش و آپلود نهایی
+# ==============================================================================
+async def process_and_upload(client, chat_id, message_id, url):
+    file_path, status_id = await download_file(url, chat_id, message_id, client)
+    
+    if file_path and os.path.exists(file_path):
+        try:
+            # ارسال فایل به صورت مستند (Document)
+            # در سلف‌بات محدودیت حجم بسیار بیشتر از ربات‌های معمولی است
+            await client.send_document(
+                chat_id, 
+                file_path, 
+                caption=f"✅ فایل با موفقیت منتقل شد.\n🔗 لینک منبع:\n{url}",
+                reply_to_message_id=message_id
+            )
+            
+            # حذف پیام وضعیت برای تمیز ماندن چت
+            if status_id:
+                try: await client.delete_messages(chat_id, [status_id])
+                except: pass
+                
+        except Exception as e:
+            error_msg = f"❌ خطا در آپلود روبیکا:\n{str(e)[:150]}"
+            if status_id: await client.edit_message_text(chat_id, status_id, error_msg)
+            else: await client.send_message(chat_id, error_msg)
+        finally:
+            # حذف فایل از روی هارد سرور برای جلوگیری از پر شدن فضا
+            if os.path.exists(file_path):
+                os.remove(file_path)
 
+# ==============================================================================
+# 🟢 هندلر پیام‌های دریافتی (فقط در پی‌وی و پیام‌های ذخیره شده)
+# ==============================================================================
+@bot.on_message_updates(filters.private)
+async def message_handler(client, update):
+    try:
+        # گرفتن متن پیام
+        message = update.message if hasattr(update, 'message') else update
+        text = str(message.text or "").strip()
+        
+        # اگر متن با http شروع شده بود، یعنی لینک دانلود است
+        if text.startswith("http"):
+            # اجرای عملیات در پس‌زمینه برای اینکه ربات متوقف نشود
+            asyncio.create_task(process_and_upload(client, message.object_guid, message.message_id, text))
+            
+        elif text == "سلام":
+            await client.send_message(message.object_guid, "سلام حامد جان! لینک فایل (حتی ۲ گیگابایتی) رو بفرست تا مستقیم برات آپلودش کنم.")
+
+    except Exception:
+        traceback.print_exc()
+
+# ==============================================================================
+# 🟢 اجرای همزمان وب‌سرور و ربات
+# ==============================================================================
 if __name__ == "__main__":
+    # اجرای وب‌سرور در یک ترد جداگانه
     threading.Thread(target=run_flask, daemon=True).start()
-    asyncio.run(start_bot())
+    
+    print("🚀 سلف‌بات در حال تلاش برای اتصال به روبیکا...")
+    # اجرای ربات
+    bot.run()