Update app.py

app.py

@@ -1,412 +1,105 @@
-"""
-🚀 Manga Engine Pro - بوت سحب المانجا الإحترافي
-✅ نظام أرشيف ذكي بقناة تيليجرام
-✅ تخزين لمرة واحدة - مشاركة للأبد
-✅ محرك Playwright لتجاوز الحمايات
-"""
-
-import os, time, threading, zipfile, io, re, json, telebot, requests
-from flask import Flask
-from playwright.sync_api import sync_playwright
-from PIL import Image
+import os, time, threading, zipfile, io, re, json
+from flask import Flask, request
+import telebot
 from telebot import types
 
 # ══════════════════════════════════════════════
-#              ⚙️  الإعدادات
+# ⚙️ الإعدادات
 # ══════════════════════════════════════════════
-BOT_TOKEN        = os.getenv("TELEGRAM_BOT_TOKEN")       # توكن البوت
-ARCHIVE_CHANNEL  = os.getenv("ARCHIVE_CHANNEL_ID")       # معرّف قناة الأرشيف مثال: -1001234567890
-ADMIN_HANDLE     = os.getenv("ADMIN_HANDLE", "@svipfast") # حساب المطور
 
-bot = telebot.TeleBot(BOT_TOKEN, threaded=False)
+BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
+ADMIN_HANDLE = os.getenv("ADMIN_HANDLE", "@svipfast")
+
+bot = telebot.TeleBot(BOT_TOKEN)
 app = Flask(__name__)
 
 # ══════════════════════════════════════════════
-#          🗄️  نظام قاعدة بيانات الأرشيف
+# 🧠 حالة المستخدم
 # ══════════════════════════════════════════════
 
-# الفهرس الداخلي: { "base_url|chapter_N" : message_id_in_channel }
-archive_index: dict = {}
-INDEX_MSG_ID: int | None = None   # معرّف رسالة الفهرس المثبّتة
-
-def _index_key(base_url: str, chapter: int) -> str:
-    return f"{base_url.rstrip('/')}|{chapter}"
-
-# ── تحميل الفهرس عند البدء ──────────────────
-def load_archive_index():
-    """يحمّل فهرس الأرشيف من الرسالة المثبّتة في قناة الأرشيف."""
-    global archive_index, INDEX_MSG_ID
-    if not ARCHIVE_CHANNEL:
-        print("⚠️  ARCHIVE_CHANNEL_ID غير مضبوط، الأرشيف معطّل.")
-        return
-    try:
-        chat = bot.get_chat(ARCHIVE_CHANNEL)
-        if chat.pinned_message:
-            INDEX_MSG_ID = chat.pinned_message.message_id
-            # الفهرس مخزّن كنص JSON في الرسالة المثبّتة
-            archive_index = json.loads(chat.pinned_message.text)
-            print(f"✅ تم تحميل الفهرس: {len(archive_index)} فصل مؤرشف.")
-        else:
-            print("ℹ️  لا يوجد فهرس مثبّت بعد، سيُنشأ عند أول أرشفة.")
-    except Exception as e:
-        print(f"❌ خطأ في تحميل الفهرس: {e}")
-
-# ── حفظ الفهرس بعد كل إضافة ────────────────
-def save_archive_index():
-    """يحفظ الفهرس الحالي في رسالة مثبّتة بقناة الأرشيف."""
-    global INDEX_MSG_ID
-    if not ARCHIVE_CHANNEL:
-        return
-    try:
-        text = json.dumps(archive_index, ensure_ascii=False)
-        # حد تيليجرام للنص 4096 حرف - نقسّم إذا تجاوز الحد
-        if len(text) > 4000:
-            # احتياط: نُرسل كملف JSON بدلاً من نص
-            file_bytes = io.BytesIO(text.encode())
-            file_bytes.name = "index.json"
-            msg = bot.send_document(ARCHIVE_CHANNEL, file_bytes, caption="#MANGA_INDEX")
-            # نخزّن فقط message_id الأخير للملف في الرسالة المثبّتة
-            if INDEX_MSG_ID:
-                bot.unpin_chat_message(ARCHIVE_CHANNEL, INDEX_MSG_ID)
-            bot.pin_chat_message(ARCHIVE_CHANNEL, msg.message_id)
-            INDEX_MSG_ID = msg.message_id
-        else:
-            if INDEX_MSG_ID:
-                bot.edit_message_text(text, ARCHIVE_CHANNEL, INDEX_MSG_ID)
-            else:
-                msg = bot.send_message(ARCHIVE_CHANNEL, text)
-                bot.pin_chat_message(ARCHIVE_CHANNEL, msg.message_id)
-                INDEX_MSG_ID = msg.message_id
-    except Exception as e:
-        print(f"❌ خطأ في حفظ الفهرس: {e}")
-
-# ── البحث في الأرشيف ─────────────────────────
-def search_archive(base_url: str, chapter: int) -> int | None:
-    """يُعيد message_id الفصل إذا كان مؤرشفاً، وإلا None."""
-    return archive_index.get(_index_key(base_url, chapter))
-
-# ── إضافة فصل للأرشيف ───────────────────────
-def add_to_archive(base_url: str, chapter: int, zip_bytes: bytes, filename: str) -> int | None:
-    """يرفع الملف لقناة الأرشيف ويحدّث الفهرس."""
-    if not ARCHIVE_CHANNEL:
-        return None
-    try:
-        caption = (
-            f"📦 <b>{filename}</b>\n"
-            f"🔑 <code>{_index_key(base_url, chapter)}</code>\n"
-            f"📅 {time.strftime('%Y-%m-%d %H:%M')}"
-        )
-        file_io = io.BytesIO(zip_bytes)
-        file_io.name = filename
-        msg = bot.send_document(
-            ARCHIVE_CHANNEL, file_io,
-            caption=caption, parse_mode="HTML"
-        )
-        archive_index[_index_key(base_url, chapter)] = msg.message_id
-        save_archive_index()
-        print(f"✅ أُرشف الفصل {chapter} بـ message_id={msg.message_id}")
-        return msg.message_id
-    except Exception as e:
-        print(f"❌ خطأ في الأرشفة: {e}")
-        return None
-
-# ── إرسال من الأرشيف مباشرةً ────────────────
-def forward_from_archive(chat_id: int, message_id: int, chapter: int):
-    """يُعيد توجيه الملف من الأرشيف للمستخدم."""
-    bot.forward_message(chat_id, ARCHIVE_CHANNEL, message_id)
+user_states = {}
 
 # ══════════════════════════════════════════════
-#          🌐  محرك السحب (Playwright)
+# 🏠 الصفحة الرئيسية
 # ══════════════════════════════════════════════
 
-def fetch_images_engine(url: str) -> list:
-    """يسحب صور المانجا من الصفحة باستخدام Playwright."""
-    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 "
-                       "(KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
-        )
-        page = context.new_page()
-        try:
-            page.goto(url, wait_until="networkidle", timeout=75000)
-            # التمرير التدريجي لتحميل الصور الكسولة
-            for _ in range(6):
-                page.mouse.wheel(0, 1500)
-                time.sleep(0.9)
-            page.wait_for_timeout(2000)
-
-            images = page.query_selector_all("img")
-            img_list = []
-            seen_srcs = set()
-
-            for img in images:
-                src = (
-                    img.get_attribute("src")
-                    or img.get_attribute("data-src")
-                    or img.get_attribute("data-lazy-src")
-                    or img.get_attribute("data-original")
-                )
-                if not src or src in seen_srcs:
-                    continue
-                # تصفية الصور الصغيرة والشعارات
-                if not any(ext in src.lower() for ext in ['.jpg', '.jpeg', '.png', '.webp', '.avif']):
-                    continue
-                if any(kw in src.lower() for kw in ['logo', 'banner', 'avatar', 'icon', 'thumb']):
-                    continue
-                seen_srcs.add(src)
-                try:
-                    res = page.request.get(src, timeout=20000)
-                    if res.status == 200 and len(res.body()) > 20000:
-                        img_list.append(Image.open(io.BytesIO(res.body())).convert('RGB'))
-                except:
-                    continue
-
-            browser.close()
-            return img_list
-        except Exception as e:
-            print(f"❌ خطأ في الجلب: {e}")
-            browser.close()
-            return []
+@app.route('/')
+def home():
+    return "✅ Bot is running with Webhook"
 
 # ══════════════════════════════════════════════
-#       🏗️  بناء الروابط وتجميع ملف ZIP
+# 🔗 Webhook endpoint
 # ══════════════════════════════════════════════
 
-def build_chapter_url(sample_url: str, chapter: int) -> str:
-    """يُنشئ رابط الفصل بناءً على نمط رابط العيّنة."""
-    base = re.sub(r'/(?:chapter-)?[\d]+$', '', sample_url.strip().rstrip('/'))
-    if "azoramoon" in sample_url or "chapter-" in sample_url:
-        return f"{base}/chapter-{chapter}"
-    return f"{base}/{chapter}"
-
-def images_to_zip_bytes(imgs: list, chapter: int) -> bytes | None:
-    """يحوّل قائمة الصور إلى ملف ZIP يحتوي PDF."""
-    if not imgs:
-        return None
-    pdf_buf = io.BytesIO()
-    imgs[0].save(pdf_buf, format='PDF', save_all=True, append_images=imgs[1:])
-    zip_buf = io.BytesIO()
-    with zipfile.ZipFile(zip_buf, 'w', zipfile.ZIP_DEFLATED) as zf:
-        zf.writestr(f"Chapter_{chapter}.pdf", pdf_buf.getvalue())
-    return zip_buf.getvalue()
+@app.route(f"/{BOT_TOKEN}", methods=["POST"])
+def webhook():
+    json_str = request.stream.read().decode("utf-8")
+    update = telebot.types.Update.de_json(json_str)
+    bot.process_new_updates([update])
+    return "OK", 200
 
 # ══════════════════════════════════════════════
-#      🤖  منطق البوت - المعالجات والأوامر
+# 🚀 /start
 # ══════════════════════════════════════════════
 
-user_states: dict = {}
-
-@app.route('/')
-def home():
-    return "<h1>🚀 Manga Engine Pro is Running</h1><p>Archive-powered manga bot.</p>"
-
-# ── /start ───────────────────────────────────
 @bot.message_handler(commands=['start'])
-def welcome(message):
-    user_states[message.chat.id] = {}
-    markup = types.InlineKeyboardMarkup(row_width=1)
-    markup.add(
-        types.InlineKeyboardButton("📥 إرسال رابط جديد",          callback_data="new_order"),
-        types.InlineKeyboardButton("📊 إحصائيات الأرشيف",         callback_data="stats"),
-        types.InlineKeyboardButton("💡 كيف أحصل على الرابط؟",     callback_data="help_link"),
-        types.InlineKeyboardButton("👨‍💻 المطور", url=f"https://t.me/{ADMIN_HANDLE.replace('@','')}"),
-    )
+def start(message):
+    markup = types.InlineKeyboardMarkup()
+    markup.add(types.InlineKeyboardButton("📥 إرسال رابط", callback_data="send_link"))
+
     bot.send_message(
         message.chat.id,
-        f"<b>🚀 مرحباً في بوت سحب المانجا الإحترافي</b>\n\n"
-        f"✨ <b>الجديد:</b> نظام أرشيف ذكي!\n"
-        f"إذا طلبت فصلاً سبق سحبه من قبل، سيُرسَل لك فوراً ⚡\n\n"
-        f"المطور: {ADMIN_HANDLE}",
-        parse_mode="HTML", reply_markup=markup
+        f"🚀 أهلاً بك!\n\nأرسل رابط المانجا وسيتم سحبه.\n\n👨‍💻 {ADMIN_HANDLE}",
+        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 == "stats":
-        total = len(archive_index)
-        bot.send_message(
-            chat_id,
-            f"📊 <b>إحصائيات الأرشيف</b>\n\n"
-            f"📦 الفصول المؤرشفة: <b>{total}</b>\n"
-            f"⚡ جاهزة للإرسال الفوري",
-            parse_mode="HTML"
-        )
-
-    elif call.data == "help_link":
-        bot.send_message(
-            chat_id,
-            "<b>💡 طريقة إرسال الرابط:</b>\n\n"
-            "انسخ رابط أي فصل من الموقع مباشرة:\n"
-            "<code>https://olympustaff.com/series/manga-name/1</code>\n\n"
-            "البوت سيكتشف تلقائياً نمط الروابط ويسحب بقية الفصول 🎯",
-            parse_mode="HTML"
-        )
+# ══════════════════════════════════════════════
+# 🔘 الأزرار
+# ══════════════════════════════════════════════
 
-    elif call.data.startswith("mode_"):
-        mode = call.data.split("_")[1]
-        user_states[chat_id]['mode'] = mode
-        prompts = {
-            "single": "🔢 أرسل <b>رقم الفصل</b>:",
-            "range":  "🔢 أرسل <b>المدى</b> (مثال: <code>1-10</code>):",
-            "auto10": "🔢 أرسل <b>رقم الفصل الأول</b> وسنحمّل 10 فصول تلقائياً:",
-        }
-        msg = bot.send_message(chat_id, prompts.get(mode, "أرسل الرقم:"), parse_mode="HTML")
-        bot.register_next_step_handler(msg, final_execution_step)
+@bot.callback_query_handler(func=lambda call: True)
+def handle_buttons(call):
+    if call.data == "send_link":
+        msg = bot.send_message(call.message.chat.id, "🔗 أرسل الرابط الآن:")
+        bot.register_next_step_handler(msg, process_link)
 
-    bot.answer_callback_query(call.id)
+# ══════════════════════════════════════════════
+# 🔗 معالجة الرابط
+# ══════════════════════════════════════════════
 
-# ── الخطوة 1: استلام الرابط ───��─────────────
-def process_url_step(message):
+def process_link(message):
     url = message.text.strip()
-    if not url.startswith("http"):
-        bot.send_message(message.chat.id, "❌ الرابط غير صحيح.")
-        return
-    user_states[message.chat.id] = {'url': url}
-    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
-    )
 
-# ── الخطوة 2: التنفيذ مع الأرشيف الذكي ──────
-def final_execution_step(message):
-    chat_id = message.chat.id
-    state   = user_states.get(chat_id, {})
-
-    if 'url' not in state:
-        bot.send_message(chat_id, "⚠️ انتهت الجلسة، ابدأ من /start")
-        return
-
-    text   = message.text.strip()
-    mode   = state['mode']
-    url    = state['url']
-    base_url = re.sub(r'/(?:chapter-)?[\d]+$', '', url.strip().rstrip('/'))
-
-    try:
-        if mode == "single":
-            start = end = int(text)
-        elif mode == "range":
-            start, end = map(int, text.split('-'))
-        elif mode == "auto10":
-            start = int(text)
-            end   = start + 9
-        else:
-            raise ValueError("mode unknown")
-    except:
-        bot.send_message(chat_id, "⚠️ تنسيق خاطئ (مثال: 5 أو 1-10)")
+    if not url.startswith("http"):
+        bot.send_message(message.chat.id, "❌ الرابط غير صالح")
         return
 
-    count = end - start + 1
-    status_msg = bot.send_message(
-        chat_id,
-        f"🔍 <b>يبحث في الأرشيف...</b>\n📋 الفصول: {start} ← {end} ({count} فصل)",
-        parse_mode="HTML"
-    )
-
-    # ── البحث في الأرشيف أولاً ──
-    cached, missing = [], []
-    for ch in range(start, end + 1):
-        mid = search_archive(base_url, ch)
-        if mid:
-            cached.append((ch, mid))
-        else:
-            missing.append(ch)
-
-    # إرسال الفصول المؤرشفة فوراً
-    if cached:
-        bot.edit_message_text(
-            f"⚡ <b>وُجد في الأرشيف {len(cached)} فصل!</b>\nيُرسَل فوراً...",
-            chat_id, status_msg.message_id, parse_mode="HTML"
-        )
-        for ch, mid in cached:
-            try:
-                forward_from_archive(chat_id, mid, ch)
-                time.sleep(0.4)  # تفادي flood
-            except Exception as e:
-                print(f"Forward error ch{ch}: {e}")
-
-    # سحب الفصول الناقصة
-    if missing:
-        bot.send_message(
-            chat_id,
-            f"🔄 <b>يسحب {len(missing)} فصل جديد...</b>\n"
-            f"الفصول: {', '.join(map(str, missing))}",
-            parse_mode="HTML"
-        )
-        for ch in missing:
-            ch_url = build_chapter_url(url, ch)
-            bot.send_chat_action(chat_id, 'upload_document')
-            imgs = fetch_images_engine(ch_url)
-            if not imgs:
-                bot.send_message(chat_id, f"❌ الفصل {ch} لم يُسحب (تحقق من الرابط)")
-                continue
+    bot.send_message(message.chat.id, f"✅ تم استلام الرابط:\n{url}")
 
-            zip_bytes = images_to_zip_bytes(imgs, ch)
-            if not zip_bytes:
-                continue
+    # (هنا تضيف محرك السحب الخاص بك)
+    bot.send_message(message.chat.id, "⚙️ جاري المعالجة...")
 
-            filename = f"Chapter_{ch}.zip"
+    time.sleep(2)
 
-            # ① أرشفة أولاً
-            archived_mid = add_to_archive(base_url, ch, zip_bytes, filename)
-
-            # ② الإرسال للمستخدم من الأرشيف أو مباشرة
-            if archived_mid:
-                forward_from_archive(chat_id, archived_mid, ch)
-            else:
-                # fallback: إرسال مباشر
-                f_io = io.BytesIO(zip_bytes)
-                f_io.name = filename
-                bot.send_document(
-                    chat_id, f_io,
-                    caption=f"📦 الفصل {ch}\n👨‍💻 {ADMIN_HANDLE}",
-                    parse_mode="HTML"
-                )
-            time.sleep(0.5)
-
-    # رسالة الانتهاء
-    summary = (
-        f"✅ <b>اكتمل التحميل!</b>\n\n"
-        f"⚡ من الأرشيف: {len(cached)}\n"
-        f"🌐 مسحوب جديد: {len(missing)}\n"
-        f"👨‍💻 {ADMIN_HANDLE}"
-    )
-    bot.send_message(chat_id, summary, parse_mode="HTML")
-    try:
-        bot.delete_message(chat_id, status_msg.message_id)
-    except:
-        pass
+    bot.send_message(message.chat.id, "✅ تم (تجربة فقط)")
 
 # ══════════════════════════════════════════════
-#                🚀  التشغيل
+# 🚀 التشغيل
 # ══════════════════════════════════════════════
 
 if __name__ == "__main__":
-    print("🔧 تثبيت Chromium...")
-    os.system("playwright install chromium")
+    print("🚀 Starting bot...")
 
-    print("📂 تحميل فهرس الأرشيف...")
-    load_archive_index()
+    # رابط Space (غيره!)
+    SPACE_URL = os.getenv("SPACE_URL")
 
-    print("🤖 تشغيل البوت...")
-    threading.Thread(
-        target=lambda: bot.infinity_polling(timeout=30, long_polling_timeout=25),
-        daemon=True
-    ).start()
+    if SPACE_URL:
+        webhook_url = f"{SPACE_URL}/{BOT_TOKEN}"
+        bot.remove_webhook()
+        time.sleep(1)
+        bot.set_webhook(url=webhook_url)
+        print(f"✅ Webhook set: {webhook_url}")
+    else:
+        print("⚠️ ضع SPACE_URL في Secrets")
 
-    print("🌐 تشغيل Flask...")
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860)