main.py

from contextlib import asynccontextmanager

import uvicorn
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from mangum import Mangum

from config.db import init_db
from models.tags_models import Tag
from routes.admin_routes import admin_router
from routes.plant_routes import plant_router
from routes.token_router import token_router
from routes.user_routes import user_router

client = {}


@asynccontextmanager
async def lifespan(app: FastAPI):
    client["connection"] = await init_db()
    yield
    client.clear()


app = FastAPI(lifespan=lifespan)
handler = Mangum(app)


def configure():
    configure_routes()
    configure_app()


def configure_routes():
    @app.get("/api", tags=["root"])
    def root():
        return "Welcome to PlantUrbanus"

    app.include_router(admin_router, prefix="/api/admin", tags=[Tag.admin])
    app.include_router(user_router, prefix="/api/users", tags=[Tag.users])
    app.include_router(plant_router, prefix="/api/plants", tags=[Tag.plants])
    # app.include_router(aroid_router, prefix="/aroids", tags=[Tag.aroids])
    app.include_router(token_router, tags=["token"])


def configure_app():
    origins = [
        "http://localhost:3000",
        "http://localhost:8000",
    ]

    app.add_middleware(
        CORSMiddleware,
        allow_origins=origins,
        allow_credentials=True,
        allow_methods=["*"],
        allow_headers=["*"],
    )


def main():
    configure()
    uvicorn.run(reload=True, app="main:app")


if __name__ == "__main__":
    main()
else:
    configure()