Update main.py

main.py

@@ -12,16 +12,15 @@ import re
 from flask import Flask
 from rubpy.bot import BotClient, filters
 from pyrogram import Client
-import yt_dlp
 
 # ==============================================================================
-# 🟢 تنظیمات اصلی ربات (توکن‌ها را اینجا وارد کنید)
+# 🟢 تنظیمات اصلی ربات
 # ==============================================================================
 app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "🚀 ربات انتقال به گیت‌هاب (پشتیبانی از وب، تلگرام و اینستاگرام) روشن است!"
+    return "🚀 ربات هوشمند انتقال به گیت‌هاب (مجهز به API رایگان ضد-تحریم) فعال است!"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860, threaded=True)
@@ -32,20 +31,15 @@ TG_BOT_TOKEN = os.environ.get("TG_BOT_TOKEN", "توکن_تلگرام").strip()
 
 GITHUB_REPO = "hajiliker6-source/internet-melli"
 
-# مقادیر عمومی API تلگرام
 TG_API_ID = 6
 TG_API_HASH = "eb06d4abfb49dc3eeb1aeb98ae0f581e"
 
-if bot_token == "توکن_روبیکا" or GITHUB_TOKEN == "توکن_گیت‌هاب":
-    print("❌ توکن روبیکا یا گیت‌هاب تنظیم نشده است! لطفاً تنظیمات بالای کد را بررسی کنید.")
-    exit()
-
 bot = BotClient(bot_token)
 BOT_GUID = None
 git_lock = asyncio.Lock()
 
 # ==============================================================================
-# 🟢 تنظیمات کلاینت تلگرام (Pyrogram)
+# 🟢 تنظیمات کلاینت تلگرام
 # ==============================================================================
 tg_client = None
 is_tg_started = False
@@ -66,54 +60,93 @@ async def ensure_tg_started():
         is_tg_started = True
 
 # ==============================================================================
-# 🟢 توابع دانلود (اینستاگرام، تلگرام و وب)
+# 🟢 توابع دانلود با API رایگان (بدون نیاز به کوکی و yt-dlp)
 # ==============================================================================
 
-# 1. تابع دانلود از اینستاگرام (و سایت‌های مشابه) با yt-dlp
-def sync_download_ytdlp(url, temp_dir):
-    """این تابع همگام است و در یک ترد مجزا اجرا می‌شود"""
-    ydl_opts = {
-        'outtmpl': f'{temp_dir}/{uuid.uuid4().hex[:6]}_%(title)s.%(ext)s',
-        'quiet': True,
-        'no_warnings': True,
-        'format': 'best'
+async def get_direct_link_from_free_api(url):
+    """استفاده از سرورهای واسط رایگان برای استخراج لینک مستقیم اینستاگرام و یوتیوب"""
+    
+    # لیست سرورهای API رایگان (اگر یکی خراب بود می‌رود سراغ بعدی)
+    api_instances = [
+        "https://api.cobalt.tools/api/json",
+        "https://co.wuk.sh/api/json",
+        "https://api.cobalt.pw/api/json"
+    ]
+    
+    headers = {
+        "Accept": "application/json",
+        "Content-Type": "application/json",
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64)",
+        "Origin": "https://cobalt.tools"
     }
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info = ydl.extract_info(url, download=True)
-        file_path = ydl.prepare_filename(info)
-        filename = os.path.basename(file_path)
-        # پاکسازی نام فایل از کاراکترهای عجیب برای گیت‌هاب
-        clean_filename = re.sub(r'[\\/*?:"<>|]', "", filename)
-        if clean_filename != filename:
-            clean_path = os.path.join(temp_dir, clean_filename)
-            os.rename(file_path, clean_path)
-            return clean_path, clean_filename
-        return file_path, filename
-
-async def download_from_instagram(url, chat_id, message_id, client):
+    
+    payload = {"url": url, "vQuality": "720"}
+    
+    async with aiohttp.ClientSession() as session:
+        for api_url in api_instances:
+            try:
+                async with session.post(api_url, json=payload, headers=headers, timeout=15) as resp:
+                    if resp.status in [200, 201]:
+                        data = await resp.json()
+                        if data.get("status") == "error":
+                            continue
+                        if "url" in data:
+                            return data["url"] # لینک مستقیم استخراج شد
+            except:
+                continue
+    return None
+
+async def download_via_free_api(url, chat_id, message_id, client):
     status_msg_id = None
+    source_name = "شبکه‌های اجتماعی"
+    if "youtube" in url or "youtu.be" in url: source_name = "یوتیوب"
+    elif "instagram" in url: source_name = "اینستاگرام"
+
     try:
-        status_msg = await client.send_message(chat_id, "⏳ در حال استخراج و دانلود رسانه از اینستاگرام...", reply_to_message_id=message_id)
+        msg_text = f"⏳ در حال دور زدن تحریم‌های {source_name} از طریق سرورهای واسط (API رایگان)..."
+        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):
             status_msg_id = status_msg.get('message_update', {}).get('message_id') or status_msg.get('message_id')
     except: pass
 
     try:
+        # مرحله اول: گرفتن لینک مستقیم از API
+        direct_url = await get_direct_link_from_free_api(url)
+        if not direct_url:
+            raise Exception("سرورهای واسط قادر به استخراج این لینک نبودند (احتمالاً پیج پرایوت است یا پست حذف شده).")
+
+        if status_msg_id:
+            try: await client.edit_message_text(chat_id, status_msg_id, "📥 لینک مستقیم استخراج شد! در حال دانلود روی سرور...")
+            except: pass
+
+        # مرحله دوم: دانلود فایل با لینک مستقیمِ به دست آمده
         temp_dir = "/app/downloads"
         os.makedirs(temp_dir, exist_ok=True)
-        
-        # اجرای کتابخانه همگام در یک ترد مجزا برای جلوگیری از قفل شدن ربات
-        file_path, filename = await asyncio.to_thread(sync_download_ytdlp, url, temp_dir)
+        filename = f"{source_name}_video_{uuid.uuid4().hex[:6]}.mp4"
+        file_path = os.path.join(temp_dir, filename)
+
+        headers = {"User-Agent": "Mozilla/5.0"}
+        async with aiohttp.ClientSession(headers=headers) as session:
+            async with session.get(direct_url, timeout=0) as resp:
+                if resp.status != 200:
+                    raise Exception("دانلود از لینک مستقیم با خطا مواجه شد.")
+                with open(file_path, 'wb') as f:
+                    async for chunk in resp.content.iter_chunked(2 * 1024 * 1024):
+                        if chunk: f.write(chunk)
+                        
         return file_path, filename, status_msg_id
+
     except Exception as e:
-        error_msg = f"❌ خطا در دانلود از اینستاگرام (احتمالاً پیج پرایوت است یا پست حذف شده):\n{str(e)[:150]}"
+        error_msg = f"❌ خطا در پردازش API واسط:\n{str(e)[:150]}"
         if status_msg_id:
             try: await client.edit_message_text(chat_id, status_msg_id, error_msg)
             except: pass
         return None, None, status_msg_id
 
-# 2. تابع دانلود از تلگرام
+# ==============================================================================
+# 🟢 تابع دانلود از تلگرام
+# ==============================================================================
 async def download_from_telegram(url, chat_id, message_id, client):
     status_msg_id = None
     try:
@@ -124,89 +157,74 @@ async def download_from_telegram(url, chat_id, message_id, client):
     except: pass
 
     try:
-        if not tg_client:
-            raise Exception("ربات تلگرام متصل نیست. لطفاً توکن تلگرام را در سورس کد قرار دهید.")
-
+        if not tg_client: raise Exception("توکن تلگرام تنظیم نشده است.")
         await ensure_tg_started()
         
         match = re.search(r"t\.me/([^/]+)/(\d+)", url)
-        if not match:
-            raise Exception("لینک تلگرام نامعتبر است.")
+        if not match: raise Exception("لینک تلگرام نامعتبر است.")
         
-        channel_username = match.group(1)
-        msg_id = int(match.group(2))
+        channel_username, msg_id = match.group(1), int(match.group(2))
         
         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
 
         msg = await tg_client.get_messages(channel_username, msg_id)
-        if not msg or not msg.media:
-            raise Exception("فایلی در این پیام یافت نشد یا پیام از کانال حذف شده است.")
+        if not msg or not msg.media: raise Exception("فایلی یافت نشد.")
         
         filename = None
         if getattr(msg, 'document', None): filename = msg.document.file_name
         elif getattr(msg, 'video', None): filename = msg.video.file_name
         elif getattr(msg, 'audio', None): filename = msg.audio.file_name
-        elif getattr(msg, 'photo', None): filename = f"tg_photo_{uuid.uuid4().hex[:6]}.jpg" # پشتیبانی از عکس تلگرام
+        elif getattr(msg, 'photo', None): filename = f"photo_{uuid.uuid4().hex[:6]}.jpg"
         
-        if not filename: filename = f"tg_media_{uuid.uuid4().hex[:6]}.dat"
+        if not filename: filename = f"tg_file_{uuid.uuid4().hex[:6]}.dat"
         
         temp_dir = "/app/downloads"
         os.makedirs(temp_dir, exist_ok=True)
         file_path = os.path.join(temp_dir, f"{uuid.uuid4().hex[:6]}_{filename}")
         
-        if status_msg_id:
-            try: await client.edit_message_text(chat_id, status_msg_id, "📥 در حال دانلود فایل از تلگرام به سرور واسط...")
-            except: pass
-        
         await tg_client.download_media(msg, file_name=file_path)
         return file_path, filename, status_msg_id
-
     except Exception as e:
-        error_msg = f"❌ خطای تلگرام:\n{str(e)[:150]}"
         if status_msg_id:
-            try: await client.edit_message_text(chat_id, status_msg_id, error_msg)
+            try: await client.edit_message_text(chat_id, status_msg_id, f"❌ خطای تلگرام: {str(e)[:100]}")
             except: pass
         return None, None, status_msg_id
 
-# 3. تابع دانلود وب (لینک مستقیم)
+# ==============================================================================
+# 🟢 تابع دانلود مستقیم (وب)
+# ==============================================================================
 async def download_large_file(url, chat_id, message_id, client):
     if "dropbox.com" in url:
-        if "dl=0" in url: url = url.replace("dl=0", "dl=1")
-        elif "?dl=" not in url and "&dl=" not in url: url += "&dl=1" if "?" in url else "?dl=1"
+        url = url.replace("dl=0", "dl=1") if "dl=0" in url else (url + ("&dl=1" if "?" in url else "?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)
     if not filename or '.' not in filename: filename = f"file_{uuid.uuid4().hex[:8]}.dat"
-        
     file_path = os.path.join(temp_dir, f"{uuid.uuid4().hex[:6]}_{filename}")
-    status_msg_id = None
     
+    status_msg_id = None
     try:
-        status_msg = await client.send_message(chat_id, "⏳ در حال دانلود فایل وب در سرور واسط...", reply_to_message_id=message_id)
+        status_msg = await client.send_message(chat_id, "⏳ در حال دانلود در سرور...", 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):
-            status_msg_id = status_msg.get('message_update', {}).get('message_id') or status_msg.get('message_id')
-    except Exception: pass
+    except: pass
 
     try:
-        headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64)"}
+        headers = {"User-Agent": "Mozilla/5.0"}
         async with aiohttp.ClientSession(headers=headers) as session:
-            async with session.get(url, timeout=0) as response:
-                if response.status != 200:
-                    return None, None, status_msg_id
+            async with session.get(url, timeout=0) as resp:
+                if resp.status != 200: return None, None, status_msg_id
                 with open(file_path, 'wb') as f:
-                    async for chunk in response.content.iter_chunked(2 * 1024 * 1024):
+                    async for chunk in resp.content.iter_chunked(2 * 1024 * 1024):
                         if chunk: f.write(chunk)
         return file_path, filename, status_msg_id
-    except Exception: return None, None, status_msg_id
+    except: return None, None, status_msg_id
 
 # ==============================================================================
-# 🟢 تابع خرد کردن فایل
+# 🟢 توابع گیت‌هاب و پردازش نهایی
 # ==============================================================================
 def split_file_for_winrar(file_path, dest_dir, base_filename):
     chunk_size = 95 * 1024 * 1024 
@@ -216,127 +234,94 @@ def split_file_for_winrar(file_path, dest_dir, base_filename):
         while True:
             chunk = f.read(chunk_size)
             if not chunk: break
-            part_name = f"{base_filename}.{part_num:03d}"
-            part_path = os.path.join(dest_dir, part_name)
-            with open(part_path, 'wb') as p:
-                p.write(chunk)
+            part_path = os.path.join(dest_dir, f"{base_filename}.{part_num:03d}")
+            with open(part_path, 'wb') as p: p.write(chunk)
             part_num += 1
 
-# ==============================================================================
-# 🟢 پردازش و آپلود در گیت‌هاب
-# ==============================================================================
 async def process_and_commit_to_github(client, chat_id, message_id, url):
+    url_l = url.lower()
     
-    url_lower = url.lower()
-    
-    # سیستم تشخیص هوشمند لینک
-    if "instagram.com/" in url_lower:
-        file_path, filename, status_msg_id = await download_from_instagram(url, chat_id, message_id, client)
-    elif "t.me/" in url_lower:
+    if any(x in url_l for x in ["youtube.com", "youtu.be", "instagram.com", "twitter.com", "x.com", "tiktok.com"]):
+        file_path, filename, status_msg_id = await download_via_free_api(url, chat_id, message_id, client)
+    elif "t.me/" in url_l:
         file_path, filename, status_msg_id = await download_from_telegram(url, chat_id, message_id, client)
     else:
         file_path, filename, status_msg_id = await download_large_file(url, chat_id, message_id, client)
     
     if not file_path or not os.path.exists(file_path):
-        if status_msg_id:
-            try: await client.edit_message_text(chat_id, status_msg_id, "❌ پردازش فایل ناموفق بود. لطفاً لینک را بررسی کنید.")
-            except: pass
         return
 
     try:
         if status_msg_id:
-            try: await client.edit_message_text(chat_id, status_msg_id, "✅ رسانه با موفقیت دریافت شد.\n✂️ در حال پارت‌بندی و انتقال به گیت‌هاب...")
+            try: await client.edit_message_text(chat_id, status_msg_id, "✅ فایل با موفقیت دریافت شد.\n✂️ انتقال به گیت‌هاب...")
             except: pass
 
         async with git_lock:
             current_timestamp = int(time.time())
-            branch_name = f"DL-{current_timestamp}-{uuid.uuid4().hex[:6]}"
+            branch_name = f"DL-{current_timestamp}-{uuid.uuid4().hex[:4]}"
             repo_url = f"https://oauth2:{GITHUB_TOKEN}@github.com/{GITHUB_REPO}.git"
             work_dir = "/app/git_workspace"
             
             if os.path.exists(work_dir): shutil.rmtree(work_dir)
-                
-            subprocess.run(["git", "clone", repo_url, work_dir], check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-            subprocess.run(["git", "config", "user.email", "bot@aialpha.local"], cwd=work_dir)
+            subprocess.run(["git", "clone", repo_url, work_dir], check=True, capture_output=True)
+            subprocess.run(["git", "config", "user.email", "bot@alpha.io"], cwd=work_dir)
             subprocess.run(["git", "config", "user.name", "AI Alpha Bot"], cwd=work_dir)
             subprocess.run(["git", "checkout", "--orphan", branch_name], cwd=work_dir, check=True)
-            subprocess.run(["git", "rm", "-rf", "."], cwd=work_dir, stderr=subprocess.DEVNULL)
+            subprocess.run(["git", "rm", "-rf", "."], cwd=work_dir, capture_output=True)
             
             file_folder = os.path.join(work_dir, filename)
             split_file_for_winrar(file_path, file_folder, filename)
             
             subprocess.run(["git", "add", "."], cwd=work_dir, check=True)
-            subprocess.run(["git", "commit", "-m", f"Add parts for {filename}"], cwd=work_dir, check=True)
+            subprocess.run(["git", "commit", "-m", f"Upload: {filename}"], cwd=work_dir, check=True)
             subprocess.run(["git", "push", "origin", branch_name], cwd=work_dir, check=True)
             
-            zip_download_link = f"https://github.com/{GITHUB_REPO}/archive/refs/heads/{branch_name}.zip"
+            zip_link = f"https://github.com/{GITHUB_REPO}/archive/refs/heads/{branch_name}.zip"
             
             success_text = (
-                f"🎉 فایل شما با موفقیت آماده شد!\n\n"
-                f"📁 **نام فایل:** {filename}\n"
-                f"🔗 **لینک دانلود مستقیم (بدون مصرف ترافیک بین‌الملل):**\n{zip_download_link}\n\n"
-                f"💡 **راهنما:** فایل بالا را دانلود کرده و استخراج (Extract) کنید.\n\n"
-                f"⚠️ **توجه مهم:**\n"
-                f"جهت حفظ کیفیت خدمات و آزادسازی فضای سرور، این فایل **دقیقاً ۲ ساعت دیگر به صورت خودکار حذف خواهد شد.** ⏳"
+                f"🎉 فایل شما آماده دانلود است!\n\n"
+                f"📁 **نام:** {filename}\n"
+                f"🔗 **لینک دانلود مستقیم (اینترنت ملی):**\n{zip_link}\n\n"
+                f"⏳ **مهلت دانلود:** ۲ ساعت"
             )
-            
             if status_msg_id:
                 try: await client.edit_message_text(chat_id, status_msg_id, success_text)
                 except: await client.send_message(chat_id, success_text)
-            else:
-                await client.send_message(chat_id, success_text)
+            else: await client.send_message(chat_id, success_text)
 
     except Exception as e:
-        error_txt = f"❌ خطای گیت‌هاب:\n{str(e)[:150]}"
         if status_msg_id:
-            try: await client.edit_message_text(chat_id, status_msg_id, error_txt)
-            except: await client.send_message(chat_id, error_txt)
-    finally:
-        try: os.remove(file_path)
-        except: pass
-        if os.path.exists("/app/git_workspace"):
-            try: shutil.rmtree("/app/git_workspace")
+            try: await client.edit_message_text(chat_id, status_msg_id, f"❌ خطای گیت‌هاب:\n{str(e)[:150]}")
             except: pass
+    finally:
+        if file_path and os.path.exists(file_path): os.remove(file_path)
+        if os.path.exists("/app/git_workspace"): shutil.rmtree("/app/git_workspace")
 
 # ==============================================================================
-# 🟢 سیستم هوشمند حذف خودکار فایل‌های منقضی شده
+# 🟢 سیستم حذف خودکار
 # ==============================================================================
 async def cleanup_expired_branches():
     while True:
         try:
             api_url = f"https://api.github.com/repos/{GITHUB_REPO}/git/refs/heads"
             headers = {"Authorization": f"token {GITHUB_TOKEN}", "Accept": "application/vnd.github.v3+json"}
-            
             async with aiohttp.ClientSession() as session:
                 async with session.get(api_url, headers=headers) as resp:
                     if resp.status == 200:
                         refs = await resp.json()
                         current_time = int(time.time())
-                        
                         for ref in refs:
                             ref_name = ref.get("ref", "")
                             if "refs/heads/DL-" in ref_name:
                                 branch_name = ref_name.split("/")[-1]
                                 parts = branch_name.split("-")
-                                
-                                if len(parts) >= 3 and parts[1].isdigit():
-                                    creation_time = int(parts[1])
-                                    if current_time - creation_time > 7200:
-                                        delete_url = f"https://api.github.com/repos/{GITHUB_REPO}/git/refs/heads/{branch_name}"
-                                        await session.delete(delete_url, headers=headers)
-                                        print(f"🗑 Branch {branch_name} automatically deleted (Expired).")
-        except Exception as e:
-            pass
+                                if len(parts) >= 2 and parts[1].isdigit():
+                                    if current_time - int(parts[1]) > 7200:
+                                        del_url = f"https://api.github.com/repos/{GITHUB_REPO}/git/refs/heads/{branch_name}"
+                                        await session.delete(del_url, headers=headers)
+        except: pass
         await asyncio.sleep(1800)
 
-def run_cleanup_system():
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    loop.run_until_complete(cleanup_expired_branches())
-
-# ==============================================================================
-# 🟢 هندلر پیام‌های روبیکا
-# ==============================================================================
 @bot.on_update(filters.private)
 async def main_handler(client, update):
     global BOT_GUID
@@ -359,16 +344,7 @@ async def main_handler(client, update):
         if not user_text_str: return
 
         if user_text_str.lower() in ["/start", "سلام"]:
-            welcome_msg = (
-                "👋 سلام!\n\n"
-                "من یک ربات انتقال‌دهنده هوشمند هستم. لینک‌های زیر را پشتیبانی می‌کنم:\n\n"
-                "🌐 **لینک دانلود مستقیم** از سایت‌ها\n"
-                "✈️ **لینک پست یا عکس کانال تلگرام** (مثال: `t.me/channel/123`)\n"
-                "📸 **لینک ریلز یا پست اینستاگرام**\n\n"
-                "لینک خود را بفرستید تا با نهایت سرعت آن را روی سرور گیت‌هاب (با اینترنت ملی) برایتان آماده کنم.\n\n"
-                "⏳ **فایل‌ها جهت مدیریت سرور، دقیقاً دو ساعت بعد از ساخته‌شدن به صورت کاملاً خودکار پاک خواهند شد.**"
-            )
-            await client.send_message(chat_id, welcome_msg)
+            await client.send_message(chat_id, "👋 سلام! لینک مورد نظرت رو بفرست:\n\n🎥 **یوتیوب / اینستاگرام / تیک‌تاک**\n✈️ **تلگرام (کانال عمومی)**\n🌐 **لینک مستقیم وب**")
             return
 
         if user_text_str.lower().startswith("http"):
@@ -377,11 +353,7 @@ async def main_handler(client, update):
     except Exception: traceback.print_exc()
 
 if __name__ == "__main__":
-    if not TG_BOT_TOKEN or TG_BOT_TOKEN == "توکن_تلگرام":
-        print("⚠️ هشدار: توکن ربات تلگرام تنظیم نشده است! دانلود از تلگرام کار نخواهد کرد.")
-        
     threading.Thread(target=run_flask, daemon=True).start()
-    threading.Thread(target=run_cleanup_system, daemon=True).start()
-    
-    print("🚀 ربات کلاستر گیت‌هاب (پشتیبانی اینستاگرام، تلگرام و وب) روشن شد...")
+    threading.Thread(target=lambda: asyncio.run(cleanup_expired_branches()), daemon=True).start()
+    print("🚀 آماده به کار با API ضد تحریم...")
     bot.run()