from fastapi import FastAPI, Request
from routes import sso, user
from config import settings
import uvicorn
from fastapi.middleware.cors import CORSMiddleware
from database import init_db
from routes import logout
from routes import boss
from services.boss_retrieval import process_boss_file
from contextlib import asynccontextmanager
import asyncio

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=[settings.frontend_url], 
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
    expose_headers=["Set-Cookie"],
)

async def boss_worker_loop():
    print("background worker started")
    while True:
        try:
            print("running boss service")
            process_boss_file()
        except Exception as e:
            print(f"worker error: {e}")
        
        # TODO: actually sync to what schedule says.
        # for now its set to 1hr just incase pretendo fuckery happens
        # (such as rotation ending)
        await asyncio.sleep(3600)

@asynccontextmanager
async def lifespan(app: FastAPI):
    task = asyncio.create_task(boss_worker_loop())
    yield
    print("shutdown: cancelling background tasks")
    task.cancel()

app = FastAPI(lifespan=lifespan)

@app.middleware("http")
async def force_cors_on_errors(request: Request, call_next):
    response = await call_next(request)
    origin = request.headers.get("origin")
    if origin == settings.frontend_url:
        response.headers["Access-Control-Allow-Origin"] = origin
        response.headers["Access-Control-Allow-Credentials"] = "true"
    return response

app.include_router(sso.router, prefix="/api/v2/sso")
app.include_router(user.router, prefix="/api/v1/users")
app.include_router(logout.router, prefix="/api/v1")
app.include_router(boss.router, prefix="/api/v1")

if __name__ == '__main__':
    uvicorn.run("main:app", host="0.0.0.0", port=settings.port, reload=True)