from fastapi import FastAPI, Request
from fastapi.templating import Jinja2Templates
from fastapi.staticfiles import StaticFiles
from contextlib import asynccontextmanager
from src.utils.logging_config import setup_logging
from alembic.config import Config
from alembic import command
import os
import logging

@asynccontextmanager
async def lifespan(app: FastAPI):
    # Startup
    setup_logging()
    logger = logging.getLogger(__name__)
    logger.info("--- Application Starting Up ---")
    
    alembic_cfg = Config("alembic.ini")
    database_url = os.getenv("DATABASE_URL")
    if database_url:
        alembic_cfg.set_main_option("sqlalchemy.url", database_url)
        try:
            command.upgrade(alembic_cfg, "head")
            logger.info("Database migrations checked/applied.")
        except Exception as e:
            logger.error(f"Error running database migrations: {e}")
    else:
        logger.warning("DATABASE_URL not set, skipping migrations.")
    
    yield
    
    logger.info("--- Application Shutting Down ---")

app = FastAPI(lifespan=lifespan)
app.mount("/static", StaticFiles(directory="../static"), name="static")
templates = Jinja2Templates(directory="templates")

from src.api import status, sync, setup, logs, metrics, activities

app.include_router(status.router, prefix="/api")
app.include_router(sync.router, prefix="/api")
app.include_router(setup.router, prefix="/api")
app.include_router(logs.router, prefix="/api")
app.include_router(metrics.router, prefix="/api")
app.include_router(activities.router, prefix="/api")

@app.get("/")
async def read_root(request: Request):
    return templates.TemplateResponse("index.html", {"request": request})

@app.get("/setup")
async def setup_page(request: Request):
    return templates.TemplateResponse("setup.html", {"request": request})