Update main.py

main.py

@@ -21,11 +21,12 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "🚀 ربات هوشمند انتقال فایل (YouTube, Instagram, Telegram, Web) روشن است!"
+    return "🚀 ربات هوشمند انتقال به گیت‌هاب (وب، تلگرام، یوتیوب و اینستاگرام) فعال است!"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860, threaded=True)
 
+# توکن‌های خود را در بخش زیر جایگذاری کنید
 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()
@@ -37,7 +38,7 @@ TG_API_ID = 6
 TG_API_HASH = "eb06d4abfb49dc3eeb1aeb98ae0f581e"
 
 if bot_token == "توکن_روبیکا" or GITHUB_TOKEN == "توکن_گیت‌هاب":
-    print("❌ تنظیمات ناقص است! توکن روبیکا یا گیت‌هاب را وارد کنید.")
+    print("❌ خطا: توکن‌های اصلی تنظیم نشده‌اند!")
     exit()
 
 bot = BotClient(bot_token)
@@ -45,13 +46,19 @@ BOT_GUID = None
 git_lock = asyncio.Lock()
 
 # ==============================================================================
-# 🟢 کلاینت تلگرام
+# 🟢 تنظیمات کلاینت تلگرام (Pyrogram)
 # ==============================================================================
 tg_client = None
 is_tg_started = False
 
 if TG_BOT_TOKEN and TG_BOT_TOKEN != "توکن_تلگرام":
-    tg_client = Client("tg_bot_session", api_id=TG_API_ID, api_hash=TG_API_HASH, bot_token=TG_BOT_TOKEN, in_memory=True)
+    tg_client = Client(
+        "tg_bot_session", 
+        api_id=TG_API_ID, 
+        api_hash=TG_API_HASH, 
+        bot_token=TG_BOT_TOKEN, 
+        in_memory=True
+    )
 
 async def ensure_tg_started():
     global is_tg_started
@@ -60,10 +67,10 @@ async def ensure_tg_started():
         is_tg_started = True
 
 # ==============================================================================
-# 🟢 توابع دانلود حرفه‌ای
+# 🟢 توابع دانلود پیشرفته
 # ==============================================================================
 
-# 1. دانلود از یوتیوب، اینستاگرام و سایر شبکه‌های اجتماعی
+# 1. تابع دانلود جهانی با yt-dlp (یوتیوب، اینستاگرام، توییتر، تیک‌تاک)
 def sync_download_ytdlp(url, temp_dir):
     ydl_opts = {
         'outtmpl': f'{temp_dir}/{uuid.uuid4().hex[:6]}_%(title)s.%(ext)s',
@@ -75,24 +82,23 @@ def sync_download_ytdlp(url, temp_dir):
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         info = ydl.extract_info(url, download=True)
         file_path = ydl.prepare_filename(info)
-        # اگر فایل mkv بود یا فرمت دیگری، yt-dlp ممکن است پسوند را تغییر دهد
-        if not os.path.exists(file_path):
-            base_path = os.path.splitext(file_path)[0]
-            for ext in ['.mp4', '.mkv', '.webm', '.flv']:
-                if os.path.exists(base_path + ext):
-                    file_path = base_path + ext
-                    break
-        
         filename = os.path.basename(file_path)
-        clean_filename = re.sub(r'[\\/*?:"<>|]', "", filename).replace(" ", "_")
-        clean_path = os.path.join(temp_dir, clean_filename)
-        os.rename(file_path, clean_path)
-        return clean_path, clean_filename
+        # پاکسازی نام فایل برای جلوگیری از خطای سیستم فایل گیت‌هاب
+        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_social_media(url, chat_id, message_id, client, platform_name):
+async def download_via_ytdlp(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, f"⏳ در حال پردازش لینک {platform_name} و دانلود با بالاترین کیفیت...", reply_to_message_id=message_id)
+        status_msg = await client.send_message(chat_id, f"⏳ در حال استخراج و دانلود از {source_name}...\n(ممکن است برای ویدیوهای حجیم کمی طول بکشد)", 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')
@@ -101,173 +107,232 @@ async def download_social_media(url, chat_id, message_id, client, platform_name)
     try:
         temp_dir = "/app/downloads"
         os.makedirs(temp_dir, exist_ok=True)
+        
+        # اجرای دانلود در ترد مجزا برای جلوگیری از فریز شدن ربات روبیکا
         file_path, filename = await asyncio.to_thread(sync_download_ytdlp, url, temp_dir)
         return file_path, filename, status_msg_id
     except Exception as e:
-        error_msg = f"❌ خطا در دانلود از {platform_name}:\nاحتمالاً محتوا حذف شده یا محدودیت جغرافیایی دارد."
+        error_msg = f"❌ خطا در دانلود از {source_name}:\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. دانلود از تلگرام (کانال‌های عمومی)
+# 2. تابع دانلود از تلگرام
 async def download_from_telegram(url, chat_id, message_id, client):
     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: pass
 
     try:
         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, m_id = match.group(1), int(match.group(2))
-        msg = await tg_client.get_messages(channel, m_id)
+        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, "⏳ در حال دریافت اطلاعات فایل از تلگرام...")
+            except: pass
+
+        msg = await tg_client.get_messages(channel_username, msg_id)
         if not msg or not msg.media: raise Exception("فایلی یافت نشد.")
         
-        filename = (getattr(msg.document, 'file_name', None) or 
-                    getattr(msg.video, 'file_name', None) or 
-                    getattr(msg.audio, 'file_name', None) or 
-                    f"photo_{uuid.uuid4().hex[:5]}.jpg")
+        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"photo_{uuid.uuid4().hex[:6]}.jpg"
+        
+        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, "📥 در حال دریافت فایل از تلگرام...")
+            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:
-        if status_msg_id: await client.edit_message_text(chat_id, status_msg_id, f"❌ خطا: {str(e)[:100]}")
+        if status_msg_id:
+            try: await client.edit_message_text(chat_id, status_msg_id, f"❌ خطای تلگرام: {str(e)[:100]}")
+            except: pass
         return None, None, status_msg_id
 
-# 3. دانلود فایل وب (لینک مستقیم)
+# 3. تابع دانلود وب (لینک مستقیم)
 async def download_large_file(url, chat_id, message_id, client):
+    if "dropbox.com" in url:
+        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)
-    filename = os.path.basename(urllib.parse.urlparse(url).path) or f"file_{uuid.uuid4().hex[:6]}.dat"
+    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
     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)
-        async with aiohttp.ClientSession() as session:
+    except: pass
+
+    try:
+        headers = {"User-Agent": "Mozilla/5.0"}
+        async with aiohttp.ClientSession(headers=headers) as session:
             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 resp.content.iter_chunked(2*1024*1024): f.write(chunk)
+                    async for chunk in resp.content.iter_chunked(2 * 1024 * 1024):
+                        if chunk: f.write(chunk)
         return file_path, filename, status_msg_id
     except: return None, None, status_msg_id
 
 # ==============================================================================
-# 🟢 عملیات گیت‌هاب و پارت‌بندی
+# 🟢 توابع گیت‌هاب و پردازش نهایی
 # ==============================================================================
-def split_file(file_path, dest_dir, base_filename):
-    chunk_size = 94 * 1024 * 1024 
+def split_file_for_winrar(file_path, dest_dir, base_filename):
+    chunk_size = 95 * 1024 * 1024 
     part_num = 1
     os.makedirs(dest_dir, exist_ok=True)
     with open(file_path, 'rb') as f:
         while True:
             chunk = f.read(chunk_size)
             if not chunk: break
-            with open(os.path.join(dest_dir, f"{base_filename}.{part_num:03d}"), '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()
-    # تشخیص پلتفرم
-    if any(x in url_l for x in ["youtube.com", "youtu.be"]):
-        file_path, filename, status_msg_id = await download_social_media(url, chat_id, message_id, client, "یوتیوب")
-    elif "instagram.com" in url_l:
-        file_path, filename, status_msg_id = await download_social_media(url, chat_id, message_id, client, "اینستاگرام")
-    elif any(x in url_l for x in ["tiktok.com", "twitter.com", "x.com"]):
-        file_path, filename, status_msg_id = await download_social_media(url, chat_id, message_id, client, "شبکه‌های اجتماعی")
+    
+    # تشخیص هوشمند منبع لینک
+    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_ytdlp(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): return
+    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✂️ در حال پارت‌بندی و انتقال به گیت‌هاب...")
+            except: pass
+
         async with git_lock:
-            ts = int(time.time())
-            branch = f"DL-{ts}-{uuid.uuid4().hex[:5]}"
+            current_timestamp = int(time.time())
+            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, capture_output=True)
-            subprocess.run(["git", "config", "user.email", "bot@ai.com"], cwd=work_dir)
-            subprocess.run(["git", "config", "user.name", "Bot"], cwd=work_dir)
-            subprocess.run(["git", "checkout", "--orphan", branch], cwd=work_dir, check=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, capture_output=True)
             
-            split_file(file_path, os.path.join(work_dir, filename), filename)
+            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", "add file"], cwd=work_dir, check=True)
-            subprocess.run(["git", "push", "origin", branch], 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)
             
-            link = f"https://github.com/{GITHUB_REPO}/archive/refs/heads/{branch}.zip"
-            msg = (f"🎉 **فایل شما آماده دانلود است!**\n\n📁 نام: {filename}\n🔗 لینک دانلود (اینترنت ملی):\n{link}\n\n"
-                   f"⏳ این لینک **۲ ساعت دیگر** به طور خودکار حذف می‌شود.")
-            await client.edit_message_text(chat_id, status_msg_id, msg)
+            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_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)
+
     except Exception as e:
-        if status_msg_id: await client.send_message(chat_id, f"❌ خطای گیت: {str(e)[:100]}")
+        if status_msg_id:
+            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_loop():
+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}"}
-            async with aiohttp.ClientSession() as sess:
-                async with sess.get(api_url, headers=headers) as r:
-                    if r.status == 200:
-                        for ref in await r.json():
-                            name = ref["ref"].split("/")[-1]
-                            if name.startswith("DL-"):
-                                try:
-                                    creation_ts = int(name.split("-")[1])
-                                    if int(time.time()) - creation_ts > 7200:
-                                        await sess.delete(f"{api_url}/{name}", headers=headers)
-                                        print(f"Deleted {name}")
-                                except: pass
+            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) >= 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)
+                                        print(f"🗑 Deleted expired branch: {branch_name}")
         except: pass
         await asyncio.sleep(1800)
 
-# ==============================================================================
-# 🟢 شروع به کار
-# ==============================================================================
 @bot.on_update(filters.private)
-async def handler(client, update):
+async def main_handler(client, update):
     global BOT_GUID
-    if not BOT_GUID:
-        me = await client.get_me()
-        BOT_GUID = me.user.user_guid
-    
-    msg = getattr(update, "message", update)
-    if not msg or msg.get('author_object_guid') == BOT_GUID: return
-    
-    text = msg.get("text", "").strip()
-    if text.lower().startswith("http"):
-        asyncio.create_task(process_and_commit_to_github(client, msg.get("author_object_guid"), msg.get("message_id"), text))
-    elif text == "/start":
-        await client.send_message(msg.get("author_object_guid"), "🚀 لینک یوتیوب، اینستاگرام، تلگرام یا لینک مستقیم بفرست تا برات آماده کنم!")
+    try:
+        if not BOT_GUID:
+            me = await client.get_me()
+            if me and hasattr(me, 'user'): BOT_GUID = getattr(me.user, 'user_guid', None)
+
+        msg_obj = getattr(update, "message", None) or getattr(update, "new_message", None)
+        author_id = getattr(update, 'author_guid', None)
+        if not author_id and msg_obj: author_id = msg_obj.get('author_object_guid') if isinstance(msg_obj, dict) else getattr(msg_obj, 'author_object_guid', None)
+        if BOT_GUID and author_id == BOT_GUID: return
+        
+        chat_id = getattr(update, 'object_guid', None) or getattr(update, 'author_guid', None) or getattr(update, "chat_id", None)
+        if not chat_id: return
+        msg_id = getattr(update, "message_id", None)
+        
+        user_text = getattr(update, "text", "") or getattr(msg_obj, "text", "")
+        user_text_str = str(user_text).strip()
+        if not user_text_str: return
+
+        if user_text_str.lower() in ["/start", "سلام"]:
+            await client.send_message(chat_id, "👋 سلام! لینک مورد نظرت رو بفرست:\n\n🎥 **یوتیوب / اینستاگرام / تیک‌تاک**\n✈️ **تلگرام (کانال عمومی)**\n🌐 **لینک مستقیم وب**\n\nفایل‌ها پارت‌بندی شده و روی گیت‌هاب (اینترنت ملی) آپلود میشن و ۲ ساعت بعد خودکار حذف میشن.")
+            return
+
+        if user_text_str.lower().startswith("http"):
+            asyncio.create_task(process_and_commit_to_github(client, chat_id, msg_id, user_text_str))
+            
+    except Exception: traceback.print_exc()
 
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
-    threading.Thread(target=lambda: asyncio.run(cleanup_loop()), daemon=True).start()
+    threading.Thread(target=lambda: asyncio.run(cleanup_expired_branches()), daemon=True).start()
+    print("🚀 ربات با پشتیبانی کامل از یوتیوب و اینستاگرام روشن شد...")
     bot.run()