Update app.py

app.py

@@ -1,187 +1,105 @@
 import os, time, threading, zipfile, io, re, telebot, requests
-from flask import Flask, render_template_string
+from flask import Flask
 from playwright.sync_api import sync_playwright
 from PIL import Image
-import img2pdf
 from telebot import types
 
 # --- الإعدادات ---
 BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
+# معرف القناة التي سيتم تخزين الملفات فيها (الأرشيف)
+ARCHIVE_CH = os.getenv("ARCHIVE_CHANNEL_ID") 
 ADMIN_HANDLE = "@svipfast"
+
 bot = telebot.TeleBot(BOT_TOKEN, threaded=False)
 app = Flask(__name__)
 
-# مخزن مؤقت لحالات المستخدمين
-user_states = {}
+# قاموس لتخزين "معرفات الملفات" (File IDs) لتجنب إعادة الرفع
+# في الإنتاج الحقيقي، يفضل استخدام قاعدة بيانات بسيطة مثل SQLite
+archive_db = {} 
 
 @app.route('/')
-def home(): return "<h1>Manga Engine Professional is Running</h1>"
+def home(): return "<h1>Archive Engine Pro is Running</h1>"
 
-# --- محرك السحب (Playwright) ---
-def fetch_images_engine(url):
+# --- وظيفة السحب (نفس المحرك السابق لكن محسن) ---
+def fetch_and_pack(url, start, end):
     with sync_playwright() as p:
         browser = p.chromium.launch(headless=True)
-        context = browser.new_context(user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36")
-        page = context.new_page()
-        try:
-            page.goto(url, wait_until="networkidle", timeout=60000)
-            # التمرير لضمان تحميل الصور بالكامل
-            for _ in range(5):
-                page.mouse.wheel(0, 1200)
-                time.sleep(0.8)
-            
-            images = page.query_selector_all("img")
-            img_list = []
-            for img in images:
-                src = img.get_attribute("src") or img.get_attribute("data-src") or img.get_attribute("data-lazy-src")
-                if src and any(ext in src.lower() for ext in ['.jpg', '.jpeg', '.png', '.webp', '.avif']):
-                    if "logo" in src.lower() or "banner" in src.lower(): continue
-                    try:
-                        res = page.request.get(src)
-                        if res.status == 200 and len(res.body()) > 25000:
-                            img_list.append(Image.open(io.BytesIO(res.body())).convert('RGB'))
-                    except: continue
-            browser.close()
-            return img_list
-        except:
-            browser.close()
-            return []
-
-# --- لوحة التحكم والترحيب ---
-@bot.message_handler(commands=['start'])
-def welcome(message):
-    user_states[message.chat.id] = {} # تصفير حالة المستخدم
-    markup = types.InlineKeyboardMarkup(row_width=1)
-    btn_start = types.InlineKeyboardButton("📥 إرسال رابط جديد", callback_data="new_order")
-    btn_help = types.InlineKeyboardButton("💡 كيف أحصل على الرابط؟", callback_data="help_link")
-    btn_admin = types.InlineKeyboardButton("👨‍💻 المطور", url=f"https://t.me/{ADMIN_HANDLE.replace('@','')}")
-    markup.add(btn_start, btn_help, btn_admin)
-    
-    bot.send_message(message.chat.id, 
-        f"<b>مرحباً بك في بوت سحب المانجا الإحترافي 🚀</b>\n\n"
-        f"النظام يعمل بمحرك محاكاة لتجاوز الحمايات.\n"
-        f"المطور المسؤول: {ADMIN_HANDLE}", 
-        parse_mode="HTML", reply_markup=markup)
-
-# --- معالجة الضغط على الأزرار ---
-@bot.callback_query_handler(func=lambda call: True)
-def handle_query(call):
-    chat_id = call.message.chat.id
-    
-    if call.data == "new_order":
-        msg = bot.send_message(chat_id, "<b>الآن، أرسل رابط المانجا (رابط أي فصل):</b>", parse_mode="HTML")
-        bot.register_next_step_handler(msg, process_url_step)
-    
-    elif call.data == "help_link":
-        help_txt = (
-            "<b>طريقة إرسال الرابط الصحيحة:</b>\n\n"
-            "انسخ رابط الفصل من الموقع مباشرة، مثال:\n"
-            "<code>https://olympustaff.com/series/manga-name/1</code>"
-        )
-        bot.send_message(chat_id, help_txt, parse_mode="HTML")
+        page = browser.new_page()
+        downloaded_files = []
+        
+        base_part = re.sub(r'/(?:chapter-)?\d+$', '', url.strip().rstrip('/'))
+        is_azora = "azoramoon" in url
 
-    # أزرار خيارات التحميل
-    elif call.data.startswith("mode_"):
-        mode = call.data.split("_")[1]
-        user_states[chat_id]['mode'] = mode
+        for i in range(start, end + 1):
+            target_url = f"{base_part}/chapter-{i}" if is_azora else f"{base_part}/{i}"
+            try:
+                page.goto(target_url, wait_until="networkidle", timeout=60000)
+                # تمرير ذكي
+                page.evaluate("window.scrollTo(0, document.body.scrollHeight/2)")
+                time.sleep(1)
+                page.evaluate("window.scrollTo(0, document.body.scrollHeight)")
+                
+                images = page.query_selector_all("img")
+                img_data = []
+                for img in images:
+                    src = img.get_attribute("src") or img.get_attribute("data-src")
+                    if src and any(x in src.lower() for x in ['.jpg', '.jpeg', '.png', '.webp']):
+                        if "logo" in src.lower() or "banner" in src.lower(): continue
+                        res = requests.get(src, timeout=10)
+                        if res.status_code == 200 and len(res.content) > 25000:
+                            img_data.append(res.content)
+                
+                if img_data:
+                    # ضغط الصور في ZIP للفصل الواحد
+                    ch_zip = f"CH_{i}.zip"
+                    with zipfile.ZipFile(ch_zip, 'w') as z:
+                        for idx, data in enumerate(img_data):
+                            z.writestr(f"{idx:03d}.jpg", data)
+                    downloaded_files.append((i, ch_zip))
+            except: continue
         
-        if mode == "single":
-            msg = bot.send_message(chat_id, "🔢 أرسل <b>رقم الفصل</b> فقط:")
-        elif mode == "range":
-            msg = bot.send_message(chat_id, "🔢 أرسل <b>المدى</b> (مثال: 1-10):")
-        elif mode == "auto10":
-            msg = bot.send_message(chat_id, "🔢 أرسل <b>رقم الفصل</b> الذي سنبدأ التحميل منه:")
-            
-        bot.register_next_step_handler(msg, final_execution_step)
-    
-    bot.answer_callback_query(call.id)
+        browser.close()
+        return downloaded_files
 
-# --- الخطوة الثانية: استلام الرابط ---
-def process_url_step(message):
-    url = message.text.strip()
-    if not url.startswith("http"):
-        bot.send_message(message.chat.id, "❌ الرابط غير صحيح، أرسل رابط يبدأ بـ http")
-        return
-    
-    user_states[message.chat.id] = {'url': url}
+# --- معالجة الطلب بنظام الأرشيف ---
+def smart_process(chat_id, url, start, end, status_msg):
+    # توليد مفتاح فريد للمانجا بناءً على الرابط
+    manga_id = re.sub(r'\W+', '', url.split('/')[-2]) 
     
-    # إظهار خيارات التحميل كأزرار
-    markup = types.InlineKeyboardMarkup(row_width=2)
-    markup.add(
-        types.InlineKeyboardButton("🎯 فصل واحد", callback_data="mode_single"),
-        types.InlineKeyboardButton("📦 حزمة فصول", callback_data="mode_range"),
-        types.InlineKeyboardButton("⏩ تحميل 10 فصول", callback_data="mode_auto10")
-    )
-    bot.send_message(message.chat.id, "<b>ممتاز! اختر الآن طريقة التحميل:</b>", parse_mode="HTML", reply_markup=markup)
+    needed_chapters = []
+    found_files = []
 
-# --- الخطوة الأخيرة: التنفيذ ---
-def final_execution_step(message):
-    chat_id = message.chat.id
-    state = user_states.get(chat_id)
-    if not state or 'url' not in state:
-        bot.send_message(chat_id, "⚠️ حدث خطأ، يرجى البدء من جديد عبر /start")
-        return
-
-    text = message.text.strip()
-    mode = state['mode']
-    url = state['url']
-
-    try:
-        if mode == "single":
-            start, end = int(text), int(text)
-        elif mode == "range":
-            start, end = map(int, text.split('-'))
-        elif mode == "auto10":
-            start = int(text)
-            end = start + 9
-        
-        status_msg = bot.send_message(chat_id, f"<b>🔄 جاري تشغيل المحرك الذكي...</b>\n📦 سحب الفصول: {start} ⬅️ {end}", parse_mode="HTML")
-        
-        # تشغيل السحب
-        zip_path = run_manga_engine(url, start, end)
-        
-        if zip_path:
-            with open(zip_path, 'rb') as f:
-                caption = (
-                    f"<b>✅ اكتملت عملية السحب بنجاح</b>\n\n"
-                    f"📂 <b>النطاق:</b> {start} إلى {end}\n"
-                    f"👨‍💻 <b>بواسطة:</b> {ADMIN_HANDLE}"
-                )
-                bot.send_document(chat_id, f, caption=caption, parse_mode="HTML")
-            os.remove(zip_path)
-            bot.delete_message(chat_id, status_msg.message_id)
+    # 1. البحث في الأرشيف أولاً
+    for i in range(start, end + 1):
+        key = f"{manga_id}_{i}"
+        if key in archive_db:
+            found_files.append(archive_db[key])
         else:
-            bot.edit_message_text(f"❌ لم ينجح المحرك في سحب الصور.\nتواصل مع الإدارة: {ADMIN_HANDLE}", chat_id, status_msg.message_id)
-            
-    except Exception as e:
-        bot.send_message(chat_id, "⚠️ خطأ في إدخال الأرقام. يرجى التأكد من كتابتها بشكل صحيح (مثال: 1-5).")
+            needed_chapters.append(i)
 
-# --- منطق بناء الروابط وضغط الملفات ---
-def run_manga_engine(sample_url, start, end):
-    base_part = re.sub(r'/(?:chapter-)?\d+$', '', sample_url.strip().rstrip('/'))
-    is_azora = "azoramoon" in sample_url
-    pdf_files = []
+    # 2. إذا كانت هناك فصول ناقصة، نسحبها
+    if needed_chapters:
+        bot.edit_message_text(f"🔍 الفصل/الفصول {needed_chapters} غير موجودة في الأرشيف.. جاري السحب الآن.", chat_id, status_msg.message_id)
+        new_files = fetch_and_pack(url, min(needed_chapters), max(needed_chapters))
+        
+        for ch_num, file_path in new_files:
+            # رفع الملف للقناة الخاصة (الأرشيف) للحفظ الدائم
+            with open(file_path, 'rb') as f:
+                archived_msg = bot.send_document(ARCHIVE_CH, f, caption=f"Archive: {manga_id} | Ch: {ch_num}")
+                # حفظ الـ File ID لاستخدامه لاحقاً دون إعادة رفع
+                file_id = archived_msg.document.file_id
+                archive_db[f"{manga_id}_{ch_num}"] = file_id
+                found_files.append(file_id)
+            os.remove(file_path)
 
-    for i in range(start, end + 1):
-        target_url = f"{base_part}/chapter-{i}" if is_azora else f"{base_part}/{i}"
-        imgs = fetch_images_engine(target_url)
-        if imgs:
-            fname = f"Chapter_{i}.pdf"
-            imgs[0].save(fname, save_all=True, append_images=imgs[1:], format='PDF')
-            pdf_files.append(fname)
-            
-    if not pdf_files: return None
-    
-    zip_name = f"Manga_Pack_{int(time.time())}.zip"
-    with zipfile.ZipFile(zip_name, 'w') as zipf:
-        for f in pdf_files:
-            zipf.write(f)
-            os.remove(f)
-    return zip_name
+    # 3. إرسال جميع الملفات للمستخدم (بسرعة البرق من سيرفرات تليجرام)
+    if found_files:
+        bot.send_message(chat_id, f"✅ تم تجهيز الحزمة لك من الأرشيف الذكي:")
+        for fid in found_files:
+            bot.send_document(chat_id, fid)
+        bot.delete_message(chat_id, status_msg.message_id)
+    else:
+        bot.edit_message_text("❌ فشل السحب بالكامل.", chat_id, status_msg.message_id)
 
-# --- التشغيل ---
-if __name__ == "__main__":
-    # تنصيب متصفح Playwright
-    os.system("playwright install chromium")
-    threading.Thread(target=lambda: bot.infinity_polling(), daemon=True).start()
-    app.run(host="0.0.0.0", port=7860)
+# --- استكمال الأزرار والـ Start كما في النسخ السابقة ---
+# ملاحظة: في execute() استدعي smart_process() بدلاً من المحرك القديم.