oneuptime/CommonServer/Middleware/ClusterKeyAuthorization.ts

import { ClusterKey as CLUSTER_KEY } from '../Config';
import {
    ExpressRequest,
    ExpressResponse,
    NextFunction,
} from '../Utils/Express';

import Response from '../Utils/Response';
import BadDataException from 'Common/Types/Exception/BadDataException';
import ObjectID from 'Common/Types/ObjectID';

export default class ClusterKeyAuthorization {
    public static async isAuthorizedServiceMiddleware(
        req: ExpressRequest,
        res: ExpressResponse,
        next: NextFunction
    ): Promise<void> {
        let clusterKey: ObjectID;

        if (req.params && req.params['clusterKey']) {
            clusterKey = new ObjectID(req.params['clusterKey']);
        } else if (req.query && req.query['clusterKey']) {
            clusterKey = new ObjectID(req.query['clusterKey'] as string);
        } else if (req.headers && req.headers['clusterkey']) {
            // Header keys are automatically transformed to lowercase
            clusterKey = new ObjectID(req.headers['clusterkey'] as string);
        } else if (req.body && req.body.clusterKey) {
            clusterKey = new ObjectID(req.body.clusterKey);
        } else {
            return Response.sendErrorResponse(
                req,
                res,
                new BadDataException('Cluster key not found.')
            );
        }

        const isAuthorized: boolean = clusterKey === CLUSTER_KEY;

        if (!isAuthorized) {
            return Response.sendErrorResponse(
                req,
                res,
                new BadDataException('Invalid cluster key provided')
            );
        }

        return next();
    }
}