import os

from dotenv import load_dotenv
from fastapi.middleware.cors import CORSMiddleware

load_dotenv()


def setup_cors(app):

    dev_origin = os.getenv("FRONTEND_URL", "http://localhost:3000")
    prod_origin = os.getenv("PROD_FRONTEND_URL")

    origins = [dev_origin, prod_origin]

    filtered_origins = []
    for origin in origins:
        if origin:
            filtered_origins.append(origin)

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