Update main.py

main.py

@@ -11,13 +11,13 @@ from flask import Flask
 from rubpy.bot import BotClient, filters
 
 # ==============================================================================
-# 🟢 تنظیمات
+# 🟢 تنظیمات و متغیرهای محیطی
 # ==============================================================================
 app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "🚀 ربات کلاستر گیت‌هاب (هماهنگ با گوشی) فعال است!"
+    return "🚀 ربات انتقال پارت‌بندی شده به گیت‌هاب (اینترنت ملی) روشن است!"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860, threaded=True)
@@ -26,18 +26,18 @@ bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
 GITHUB_TOKEN = os.environ.get("GITHUB_TOKEN", "").strip()
 GITHUB_REPO = "Hamed744/internet-melli"
 
+if not bot_token or not GITHUB_TOKEN:
+    print("❌ توکن روبیکا یا گیت‌هاب وارد نشده است!")
+    exit()
+
 bot = BotClient(bot_token)
-git_lock = asyncio.Lock()
+BOT_GUID = None
 
-# ==============================================================================
-# 🟢 تابع کمکی محاسبه حجم
-# ==============================================================================
-def get_size_format(file_path):
-    size_bytes = os.path.getsize(file_path)
-    return f"{size_bytes / (1024 * 1024):.2f} MB"
+# قفل برای جلوگیری از تداخل کامیت‌های همزمان گیت
+git_lock = asyncio.Lock()
 
 # ==============================================================================
-# 🟢 دانلود فایل
+# 🟢 تابع دانلود (تغییری نکرده است)
 # ==============================================================================
 async def download_large_file(url, chat_id, message_id, client):
     if "dropbox.com" in url:
@@ -50,114 +50,161 @@ async def download_large_file(url, chat_id, message_id, client):
     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_id = getattr(status_msg, 'message_id', None) or status_msg.get('message_id')
-    except: pass
+        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
 
     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 response:
-                if response.status != 200: return None, None, status_msg_id
+                if response.status != 200:
+                    return None, None, status_msg_id
                 with open(file_path, 'wb') as f:
-                    async for chunk in response.content.iter_chunked(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
 
 # ==============================================================================
-# 🟢 خرد کردن استاندارد فایل برای اپلیکیشن‌های گوشی (با 7z)
+# 🟢 تابع خرد کردن فایل برای WinRAR (.001, .002, ...)
 # ==============================================================================
-def split_file_with_7z(file_path, dest_dir, base_filename):
-    os.makedirs(dest_dir, exist_ok=True)
-    archive_path = os.path.join(dest_dir, f"{base_filename}.7z")
-    source_dir = os.path.dirname(file_path)
-    source_file = os.path.basename(file_path)
+def split_file_for_winrar(file_path, dest_dir, base_filename):
+    chunk_size = 95 * 1024 * 1024  # محدودیت 95 مگابایتی برای گیت‌هاب
+    part_num = 1
     
-    # -v95m: حجم هر پارت ۹۵ مگابایت
-    # -mx=0: سطح فشرده‌سازی صفر (فقط خرد می‌کند تا سرعت سرور پایین نیاید)
-    cmd = ['7z', 'a', '-v95m', '-mx=0', archive_path, source_file]
+    os.makedirs(dest_dir, exist_ok=True)
     
-    # اجرا در پوشه سورس تا مسیرهای تو در تو داخل فایل زیپ ایجاد نشود
-    subprocess.run(cmd, cwd=source_dir, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    with open(file_path, 'rb') as f:
+        while True:
+            chunk = f.read(chunk_size)
+            if not chunk: break
+            
+            # تولید نام پارت به سبک WinRAR (مثال: video.mp4.001)
+            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(client, chat_id, message_id, url):
+async def process_and_commit_to_github(client, chat_id, message_id, url):
     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: await client.edit_message_text(chat_id, status_msg_id, "❌ خطای دانلود.")
+        if status_msg_id:
+            try: await client.edit_message_text(chat_id, status_msg_id, "❌ خطا در دانلود فایل اصلی.")
+            except: pass
         return
 
-    file_size_str = get_size_format(file_path)
-
     try:
         if status_msg_id:
-            await client.edit_message_text(chat_id, status_msg_id, f"✅ دانلود شد ({file_size_str})\n✂️ در حال ساخت پارت‌های استاندارد گوشی و Push به گیت‌هاب...")
+            try: await client.edit_message_text(chat_id, status_msg_id, "✅ دانلود تکمیل شد.\n✂️ در حال پارت‌بندی و انتقال به مخزن گیت‌هاب (این مرحله زمان‌بر است)...")
+            except: pass
 
+        # استفاده از قفل تا عملیات گیت برای لینک‌های مختلف تداخل پیدا نکند
         async with git_lock:
             tag_name = f"DL-{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)
+            # ۱. پاکسازی محیط کاری قبلی
+            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", "clone", "--depth", "1", repo_url, work_dir], check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-            subprocess.run(["git", "config", "user.email", "bot@alpha.com"], cwd=work_dir)
-            subprocess.run(["git", "config", "user.name", "Bot"], cwd=work_dir)
+            # تنظیمات اولیه کاربر گیت برای کامیت
+            subprocess.run(["git", "config", "user.email", "bot@aialpha.local"], cwd=work_dir)
+            subprocess.run(["git", "config", "user.name", "AI Alpha Bot"], cwd=work_dir)
             
-            # خرد کردن استاندارد
-            target_folder = os.path.join(work_dir, tag_name)
-            split_file_with_7z(file_path, target_folder, filename)
+            # ۳. ساخت یک پوشه مخصوص این فایل در مخزن و خرد کردن فایل داخل آن
+            file_folder = os.path.join(work_dir, tag_name)
+            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} ({file_size_str})"], cwd=work_dir, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+            subprocess.run(["git", "commit", "-m", f"Add parts for {filename}"], cwd=work_dir, check=True)
+            
+            # ۵. ساخت تگ و پوش کردن آن به سرور گیت‌هاب
             subprocess.run(["git", "tag", tag_name], cwd=work_dir, check=True)
-            subprocess.run(["git", "push", "origin", "main", "--tags"], cwd=work_dir, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+            subprocess.run(["git", "push", "origin", "main"], cwd=work_dir, check=True)
+            subprocess.run(["git", "push", "origin", tag_name], cwd=work_dir, check=True)
             
-            link = f"https://github.com/{GITHUB_REPO}/archive/refs/tags/{tag_name}.zip"
+            # ۶. آماده‌سازی لینک نهایی اینترنت ملی
+            zip_download_link = f"https://github.com/{GITHUB_REPO}/archive/refs/tags/{tag_name}.zip"
             
-            final_text = (
-                f"🎉 انتقال با موفقیت انجام شد!\n\n"
-                f"📦 نام فایل: {filename}\n"
-                f"📊 حجم کل: {file_size_str}\n"
-                f"🔗 لینک دانلود (اینترنت ملی):\n{link}\n\n"
-                f"📱 راهنمای استخراج در گوشی:\n"
-                f"۱. فایل Zip را دانلود و استخراج کنید.\n"
-                f"۲. وارد پوشه شوید و روی پارت اول (`001.`) با برنامه RAR یا ZArchiver ضربه بزنید تا به صورت خودکار فایل کامل را به شما بدهد."
+            success_text = (
+                f"🎉 فایل با موفقیت در گیت‌هاب ذخیره شد!\n\n"
+                f"📁 نام: {filename}\n"
+                f"🔗 لینک دانلود مستقیم (با اینترنت ملی):\n{zip_download_link}\n\n"
+                f"💡 راهنما: فایل Zip بالا را دانلود کنید. داخل آن پارت‌هایی با پسوند 001 و 002 وجود دارد. روی پارت 001 کلیک راست کرده و با WinRAR گزینه Extract را بزنید تا فایل اصلی به صورت کامل استخراج شود."
             )
-            if status_msg_id: await client.edit_message_text(chat_id, status_msg_id, final_text)
+            
+            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:
-        err = str(e)[:150]
-        if status_msg_id: await client.edit_message_text(chat_id, status_msg_id, f"❌ خطای عملیات:\n{err}")
+        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:
-        if 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
 
 # ==============================================================================
-# 🟢 اجرا
+# 🟢 هندلر روبیکا
 # ==============================================================================
 @bot.on_update(filters.private)
 async def main_handler(client, update):
-    msg = getattr(update, "message", None) or getattr(update, "new_message", None)
-    if not msg: return
-    
-    text = str(msg.get("text", "")).strip()
-    chat_id = msg.get("author_object_guid")
-    msg_id = msg.get("message_id")
-    
-    if text.lower().startswith("http"):
-        asyncio.create_task(process_and_commit(client, chat_id, msg_id, text))
+    global BOT_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, "👋 سلام! لینک مستقیم هر فایلی رو بفرست تا برات خرد کنم و بفرستم روی گیت‌هاب (قابل دانلود با اینترنت ملی).")
+            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()
+    print("🚀 ربات کلاستر گیت‌هاب روشن شد...")
     bot.run()