Update main.py

main.py

@@ -703,17 +703,132 @@ HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 # 🟢 پارت 11: آپلودر دوگانه هوشمند (آپلود در بله + بکاپ خودکار در روبیکا)
 # ==============================================================================
 BALE_BOT_TOKEN = os.environ.get("BALE_BOT_TOKEN", "").strip()
+import shutil
+import uuid
+import asyncio
+import os
+import aiohttp
+from pydub import AudioSegment
+
+async def official_rubika_upload(chat_id, target_path, file_type, caption):
+    """اجرای دقیق 3 مرحله مستندات رسمی روبیکا به همراه تشخیص و تبدیل فرمت‌های تقلبی"""
+    global bot_token
+    if not bot_token: return False, "توکن ربات یافت نشد", target_path
+
+    api_file_type = "File"
+    if file_type in ["photo", "Image", "image"]: api_file_type = "Image"
+    elif file_type in ["voice", "Voice", "audio", "Music", "music"]: api_file_type = "Music"
+
+    original_path = target_path
+
+    # 🟢 تشخیص فایل‌های صوتی که پسوند mp3 دارند اما در واقع wav هستند
+    if api_file_type == "Music":
+        base_name, ext = os.path.splitext(target_path)
+        needs_conversion = (ext.lower() != '.mp3')
+        
+        if not needs_conversion:
+            try:
+                with open(target_path, 'rb') as f:
+                    header = f.read(4)
+                    if header.startswith(b'RIFF') or header.startswith(b'OggS') or header.startswith(b'fLaC'):
+                        needs_conversion = True
+            except:
+                pass
+                
+        if needs_conversion:
+            mp3_path = f"{base_name}_conv_{uuid.uuid4().hex[:4]}.mp3"
+            def convert_to_mp3():
+                try:
+                    audio = AudioSegment.from_file(original_path)
+                    audio.export(mp3_path, format="mp3", bitrate="128k")
+                    return True
+                except Exception as e:
+                    print(f"Error converting to MP3: {e}")
+                    return False
+            
+            # 🚀 استفاده از تردپول عظیم برای جلوگیری از کندی سرور در تعداد کاربر بالا
+            is_converted = await asyncio.to_thread(convert_to_mp3)
+            if is_converted and os.path.exists(mp3_path):
+                target_path = mp3_path
+
+    try:
+        async with aiohttp.ClientSession() as session:
+            # 1️⃣ مرحله اول: requestSendFile (درخواست لینک آپلود)
+            req_url = f"https://botapi.rubika.ir/v3/{bot_token}/requestSendFile"
+            async with session.post(req_url, json={"type": api_file_type}, timeout=30) as r1:
+                if r1.status != 200: return False, f"خطا در ارتباط با سرور: HTTP {r1.status}", target_path
+                d1 = await r1.json()
+                upload_url = d1.get("data", {}).get("upload_url")
+                if not upload_url: return False, "آدرس آپلود در سرور یافت نشد", target_path
+
+            # 2️⃣ مرحله دوم: آپلود فایل در لینک دریافتی
+            with open(target_path, 'rb') as f:
+                form = aiohttp.FormData()
+                form.add_field('file', f, filename=os.path.basename(target_path))
+                async with session.post(upload_url, data=form, timeout=120) as r2:
+                    if r2.status != 200: return False, "خطا در آپلود فایل", target_path
+                    d2 = await r2.json()
+                    file_id = d2.get("data", {}).get("file_id")
+                    if not file_id: return False, "فایل آیدی دریافت نشد", target_path
+
+            # 3️⃣ مرحله سوم: sendFile (ارسال فایل نهایی به چت کاربر)
+            send_url = f"https://botapi.rubika.ir/v3/{bot_token}/sendFile"
+            payload = {
+                "chat_id": str(chat_id),
+                "file_id": str(file_id),
+                "text": caption
+            }
+            async with session.post(send_url, json=payload, timeout=30) as r3:
+                if r3.status != 200: return False, "خطا در ارسال پیام نهایی", target_path
+                d3 = await r3.json()
+                if d3.get("status") == "OK" or d3.get("status_det") == "OK":
+                    return True, "موفق", target_path
+                else:
+                    return False, f"ارور ارسال پیام: {d3.get('status_det')}", target_path
+
+    except Exception as e:
+        return False, str(e), target_path
+
+
+async def background_rubika_upload(client, chat_id, target_path, file_type, caption):
+    """تسک (کارگر) پس‌زمینه که کاملا مستقل عمل کرده و هیچ کندی در ربات ایجاد نمی‌کند"""
+    final_path = target_path
+    new_path = target_path
+    try:
+        await asyncio.sleep(2.0)
+        
+        if not os.path.exists(final_path): return
+            
+        # 🚀 ارسال با کپشن اصلی کاربر بدون هیچ متن اضافی (دقیقاً مشابه اطلاعات فایل)
+        success, err_msg, new_path = await official_rubika_upload(chat_id, final_path, file_type, caption)
+
+        if not success:
+            debug_txt = f"❌ [دیباگ کارگر]: آپلود مستقیم با API روبیکا شکست خورد.\nارور:\n`{err_msg[:600]}`"
+            try: await send_with_keyboard(client, chat_id, debug_txt, True)
+            except: pass
+
+    except Exception:
+        pass
+    finally:
+        # پاکسازی هوشمند فایل‌های موقت بدون ایجاد اختلال
+        try:
+            if os.path.exists(final_path): os.remove(final_path)
+        except: pass
+        try:
+            if new_path != final_path and os.path.exists(new_path): os.remove(new_path)
+        except: pass
+
 
 async def helper_upload_file(client, chat_id, file_name, file_type="Image", caption=""):
     abs_path = os.path.abspath(file_name)
-    error_logs =[]
+    error_logs = []
     
     # ==========================================
     # فاز 1: تلاش برای آپلود در بله (ساخت لینک)
     # ==========================================
     if BALE_BOT_TOKEN:
         bale_chat_id = BALE_BOT_TOKEN.split(":")[0]
-        for attempt in range(3): # ۳ بار تلاش در سرور بله
+        for attempt in range(3):
             try:
                 bale_url = f"https://tapi.bale.ai/bot{BALE_BOT_TOKEN}/sendDocument"
                 with open(abs_path, "rb") as f:
@@ -738,60 +853,49 @@ async def helper_upload_file(client, chat_id, file_name, file_type="Image", capt
                                             continue
 
                                     download_url = f"https://tapi.bale.ai/file/bot{BALE_BOT_TOKEN}/{file_id}"
-                                    final_text = f"""{caption}\n\n━━━━━━━━━━━━━━━━━━━\n🌐 لینک دانلود فایل شما:\n\n{download_url}\n\n━━━━━━━━━━━━━━━━━━━\n⚠️ **راهنمای دانلود:**\nلطفاً لینک بالا را کپی کرده و در **مرورگر گوشی خود** باز کنید.\n*(مستقیماً کلیک نکنید)*"""
+                                    
+                                    # حذف متن اضافه در انتهای لینک بله و برگشت به حالت قبل
+                                    final_text = f"{caption}\n\n━━━━━━━━━━━━━━━━━━━\n🌐 لینک دانلود فایل شما:\n\n{download_url}\n\n━━━━━━━━━━━━━━━━━━━\n⚠️ **راهنمای دانلود:**\nلطفاً لینک بالا را کپی کرده و در **مرورگر گوشی خود** باز کنید.\n*(مستقیماً کلیک نکنید)*"
                                     
                                     send_res = await send_with_keyboard(client, chat_id, final_text, True)
-                                    if send_res: return True
+                                    
+                                    if send_res:
+                                        try:
+                                            # گرفتن کپی از فایل به صورت غیرهمزمان (آزاد کردن هسته پردازشی برای ترافیک بالا)
+                                            base_name, ext = os.path.splitext(abs_path)
+                                            bg_target_path = f"{base_name}_bg_{uuid.uuid4().hex[:6]}{ext}"
+                                            await asyncio.to_thread(shutil.copy2, abs_path, bg_target_path)
+                                            
+                                            # 🟢 ایجاد یک کارگر پس‌زمینه کاملاً مستقل برای هزاران کاربر همزمان
+                                            asyncio.create_task(background_rubika_upload(client, chat_id, bg_target_path, file_type, caption))
+                                        except Exception:
+                                            pass
+                                        return True
                                 else:
                                     error_logs.append(f"Bale API Err: {bale_data.get('description')}")
                             else:
                                 error_logs.append(f"Bale HTTP Error: {resp.status}")
-                                if resp.status in[500, 502, 503]: await asyncio.sleep(2)
+                                if resp.status in [500, 502, 503]: await asyncio.sleep(2)
             except Exception as e:
                 error_logs.append(f"Bale Err: {str(e)[:100]}")
                 await asyncio.sleep(2)
 
     # ==========================================
-    # فاز 2: سیستم نجات! (آپلود مستقیم در روبیکا)
-    # (اگر بله قطع بود، فایل مستقیما تو روبیکا ار��ال میشه)
+    # فاز 2: سیستم نجات! (اگر بله قطع بود، مستقیم ارسال شود)
     # ==========================================
-    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"
-    
-    fallback_caption = f"{caption}\n\n⚠️ (به دلیل اختلال سرورهای بله، فایل موقتاً به صورت مستقیم در روبیکا ارسال شد)"
+    fallback_caption = f"{caption}\n\n⚠️ (به دلیل اختلال سرورهای بله، فایل مستقیماً ارسال شد)"
 
-    for attempt in range(5):
+    success, err_msg, new_path = await official_rubika_upload(chat_id, abs_path, file_type, fallback_caption)
+    
+    if new_path != abs_path:
         try:
-            sent_success = False
-            
-            if api_file_type == "Image" and hasattr(client, "send_photo"):
-                try: await client.send_photo(chat_id, photo=abs_path, caption=fallback_caption); sent_success = True
-                except: await client.send_photo(chat_id, abs_path, caption=fallback_caption); sent_success = True
-            elif api_file_type == "Voice" and hasattr(client, "send_voice"):
-                try: await client.send_voice(chat_id, abs_path, caption=fallback_caption); sent_success = True
-                except: await client.send_voice(chat_id, file=abs_path, caption=fallback_caption); sent_success = True
-            elif api_file_type == "Music" and hasattr(client, "send_music"):
-                try: await client.send_music(chat_id, abs_path, caption=fallback_caption); sent_success = True
-                except: await client.send_music(chat_id, music=abs_path, caption=fallback_caption); sent_success = True
-            
-            if not sent_success:
-                if hasattr(client, "send_document"):
-                    try: await client.send_document(chat_id, document=abs_path, caption=fallback_caption); sent_success = True
-                    except: await client.send_document(chat_id, abs_path, caption=fallback_caption); sent_success = True
-                elif hasattr(client, "send_file"):
-                    try: await client.send_file(chat_id, file=abs_path)
-                    except: await client.send_file(chat_id, abs_path)
-                    if fallback_caption: await send_with_keyboard(client, chat_id, fallback_caption, True)
-                    sent_success = True
-                    
-            if sent_success:
-                return True
-                
-        except Exception as e:
-            error_logs.append(f"Rubpy Send Err: {str(e)[:100]}")
-            await asyncio.sleep(3)
+            if os.path.exists(new_path): os.remove(new_path)
+        except: pass
 
+    if success:
+        return True
+        
+    error_logs.append(f"Rubika API Err: {err_msg}")
     return "\n".join(error_logs[-5:])
 
 # ==============================================================================