Update main.py

main.py

@@ -1,96 +1,115 @@
+import os
+import re
 import asyncio
-import logging
-from pyrogram import Client, filters
-from motor.motor_asyncio import AsyncIOMotorClient
-from aiohttp import web
-from config import Config
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.responses import StreamingResponse, JSONResponse
+from pyrogram import Client
+import uvicorn
+from contextlib import asynccontextmanager
 
-# Logging သတ်မှတ်ခြင်း
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+# --- Config ---
+API_ID = os.getenv("API_ID")
+API_HASH = os.getenv("API_HASH")
+STRING_SESSION = os.getenv("STRING_SESSION")
+ACCESS_KEY = os.getenv("ACCESS_KEY", "0000")
 
-# --- Database Initialize ---
-db_client = AsyncIOMotorClient(Config.DATABASE_URL)
-db = db_client[Config.DATABASE_NAME]
-collection = db[Config.COLLECTION_NAME]
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Bot ကို စတင်ခြင်း
+    user_bot = Client(
+        "tg_streamer",
+        api_id=API_ID,
+        api_hash=API_HASH,
+        session_string=STRING_SESSION,
+        in_memory=True,
+        workers=16
+    )
+    await user_bot.start()
+    app.state.tg = user_bot
+    print("🚀 Ultimate Smooth Streamer is Online!")
+    yield
+    # Bot ကို ပိတ်ခြင်း
+    await user_bot.stop()
 
-# --- Telegram Clients ---
-# Bot Client (Bot Token ဖြင့်)
-bot = Client(
-    "TgIndexBot",
-    api_id=Config.API_ID,
-    api_hash=Config.API_HASH,
-    bot_token=Config.BOT_TOKEN
-)
+app = FastAPI(lifespan=lifespan)
 
-# User Client (Session String ဖြင့် - Channel history ဖတ်ရန်)
-user_bot = Client(
-    "TgUserSession",
-    api_id=Config.API_ID,
-    api_hash=Config.API_HASH,
-    session_string=Config.SESSION
-)
+# --- Cron-job.org အတွက် Health Check Endpoint ---
+@app.get("/")
+async def health_check():
+    return JSONResponse(content={"status": "running", "message": "Server is alive!"})
 
-# --- Bot Commands ---
+def parse_tg_link(link: str):
+    match = re.search(r't\.me/(?:c/)?([^/]+)/(\d+)', link)
+    if match:
+        chat_id = match.group(1)
+        if chat_id.isdigit():
+            chat_id = int("-100" + chat_id)
+        return chat_id, int(match.group(2))
+    return None, None
 
-@bot.on_message(filters.command("start") & filters.private)
-async def start_cmd(client, message):
-    if message.from_user.id in Config.OWNER_ID:
-        await message.reply_text(f"မင်္ဂလာပါ Owner {message.from_user.first_name}!\nBot အောင်မြင်စွာ တက်လာပါပြီ။")
-    else:
-        await message.reply_text("သင်သည် ဤ Bot ၏ Owner မဟုတ်ပါ။")
+@app.get("/stream")
+async def stream_telegram(request: Request, url: str, key: str = None):
+    if key != ACCESS_KEY:
+        raise HTTPException(status_code=403, detail="Invalid Key")
 
-@bot.on_message(filters.command("index") & filters.private)
-async def index_files(client, message):
-    if message.from_user.id not in Config.OWNER_ID:
-        return
-    
-    await message.reply_text("Indexing စတင်နေပါပြီ... ခေတ္တစောင့်ပါ။")
+    chat_id, msg_id = parse_tg_link(url)
+    if not chat_id:
+        raise HTTPException(status_code=400, detail="Invalid URL")
+
+    client = request.app.state.tg
     
-    count = 0
-    async for msg in user_bot.get_chat_history(Config.INDEX_CHANNEL):
-        if msg.media:
-            # Media file data ကို database ထဲသိမ်းခြင်း
-            file_data = {
-                "message_id": msg.id,
-                "file_name": getattr(msg.document or msg.video or msg.audio, 'file_name', 'Unknown'),
-                "file_type": str(msg.media),
-                "caption": msg.caption or ""
-            }
-            await collection.update_one({"message_id": msg.id}, {"$set": file_data}, upsert=True)
-            count += 1
-            
-    await message.reply_text(f"အောင်မြင်စွာ Index လုပ်ပြီးပါပြီ။ ဖိုင်ပေါင်း {count} ဖိုင် သိမ်းဆည်းခဲ့သည်။")
+    try:
+        msg = await client.get_messages(chat_id, msg_id)
+        media = msg.video or msg.document or msg.audio or msg.animation
+        if not media: raise Exception()
+    except:
+        raise HTTPException(status_code=404, detail="File not found")
 
-# --- Web Server (for Health Check/Hugging Face) ---
+    file_size = media.file_size
+    range_header = request.headers.get("range")
+    
+    start_byte = 0
+    if range_header:
+        match = re.search(r'bytes=(\d+)-', range_header)
+        if match:
+            start_byte = int(match.group(1))
 
-async def handle_web(request):
-    return web.Response(text="🚀 TgindexPro is Running!")
+    async def refined_generator():
+        offset_in_mb = start_byte // (1024 * 1024)
+        skip_bytes = start_byte % (1024 * 1024)
+        
+        try:
+            generator = client.stream_media(msg, offset=offset_in_mb, limit=30)
+            first_chunk = True
+            async for chunk in generator:
+                if first_chunk:
+                    if len(chunk) > skip_bytes:
+                        yield chunk[skip_bytes:]
+                    first_chunk = False
+                else:
+                    yield chunk
+        except Exception as e:
+            print(f"Stream interrupted: {e}")
+            return
 
-async def start_webserver():
-    app = web.Application()
-    app.router.add_get("/", handle_web)
-    runner = web.AppRunner(app)
-    await runner.setup()
-    site = web.TCPSite(runner, "0.0.0.0", Config.PORT)
-    await site.start()
-    logger.info(f"Web server started on port {Config.PORT}")
+    headers = {
+        "Content-Type": getattr(media, "mime_type", "video/mp4"),
+        "Accept-Ranges": "bytes",
+        "Access-Control-Allow-Origin": "*",
+        "Cache-Control": "public, max-age=3600",
+    }
 
-# --- Main Function ---
+    if range_header:
+        headers["Content-Range"] = f"bytes {start_byte}-{file_size-1}/{file_size}"
+        status_code = 206
+    else:
+        status_code = 200
 
-async def main():
-    logger.info("Starting TgindexPro...")
-    
-    # Web Server run ခြင်း
-    await start_webserver()
-    
-    # Bot နှင့် User Client တို့ကို run ခြင်း
-    await bot.start()
-    await user_bot.start()
-    
-    logger.info("Bot and User Session are online!")
-    await asyncio.Event().wait()
+    return StreamingResponse(
+        refined_generator(),
+        status_code=status_code,
+        headers=headers
+    )
 
 if __name__ == "__main__":
-    loop = asyncio.get_event_loop()
-    loop.run_until_complete(main())
+    uvicorn.run(app, host="0.0.0.0", port=7860)