Spaces:

Opera8
/

Downloadbotmelli

Sleeping

App Files Files Community

Opera8 commited on May 9

Commit

7ed6519

verified ·

1 Parent(s): 1b50641

Update main.py

Browse files

Files changed (1) hide show

main.py +51 -123

main.py CHANGED Viewed

@@ -32,19 +32,16 @@ def custom_input(prompt):
     print(f"[سیستم لاگین] درخواست از سرور: {prompt}")
     prompt_lower = prompt.lower()
-    # مرحله اول: گرفتن شماره موبایل
     if "شماره" in prompt_lower or "phone" in prompt_lower or "number" in prompt_lower:
         login_state["step"] = "phone"
         while login_state["step"] != "processing_phone":
             time.sleep(1)
         return login_state["phone"]
-    # 🟢 مرحله مهم (حل مشکل کرش): تایید کردن اتوماتیک صحت شماره موبایل
     elif "correct" in prompt_lower or "y or n" in prompt_lower or "y/n" in prompt_lower:
         print("[سیستم لاگین] تایید خودکار شماره (y)")
         return "y"
-    # مرحله دوم: گرفتن کد تایید پیامک شده
     elif "کد" in prompt_lower or "code" in prompt_lower or "verify" in prompt_lower:
         login_state["step"] = "code"
         while login_state["step"] != "processing_code":
@@ -52,11 +49,10 @@ def custom_input(prompt):
         threading.Thread(target=wait_and_success).start()
         return login_state["code"]
-    # بستن راه کرش برای پسورد دو مرحله‌ای یا هر سوال پیش‌بینی نشده‌ی دیگر
     return ""
 def wait_and_success():
-    time.sleep(6) # صبر برای ساخت فایل سشن توسط روبیکا
     if os.path.exists("my_userbot.sqlite"):
         try:
             with open("my_userbot.sqlite", "rb") as f:
@@ -68,7 +64,7 @@ def wait_and_success():
 builtins.input = custom_input
 # ==============================================================================
-# 🟢 رابط کاربری وب (پنل مدیریت شیک در هاگینگ فیس)
 # ==============================================================================
 app = Flask(__name__)
@@ -87,52 +83,40 @@ HTML_TEMPLATE = """
         .card { background: white; padding: 40px; border-radius: 15px; box-shadow: 0 10px 20px rgba(0,0,0,0.1); width: 100%; max-width: 450px; text-align: center; }
         h2 { color: #2c3e50; margin-bottom: 15px; }
         p { color: #555; font-size: 14px; margin-bottom: 25px; line-height: 1.6; }
-        input[type="text"] { width: 85%; padding: 12px; margin-bottom: 20px; border: 2px solid #ddd; border-radius: 8px; font-size: 16px; text-align: center; direction: ltr; transition: border 0.3s; }
-        input[type="text"]:focus { border-color: #007bff; outline: none; }
-        button { background-color: #007bff; color: white; border: none; padding: 12px 20px; width: 95%; border-radius: 8px; font-size: 16px; font-weight: bold; cursor: pointer; transition: background 0.3s; }
-        button:hover { background-color: #0056b3; }
         .loader { border: 4px solid #f3f3f3; border-top: 4px solid #007bff; border-radius: 50%; width: 40px; height: 40px; animation: spin 1s linear infinite; margin: 0 auto 20px auto; }
         @keyframes spin { 0% { transform: rotate(0deg); } 100% { transform: rotate(360deg); } }
-        textarea { width: 90%; height: 120px; border: 2px solid #ddd; border-radius: 8px; padding: 10px; font-size: 12px; direction: ltr; text-align: left; resize: none; margin-bottom: 20px; background: #f8f9fa; cursor: pointer;}
-        .success { color: #28a745; font-weight: bold; font-size: 20px; margin-bottom: 15px;}
-        .warning { color: #dc3545; font-size: 12px; font-weight: bold; }
     </style>
 </head>
 <body>
     <div class="card">
         <h2>🚀 پنل اتصال ربات ۲ گیگابایتی</h2>
         {% if state.step == 'phone' %}
-            <p>برای فعال‌سازی ربات، <b>شماره موبایل</b> اکانت روبیکای خود را وارد کنید.</p>
             <form action="/submit_phone" method="POST">
-                <input type="text" name="phone" placeholder="989123456789" required autocomplete="off">
                 <button type="submit">دریافت کد تایید</button>
             </form>
         {% elif state.step == 'processing_phone' %}
-            <div class="loader"></div>
-            <p>در حال برقراری ارتباط با سرورهای روبیکا...</p>
         {% elif state.step == 'code' %}
-            <p><b>کد ۵ رقمی</b> ارسال شده به پیام‌های روبیکا را وارد کنید.</p>
             <form action="/submit_code" method="POST">
-                <input type="text" name="code" placeholder="12345" required autocomplete="off">
                 <button type="submit">تایید و ورود</button>
             </form>
         {% elif state.step == 'processing_code' %}
-            <div class="loader"></div>
-            <p>در حال تایید کد و ساخت نشست (Session)...</p>
         {% elif state.step == 'success' %}
-            <div class="success">✅ ربات با موفقیت متصل شد!</div>
-            <p>متن زیر را کپی کنید و در تنظیمات اسپیس (Variables)، در متغیری به نام <b>SESSION_STRING</b> قرار دهید تا با ری‌استارت شدن سرور، ربات خاموش نشود:</p>
-            <textarea readonly onclick="this.select()">{{ state.session_string }}</textarea>
-            <p class="warning">⚠️ این متن محرمانه است، آن را به هیچکس ندهید.</p>
         {% elif state.step == 'logged_in' %}
-            <div class="success">✅ ربات روشن و آماده کار است!</div>
-            <p>اکنون می‌توانید لینک‌های خود را در روبیکا (پیام‌های ذخیره شده) ارسال کنید تا فایل تا حجم ۲ گیگابایت مستقیماً آپلود شود.</p>
         {% endif %}
     </div>
 </body>
@@ -163,116 +147,60 @@ def run_flask():
     app.run(host="0.0.0.0", port=7860, threaded=True)
 # ==============================================================================
-# 🟢 سیستم لود کردن نشست (در صورت وجود)
-# ==============================================================================
-session_string = os.environ.get("SESSION_STRING", "")
-session_filename = "my_userbot.sqlite"
-if session_string:
-    with open(session_filename, "wb") as f:
-        f.write(base64.b64decode(session_string))
-    login_state["step"] = "logged_in"
-    print("✅ فایل سشن با موفقیت از متغیر محیطی بارگذاری شد.")
-elif os.path.exists(session_filename) and os.path.getsize(session_filename) > 0:
-    login_state["step"] = "logged_in"
-else:
-    login_state["step"] = "phone"
-bot = Client("my_userbot")
-# ==============================================================================
-# 🟢 توابع دانلود و آپلود 2 گیگابایتی
 # ==============================================================================
 async def download_large_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)
-    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}")
-    status_msg_id = None
     try:
-        status_msg = await client.send_message(chat_id, "⏳ در حال دانلود فایل از اینترنت آزاد...\n(محدودیت آپلود در این اکانت: تا 2 گیگابایت)", reply_to_message_id=message_id)
-        status_msg_id = getattr(status_msg, 'message_id', None) or (status_msg.get('message_update', {}).get('message_id') if isinstance(status_msg, dict) else None)
-    except: pass
-    try:
-        headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"}
-        async with aiohttp.ClientSession(headers=headers) as session:
             async with session.get(url, timeout=0) 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(2 * 1024 * 1024):
                         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:
-        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_url_and_upload(client, chat_id, message_id, url):
-    file_path, status_msg_id = await download_large_file(url, chat_id, message_id, client)
-    if file_path and os.path.exists(file_path):
         try:
-            caption_text = f"📥 دانلود مستقیم:\n🔗 {url}"
-            upload_success = False
-            for attempt in range(5):
-                try:
-                    try: await client.send_document(chat_id, document=file_path, caption=caption_text)
-                    except TypeError: await client.send_document(chat_id, file=file_path, caption=caption_text)
-                    except: await client.send_document(chat_id, file_path, caption=caption_text)
-                    upload_success = True
-                    break
-                except Exception:
-                    await asyncio.sleep(4)
-            if not upload_success: raise Exception("سرور روبیکا فایل را پس از ۵ بار تلاش نپذیرفت.")
-            if status_msg_id:
-                try: await client.delete_messages(chat_id, [status_msg_id])
-                except: pass
-        except Exception as e:
-            error_txt = f"❌ خطا در عملیات آپلود:\n{str(e)[:150]}"
-            if status_msg_id: await client.edit_message_text(chat_id, status_msg_id, error_txt)
-            else: await client.send_message(chat_id, error_txt)
         finally:
-            try: os.remove(file_path)
-            except: pass
 # ==============================================================================
-# 🟢 هندلر دریافت پیام (مخصوص یوزربات)
 # ==============================================================================
 @bot.on_message_updates(filters.private)
 async def main_handler(client, update):
     try:
-        msg_obj = getattr(update, "message", None) or getattr(update, "new_message", None) or update
-        chat_id = getattr(update, 'object_guid', None) or getattr(update, 'author_guid', None) or getattr(update, "chat_id", None)
-        msg_id = getattr(update, "message_id", None) or (msg_obj.get("message_id") if isinstance(msg_obj, dict) else getattr(msg_obj, "message_id", None))
-        user_text_str = str(getattr(update, "text", "") or getattr(msg_obj, "text", "")).strip()
-        if not user_text_str: return
-        if user_text_str.lower() in ["/start", "سلام"]:
-            await client.send_message(chat_id, "👋 سلام مدیر! لینک فایلت رو بفرست تا با اکانت خودت تا حجم ۲ گیگ برات آپلودش کنم.")
-            return
-        if user_text_str.lower().startswith("http"):
-            asyncio.create_task(process_url_and_upload(client, chat_id, msg_id, user_text_str))
-    except Exception: traceback.print_exc()
-# ==============================================================================
-# 🟢 نقطه شروع
-# ==============================================================================
 if __name__ == "__main__":
-    # بالا آوردن وب‌سرور پن

     print(f"[سیستم لاگین] درخواست از سرور: {prompt}")
     prompt_lower = prompt.lower()
     if "شماره" in prompt_lower or "phone" in prompt_lower or "number" in prompt_lower:
         login_state["step"] = "phone"
         while login_state["step"] != "processing_phone":
             time.sleep(1)
         return login_state["phone"]
     elif "correct" in prompt_lower or "y or n" in prompt_lower or "y/n" in prompt_lower:
         print("[سیستم لاگین] تایید خودکار شماره (y)")
         return "y"
     elif "کد" in prompt_lower or "code" in prompt_lower or "verify" in prompt_lower:
         login_state["step"] = "code"
         while login_state["step"] != "processing_code":
         threading.Thread(target=wait_and_success).start()
         return login_state["code"]
     return ""
 def wait_and_success():
+    time.sleep(6)
     if os.path.exists("my_userbot.sqlite"):
         try:
             with open("my_userbot.sqlite", "rb") as f:
 builtins.input = custom_input
 # ==============================================================================
+# 🟢 رابط کاربری وب
 # ==============================================================================
 app = Flask(__name__)
         .card { background: white; padding: 40px; border-radius: 15px; box-shadow: 0 10px 20px rgba(0,0,0,0.1); width: 100%; max-width: 450px; text-align: center; }
         h2 { color: #2c3e50; margin-bottom: 15px; }
         p { color: #555; font-size: 14px; margin-bottom: 25px; line-height: 1.6; }
+        input[type="text"] { width: 85%; padding: 12px; margin-bottom: 20px; border: 2px solid #ddd; border-radius: 8px; font-size: 16px; text-align: center; direction: ltr; }
+        button { background-color: #007bff; color: white; border: none; padding: 12px 20px; width: 95%; border-radius: 8px; font-size: 16px; font-weight: bold; cursor: pointer; }
         .loader { border: 4px solid #f3f3f3; border-top: 4px solid #007bff; border-radius: 50%; width: 40px; height: 40px; animation: spin 1s linear infinite; margin: 0 auto 20px auto; }
         @keyframes spin { 0% { transform: rotate(0deg); } 100% { transform: rotate(360deg); } }
+        textarea { width: 90%; height: 120px; border: 2px solid #ddd; border-radius: 8px; padding: 10px; font-size: 12px; direction: ltr; }
+        .success { color: #28a745; font-weight: bold; font-size: 20px; }
     </style>
 </head>
 <body>
     <div class="card">
         <h2>🚀 پنل اتصال ربات ۲ گیگابایتی</h2>
         {% if state.step == 'phone' %}
+            <p>شماره موبایل اکانت روبیکای خود را وارد کنید.</p>
             <form action="/submit_phone" method="POST">
+                <input type="text" name="phone" placeholder="989123456789" required>
                 <button type="submit">دریافت کد تایید</button>
             </form>
         {% elif state.step == 'processing_phone' %}
+            <div class="loader"></div><p>در حال ارتباط با روبیکا...</p>
         {% elif state.step == 'code' %}
+            <p>کد ۵ رقمی روبیکا را وارد کنید.</p>
             <form action="/submit_code" method="POST">
+                <input type="text" name="code" placeholder="12345" required>
                 <button type="submit">تایید و ورود</button>
             </form>
         {% elif state.step == 'processing_code' %}
+            <div class="loader"></div><p>در حال تایید کد...</p>
         {% elif state.step == 'success' %}
+            <div class="success">✅ متصل شد!</div>
+            <p>این متن را در متغیر <b>SESSION_STRING</b> قرار دهید:</p>
+            <textarea readonly>{{ state.session_string }}</textarea>
         {% elif state.step == 'logged_in' %}
+            <div class="success">✅ ربات فعال است!</div>
+            <p>اکنون لینک‌های خود را در روبیکا ارسال کنید.</p>
         {% endif %}
     </div>
 </body>
     app.run(host="0.0.0.0", port=7860, threaded=True)
 # ==============================================================================
+# 🟢 توابع دانلود و آپلود (Userbot)
 # ==============================================================================
 async def download_large_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)
+    filename = f"file_{uuid.uuid4().hex[:6]}.dat"
+    file_path = os.path.join(temp_dir, filename)
     try:
+        status_msg = await client.send_message(chat_id, "⏳ در حال دانلود فایل (تا 2 گیگ)...", reply_to_message_id=message_id)
+        status_id = status_msg.message_id if hasattr(status_msg, 'message_id') else status_msg.get('message_id')
+        async with aiohttp.ClientSession() as session:
             async with session.get(url, timeout=0) as response:
                 with open(file_path, 'wb') as f:
                     async for chunk in response.content.iter_chunked(2 * 1024 * 1024):
                         if chunk: f.write(chunk)
+        await client.edit_message_text(chat_id, status_id, "✅ دانلود کامل شد. در حال آپلود در روبیکا...")
+        return file_path, status_id
+    except: return None, None
 async def process_url_and_upload(client, chat_id, message_id, url):
+    file_path, status_id = await download_large_file(url, chat_id, message_id, client)
+    if file_path:
         try:
+            await client.send_document(chat_id, file_path, caption=f"📥 دانلود مستقیم:\n🔗 {url}")
+            if status_id: await client.delete_messages(chat_id, [status_id])
+        except: pass
         finally:
+            if os.path.exists(file_path): os.remove(file_path)
 # ==============================================================================
+# 🟢 سیستم لود و اجرای اصلی
 # ==============================================================================
+session_string = os.environ.get("SESSION_STRING", "")
+if session_string:
+    with open("my_userbot.sqlite", "wb") as f:
+        f.write(base64.b64decode(session_string))
+    login_state["step"] = "logged_in"
+else:
+    login_state["step"] = "phone"
+bot = Client("my_userbot")
 @bot.on_message_updates(filters.private)
 async def main_handler(client, update):
     try:
+        msg = update.message if hasattr(update, 'message') else update
+        text = msg.text
+        if text and text.startswith("http"):
+            asyncio.create_task(process_url_and_upload(client, msg.object_guid, msg.message_id, text))
+    except: pass
 if __name__ == "__main__":
+    threading.Thread(target=run_flask, daemon=True).start()
+    print("🚀 پنل وب روشن شد...")
+    bot.run()