Update main.py

main.py

@@ -15,13 +15,13 @@ from pyrogram import Client
 import yt_dlp
 
 # ==============================================================================
-# 🟢 تنظیمات اصلی ربات
+# 🟢 تنظیمات اصلی ربات (توکن‌ها را اینجا وارد کنید)
 # ==============================================================================
 app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "🚀 ربات هوشمند انتقال به گیت‌هاب (ضد-تحریم یوتیوب و اینستاگرام) فعال است!"
+    return "🚀 ربات انتقال به گیت‌هاب (پشتیبانی از وب، تلگرام و اینستاگرام) روشن است!"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860, threaded=True)
@@ -32,11 +32,12 @@ 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("❌ خطا: توکن‌های اصلی تنظیم نشده‌اند!")
+    print("❌ توکن روبیکا یا گیت‌هاب تنظیم نشده است! لطفاً تنظیمات بالای کد را بررسی کنید.")
     exit()
 
 bot = BotClient(bot_token)
@@ -65,52 +66,34 @@ async def ensure_tg_started():
         is_tg_started = True
 
 # ==============================================================================
-# 🟢 توابع دانلود پیشرفته (آپدیت ضد-ربات)
+# 🟢 توابع دانلود (اینستاگرام، تلگرام و وب)
 # ==============================================================================
 
+# 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': 'b',
-        'nocheckcertificate': True, # حل مشکل قطعی SSL
-        'source_address': '0.0.0.0', # اجبار به استفاده از IPv4 برای جلوگیری از بلاک شدن دیتاسنتر
-        'http_headers': {
-            # شبیه‌سازی مرورگر سافاری در آیفون برای دور زدن دیوار امنیتی اینستاگرام
-            'User-Agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 16_6 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.6 Mobile/15E148 Safari/604.1',
-            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
-            'Accept-Language': 'en-us,en;q=0.5',
-            'Sec-Fetch-Mode': 'navigate'
-        },
-        'extractor_args': {
-            'youtube': {
-                # استفاده از کلاینت‌های مختلف برای دور زدن ربات یوتیوب
-                'player_client': ['ios', 'android', 'web'] 
-            }
-        }
+        'format': 'best'
     }
-    
     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_via_ytdlp(url, chat_id, message_id, client):
+async def download_from_instagram(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"⏳ در حال استخراج و دور زدن تحریم‌های {source_name}...\n(ممکن است کمی طول بکشد)", 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')
@@ -120,10 +103,11 @@ async def download_via_ytdlp(url, chat_id, message_id, client):
         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"❌ متأسفانه سرور توسط {source_name} مسدود شده است:\n{str(e)[:150]}"
+        error_msg = f"❌ خطا در دانلود از اینستاگرام (احتمالاً پیج پرایوت است یا پست حذف شده):\n{str(e)[:150]}"
         if status_msg_id:
             try: await client.edit_message_text(chat_id, status_msg_id, error_msg)
             except: pass
@@ -140,76 +124,89 @@ 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, msg_id = match.group(1), int(match.group(2))
+        channel_username = match.group(1)
+        msg_id = 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"photo_{uuid.uuid4().hex[:6]}.jpg"
+        elif getattr(msg, 'photo', None): filename = f"tg_photo_{uuid.uuid4().hex[:6]}.jpg" # پشتیبانی از عکس تلگرام
         
-        if not filename: filename = f"tg_file_{uuid.uuid4().hex[:6]}.dat"
+        if not filename: filename = f"tg_media_{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:
+        error_msg = f"❌ خطای تلگرام:\n{str(e)[:150]}"
         if status_msg_id:
-            try: await client.edit_message_text(chat_id, status_msg_id, f"❌ خطای تلگرام: {str(e)[:100]}")
+            try: await client.edit_message_text(chat_id, status_msg_id, error_msg)
             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:
-        url = url.replace("dl=0", "dl=1") if "dl=0" in url else (url + ("&dl=1" if "?" in url else "?dl=1"))
+        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"
 
     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
+    
     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)
-    except: pass
+        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
 
     try:
-        headers = {"User-Agent": "Mozilla/5.0"}
+        headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64)"}
         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
+            async with session.get(url, timeout=0) as response:
+                if response.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):
+                    async for chunk in response.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
+    except Exception: return None, None, status_msg_id
 
 # ==============================================================================
-# 🟢 توابع گیت‌هاب و پردازش نهایی
+# 🟢 تابع خرد کردن فایل
 # ==============================================================================
 def split_file_for_winrar(file_path, dest_dir, base_filename):
     chunk_size = 95 * 1024 * 1024 
@@ -219,98 +216,127 @@ def split_file_for_winrar(file_path, dest_dir, base_filename):
         while True:
             chunk = f.read(chunk_size)
             if not chunk: break
-            part_path = os.path.join(dest_dir, f"{base_filename}.{part_num:03d}")
-            with open(part_path, 'wb') as p: p.write(chunk)
+            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_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", "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:
+    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:
         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, "❌ پردازش لینک با خطا مواجه شد.")
+            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[:4]}"
+            branch_name = f"DL-{current_timestamp}-{uuid.uuid4().hex[:6]}"
             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@alpha.io"], cwd=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", "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)
+            subprocess.run(["git", "rm", "-rf", "."], cwd=work_dir, stderr=subprocess.DEVNULL)
             
             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"Upload: {filename}"], cwd=work_dir, check=True)
+            subprocess.run(["git", "commit", "-m", f"Add parts for {filename}"], cwd=work_dir, check=True)
             subprocess.run(["git", "push", "origin", branch_name], cwd=work_dir, check=True)
             
-            zip_link = f"https://github.com/{GITHUB_REPO}/archive/refs/heads/{branch_name}.zip"
+            zip_download_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"⏳ **مهلت دانلود:** این لینک دقیقاً ۲ ساعت دیگر برای همیشه حذف می‌شود."
+                f"🎉 فایل شما با موفقیت آماده شد!\n\n"
+                f"📁 **نام فایل:** {filename}\n"
+                f"🔗 **لینک دانلود مستقیم (بدون مصرف ترافیک بین‌الملل):**\n{zip_download_link}\n\n"
+                f"💡 **راهنما:** فایل بالا را دانلود کرده و استخراج (Extract) کنید.\n\n"
+                f"⚠️ **توجه مهم:**\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, f"❌ خطای گیت‌هاب:\n{str(e)[:150]}")
-            except: pass
+            try: await client.edit_message_text(chat_id, status_msg_id, error_txt)
+            except: await client.send_message(chat_id, error_txt)
     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")
+        try: os.remove(file_path)
+        except: pass
+        if os.path.exists("/app/git_workspace"):
+            try: shutil.rmtree("/app/git_workspace")
+            except: pass
 
 # ==============================================================================
-# 🟢 پاکسازی و اجرا
+# 🟢 سیستم هوشمند حذف خودکار فایل‌های منقضی شده
 # ==============================================================================
 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) >= 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
+                                
+                                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
         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
@@ -333,7 +359,16 @@ async def main_handler(client, update):
         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فایل‌ها پارت‌بندی شده و روی گیت‌هاب (اینترنت ملی) آپلود میشن و ۲ ساعت بعد خودکار حذف میشن.")
+            welcome_msg = (
+                "👋 سلام!\n\n"
+                "من یک ربات انتقال‌دهنده هوشمند هستم. لینک‌های زیر را پشتیبانی می‌کنم:\n\n"
+                "🌐 **لینک دانلود مستقیم** از سایت‌ها\n"
+                "✈️ **لینک پست یا عکس کانال تلگرام** (مثال: `t.me/channel/123`)\n"
+                "📸 **لینک ریلز یا پست اینستاگرام**\n\n"
+                "لینک خود را بفرستید تا با نهایت سرعت آن را روی سرور گیت‌هاب (با اینترنت ملی) برایتان آماده کنم.\n\n"
+                "⏳ **فایل‌ها جهت مدیریت سرور، دقیقاً دو ساعت بعد از ساخته‌شدن به صورت کاملاً خودکار پاک خواهند شد.**"
+            )
+            await client.send_message(chat_id, welcome_msg)
             return
 
         if user_text_str.lower().startswith("http"):
@@ -342,7 +377,11 @@ 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=lambda: asyncio.run(cleanup_expired_branches()), daemon=True).start()
-    print("🚀 ربات با پشتیبانی کامل از یوتیوب و اینستاگرام روشن شد...")
+    threading.Thread(target=run_cleanup_system, daemon=True).start()
+    
+    print("🚀 ربات کلاستر گیت‌هاب (پشتیبانی اینستاگرام، تلگرام و وب) روشن شد...")
     bot.run()