from fastapi import FastAPI
from src.api.v1 import tasks
from src.api.v1.auth import router as auth_router
from src.auth.middleware import JWTMiddleware
from fastapi.middleware.cors import CORSMiddleware
from src.core.database import create_db_and_tables
from contextlib import asynccontextmanager


@asynccontextmanager
async def lifespan(app: FastAPI):
    # Startup
    create_db_and_tables()
    yield
    # Shutdown (if needed)


app = FastAPI(title="Todo API", version="1.0.0", lifespan=lifespan)


# Add JWT authentication middleware
app.add_middleware(JWTMiddleware)

app.add_middleware(
    CORSMiddleware,
    allow_origins=[
        "http://localhost:3000",
        "https://ahmed-hackathon-ii-phase-ii.vercel.app",
    ],  # frontend
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# Include API routes
app.include_router(tasks.router, prefix="/api/v1/tasks", tags=["tasks"])
app.include_router(auth_router, prefix="/api/v1", tags=["auth"])
# Also include routes at root level for backward compatibility
app.include_router(auth_router, tags=["auth-root"])
app.include_router(tasks.router, prefix="/tasks", tags=["tasks-root"])


@app.get("/")
def read_root():
    return {"Hello": "World"}


if __name__ == "__main__":
    import uvicorn

    uvicorn.run(app, host="0.0.0.0", port=7860)