Update main.py

main.py

@@ -698,27 +698,32 @@ HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
 HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
 # ==============================================================================
-# 🟢 پارت 11: آپلودر خالص و مستقیم بر اساس مستندات رسمی API روبیکا (بدون باگ کتابخانه)
+# 🟢 پارت 11: سیستم آپلود ابری نوین (استفاده از سرورهای بله به عنوان S3 پرسرعت)
 # ==============================================================================
 import asyncio
 import os
 import aiohttp
 
-# 🟢 ۱. مدیریت صف آپلود برای جلوگیری از خطای TOO_REQUESTS
+# توکن ربات تستی بله شما (آپلودسنتر)
+BALE_TOKEN = "2030030646:ioHMqrCsw0PwK5hTg_1Av1rsXZMiu-LETBI"
+# در پیام‌رسان‌های تلگرام و بله، آیدی عددی ربات همان بخش اول توکن است
+BALE_CHAT_ID = BALE_TOKEN.split(":")[0] 
+
+# 🟢 ۱. مدیریت صف آپلود ابری
 _upload_lock = None
 _shared_session = None
 
 def get_upload_lock():
     global _upload_lock
     if _upload_lock is None:
-        # قفل ارسال را روی 3 می‌گذاریم تا فایل‌ها با نظم و طبق استاندارد سرور روبیکا ارسال شوند
-        _upload_lock = asyncio.Semaphore(3) 
+        # ظرفیت روی 20 تنظیم شد (ارسال موشکی و بدون ارور)
+        _upload_lock = asyncio.Semaphore(20) 
     return _upload_lock
 
 async def get_shared_session():
     global _shared_session
     if _shared_session is None or _shared_session.closed:
-        connector = aiohttp.TCPConnector(limit=50, keepalive_timeout=60)
+        connector = aiohttp.TCPConnector(limit=100, keepalive_timeout=60)
         _shared_session = aiohttp.ClientSession(connector=connector)
     return _shared_session
 
@@ -733,124 +738,99 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
         return f"File Read Error: {e}"
 
     error_logs = []
-    bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
-    
-    # تعیین نوع دقیق فایل طبق Enum مستندات روبیکا (FileTypeEnum)
-    api_file_type = "Image" if file_type in ["photo", "Image", "image"] else "Voice" if file_type in ["voice", "Voice", "audio"] else file_type
-    if api_file_type == "Voice" and not abs_path.lower().endswith('.ogg'): 
-        api_file_type = "Music"
-    if api_file_type == "File" and abs_path.lower().endswith('.mp3'): 
-        api_file_type = "Music"
-        
+    rubika_bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
+
     async with get_upload_lock():
         session = await get_shared_session()
-        backoff_time = 5 
+        download_link = None
         
-        # 🚀 فاز اصلی: استفاده ۱۰۰ درصدی از API خام روبیکا طبق مستندات
-        for attempt in range(15): 
+        # 🚀 فاز اول: آپلود فایل به سرورهای بله و دریافت لینک مستقیم
+        for attempt in range(3):
             try:
-                # ---------------------------------------------------------
-                # مرحله ۱: متد requestSendFile (دریافت آدرس آپلود)
-                # ---------------------------------------------------------
-                req_resp = await session.post(
-                    f"https://botapi.rubika.ir/v3/{bot_token}/requestSendFile", 
-                    json={"type": api_file_type},
-                    timeout=aiohttp.ClientTimeout(total=20)
-                )
-                if req_resp.status != 200:
-                    error_logs.append(f"reqSend HTTP {req_resp.status}")
-                    await asyncio.sleep(3)
-                    continue
-                    
-                req_data = await req_resp.json()
-                if req_data.get("status") in ["TOO_REQUESTS", "TOO_REQ"]:
-                    error_logs.append(f"TOO_REQ (گرفتن لینک - صبر {backoff_time}s)")
-                    await asyncio.sleep(backoff_time)
-                    backoff_time += 5 
-                    continue
-
-                upload_url = req_data.get("data", {}).get("upload_url")
-                if not upload_url: 
-                    error_logs.append("لینک آپلود دریافت نشد")
-                    await asyncio.sleep(3)
-                    continue
-
-                # ---------------------------------------------------------
-                # مرحله ۲: آپلود فایل (multipart/form-data)
-                # ---------------------------------------------------------
+                # ارسال فایل به چت خودِ ربات در بله
                 form = aiohttp.FormData()
-                form.add_field('file', file_bytes, filename=os.path.basename(abs_path))
+                form.add_field('chat_id', BALE_CHAT_ID)
+                form.add_field('document', file_bytes, filename=os.path.basename(abs_path))
                 
-                up_resp = await session.post(
-                    upload_url, 
-                    data=form, 
-                    timeout=aiohttp.ClientTimeout(total=120)
-                )
-                if up_resp.status != 200:
-                    error_logs.append(f"Upload HTTP {up_resp.status}")
-                    await asyncio.sleep(3)
-                    continue
+                bale_upload_url = f"https://tapi.bale.ai/bot{BALE_TOKEN}/sendDocument"
+                async with session.post(bale_upload_url, data=form, timeout=aiohttp.ClientTimeout(total=60)) as resp:
+                    if resp.status == 200:
+                        b_data = await resp.json()
+                        file_id = b_data.get('result', {}).get('document', {}).get('file_id')
+                        
+                        if file_id:
+                            # درخواست گرفتن مسیر فایل از سرور بله
+                            bale_file_url = f"https://tapi.bale.ai/bot{BALE_TOKEN}/getFile?file_id={file_id}"
+                            async with session.get(bale_file_url) as f_resp:
+                                if f_resp.status == 200:
+                                    f_data = await f_resp.json()
+                                    file_path = f_data.get('result', {}).get('file_path')
+                                    if file_path:
+                                        # ساخت لینک دانلود مستقیم نهایی
+                                        download_link = f"https://tapi.bale.ai/file/bot{BALE_TOKEN}/{file_path}"
+                                        break
+            except Exception as e:
+                error_logs.append(f"Bale Upload Error (Attempt {attempt+1}): {str(e)[:30]}")
+                await asyncio.sleep(2)
 
-                up_data = await up_resp.json()
-                if up_data.get("status") in ["TOO_REQUESTS", "TOO_REQ"]:
-                    error_logs.append(f"TOO_REQ (آپلود - صبر {backoff_time}s)")
-                    await asyncio.sleep(backoff_time)
-                    backoff_time += 5
-                    continue
+        # 🚀 فاز دوم: ارسال پیام متنی زیبا به کاربر در روبیکا
+        if download_link:
+            # 🟢 دیزاین جدید پیام با خطوط جداکننده و توضیحات کامل
+            final_caption = f"""{caption}
 
-                final_file_id = None
-                if up_data.get("status") in ["OK", "OK_DET"]:
-                    # استخراج file_id با توجه به تفاوت‌های ساختاری سرور روبیکا
-                    if "data" in up_data and isinstance(up_data["data"], dict):
-                        final_file_id = up_data["data"].get("file_id") or up_data["data"].get("id")
-                    else:
-                        final_file_id = up_data.get("file_id")
-                else:
-                    error_logs.append(f"Up Error: {up_data.get('status')}")
-                            
-                # ---------------------------------------------------------
-                # مرحله ۳: متد sendFile (ارسال فایل نهایی به کاربر)
-                # ---------------------------------------------------------
-                if final_file_id:
-                    send_payload = {
-                        "chat_id": str(chat_id),
-                        "file_id": str(final_file_id),
-                        "text": caption, # طبق مستندات شما، نام فیلد متن text است نه caption
-                        "chat_keypad_type": "New",
-                        "chat_keypad": MAIN_KEYPAD_DICT
-                    }
+➖➖➖➖➖➖➖➖➖➖
+🌐 **لینک دانلود مستقیم فایل:**
+`{download_link}`
+➖➖➖➖➖➖➖➖➖➖
+
+⚠️ **راهنمای دانلود:**
+لطفاً لینک بالا را کپی کرده و در مرورگر گوشی خود (مثل کروم، فایرفاکس یا سافاری) باز کنید تا فایلتان دانلود شود. (چون روبیکا اجازه نمی‌دهد، مستقیماً از اینجا روی لینک کلیک نکنید).
+
+🇮🇷 این لینک روی دامنه سایت پیام‌رسان داخلی «بله» قرار دارد و به راحتی با اینترنت ملی (بدون فیلترشکن) با بالاترین سرعت دانلود می‌شود.
+
+⏳ *پ.ن: در صورتی که محدودیت‌های روبیکا در دریافت مستقیم فایل برطرف شود، سیستم را تغییر می‌دهیم تا دوباره فایل‌ها را به صورت مستقیم داخل همین ربات دریافت کنید.*"""
+            
+            # ارسال مستقیم متنی از طریق API خام روبیکا
+            send_payload = {
+                "chat_id": str(chat_id),
+                "text": final_caption,
+                "chat_keypad_type": "New",
+                "chat_keypad": MAIN_KEYPAD_DICT
+            }
+            
+            for attempt in range(3):
+                try:
                     send_resp = await session.post(
-                        f"https://botapi.rubika.ir/v3/{bot_token}/sendFile", 
+                        f"https://botapi.rubika.ir/v3/{rubika_bot_token}/sendMessage", 
                         json=send_payload,
-                        timeout=aiohttp.ClientTimeout(total=20)
+                        timeout=aiohttp.ClientTimeout(total=15)
                     )
                     s_data = await send_resp.json()
                     
                     if s_data.get("status") == "OK": 
-                        await asyncio.sleep(1)
                         return True
-                    elif s_data.get("status") in ["TOO_REQUESTS", "TOO_REQ"]:
-                        error_logs.append(f"TOO_REQ (ارسال نهایی - صبر {backoff_time}s)")
-                        await asyncio.sleep(backoff_time)
-                        backoff_time += 5
-                        continue
                     else:
-                        error_logs.append(f"Send Error: {s_data.get('status')}")
-                else:
-                    if not final_file_id and "INVALID_INPUT" in str(up_data):
-                        api_file_type = "File"
-                        error_logs.append("تغییر اجباری نوع فایل")
+                        error_logs.append(f"Rubika Send API Error: {s_data.get('status')}")
+                except Exception as e:
+                    error_logs.append(f"Rubika Send HTTP Error: {str(e)[:30]}")
+                    await asyncio.sleep(2)
 
+            # بک‌آپ با کتابخانه در صورت کار نکردن API خام
+            try:
+                await client.send_message(chat_id, final_caption)
+                return True
             except Exception as e:
-                error_logs.append(f"تلاش {attempt+1} خطا: {str(e)[:25]}")
-                await asyncio.sleep(3)
+                error_logs.append(f"Rubika Fallback Error: {str(e)[:30]}")
+                
+        else:
+            error_logs.append("خطا در دریافت لینک دانلود از سرور ابری بله.")
 
-    return "❌ خطا در آپلود (API خام):\n" + "\n".join(error_logs[-5:])
+    return "❌ خطا در سیستم آپلود ابری:\n" + "\n".join(error_logs[-3:])
 
 # ==============================================================================
 # 🟢 ۲. تعریف کارگرهای پس‌زمینه عمومی (مشترک برای همه بخش‌ها)
 # ==============================================================================
-_general_upload_workers = asyncio.Semaphore(50) # هسته ربات آزاد است، ظرفیت روی 50 تنظیم شد
+_general_upload_workers = asyncio.Semaphore(20) # ظرفیت 20 عملیات پس‌زمینه همزمان
 
 async def background_general_uploader(client, chat_id, file_name, file_type, caption_text, str_chat_id, credit_key=None):
     async with _general_upload_workers:
@@ -861,15 +841,13 @@ async def background_general_uploader(client, chat_id, file_name, file_type, cap
                 if not creds.get("is_premium") and credit_key:
                     user_credits_db[str_chat_id][credit_key] += 1
                     save_db(user_credits_db)
-                await send_with_keyboard(client, chat_id, f"❌ سرور ارسال گزارش داد:\nمتأسفانه روبیکا اجازه آپلود را نداد و سهمیه شما عودت داده شد.\n`{str(res)[:500]}`", True)
+                await send_with_keyboard(client, chat_id, f"❌ سیستم توزیع ابری گزارش داد:\nمتأسفانه ایجاد لینک دانلود با مشکل مواجه شد و اعتبار شما بازگردانده شد.\n`{str(res)[:500]}`", True)
         except: pass
         finally:
             if os.path.exists(file_name): os.remove(file_name)
 
 
 
-
-
 # ==============================================================================
 # 🟢 پارت 12: توابع تغییر صدا و لیست گویندگان
 # ==============================================================================