Update app.py

app.py

@@ -1,571 +1,29 @@
+from fastapi import FastAPI, Request
 import os
-import asyncio
-import logging
 from datetime import datetime
-from pathlib import Path
-from telethon import TelegramClient, events
-from telethon.sessions import StringSession
-from telethon.network import ConnectionTcpFull
-import nest_asyncio
-import aiohttp
 
-# فعال کردن nest_asyncio
-nest_asyncio.apply()
+app = FastAPI()
 
-# تنظیمات لاگ
-logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
-)
-logger = logging.getLogger(__name__)
+SAVE_DIR = "files"
+os.makedirs(SAVE_DIR, exist_ok=True)
 
-# تنظیمات از متغیرهای محیطی
-API_ID = os.environ.get("API_ID")
-API_HASH = os.environ.get("API_HASH")
-BOT_TOKEN = os.environ.get("BOT_TOKEN")
 
-# تنظیمات Cloudflare Proxy
-CF_PROXY_ENABLED = os.environ.get("CF_PROXY_ENABLED", "true").lower() == "true"
-CF_WORKER_URL = os.environ.get("CF_WORKER_URL", "https://danitell.sararoid85.workers.dev")
+@app.get("/")
+def home():
+    return {"status": "running"}
 
-# مسیر ذخیره‌سازی در فضای HuggingFace
-BASE_DIR = Path("/tmp/telegram_bot_files")
-BASE_DIR.mkdir(parents=True, exist_ok=True)
 
-# دیکشنری برای ذخیره اطلاعات کاربران
-user_data = {}
+@app.post("/upload")
+async def upload(request: Request):
+    data = await request.body()
 
-def check_config():
-    """بررسی تنظیمات لازم"""
-    missing_configs = []
-    
-    if not API_ID:
-        missing_configs.append("API_ID")
-    if not API_HASH:
-        missing_configs.append("API_HASH")
-    if not BOT_TOKEN:
-        missing_configs.append("BOT_TOKEN")
-    
-    if missing_configs:
-        logger.error(f"❌ تنظیمات زیر تنظیم نشده‌اند: {', '.join(missing_configs)}")
-        return False
-    
-    logger.info("✅ تنظیمات اولیه بررسی شد.")
-    return True
+    filename = request.headers.get("X-Filename")
+    if not filename:
+        filename = f"file_{datetime.now().timestamp()}"
 
-class CustomTelegramClient(TelegramClient):
-    """کلاینت تلگرام با پشتیبانی از Cloudflare Proxy"""
-    
-    async def _connect(self):
-        """اتصال با استفاده از Cloudflare Proxy"""
-        if CF_PROXY_ENABLED and CF_WORKER_URL:
-            logger.info(f"🌐 استفاده از Cloudflare Proxy: {CF_WORKER_URL}")
-            
-            # ایجاد session با proxy
-            connector = aiohttp.TCPConnector(ssl=False)
-            session = aiohttp.ClientSession(connector=connector)
-            
-            # اتصال مستقیم با تنظیمات مخصوص
-            await super()._connect()
-            
-        else:
-            # اتصال مستقیم
-            await super()._connect()
+    path = os.path.join(SAVE_DIR, filename)
 
-class TelegramFileBot:
-    def __init__(self):
-        self.client = None
-        self.running = False
-        
-    async def start(self):
-        """شروع ربات"""
-        try:
-            logger.info("🚀 در حال راه‌اندازی ربات...")
-            
-            # تنظیمات اتصال ویژه برای Cloudflare Proxy
-            connection_params = {
-                'connection': ConnectionTcpFull,
-                'use_ipv6': False,
-                'timeout': 60,
-                'retry_delay': 3,
-                'connection_retries': 10,
-                'request_retries': 5
-            }
-            
-            # اگر Cloudflare Proxy فعال باشد
-            if CF_PROXY_ENABLED:
-                # تغییر آدرس سرور تلگرام به مسیر از طریق Worker
-                # این یک ترفند است: ما سرور را به Worker خود تغییر می‌دهیم
-                connection_params['connection'] = ConnectionTcpFull
-                
-                # استفاده از کلاینت معمولی اما با timeout بیشتر
-                logger.info(f"🔗 اتصال از طریق Cloudflare Worker: {CF_WORKER_URL}")
-            
-            # ایجاد کلاینت تلگرام
-            self.client = TelegramClient(
-                StringSession(),
-                int(API_ID),
-                API_HASH,
-                **connection_params
-            )
-            
-            # ترفند: استفاده از MTProto proxy (اگر Worker از آن پشتیبانی کند)
-            if CF_PROXY_ENABLED:
-                try:
-                    # تست اتصال
-                    logger.info("🔍 در حال تست اتصال به تلگرام...")
-                    
-                    # تنظیم session به صورت دستی
-                    self.client.session.set_dc(2, '149.154.167.50', 443)
-                    
-                except Exception as e:
-                    logger.warning(f"⚠️ خطا در تنظیم proxy: {e}")
-            
-            logger.info("🔗 در حال اتصال به سرور تلگرام...")
-            
-            # شروع ربات
-            await self.client.start(bot_token=BOT_TOKEN)
-            
-            if not await self.client.is_user_authorized():
-                logger.error("❌ احراز هویت ناموفق بود!")
-                return
-            
-            self.running = True
-            
-            me = await self.client.get_me()
-            logger.info(f"✅ ربات با موفقیت شروع به کار کرد: @{me.username} (ID: {me.id})")
-            logger.info(f"📁 مسیر ذخیره‌سازی: {BASE_DIR}")
-            
-            # نمایش اطلاعات
-            await self.show_storage_info()
-            
-            # ثبت هندلرها
-            await self._setup_handlers()
-            
-            # تنظیم دستورات ربات
-            await self.setup_bot_commands()
-            
-            logger.info("🤖 ربات آماده دریافت فایل‌ها است...")
-            logger.info(f"👤 آدرس ربات: https://t.me/{me.username}")
-            
-            # نمایش دستورات
-            logger.info("📋 دستورات فعال: /start, /help, /list, /download, /clean, /stats")
-            
-            # اجرای ربات
-            await self.client.run_until_disconnected()
-            
-        except asyncio.TimeoutError:
-            logger.error("⏰ Timeout در اتصال به تلگرام.")
-            logger.info("💡 راه‌حل‌های پیشنهادی:")
-            logger.info("1. مطمئن شوید Cloudflare Worker درست تنظیم شده")
-            logger.info("2. آدرس Worker را در CF_WORKER_URL قرار دهید")
-            logger.info("3. از یک پروکسی SOCKS5 جایگزین استفاده کنید")
-            
-        except Exception as e:
-            logger.error(f"❌ خطا در شروع ربات: {str(e)}", exc_info=True)
-            self.running = False
-    
-    async def setup_bot_commands(self):
-        """تنظیم دستورات ربات در تلگرام"""
-        try:
-            commands = [
-                ('start', 'شروع کار با ربات'),
-                ('help', 'راهنمای کامل'),
-                ('list', 'نمایش فایل‌های شما'),
-                ('download', 'دانلود فایل با کد'),
-                ('clean', 'پاکسازی فایل‌های شما'),
-                ('stats', 'آمار ربات')
-            ]
-            
-            await self.client(
-                events.bots.SetBotCommandsRequest(
-                    scope=events.bots.BotCommandScopeDefault(),
-                    lang_code='fa',
-                    commands=[events.bots.BotCommand(command, description) for command, description in commands]
-                )
-            )
-            logger.info("✅ دستورات ربات در تلگرام تنظیم شد.")
-        except Exception as e:
-            logger.warning(f"⚠️ خطا در تنظیم دستورات: {e}")
-    
-    async def show_storage_info(self):
-        """نمایش اطلاعات فضای ذخیره‌سازی"""
-        try:
-            total_size = 0
-            file_count = 0
-            
-            for file_path in BASE_DIR.glob("*"):
-                if file_path.is_file():
-                    total_size += file_path.stat().st_size
-                    file_count += 1
-            
-            logger.info(f"💾 فضای ذخیره‌سازی:")
-            logger.info(f"   📁 تعداد فایل‌ها: {file_count}")
-            logger.info(f"   📊 حجم استفاده شده: {total_size / 1024 / 1024:.2f} MB")
-            
-        except Exception as e:
-            logger.error(f"خطا در دریافت اطلاعات فضای ذخیره‌سازی: {e}")
-    
-    async def _setup_handlers(self):
-        """تنظیم هندلرهای ربات"""
-        
-        @self.client.on(events.NewMessage(pattern='/start'))
-        async def start_handler(event):
-            """هندلر دستور /start"""
-            try:
-                user = await event.get_sender()
-                logger.info(f"👤 کاربر {user.id} ({user.first_name}) /start را فرستاد")
-                
-                welcome_msg = """
-                🌟 **به ربات آپلود فایل خوش آمدید!**
+    with open(path, "wb") as f:
+        f.write(data)
 
-                **📌 قابلیت‌ها:**
-                • دریافت انواع فایل‌ها (تا 2GB)
-                • ذخیره فایل در فضای ابری
-                • دانلود فایل با کد یکتا
-
-                **🚀 نحوه استفاده:**
-                1. فایل خود را ارسال کنید
-                2. ربات کد فایل را به شما می‌دهد
-                3. با دستور /download کد فایل را دانلود کنید
-
-                **📋 دستورات:**
-                /start - شروع کار
-                /help - راهنمای کامل
-                /list - نمایش فایل‌های شما
-                /clean - پاکسازی فایل‌های قدیمی
-                /stats - آمار ربات
-
-                **⚠️ توجه:**
-                • فایل‌ها به صورت موقت ذخیره می‌شوند
-                • بعد از ری‌استارت Space فایل‌ها پاک می‌شوند
-                """
-                await event.reply(welcome_msg)
-                
-            except Exception as e:
-                logger.error(f"خطا در start_handler: {e}")
-        
-        @self.client.on(events.NewMessage(pattern='/help'))
-        async def help_handler(event):
-            """هندلر دستور /help"""
-            help_text = """
-            **📖 راهنمای کامل ربات:**
-
-            **🔹 نحوه کار:**
-            1. فایلی برای ربات بفرستید (هر نوعی)
-            2. ربات آن را ذخیره می‌کند و یک کد به شما می‌دهد
-            3. با دستور /download می‌توانید فایل را دانلود کنید
-
-            **🔹 مثال:**
-            ```
-            /download 123456_abc
-            ```
-
-            **🔹 دستورات موجود:**
-            • `/list` - نمایش 10 فایل آخر شما
-            • `/download کد` - دانلود فایل با کد
-            • `/clean` - پاک کردن تمام فایل‌های شما
-            • `/stats` - مشاهده آمار ربات
-
-            **🔹 نکات مهم:**
-            • حداکثر حجم فایل: 2GB
-            • فایل‌ها در فضای /tmp ذخیره می‌شوند
-            • کد فایل را ذخیره کنید
-            """
-            await event.reply(help_text)
-        
-        @self.client.on(events.NewMessage(pattern='/list'))
-        async def list_handler(event):
-            """نمایش فایل‌های کاربر"""
-            try:
-                user_id = event.sender_id
-                
-                if user_id in user_data and user_data[user_id].get("files"):
-                    files = user_data[user_id]["files"]
-                    
-                    if not files:
-                        await event.reply("📭 **هیچ فایلی از شما آپلود نشده است.**")
-                        return
-                    
-                    response = "📁 **فایل‌های شما:**\n\n"
-                    for idx, file_info in enumerate(files[-10:], 1):
-                        response += f"**{idx}. {file_info['name']}**\n"
-                        response += f"   📏 حجم: {file_info['size_mb']:.2f} MB\n"
-                        response += f"   🕐 زمان: {file_info['time']}\n"
-                        response += f"   🔗 کد: `{file_info['code']}`\n\n"
-                    
-                    response += "📥 برای دانلود:\n"
-                    response += "```\n"
-                    response += "/download کد_فایل\n"
-                    response += "```\n\n"
-                    response += "مثال: `/download abc123`"
-                    
-                    await event.reply(response)
-                else:
-                    await event.reply("📭 **هیچ فایلی از شما آپلود نشده است.**")
-                    
-            except Exception as e:
-                logger.error(f"خطا در list_handler: {e}")
-        
-        @self.client.on(events.NewMessage(pattern='/clean'))
-        async def clean_handler(event):
-            """پاکسازی فایل‌های کاربر"""
-            try:
-                user_id = event.sender_id
-                
-                if user_id in user_data and "files" in user_data[user_id]:
-                    file_count = len(user_data[user_id]["files"])
-                    
-                    # پاک کردن فایل‌های فیزیکی
-                    deleted_count = 0
-                    for file_info in user_data[user_id]["files"]:
-                        file_path = BASE_DIR / file_info["code"]
-                        if file_path.exists():
-                            try:
-                                file_path.unlink()
-                                deleted_count += 1
-                            except Exception as e:
-                                logger.error(f"خطا در پاک کردن فایل {file_info['code']}: {e}")
-                    
-                    # پاک کردن داده‌های کاربر
-                    user_data[user_id]["files"] = []
-                    
-                    await event.reply(f"🗑️ **{deleted_count} فایل از {file_count} با موفقیت پاک شدند.**")
-                else:
-                    await event.reply("📭 **هیچ فایلی برای پاک کردن وجود ندارد.**")
-                    
-            except Exception as e:
-                logger.error(f"خطا در clean_handler: {e}")
-        
-        @self.client.on(events.NewMessage(pattern='/stats'))
-        async def stats_handler(event):
-            """نمایش آمار ربات"""
-            try:
-                total_files = 0
-                total_size = 0
-                total_users = len(user_data)
-                
-                for user_id, data in user_data.items():
-                    if "files" in data:
-                        total_files += len(data["files"])
-                        for file_info in data["files"]:
-                            total_size += file_info.get("size_bytes", 0)
-                
-                # محاسبه فضای باقی‌مانده
-                try:
-                    statvfs = os.statvfs('/tmp')
-                    free_space = (statvfs.f_bavail * statvfs.f_frsize) / 1024 / 1024 / 1024  # GB
-                    total_space = (statvfs.f_blocks * statvfs.f_frsize) / 1024 / 1024 / 1024  # GB
-                    used_space = total_space - free_space
-                except:
-                    free_space = used_space = total_space = 0
-                
-                stats_msg = f"""
-                📊 **آمار ربات:**
-
-                👥 کاربران فعال: {total_users}
-                📁 فایل‌های آپلود شده: {total_files}
-                💾 حجم کل فایل‌ها: {total_size / 1024 / 1024:.2f} MB
-
-                🗂️ **فضای ذخیره‌سازی:**
-                📊 استفاده شده: {used_space:.2f} GB
-                📈 کل فضای /tmp: {total_space:.2f} GB
-                📉 فضای آزاد: {free_space:.2f} GB
-
-                ⚠️ **توجه:** فایل‌ها در `/tmp` ذخیره می‌شوند.
-                """
-                await event.reply(stats_msg)
-                
-            except Exception as e:
-                logger.error(f"خطا در stats_handler: {e}")
-        
-        @self.client.on(events.NewMessage(pattern=r'/download\s+\S+'))
-        async def download_handler(event):
-            """دانلود فایل با کد"""
-            try:
-                parts = event.message.text.split()
-                if len(parts) < 2:
-                    await event.reply("❌ **لطفا کد فایل را وارد کنید.**\nمثال: `/download abc123`")
-                    return
-                
-                file_code = parts[1].strip()
-                file_path = BASE_DIR / file_code
-                
-                if not file_path.exists():
-                    await event.reply("❌ **فایل یافت نشد.**\nمطمئن شوید کد را درست وارد کرده‌اید.")
-                    return
-                
-                # پیدا کردن اطلاعات فایل
-                file_info = None
-                for user_id, data in user_data.items():
-                    if "files" in data:
-                        for f in data["files"]:
-                            if f["code"] == file_code:
-                                file_info = f
-                                break
-                
-                if file_info:
-                    # ارسال فایل به کاربر
-                    caption = f"📄 **{file_info['name']}**\n📏 حجم: {file_info['size_mb']:.2f} MB\n🕐 زمان آپلود: {file_info['time']}"
-                    await event.reply("📤 **در حال ارسال فایل...**")
-                    
-                    # ارسال فایل
-                    await event.respond(
-                        file=file_path,
-                        caption=caption,
-                        force_document=True
-                    )
-                    
-                    logger.info(f"فایل ارسال شد: {file_code} به کاربر {event.sender_id}")
-                else:
-                    await event.reply("⚠️ **اطلاعات فایل یافت نشد.**")
-                    
-            except Exception as e:
-                logger.error(f"خطا در دانلود: {e}", exc_info=True)
-                await event.reply(f"❌ **خطا در پردازش:** {str(e)}")
-        
-        @self.client.on(events.NewMessage)
-        async def message_handler(event):
-            """هندلر پیام‌های معمولی و فایل‌ها"""
-            # اگر پیام دستور باشد، نادیده بگیر
-            if event.message.text and event.message.text.startswith('/'):
-                return
-            
-            user_id = event.sender_id
-            
-            # اگر پیام حاوی فایل باشد
-            if event.file:
-                await self._handle_file(event, user_id)
-            else:
-                # پاسخ به پیام متنی
-                await event.reply("📝 **لطفا یک فایل ارسال کنید.**\nبرای راهنمایی /help را ارسال کنید.")
-    
-        async def _handle_file(self, event, user_id):
-            """مدیریت دریافت و ذخیره فایل"""
-            try:
-                # اطلاعات فایل
-                file_name = event.file.name or f"file_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
-                file_size = event.file.size
-                
-                # بررسی حجم فایل (حداکثر 2GB)
-                max_size = 2 * 1024 * 1024 * 1024  # 2GB
-                if file_size > max_size:
-                    await event.reply(f"❌ **حجم فایل بسیار بزرگ است!**\nحداکثر حجم مجاز: 2GB\nحجم فایل شما: {file_size / 1024 / 1024 / 1024:.2f} GB")
-                    return
-                
-                # ایجاد کد یکتا برای فایل
-                timestamp = int(datetime.now().timestamp())
-                file_code = f"{user_id}_{timestamp}"
-                save_path = BASE_DIR / file_code
-                
-                # اطلاع‌رسانی شروع دریافت
-                status_msg = await event.reply(
-                    f"📥 **دریافت فایل...**\n\n"
-                    f"📄 نام: `{file_name}`\n"
-                    f"📏 حجم: {file_size / 1024 / 1024:.2f} MB\n"
-                    f"⏳ لطفا صبر کنید..."
-                )
-                
-                # دانلود فایل
-                download_path = await event.download_media(file=str(save_path))
-                
-                # ذخیره اطلاعات فایل
-                file_info = {
-                    "name": file_name,
-                    "code": file_code,
-                    "path": str(download_path),
-                    "size_bytes": file_size,
-                    "size_mb": file_size / 1024 / 1024,
-                    "time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-                    "message_id": event.id
-                }
-                
-                # ایجاد یا به‌روزرسانی داده کاربر
-                if user_id not in user_data:
-                    user_data[user_id] = {"files": []}
-                
-                user_data[user_id]["files"].append(file_info)
-                
-                # فقط آخرین 20 فایل را نگه دار
-                if len(user_data[user_id]["files"]) > 20:
-                    # حذف قدیمی‌ترین فایل
-                    old_file = user_data[user_id]["files"].pop(0)
-                    old_path = BASE_DIR / old_file["code"]
-                    if old_path.exists():
-                        old_path.unlink()
-                        logger.info(f"فایل قدیمی پاک شد: {old_file['code']}")
-                
-                # پاسخ موفقیت‌آمیز
-                success_msg = (
-                    f"✅ **فایل با موفقیت دریافت شد!**\n\n"
-                    f"📄 نام فایل: `{file_name}`\n"
-                    f"📏 حجم فایل: {file_size / 1024 / 1024:.2f} MB\n"
-                    f"🔢 کد فایل: `{file_code}`\n\n"
-                    f"📥 **برای دانلود:**\n"
-                    f"```\n"
-                    f"/download {file_code}\n"
-                    f"```\n\n"
-                    f"📋 **تعداد فایل‌های شما:** {len(user_data[user_id]['files'])}"
-                )
-                
-                await status_msg.edit(success_msg)
-                
-                logger.info(f"فایل دریافت شد: {file_name} ({file_code}) توسط کاربر {user_id}")
-                
-            except Exception as e:
-                logger.error(f"خطا در دریافت فایل: {e}", exc_info=True)
-                try:
-                    await event.reply(f"❌ **خطا در دریافت فایل:**\n{str(e)}")
-                except:
-                    pass
-
-async def main():
-    """تابع اصلی اجرای ربات"""
-    # بررسی تنظیمات
-    if not check_config():
-        logger.error("❌ لطفاً تنظیمات را در Secrets تنظیم کنید.")
-        return
-    
-    # ایجاد مسیر ذخیره‌سازی
-    BASE_DIR.mkdir(parents=True, exist_ok=True)
-    
-    # نمایش اطلاعات راه‌اندازی
-    logger.info("=" * 60)
-    logger.info("🤖 ربات آپلود فایل تلگرام با Cloudflare Proxy")
-    logger.info(f"📁 مسیر ذخیره‌سازی: {BASE_DIR}")
-    logger.info(f"🌐 Cloudflare Proxy: {'فعال ✅' if CF_PROXY_ENABLED else 'غیرفعال ❌'}")
-    
-    if CF_PROXY_ENABLED and CF_WORKER_URL:
-        logger.info(f"   🔗 Worker URL: {CF_WORKER_URL}")
-    
-    # محاسبه فضای موجود
-    try:
-        statvfs = os.statvfs('/tmp')
-        free_gb = (statvfs.f_bavail * statvfs.f_frsize) / 1024 / 1024 / 1024
-        total_gb = (statvfs.f_blocks * statvfs.f_frsize) / 1024 / 1024 / 1024
-        logger.info(f"💾 فضای /tmp: {free_gb:.2f} GB آزاد از {total_gb:.2f} GB")
-    except:
-        logger.info("💾 فضای /tmp: نامشخص")
-    
-    logger.info("=" * 60)
-    
-    # ایجاد و اجرای ربات
-    bot = TelegramFileBot()
-    
-    try:
-        await bot.start()
-    except KeyboardInterrupt:
-        logger.info("🛑 ربات متوقف شد.")
-    except Exception as e:
-        logger.error(f"❌ خطای غیرمنتظره: {str(e)}", exc_info=True)
-        
-        # راه‌حل‌های پیشنهادی
-        logger.info("💡 **راه‌حل‌های پیشنهادی:**")
-        logger.info("1. مطمئن شوید Cloudflare Worker درست تنظیم شده است")
-        logger.info("2. Worker باید به api.telegram.org پروکسی کند")
-        logger.info("3. می‌توانید از پروکسی SOCKS5 جایگزین استفاده کنید")
-        logger.info("4. API credentials را دوباره بررسی کنید")
-
-# اجرای ربات
-if __name__ == "__main__":
-    asyncio.run(main())
+    return {"saved_as": filename}