Update main.py

main.py

@@ -20,7 +20,7 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "🚀 ربات هوشمند انتقال به گیت‌هاب (مجهز به APIهای جدید و تلگرام) فعال است!"
+    return "🚀 ربات هوشمند انتقال به گیت‌هاب (مجهز به کبالت و تلگرام) فعال است!"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860, threaded=True)
@@ -29,9 +29,6 @@ bot_token = os.environ.get("RUBIKA_AUTH", "توکن_روبیکا").strip()
 GITHUB_TOKEN = os.environ.get("GITHUB_TOKEN", "توکن_گیت‌هاب").strip()
 TG_BOT_TOKEN = os.environ.get("TG_BOT_TOKEN", "توکن_تلگرام").strip()
 
-# توکن مربوط به RapidAPI برای دانلود یوتیوب (مهم)
-RAPIDAPI_KEY = os.environ.get("RAPIDAPI_KEY", "b448420296msh34bbdc459d62fa0p124de1jsn8778ca694e65").strip()
-
 GITHUB_REPO = "hajiliker6-source/internet-melli"
 
 TG_API_ID = 6
@@ -63,12 +60,12 @@ async def ensure_tg_started():
         is_tg_started = True
 
 # ==============================================================================
-# 🟢 تابع اختصاصی دانلود یوتیوب از طریق RapidAPI (با لاگ‌گیر دقیق)
+# 🟢 تابع اختصاصی دانلود یوتیوب از طریق Cobalt (کاملا رایگان و بدون کلید)
 # ==============================================================================
-async def download_youtube_rapidapi(url, chat_id, message_id, client):
+async def download_youtube_cobalt(url, chat_id, message_id, client):
     status_msg_id = None
     try:
-        msg_text = "⏳ در حال ارسال درخواست به سرور قدرتمند پردازش یوتیوب..."
+        msg_text = "⏳ در حال استخراج لینک یوتیوب از سرورهای ضد-تحریم کبالت..."
         status_msg = await client.send_message(chat_id, msg_text, reply_to_message_id=message_id)
         status_msg_id = getattr(status_msg, 'message_id', None)
         if not status_msg_id and isinstance(status_msg, dict):
@@ -76,81 +73,42 @@ async def download_youtube_rapidapi(url, chat_id, message_id, client):
     except: pass
 
     try:
-        if RAPIDAPI_KEY == "توکن_رپید_ای_پی_آی" or not RAPIDAPI_KEY:
-            raise Exception("کلید RAPIDAPI_KEY تنظیم نشده است! بدون این کلید، API کار نخواهد کرد.")
-
-        API_BASE = "https://yt-video-audio-downloader-api.p.rapidapi.com"
+        api_url = "https://api.cobalt.tools/api/json"
         headers = {
+            "Accept": "application/json",
             "Content-Type": "application/json",
-            "X-RapidAPI-Key": RAPIDAPI_KEY,
-            "X-RapidAPI-Host": "yt-video-audio-downloader-api.p.rapidapi.com",
             "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
         }
+        payload = {
+            "url": url,
+            "vCodec": "h264",
+            "vQuality": "720",
+            "filenamePattern": "classic"
+        }
 
         async with aiohttp.ClientSession(headers=headers) as session:
-            # مرحله 1: شروع دانلود (مسیر نهایی و قطعی: /download)
-            payload = {"url": url, "format": "mp4", "quality": 720}
-            async with session.post(f"{API_BASE}/download", json=payload) as resp:
-                
+            # مرحله ۱: ارسال درخواست به کبالت برای دور زدن محدودیت‌های یوتیوب
+            async with session.post(api_url, json=payload) as resp:
                 if resp.status != 200:
                     server_log = await resp.text()
                     error_message = (
-                        f"خطا در ارتباط با سرور یوتیوب.\n"
+                        f"خطا در ارتباط با سرور کبالت.\n"
                         f"کد ارور: {resp.status}\n\n"
-                        f"📝 لاگ سرور RapidAPI:\n{server_log}"
+                        f"📝 لاگ سرور:\n{server_log}"
                     )
                     raise Exception(error_message)
                     
-                start_data = await resp.json()
-
-            download_url = None
-            if start_data.get("directDownload"):
-                download_url = start_data["downloadUrl"]
-            else:
-                job_id = start_data.get("jobId")
-                if not job_id:
-                    raise Exception(f"سرور شناسه Job ID را برنگرداند. پاسخ سرور:\n{start_data}")
-                
-                # خواندن داینامیک مسیر وضعیت
-                status_path = start_data.get("statusUrl")
-                if not status_path:
-                    status_path = f"/status/{job_id}" # بکاپ اگر سرور لینک نداد
-                    
-                if status_path.startswith("/"):
-                    status_url = API_BASE + status_path
-                else:
-                    status_url = f"{API_BASE}/{status_path}"
+                data = await resp.json()
                 
-                if status_msg_id:
-                    try: await client.edit_message_text(chat_id, status_msg_id, "⏳ ویدیو در سرور رپید در حال پردازش است (در حال بررسی وضعیت...)")
-                    except: pass
-
-                # مرحله 2: بررسی وضعیت پردازش
-                while True:
-                    await asyncio.sleep(3)
-                    async with session.get(status_url) as status_resp:
-                        if status_resp.status != 200:
-                            status_log = await status_resp.text()
-                            raise Exception(f"خطا در بررسی وضعیت (کد {status_resp.status}).\nلاگ:\n{status_log}")
-                            
-                        status_data = await status_resp.json()
-                        if status_data.get("status") == "completed":
-                            download_url = status_data["downloadUrl"]
-                            break
-                        elif status_data.get("status") == "error":
-                            raise Exception("خطای پردازش داخلی سرور: " + status_data.get("error", "نامشخص"))
-
+            download_url = data.get("url")
             if not download_url:
-                raise Exception("دریافت لینک نهایی دانلود از سرور با شکست مواجه شد.")
-            
-            if download_url.startswith("/"):
-                download_url = API_BASE + download_url
+                raise Exception(f"سرور کبالت لینک دانلود را برنگرداند. پاسخ:\n{data}")
 
             if status_msg_id:
-                try: await client.edit_message_text(chat_id, status_msg_id, "📥 پردازش موفق! در حال دانلود ویدیو از سرور واسط...")
+                try: await client.edit_message_text(chat_id, status_msg_id, "📥 لینک استخراج شد! در حال دانلود ویدیو روی سرور...")
                 except: pass
 
-            # مرحله 3: دانلود فایل نهایی 
+            # مرحله ۲: دانلود فایل نهایی روی سرور
             temp_dir = "/app/downloads"
             os.makedirs(temp_dir, exist_ok=True)
             filename = f"youtube_{uuid.uuid4().hex[:6]}.mp4"
@@ -167,9 +125,9 @@ async def download_youtube_rapidapi(url, chat_id, message_id, client):
             return file_path, filename, status_msg_id
 
     except Exception as e:
-        error_msg = f"❌ خطای یوتیوب:\n{str(e)}"
+        error_msg = f"❌ خطای یوتیوب (کبالت):\n{str(e)[:1000]}"
         if status_msg_id:
-            try: await client.edit_message_text(chat_id, status_msg_id, error_msg[:1000])
+            try: await client.edit_message_text(chat_id, status_msg_id, error_msg)
             except: pass
         return None, None, status_msg_id
 
@@ -178,7 +136,6 @@ async def download_youtube_rapidapi(url, chat_id, message_id, client):
 # ==============================================================================
 async def get_direct_link_from_free_api(url):
     is_ig = "instagram.com" in url.lower()
-    
     if not is_ig: return None
 
     async with aiohttp.ClientSession() as session:
@@ -345,8 +302,9 @@ def split_file_for_winrar(file_path, dest_dir, base_filename):
 async def process_and_commit_to_github(client, chat_id, message_id, url):
     url_l = url.lower()
     
+    # 🟢 هدایت لینک‌های یوتیوب به سمت کبالت
     if any(x in url_l for x in ["youtube.com", "youtu.be"]):
-        file_path, filename, status_msg_id = await download_youtube_rapidapi(url, chat_id, message_id, client)
+        file_path, filename, status_msg_id = await download_youtube_cobalt(url, chat_id, message_id, client)
     elif "instagram.com" in url_l:
         file_path, filename, status_msg_id = await download_via_free_api(url, chat_id, message_id, client)
     elif "t.me/" in url_l:
@@ -461,5 +419,5 @@ async def main_handler(client, update):
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
     threading.Thread(target=lambda: asyncio.run(cleanup_expired_branches()), daemon=True).start()
-    print("🚀 آماده به کار با APIهای جایگزین و RapidAPI...")
+    print("🚀 آماده به کار با APIهای جایگزین و Cobalt...")
     bot.run()