Update app.py

app.py

@@ -1,66 +1,40 @@
 import os
 import uuid
-import threading
 import time
+import asyncio
 import fitz  # PyMuPDF
-from fastapi import FastAPI
+import logging
+from contextlib import asynccontextmanager
+
+from fastapi import FastAPI, Request, Response
 from fastapi.responses import FileResponse
 import uvicorn
-import logging
+
 from telegram import Update, ReplyKeyboardMarkup, KeyboardButton
 from telegram.ext import Application, CommandHandler, MessageHandler, filters, ContextTypes
 
-# --- فعال‌سازی سیستم گزارش‌گیری ---
-logging.basicConfig(
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
-    level=logging.INFO
-)
+# --- تنظیمات لاگ و متغیرها ---
+logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 TOKEN = os.environ.get("TOKEN")
 DOWNLOAD_DIR = "downloads"
 SPACE_URL = "https://barnamemaker-pdfmeger.hf.space"
-MERGED_FILE_TTL = 3600        # مدت نگهداری فایل‌های ادغام‌شده (ثانیه)
+MERGED_FILE_TTL = 3600  # ۱ ساعت
 
 os.makedirs(DOWNLOAD_DIR, exist_ok=True)
-
-# user_id -> list of file paths
 user_files: dict[int, list[str]] = {}
-# قفل برای جلوگیری از race condition
-user_locks: dict[int, threading.Lock] = {}
+user_locks: dict[int, asyncio.Lock] = {}
 
-app_web = FastAPI()
-
-def get_user_lock(user_id: int) -> threading.Lock:
+def get_user_lock(user_id: int) -> asyncio.Lock:
     if user_id not in user_locks:
-        user_locks[user_id] = threading.Lock()
+        user_locks[user_id] = asyncio.Lock()
     return user_locks[user_id]
 
-# ───────────────────────── Web server ─────────────────────────
-
-@app_web.get("/")
-async def root():
-    return {"message": "Bot is running!"}
-
-@app_web.get("/download/{filename}")
-async def get_file(filename: str):
-    # جلوگیری از path traversal
-    if ".." in filename or "/" in filename:
-        return {"error": "Invalid filename"}
-    file_path = os.path.join(DOWNLOAD_DIR, filename)
-    if os.path.exists(file_path):
-        return FileResponse(file_path, media_type="application/pdf")
-    return {"error": "File not found"}
-
-def run_web():
-    uvicorn.run(app_web, host="0.0.0.0", port=7860)
-
-# ───────────────────────── Cleanup ─────────────────────────
-
-def cleanup_files():
-    """هر ساعت فایل‌های قدیمی‌تر از TTL را پاک می‌کند."""
+# --- تسک پاکسازی در پس‌زمینه ---
+async def cleanup_task():
     while True:
-        time.sleep(600)  # هر ۱۰ دقیقه چک کن
+        await asyncio.sleep(600)
         now = time.time()
         for f in os.listdir(DOWNLOAD_DIR):
             path = os.path.join(DOWNLOAD_DIR, f)
@@ -71,8 +45,7 @@ def cleanup_files():
             except Exception as e:
                 logger.warning(f"Could not remove {f}: {e}")
 
-# ───────────────────────── UI helpers ─────────────────────────
-
+# --- منوی دکمه‌ها ---
 def get_menu():
     keyboard = [
         [KeyboardButton("📄 ادغام فایل‌ها")],
@@ -86,17 +59,14 @@ HELP_TEXT = (
     "۲. روی «📄 ادغام فایل‌ها» کلیک کنید.\n"
     "۳. لینک دانلود فایل ادغام‌شده را دریافت کنید.\n\n"
     "برای پاک کردن صف: «🗑 پاک کردن صف»\n"
-    "فایل‌ها به ترتیب ارسال ادغام می‌شوند."
 )
 
-# ───────────────────────── Handlers ─────────────────────────
-
+# --- منطق ربات (Handlers) ---
 async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
     user_id = update.effective_user.id
     logger.info(f"User {user_id} started the bot.")
     await update.message.reply_text(
-        "سلام! 👋\nفایل‌های PDF خود را بفرستید و سپس «📄 ادغام فایل‌ها» را بزنید.\n\n"
-        "برای راهنما: ❓ راهنما",
+        "سلام! 👋\nفایل‌های PDF خود را بفرستید و سپس «📄 ادغام فایل‌ها» را بزنید.",
         reply_markup=get_menu()
     )
 
@@ -104,13 +74,12 @@ async def handle_document(update: Update, context: ContextTypes.DEFAULT_TYPE):
     user_id = update.effective_user.id
     doc = update.message.document
 
-    # بررسی نوع فایل
     if doc.mime_type != "application/pdf":
         await update.message.reply_text("⚠️ فقط فایل PDF قبول می‌شود.", reply_markup=get_menu())
         return
 
     lock = get_user_lock(user_id)
-    with lock:
+    async with lock:
         if user_id not in user_files:
             user_files[user_id] = []
 
@@ -122,149 +91,124 @@ async def handle_document(update: Update, context: ContextTypes.DEFAULT_TYPE):
         try:
             await file.download_to_drive(file_path)
         except Exception as e:
-            logger.error(f"Download error for user {user_id}: {e}")
-            await update.message.reply_text("❌ خطا در دریافت فایل. دوباره امتحان کنید.", reply_markup=get_menu())
+            logger.error(f"Download error: {e}")
+            await update.message.reply_text("❌ خطا در دریافت فایل.", reply_markup=get_menu())
             return
 
-        # اعتبارسنجی PDF (جلوگیری از فایل‌های خراب)
         try:
             with fitz.open(file_path) as test_doc:
                 if test_doc.page_count == 0:
                     raise ValueError("Empty PDF")
         except Exception:
             os.remove(file_path)
-            await update.message.reply_text("⚠️ فایل PDF معتبر نیست یا خراب است.", reply_markup=get_menu())
+            await update.message.reply_text("⚠️ فایل PDF معتبر نیست.", reply_markup=get_menu())
             return
 
         user_files[user_id].append(file_path)
         count = len(user_files[user_id])
 
-    await update.message.reply_text(
-        f"✅ فایل {count} دریافت شد: {doc.file_name}",
-        reply_markup=get_menu()
-    )
+    await update.message.reply_text(f"✅ فایل {count} دریافت شد: {doc.file_name}", reply_markup=get_menu())
 
 async def merge_pdfs(update: Update, context: ContextTypes.DEFAULT_TYPE):
     user_id = update.effective_user.id
     lock = get_user_lock(user_id)
 
-    with lock:
+    async with lock:
         files = list(user_files.get(user_id, []))
 
     if not files:
-        await update.message.reply_text("⚠️ صف شما خالی است! ابتدا فایل‌های PDF بفرستید.", reply_markup=get_menu())
+        await update.message.reply_text("⚠️ صف شما خالی است!", reply_markup=get_menu())
         return
 
     if len(files) == 1:
         await update.message.reply_text("⚠️ حداقل ۲ فایل برای ادغام لازم است.", reply_markup=get_menu())
         return
 
-    await update.message.reply_text(f"⏳ در حال ادغام {len(files)} فایل...", reply_markup=get_menu())
-
+    msg = await update.message.reply_text(f"⏳ در حال ادغام {len(files)} فایل...")
     output_filename = f"merged_{user_id}_{uuid.uuid4().hex[:8]}.pdf"
     output_path = os.path.join(DOWNLOAD_DIR, output_filename)
 
     try:
         result = fitz.open()
         for pdf_path in files:
-            if not os.path.exists(pdf_path):
-                logger.warning(f"Missing file during merge: {pdf_path}")
-                continue
+            if not os.path.exists(pdf_path): continue
             with fitz.open(pdf_path) as doc:
                 result.insert_pdf(doc)
 
-        if result.page_count == 0:
-            result.close()
-            await update.message.reply_text("❌ هیچ صفحه‌ای برای ادغام یافت نشد.", reply_markup=get_menu())
-            return
-
         result.save(output_path)
         result.close()
 
-        # پاک کردن فایل‌های موقت کاربر
-        with lock:
+        async with lock:
             for f in user_files.get(user_id, []):
-                try:
-                    if os.path.exists(f):
-                        os.remove(f)
-                except Exception as e:
-                    logger.warning(f"Could not delete temp file {f}: {e}")
+                if os.path.exists(f): os.remove(f)
             user_files[user_id] = []
 
         link = f"{SPACE_URL}/download/{output_filename}"
-        size_kb = os.path.getsize(output_path) // 1024
-        await update.message.reply_text(
-            f"🎉 ادغام موفق!\n"
-            f"📄 تعداد فایل‌ها: {len(files)}\n"
-            f"📦 حجم خروجی: {size_kb} KB\n\n"
-            f"📥 لینک دانلود:\n{link}",
-            reply_markup=get_menu()
-        )
+        await msg.edit_text(f"🎉 ادغام موفق!\n📥 لینک دانلود:\n{link}")
 
     except Exception as e:
-        logger.error(f"Error merging PDFs for user {user_id}: {e}")
-        # پاک کردن فایل ناقص
-        if os.path.exists(output_path):
-            try:
-                os.remove(output_path)
-            except Exception:
-                pass
-        await update.message.reply_text(
-            "❌ خطایی در پردازش رخ داد. لطفاً دوباره امتحان کنید.",
-            reply_markup=get_menu()
-        )
+        logger.error(f"Error merging: {e}")
+        await msg.edit_text("❌ خطایی در پردازش رخ داد.")
 
 async def clear_queue(update: Update, context: ContextTypes.DEFAULT_TYPE):
     user_id = update.effective_user.id
     lock = get_user_lock(user_id)
-
-    with lock:
+    async with lock:
         files = list(user_files.get(user_id, []))
         user_files[user_id] = []
-
-    # پاک کردن فایل‌های دانلودشده از دیسک
-    removed = 0
+    
     for f in files:
-        try:
-            if os.path.exists(f):
-                os.remove(f)
-                removed += 1
-        except Exception as e:
-            logger.warning(f"Could not delete {f}: {e}")
-
-    await update.message.reply_text(
-        f"🗑 صف پاک شد. ({removed} فایل حذف شد)",
-        reply_markup=get_menu()
-    )
+        if os.path.exists(f): os.remove(f)
+    await update.message.reply_text("🗑 صف پاک شد.", reply_markup=get_menu())
 
 async def handle_text(update: Update, context: ContextTypes.DEFAULT_TYPE):
     text = update.message.text
-    if text == "📄 ادغام فایل‌ها":
-        await merge_pdfs(update, context)
-    elif text == "🗑 پاک کردن صف":
-        await clear_queue(update, context)
-    elif text == "❓ راهنما":
-        await update.message.reply_text(HELP_TEXT, reply_markup=get_menu())
-    else:
-        count = len(user_files.get(update.effective_user.id, []))
-        await update.message.reply_text(
-            f"لطفاً از منو استفاده کنید.\n📂 فایل‌های در صف: {count}",
-            reply_markup=get_menu()
-        )
-
-# ───────────────────────── Main ─────────────────────────
+    if text == "📄 ادغام فایل‌ها": await merge_pdfs(update, context)
+    elif text == "🗑 پاک کردن صف": await clear_queue(update, context)
+    elif text == "❓ راهنما": await update.message.reply_text(HELP_TEXT, reply_markup=get_menu())
+
+# --- تنظیمات اتصال اصلی (Webhook) ---
+ptb_app = Application.builder().token(TOKEN).build()
+ptb_app.add_handler(CommandHandler("start", start))
+ptb_app.add_handler(MessageHandler(filters.Document.PDF, handle_document))
+ptb_app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_text))
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # تنظیم Webhook در هنگام روشن شدن سرور
+    await ptb_app.bot.set_webhook(url=f"{SPACE_URL}/webhook", drop_pending_updates=True)
+    async with ptb_app:
+        await ptb_app.start()
+        task = asyncio.create_task(cleanup_task())
+        yield
+        # خاموش کردن امن
+        task.cancel()
+        await ptb_app.stop()
+
+app_web = FastAPI(lifespan=lifespan)
+
+# مسیری که تلگرام پیام‌ها را به آن می‌فرستد
+@app_web.post("/webhook")
+async def telegram_webhook(request: Request):
+    req_json = await request.json()
+    update = Update.de_json(req_json, ptb_app.bot)
+    await ptb_app.process_update(update)
+    return Response(status_code=200)
 
-if __name__ == "__main__":
-    if not TOKEN:
-        raise RuntimeError("متغیر محیطی TOKEN تنظیم نشده است!")
-
-    threading.Thread(target=run_web, daemon=True).start()
-    threading.Thread(target=cleanup_files, daemon=True).start()
+@app_web.get("/")
+async def root():
+    return {"message": "Bot Webhook is Active!"}
 
-    app = Application.builder().token(TOKEN).build()
-    app.add_handler(CommandHandler("start", start))
-    app.add_handler(MessageHandler(filters.Document.PDF, handle_document))
-    app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_text))
+@app_web.get("/download/{filename}")
+async def get_file(filename: str):
+    if ".." in filename or "/" in filename: return {"error": "Invalid"}
+    file_path = os.path.join(DOWNLOAD_DIR, filename)
+    if os.path.exists(file_path):
+        return FileResponse(file_path, media_type="application/pdf")
+    return {"error": "Not found"}
 
-    logger.info("Bot is polling...")
-    app.run_polling(drop_pending_updates=True)
+if __name__ == "__main__":
+    if not TOKEN:
+        raise RuntimeError("TOKEN is missing!")
+    # اجرای سرور واحد وب
+    uvicorn.run(app_web, host="0.0.0.0", port=7860)