from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from core.settings import app_settings
from utils import logger


# type: ignore


def add_cors_middleware(app: FastAPI):
    """
    添加跨域中间件

    :param app: FastAPI对象
    :return:
    """
    if app_settings.origins and len(app_settings.origins) > 0:
        logger.debug(f"允许跨域访问的域名: {app_settings.origins}")
        # 后台api允许跨域
        app.add_middleware(
            CORSMiddleware,  # type: ignore
            allow_origins=app_settings.origins,
            allow_credentials=True,
            allow_methods=["*"],
            allow_headers=["*"],
        )