Update main.py

main.py

@@ -698,111 +698,174 @@ HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
 HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
 # ==============================================================================
-# 🟢 پارت 11: آپلودر با استفاده انحصاری از کتابخانه رسمی روب‌پای (Rubpy)
+# 🟢 پارت 11: آپلودر خالص و مستقیم بر اساس مستندات رسمی API روبیکا (بدون باگ کتابخانه)
 # ==============================================================================
 import asyncio
 import os
+import aiohttp
 
-# 🟢 ۱. مدیریت صف آپلود
+# 🟢 ۱. مدیریت صف آپلود برای جلوگیری از خطای TOO_REQUESTS
 _upload_lock = None
+_shared_session = None
 
 def get_upload_lock():
     global _upload_lock
     if _upload_lock is None:
-        # برای اینکه کتابخانه کرش نکند، فایل‌ها را 2 تا 2 تا به کتابخانه می‌سپاریم
-        _upload_lock = asyncio.Semaphore(10) 
+        # قفل ارسال را روی 3 می‌گذاریم تا فایل‌ها با نظم و طبق استاندارد سرور روبیکا ارسال شوند
+        _upload_lock = asyncio.Semaphore(3) 
     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)
+        _shared_session = aiohttp.ClientSession(connector=connector)
+    return _shared_session
+
 async def helper_upload_file(client, chat_id, file_name, file_type="Image", caption=""):
     abs_path = os.path.abspath(file_name)
     if not os.path.exists(abs_path):
         return "File Not Found"
 
+    try:
+        file_bytes = await asyncio.to_thread(sync_read_file, abs_path)
+    except Exception as e:
+        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"
-
+        
     async with get_upload_lock():
-        backoff_time = 10 # زمان صبر اولیه در صورت برخورد به محدودیت
+        session = await get_shared_session()
+        backoff_time = 5 
         
-        # ربات ۵ بار تلاش می‌کند تا کتابخانه کار را انجام دهد
-        for attempt in range(5): 
+        # 🚀 فاز اصلی: استفاده ۱۰۰ درصدی از API خام روبیکا طبق مستندات
+        for attempt in range(15): 
             try:
-                if api_file_type == "Image":
-                    if hasattr(client, "send_photo"):
-                        try:
-                            await client.send_photo(chat_id, photo=abs_path, caption=caption)
-                        except TypeError:
-                            await client.send_photo(chat_id, abs_path, caption=caption)
-                        return True
-                        
-                elif api_file_type in ["Music", "Voice"]:
-                    if hasattr(client, "send_music"):
-                        try:
-                            await client.send_music(chat_id, abs_path, caption=caption)
-                        except TypeError:
-                            await client.send_music(chat_id, music=abs_path, caption=caption)
-                        return True
-                        
-                else:
-                    if hasattr(client, "send_document"):
-                        try:
-                            await client.send_document(chat_id, document=abs_path, caption=caption)
-                        except TypeError:
-                            await client.send_document(chat_id, abs_path, caption=caption)
-                        return True
-                    elif hasattr(client, "send_file"):
-                        try:
-                            await client.send_file(chat_id, file=abs_path)
-                        except TypeError:
-                            await client.send_file(chat_id, abs_path)
-                        if caption:
-                            await send_with_keyboard(client, chat_id, caption, True)
-                        return True
-                        
-                error_logs.append("متد مناسب در کتابخانه یافت نشد.")
-                break # اگر متد کلاً وجود نداشت، از حلقه خارج می‌شویم
-                
-            except Exception as e:
-                err_str = str(e)
-                error_logs.append(f"تلاش {attempt+1} با کتابخانه: {err_str[:40]}")
+                # ---------------------------------------------------------
+                # مرحله ۱: متد 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))
                 
-                # اگر کتابخانه ارور TOO_REQUESTS داد، صبر می‌کنیم
-                if "TOO_REQ" in err_str.upper() or "429" in err_str:
+                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
+
+                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 += 15 # دفعه بعد بیشتر صبر می‌کند
+                    backoff_time += 5
+                    continue
+
+                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:
-                    await asyncio.sleep(4)
+                    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
+                    }
+                    send_resp = await session.post(
+                        f"https://botapi.rubika.ir/v3/{bot_token}/sendFile", 
+                        json=send_payload,
+                        timeout=aiohttp.ClientTimeout(total=20)
+                    )
+                    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("تغییر اجباری نوع فایل")
 
-    return "❌ خطا در آپلود با کتابخانه روب‌پای:\n" + "\n".join(error_logs[-3:])
+            except Exception as e:
+                error_logs.append(f"تلاش {attempt+1} خطا: {str(e)[:25]}")
+                await asyncio.sleep(3)
 
+    return "❌ خطا در آپلود (API خام):\n" + "\n".join(error_logs[-5:])
 
 # ==============================================================================
 # 🟢 ۲. تعریف کارگرهای پس‌زمینه عمومی (مشترک برای همه بخش‌ها)
 # ==============================================================================
-_general_upload_workers = asyncio.Semaphore(10) # ظرفیت پردازش پس‌زمینه
+_general_upload_workers = asyncio.Semaphore(50) # هسته ربات آزاد است، ظرفیت روی 50 تنظیم شد
 
 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:
         try:
             res = await helper_upload_file(client, chat_id, file_name, file_type, caption_text)
             if res is not True:
-                # عودت اعتبار در صورت شکست قطعی آپلود
                 creds = get_user_credits(str_chat_id)
                 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آپلود {file_type} انجام نشد و اعتبار شما برگشت داده شد.\n`{str(res)[:500]}`", True)
-        except Exception as e: 
-            pass
+                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)
+            if os.path.exists(file_name): os.remove(file_name)
+