Create main.py

main.py

@@ -0,0 +1,113 @@
+# filename: main.py
+
+import logging
+import asyncio
+import os
+from dotenv import load_dotenv
+
+# IMPORTANT: Load environment variables from .env file at the very beginning
+load_dotenv()
+
+from fastapi import FastAPI
+from fastapi.responses import FileResponse, PlainTextResponse
+import uvicorn
+
+# Now, import our application modules AFTER loading the environment
+import config
+from core.bot import bot, start_bot_runtime
+from database import manager as db_manager
+
+# Import all handler modules so Telethon can register the event handlers
+from handlers import start, admin, user, links
+
+# --- Logging Configuration ---
+# Creates the logs directory if it doesn't exist
+os.makedirs("logs", exist_ok=True)
+
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.StreamHandler(), # To show logs in the console
+        logging.FileHandler("logs/bot.log", mode='a', encoding='utf-8') # To save logs to a file
+    ]
+)
+logger = logging.getLogger(__name__)
+
+
+# --- FastAPI Application Setup ---
+app = FastAPI(
+    title="Terabox Bot Service",
+    description="The backend service for the Terabox Downloader Telegram Bot.",
+    version="1.0.0"
+)
+
+
+# --- FastAPI Events: Startup and Shutdown ---
+@app.on_event("startup")
+async def startup_event():
+    """
+    This function runs once when the application starts.
+    It initializes everything needed for the bot to run.
+    """
+    logger.info("===== APPLICATION STARTUP =====")
+    
+    # 1. Initialize the database and create indexes for fast queries
+    await db_manager.setup_database_indexes()
+    
+    # 2. Start the Telethon client
+    logger.info("Starting Telegram client...")
+    await bot.start(bot_token=config.BOT_TOKEN)
+    me = await bot.get_me()
+    logger.info(f"Bot client started successfully as @{me.username}")
+    
+    # 3. Start the bot's core engine (workers, scheduler)
+    start_bot_runtime()
+    
+    # 4. Run the bot's main event loop in the background
+    asyncio.create_task(bot.run_until_disconnected())
+    logger.info("Bot is now running and listening for events...")
+
+
+@app.on_event("shutdown")
+async def shutdown_event():
+    """This function runs once when the application is shutting down."""
+    logger.info("===== APPLICATION SHUTDOWN =====")
+    if bot.is_connected():
+        await bot.disconnect()
+        logger.info("Bot client has been disconnected.")
+
+
+# --- FastAPI Routes (Web Endpoints) ---
+@app.get("/", response_class=FileResponse)
+async def read_index():
+    """
+    Serves the main web interface if index.html exists.
+    Otherwise, returns a simple status message.
+    """
+    index_path = "index.html"
+    if os.path.exists(index_path):
+        return FileResponse(index_path)
+    return PlainTextResponse("The bot service is running. The web UI (index.html) was not found.")
+
+
+@app.get("/health", response_class=PlainTextResponse)
+async def health_check():
+    """A simple endpoint for health checks and monitoring."""
+    if bot.is_connected():
+        return PlainTextResponse("✅ OK: Bot service is running and connected to Telegram.")
+    else:
+        return PlainTextResponse("❌ ERROR: Bot service is running but not connected to Telegram.", status_code=503)
+
+
+# --- Main Execution Block ---
+if __name__ == "__main__":
+    logger.info("Starting Uvicorn server...")
+    # Uvicorn runs the FastAPI application.
+    # 'reload=True' is for development; you might remove it in production.
+    uvicorn.run(
+        "main:app",
+        host="0.0.0.0",
+        port=8080, # Common port for web services
+        reload=True # Automatically restarts the server when you save a file
+    )