import fastapi
import os


QUEUES_TOKEN = os.getenv('QUEUES_TOKEN')

class CheckToken:
    async def __call__(self, 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)