import uvicorn
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from Config.uvicorn_log_config import uvicorn_log_config
from Context.common import conf, common_db
from Mods.EditRecord import Router as EditRecordRouter
from Mods.User import Router as UserRouter
from Mods.BasicInformation import Router as BasicInformationRouter
from Mods.OtherInformation import Router as OtherInformationRouter
from Mods.CommonInformation import Router as CommonInformationRouter
from Mods.RegionalEconomies import Router as RegionalEconomiesRouter
from Utils.RecordUtils import record_middleware

common_db.init_database()
app = FastAPI(
    title="城投调研工具后端",
    description="城投调研工具后端",
    version="v1.0.0",
    docs_url="/uir/docs",
    openapi_url="/uir/openapi.json"
)
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.middleware('http')(record_middleware)
app.router.prefix = "/api/urban_investment_research"
app.include_router(EditRecordRouter.router)
app.include_router(UserRouter.router)
app.include_router(BasicInformationRouter.router)
app.include_router(OtherInformationRouter.router)
app.include_router(CommonInformationRouter.router)
app.include_router(RegionalEconomiesRouter.router)
print(f"http://127.0.0.1:{conf['app']['port']}/docs")
if __name__ == '__main__':
    uvicorn.run(app, log_config=uvicorn_log_config, port=int(conf['app']['port']))