perf: cache only JSON payload validation functions

[e-mobility-charging-stations-simulator.git] / src / charging-station / ocpp / 2.0 / OCPP20ResponseService.ts

// Partial Copyright Jerome Benoit. 2021-2024. All Rights Reserved.

import type { ValidateFunction } from 'ajv'

import { OCPP20ServiceUtils } from './OCPP20ServiceUtils.js'
import { type ChargingStation, addConfigurationKey } from '../../../charging-station/index.js'
import { OCPPError } from '../../../exception/index.js'
import {
  ErrorType,
  type JsonType,
  type OCPP20BootNotificationResponse,
  type OCPP20ClearCacheResponse,
  type OCPP20HeartbeatResponse,
  OCPP20IncomingRequestCommand,
  OCPP20OptionalVariableName,
  OCPP20RequestCommand,
  type OCPP20StatusNotificationResponse,
  OCPPVersion,
  RegistrationStatusEnumType,
  type ResponseHandler
} from '../../../types/index.js'
import { logger } from '../../../utils/index.js'
import { OCPPResponseService } from '../OCPPResponseService.js'

const moduleName = 'OCPP20ResponseService'

export class OCPP20ResponseService extends OCPPResponseService {
  public jsonSchemasIncomingRequestResponseValidateFunction: Map<
  OCPP20IncomingRequestCommand,
  ValidateFunction<JsonType>
  >

  protected jsonSchemasValidateFunction: Map<OCPP20RequestCommand, ValidateFunction<JsonType>>
  private readonly responseHandlers: Map<OCPP20RequestCommand, ResponseHandler>

  public constructor () {
    // if (new.target.name === moduleName) {
    //   throw new TypeError(`Cannot construct ${new.target.name} instances directly`)
    // }
    super(OCPPVersion.VERSION_20)
    this.responseHandlers = new Map<OCPP20RequestCommand, ResponseHandler>([
      [
        OCPP20RequestCommand.BOOT_NOTIFICATION,
        this.handleResponseBootNotification.bind(this) as ResponseHandler
      ],
      [OCPP20RequestCommand.HEARTBEAT, this.emptyResponseHandler],
      [OCPP20RequestCommand.STATUS_NOTIFICATION, this.emptyResponseHandler]
    ])
    this.jsonSchemasValidateFunction = new Map<OCPP20RequestCommand, ValidateFunction<JsonType>>([
      [
        OCPP20RequestCommand.BOOT_NOTIFICATION,
        this.ajv
          .compile(
            OCPP20ServiceUtils.parseJsonSchemaFile<OCPP20BootNotificationResponse>(
              'assets/json-schemas/ocpp/2.0/BootNotificationResponse.json',
              moduleName,
              'constructor'
            )
          )
          .bind(this)
      ],
      [
        OCPP20RequestCommand.HEARTBEAT,
        this.ajv
          .compile(
            OCPP20ServiceUtils.parseJsonSchemaFile<OCPP20HeartbeatResponse>(
              'assets/json-schemas/ocpp/2.0/HeartbeatResponse.json',
              moduleName,
              'constructor'
            )
          )
          .bind(this)
      ],
      [
        OCPP20RequestCommand.STATUS_NOTIFICATION,
        this.ajv
          .compile(
            OCPP20ServiceUtils.parseJsonSchemaFile<OCPP20StatusNotificationResponse>(
              'assets/json-schemas/ocpp/2.0/StatusNotificationResponse.json',
              moduleName,
              'constructor'
            )
          )
          .bind(this)
      ]
    ])
    this.jsonSchemasIncomingRequestResponseValidateFunction = new Map<
    OCPP20IncomingRequestCommand,
    ValidateFunction<JsonType>
    >([
      [
        OCPP20IncomingRequestCommand.CLEAR_CACHE,
        this.ajv
          .compile(
            OCPP20ServiceUtils.parseJsonSchemaFile<OCPP20ClearCacheResponse>(
              'assets/json-schemas/ocpp/2.0/ClearCacheResponse.json',
              moduleName,
              'constructor'
            )
          )
          .bind(this)
      ]
    ])
    this.validatePayload = this.validatePayload.bind(this)
  }

  public async responseHandler<ReqType extends JsonType, ResType extends JsonType>(
    chargingStation: ChargingStation,
    commandName: OCPP20RequestCommand,
    payload: ResType,
    requestPayload: ReqType
  ): Promise<void> {
    if (chargingStation.isRegistered() || commandName === OCPP20RequestCommand.BOOT_NOTIFICATION) {
      if (
        this.responseHandlers.has(commandName) &&
        OCPP20ServiceUtils.isRequestCommandSupported(chargingStation, commandName)
      ) {
        try {
          this.validatePayload(chargingStation, commandName, payload)
          // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
          await this.responseHandlers.get(commandName)!(chargingStation, payload, requestPayload)
        } catch (error) {
          logger.error(
            `${chargingStation.logPrefix()} ${moduleName}.responseHandler: Handle response error:`,
            error
          )
          throw error
        }
      } else {
        // Throw exception
        throw new OCPPError(
          ErrorType.NOT_IMPLEMENTED,
          `${commandName} is not implemented to handle response PDU ${JSON.stringify(
            payload,
            undefined,
            2
          )}`,
          commandName,
          payload
        )
      }
    } else {
      throw new OCPPError(
        ErrorType.SECURITY_ERROR,
        `${commandName} cannot be issued to handle response PDU ${JSON.stringify(
          payload,
          undefined,
          2
        )} while the charging station is not registered on the central server.`,
        commandName,
        payload
      )
    }
  }

  private validatePayload (
    chargingStation: ChargingStation,
    commandName: OCPP20RequestCommand,
    payload: JsonType
  ): boolean {
    if (this.jsonSchemasValidateFunction.has(commandName)) {
      return this.validateResponsePayload(chargingStation, commandName, payload)
    }
    logger.warn(
      `${chargingStation.logPrefix()} ${moduleName}.validatePayload: No JSON schema validation function found for command '${commandName}' PDU validation`
    )
    return false
  }

  private handleResponseBootNotification (
    chargingStation: ChargingStation,
    payload: OCPP20BootNotificationResponse
  ): void {
    if (payload.status === RegistrationStatusEnumType.ACCEPTED) {
      addConfigurationKey(
        chargingStation,
        OCPP20OptionalVariableName.HeartbeatInterval,
        payload.interval.toString(),
        {},
        { overwrite: true, save: true }
      )
      OCPP20ServiceUtils.startHeartbeatInterval(chargingStation, payload.interval)
    }
    if (Object.values(RegistrationStatusEnumType).includes(payload.status)) {
      const logMsg = `${chargingStation.logPrefix()} Charging station in '${
        payload.status
      }' state on the central server`
      payload.status === RegistrationStatusEnumType.REJECTED
        ? logger.warn(logMsg)
        : logger.info(logMsg)
    } else {
      logger.error(
        `${chargingStation.logPrefix()} Charging station boot notification response received: %j with undefined registration status`,
        payload
      )
    }
  }
}