import fastapi
import uvicorn
import os

from app.routers import take
from app.routers import put
from app.routers import finish

from app.storage import mongo


QUEUES_TOKEN = os.getenv('QUEUES_TOKEN')

app = fastapi.FastAPI()


@app.middleware('http')
async def check_token(request: fastapi.Request, call_next):
    if QUEUES_TOKEN:
        token = request.headers.get('X-Queues-Token')
        if not token or token != QUEUES_TOKEN:
            raise fastapi.HTTPException(403)
    return await call_next(request)


app.include_router(take.router)
app.include_router(put.router)
app.include_router(finish.router)

mongo.create_indexes()


if __name__ == '__main__':
    uvicorn.run(app, host="0.0.0.0", port=1238)