oneuptime/Ingestor/Middleware/ProbeAuthorization.ts

import { ProbeExpressRequest } from "../Types/Request";
import OneUptimeDate from "Common/Types/Date";
import Dictionary from "Common/Types/Dictionary";
import BadDataException from "Common/Types/Exception/BadDataException";
import { JSONObject } from "Common/Types/JSON";
import ObjectID from "Common/Types/ObjectID";
import { ClusterKey as ONEUPTIME_SECRET } from "CommonServer/EnvironmentConfig";
import ProbeService from "CommonServer/Services/ProbeService";
import { ExpressResponse, NextFunction } from "CommonServer/Utils/Express";
import Response from "CommonServer/Utils/Response";
import Probe from "Model/Models/Probe";

export default class ProbeAuthorization {
  public static getClusterKeyHeaders(): Dictionary<string> {
    return {
      clusterkey: ONEUPTIME_SECRET.toString(),
    };
  }

  public static async isAuthorizedServiceMiddleware(
    req: ProbeExpressRequest,
    res: ExpressResponse,
    next: NextFunction,
  ): Promise<void> {
    const data: JSONObject = req.body;

    if (!data["probeId"] || !data["probeKey"]) {
      return Response.sendErrorResponse(
        req,
        res,
        new BadDataException("ProbeId or ProbeKey is missing"),
      );
    }

    const probeId: ObjectID = new ObjectID(data["probeId"] as string);

    const probeKey: string = data["probeKey"] as string;

    const probe: Probe | null = await ProbeService.findOneBy({
      query: {
        _id: probeId.toString(),
        key: probeKey,
      },
      select: {
        _id: true,
      },
      props: {
        isRoot: true,
      },
    });

    if (!probe) {
      return Response.sendErrorResponse(
        req,
        res,
        new BadDataException("Invalid Probe ID or Probe Key"),
      );
    }

    await ProbeService.updateOneById({
      id: probeId,
      data: {
        lastAlive: OneUptimeDate.getCurrentDate(),
      },
      props: {
        isRoot: true,
      },
    });

    req.probe = probe;

    return next();
  }
}