README.md: document added UI protocol features

[e-mobility-charging-stations-simulator.git] / src / charging-station / ui-server / ui-services / AbstractUIService.ts

import type { RawData } from 'ws';

import BaseError from '../../../exception/BaseError';
import { Bootstrap } from '../../../internal';
import type { JsonType } from '../../../types/JsonType';
import {
  ProcedureName,
  ProtocolRequest,
  ProtocolRequestHandler,
  ProtocolVersion,
  RequestPayload,
  ResponsePayload,
  ResponseStatus,
} from '../../../types/UIProtocol';
import type {
  BroadcastChannelProcedureName,
  BroadcastChannelRequestPayload,
} from '../../../types/WorkerBroadcastChannel';
import logger from '../../../utils/Logger';
import Utils from '../../../utils/Utils';
import UIServiceWorkerBroadcastChannel from '../../UIServiceWorkerBroadcastChannel';
import type { AbstractUIServer } from '../AbstractUIServer';

const moduleName = 'AbstractUIService';

export default abstract class AbstractUIService {
  protected readonly requestHandlers: Map<ProcedureName, ProtocolRequestHandler>;
  private readonly version: ProtocolVersion;
  private readonly uiServer: AbstractUIServer;
  private readonly uiServiceWorkerBroadcastChannel: UIServiceWorkerBroadcastChannel;
  private readonly broadcastChannelRequests: Map<string, number>;

  constructor(uiServer: AbstractUIServer, version: ProtocolVersion) {
    this.version = version;
    this.uiServer = uiServer;
    this.requestHandlers = new Map<ProcedureName, ProtocolRequestHandler>([
      [ProcedureName.LIST_CHARGING_STATIONS, this.handleListChargingStations.bind(this)],
      [ProcedureName.START_SIMULATOR, this.handleStartSimulator.bind(this)],
      [ProcedureName.STOP_SIMULATOR, this.handleStopSimulator.bind(this)],
    ]);
    this.uiServiceWorkerBroadcastChannel = new UIServiceWorkerBroadcastChannel(this);
    this.broadcastChannelRequests = new Map<string, number>();
  }

  public async requestHandler(request: RawData | JsonType): Promise<void> {
    let messageId: string;
    let command: ProcedureName;
    let requestPayload: RequestPayload | undefined;
    let responsePayload: ResponsePayload;
    try {
      [messageId, command, requestPayload] = this.requestValidation(request);

      if (this.requestHandlers.has(command) === false) {
        throw new BaseError(
          `${command} is not implemented to handle message payload ${JSON.stringify(
            requestPayload,
            null,
            2
          )}`
        );
      }

      // Call the request handler to build the response payload
      responsePayload = await this.requestHandlers.get(command)(messageId, requestPayload);
    } catch (error) {
      // Log
      logger.error(`${this.logPrefix(moduleName, 'messageHandler')} Handle request error:`, error);
      responsePayload = {
        status: ResponseStatus.FAILURE,
        command,
        requestPayload,
        responsePayload,
        errorMessage: (error as Error).message,
        errorStack: (error as Error).stack,
      };
    }
    // Send response for payload not forwarded to broadcast channel
    if (responsePayload !== undefined) {
      this.sendResponse(messageId ?? 'error', responsePayload);
    }
  }

  public sendRequest(
    messageId: string,
    procedureName: ProcedureName,
    requestPayload: RequestPayload
  ): void {
    this.uiServer.sendRequest(
      this.uiServer.buildProtocolRequest(messageId, procedureName, requestPayload)
    );
  }

  public sendResponse(messageId: string, responsePayload: ResponsePayload): void {
    this.uiServer.sendResponse(this.uiServer.buildProtocolResponse(messageId, responsePayload));
  }

  public logPrefix(modName: string, methodName: string): string {
    return this.uiServer.logPrefix(modName, methodName, this.version);
  }

  public deleteBroadcastChannelRequest(uuid: string): void {
    this.broadcastChannelRequests.delete(uuid);
  }

  public getBroadcastChannelExpectedResponses(uuid: string): number {
    return this.broadcastChannelRequests.get(uuid) ?? 0;
  }

  protected sendBroadcastChannelRequest(
    uuid: string,
    procedureName: BroadcastChannelProcedureName,
    payload: BroadcastChannelRequestPayload
  ): void {
    if (!Utils.isEmptyArray(payload.hashIds)) {
      payload.hashIds = payload.hashIds
        .map((hashId) => {
          if (this.uiServer.chargingStations.has(hashId) === true) {
            return hashId;
          }
          logger.warn(
            `${this.logPrefix(
              moduleName,
              'sendBroadcastChannelRequest'
            )} Charging station with hashId '${hashId}' not found`
          );
        })
        .filter((hashId) => hashId !== undefined);
    }
    const expectedNumberOfResponses = !Utils.isEmptyArray(payload.hashIds)
      ? payload.hashIds.length
      : this.uiServer.chargingStations.size;
    this.uiServiceWorkerBroadcastChannel.sendRequest([uuid, procedureName, payload]);
    this.broadcastChannelRequests.set(uuid, expectedNumberOfResponses);
  }

  // Validate the raw data received from the UI server
  private requestValidation(rawData: RawData | JsonType): ProtocolRequest {
    // logger.debug(
    //   `${this.logPrefix(
    //     moduleName,
    //     'requestValidation'
    //   )} Data received in string format: ${rawData.toString()}`
    // );

    const data = JSON.parse(rawData.toString()) as JsonType[];

    if (Array.isArray(data) === false) {
      throw new BaseError('UI protocol request is not an array');
    }

    if (data.length !== 3) {
      throw new BaseError('UI protocol request is malformed');
    }

    return data as ProtocolRequest;
  }

  private handleListChargingStations(): ResponsePayload {
    // TODO: remove cast to unknown
    return {
      status: ResponseStatus.SUCCESS,
      ...[...this.uiServer.chargingStations.values()],
    } as unknown as ResponsePayload;
  }

  private async handleStartSimulator(): Promise<ResponsePayload> {
    await Bootstrap.getInstance().start();
    return { status: ResponseStatus.SUCCESS };
  }

  private async handleStopSimulator(): Promise<ResponsePayload> {
    await Bootstrap.getInstance().stop();
    return { status: ResponseStatus.SUCCESS };
  }
}