Update main.py

main.py

@@ -5,70 +5,39 @@ import threading
 import uuid
 import urllib.parse
 import traceback
-import math
-import subprocess
-import glob
-import json
 from flask import Flask
+from rubpy.bot import BotClient, filters
 
 # ==============================================================================
-# 🟢 تنظیمات و ایمپورت‌ها (مشابه نسخه آلفا پرو شما)
+# 🟢 تنظیمات و متغیرهای محیطی
 # ==============================================================================
-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)
 
 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:
-    print("❌ توکن ربات وارد نشده است!")
+    print("❌ توکن ربات روبیکا وارد نشده است!")
+    exit()
+if not GITHUB_TOKEN:
+    print("❌ توکن گیت‌هاب وارد نشده است! حتماً در بخش Secrets اضافه کنید.")
     exit()
 
 bot = BotClient(bot_token)
 BOT_GUID = None
 
 # ==============================================================================
-# 🟢 توابع کمکی FFmpeg برای تشخیص و برش ویدیو
-# ==============================================================================
-def get_video_info(file_path):
-    try:
-        cmd = ['ffprobe', '-v', 'quiet', '-print_format', 'json', '-show_format', '-show_streams', file_path]
-        result = subprocess.check_output(cmd).decode('utf-8')
-        data = json.loads(result)
-        duration = float(data.get('format', {}).get('duration', 0.0))
-        has_video = any(stream.get('codec_type') == 'video' for stream in data.get('streams', []))
-        return duration, has_video
-    except Exception:
-        return 0.0, False
-
-def split_video(file_path, segment_time):
-    output_pattern = f"{file_path}_part%03d.mp4" 
-    cmd = [
-        'ffmpeg', '-i', file_path,
-        '-c', 'copy', 
-        '-map', '0',
-        '-segment_time', str(segment_time),
-        '-f', 'segment',
-        '-reset_timestamps', '1',
-        output_pattern
-    ]
-    try:
-        subprocess.run(cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-        return sorted(glob.glob(f"{file_path}_part*.mp4"))
-    except Exception:
-        return []
-
-# ==============================================================================
-# 🟢 تابع دانلود (با قابلیت تبدیل خودکار لینک دراپ‌باکس به لینک مستقیم)
+# 🟢 دانلود فایل از لینک به سرور واسط
 # ==============================================================================
 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")
@@ -87,126 +56,124 @@ async def download_large_file(url, chat_id, message_id, client):
     status_msg_id = None
     
     try:
-        status_msg = await client.send_message(chat_id, "⏳ در حال دانلود فایل از اینترنت آزاد (Dropbox/Global) به سرور...\n(این مرحله بسته به حجم ممکن است طول بکشد)", reply_to_message_id=message_id)
+        status_msg = await client.send_message(
+            chat_id, 
+            "⏳ در حال دانلود فایل از اینترنت آزاد به سرور...\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')
     except Exception: pass
 
     try:
-        headers = {"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"}
-        
+        headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"}
         async with aiohttp.ClientSession(headers=headers) as session:
             async with session.get(url, timeout=0) as response:
                 if response.status != 200:
                     if status_msg_id: 
                         try: await client.edit_message_text(chat_id, status_msg_id, f"❌ خطا در دسترسی به لینک! کد وضعیت: {response.status}")
                         except: pass
-                    return None, status_msg_id
+                    return None, None, status_msg_id
                 
                 with open(file_path, 'wb') as f:
                     async for chunk in response.content.iter_chunked(2 * 1024 * 1024):
                         if chunk: f.write(chunk)
                             
         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
             
-        return file_path, status_msg_id
+        return file_path, filename, status_msg_id
     except Exception as e:
         if status_msg_id:
             try: await client.edit_message_text(chat_id, status_msg_id, f"❌ خطای شبکه هنگام دانلود:\n{str(e)[:100]}")
             except: pass
-        return None, status_msg_id
+        return None, None, status_msg_id
 
 # ==============================================================================
-# 🟢 تابع آپلود و پارت‌بندی (هوشمند)
+# 🟢 آپلود مستقیم فایل به عنوان GitHub Release
 # ==============================================================================
-async def process_url_and_upload(client, chat_id, message_id, url):
-    file_path, status_msg_id = await download_large_file(url, chat_id, message_id, client)
+async def upload_to_github_release(file_path, filename, chat_id, status_msg_id, client):
+    tag_name = f"upload-{uuid.uuid4().hex[:8]}"
+    release_name = f"فایل انتقالی: {filename}"
     
-    if file_path and os.path.exists(file_path):
-        try:
-            file_size_bytes = os.path.getsize(file_path)
-            file_size_mb = file_size_bytes / (1024 * 1024)
-            caption_text = f"📥 فایل شما آماده شد!\n🔗 لینک منبع: {url}"
+    headers = {
+        "Authorization": f"token {GITHUB_TOKEN}",
+        "Accept": "application/vnd.github.v3+json"
+    }
+    
+    try:
+        async with aiohttp.ClientSession(headers=headers) as session:
+            # ۱. ساخت یک Release جدید در گیت‌هاب
+            create_release_url = f"https://api.github.com/repos/{GITHUB_REPO}/releases"
+            release_data = {
+                "tag_name": tag_name,
+                "name": release_name,
+                "body": "آپلود شده توسط ربات روبیکا."
+            }
+            
+            async with session.post(create_release_url, json=release_data) as resp:
+                if resp.status != 201:
+                    raise Exception(f"خطا در ساخت Release: {await resp.text()}")
+                release_info = await resp.json()
+                release_id = release_info.get("id")
+                html_url = release_info.get("html_url") # لینک صفحه ریلیز
+
+            # ۲. آپلود فایل در Release (به عنوان Asset)
+            if status_msg_id:
+                try: await client.edit_message_text(chat_id, status_msg_id, "📦 Release ساخته شد. در حال انتقال فایل به گیت‌هاب (تا 2 گیگابایت)...")
+                except: pass
 
-            if file_size_mb > 49:
-                duration, has_video = get_video_info(file_path)
-                
-                # حالت اول: ویدیو است -> برش زمانی
-                if has_video and duration > 0:
-                    await client.send_message(chat_id, f"🎬 ویدیوی حجیم ({file_size_mb:.1f} مگابایت) شناسایی شد.\n✂️ در حال برش به پارت‌های MP4 قابل پخش...")
-                    parts_needed = math.ceil(file_size_mb / 40.0) 
-                    segment_time = math.ceil(duration / parts_needed)
-                    segments = split_video(file_path, segment_time)
+            upload_url = f"https://uploads.github.com/repos/{GITHUB_REPO}/releases/{release_id}/assets?name={urllib.parse.quote(filename)}"
+            
+            # باز کردن فایل برای استریم
+            with open(file_path, 'rb') as f:
+                upload_headers = headers.copy()
+                upload_headers["Content-Type"] = "application/octet-stream"
+                async with session.post(upload_url, data=f, headers=upload_headers) as upload_resp:
+                    if upload_resp.status != 201:
+                        raise Exception(f"خطا در آپلود فایل: {await upload_resp.text()}")
+                    asset_info = await upload_resp.json()
                     
-                    if segments:
-                        for i, seg_path in enumerate(segments):
-                            part_caption = f"{caption_text}\n\n🎬 قسمت {i+1} از {len(segments)}"
-                            upload_success = False
-                            for attempt in range(5):
-                                try:
-                                    try: await client.send_file(chat_id, file=seg_path, text=part_caption)
-                                    except: await client.send_file(chat_id, seg_path, text=part_caption)
-                                    upload_success = True
-                                    break
-                                except Exception: await asyncio.sleep(4)
-                            try: os.remove(seg_path)
-                            except: pass
-                    else: raise Exception("خطا در برش ویدیو")
-
-                # حالت دوم: غیر ویدیویی است -> پارت‌بندی دیتایی
-                else:
-                    await client.send_message(chat_id, f"⚠️ فایل حجیم ({file_size_mb:.1f} مگابایت) تشخیص داده شد.\n📦 در حال تقسیم به پارت‌های ۴۹ مگابایتی...")
-                    chunk_size = 49 * 1024 * 1024 
-                    total_parts = math.ceil(file_size_bytes / chunk_size)
-                    with open(file_path, 'rb') as f:
-                        for i in range(1, total_parts + 1):
-                            chunk = f.read(chunk_size)
-                            if not chunk: break
-                            part_path = f"{file_path}.part{i}"
-                            with open(part_path, 'wb') as chunk_file: chunk_file.write(chunk)
-                            part_caption = f"{caption_text}\n\n📦 پارت {i} از {total_parts}"
-                            upload_success = False
-                            for attempt in range(5):
-                                try:
-                                    try: await client.send_file(chat_id, file=part_path, text=part_caption)
-                                    except: await client.send_file(chat_id, part_path, text=part_caption)
-                                    upload_success = True
-                                    break
-                                except Exception: await asyncio.sleep(4)
-                            try: os.remove(part_path)
-                            except: pass
+            # ارسال موفقیت آمیز لینک
+            success_text = (
+                f"🎉 آپلود با موفقیت انجام شد!\n\n"
+                f"📁 نام فایل: {filename}\n"
+                f"🔗 لینک صفحه فایل:\n{html_url}\n\n"
+                f"⚠️ نکته: برای دانلود، کافیست در مرورگر خود وارد اکانت گیت‌هاب شوید و در صفحه بالا، روی نام فایل در بخش Assets کلیک کنید."
+            )
             
-            # حالت عادی: زیر 50 مگابایت
+            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:
-                upload_success = False
-                for attempt in range(5):
-                    try:
-                        try: await client.send_file(chat_id, file=file_path, text=caption_text)
-                        except: await client.send_file(chat_id, file_path, text=caption_text)
-                        upload_success = True
-                        break
-                    except Exception: await asyncio.sleep(4)
-                if not upload_success: raise Exception("روبیکا فایل را نپذیرفت")
+                await client.send_message(chat_id, success_text)
 
-            if status_msg_id:
-                try: await client.delete_messages(chat_id, [status_msg_id])
-                except: pass
-                
-        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)
-            else: await client.send_message(chat_id, error_txt)
+    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)
+        else:
+            await client.send_message(chat_id, error_txt)
+
+# ==============================================================================
+# 🟢 پردازش نهایی لینک
+# ==============================================================================
+async def process_url_and_upload(client, chat_id, message_id, url):
+    file_path, filename, status_msg_id = await download_large_file(url, chat_id, message_id, client)
+    
+    if file_path and os.path.exists(file_path):
+        try:
+            await upload_to_github_release(file_path, filename, chat_id, status_msg_id, client)
         finally:
+            # پاک کردن فایل از روی سرور واسط پس از ا��مام کار
             try: os.remove(file_path)
             except: pass
 
 # ==============================================================================
-# 🟢 هندلر اصلی
+# 🟢 هندلر اصلی روبیکا
 # ==============================================================================
 @bot.on_update(filters.private)
 async def main_handler(client, update):
@@ -231,7 +198,7 @@ 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, "👋 سلام! لینک مستقیم هر فایلی (حتی Dropbox) رو بفرست تا برات تو روبیکا آپلود کنم.")
+            await client.send_message(chat_id, "👋 سلام! لینک مستقیم هر فایلی رو بفرست تا برات تو مخزن شخصی گیت‌هاب آپلود کنم (تا ۲ گیگابایت).")
             return
 
         if user_text_str.lower().startswith("http"):
@@ -243,5 +210,5 @@ async def main_handler(client, update):
 
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
-    print("🚀 ربات با قابلیت دراپ‌باکس روشن شد...")
+    print("🚀 ربات انتقال به گیت‌هاب روشن شد...")
     bot.run()