-import AbstractUIService from './ui-services/AbstractUIService';
-import { Server as HttpServer } from 'http';
-import { ProtocolVersion } from '../../types/UIProtocol';
-import { Server as WSServer } from 'ws';
+import { type IncomingMessage, Server, type ServerResponse } from 'node:http';
+
+import type { WebSocket } from 'ws';
+
+import type { AbstractUIService } from './ui-services/AbstractUIService';
+import { UIServiceFactory } from './ui-services/UIServiceFactory';
+import { BaseError } from '../../exception';
+import {
+ AuthenticationType,
+ type ChargingStationData,
+ type ProcedureName,
+ type ProtocolRequest,
+ type ProtocolResponse,
+ ProtocolVersion,
+ type RequestPayload,
+ type ResponsePayload,
+ type UIServerConfiguration,
+} from '../../types';
export abstract class AbstractUIServer {
- public readonly chargingStations: Set<string>;
+ public readonly chargingStations: Map<string, ChargingStationData>;
+ protected readonly httpServer: Server;
+ protected readonly responseHandlers: Map<string, ServerResponse | WebSocket>;
protected readonly uiServices: Map<ProtocolVersion, AbstractUIService>;
- protected server: WSServer | HttpServer;
- public constructor() {
- this.chargingStations = new Set<string>();
+ public constructor(protected readonly uiServerConfiguration: UIServerConfiguration) {
+ this.chargingStations = new Map<string, ChargingStationData>();
+ this.httpServer = new Server();
+ this.responseHandlers = new Map<string, ServerResponse | WebSocket>();
this.uiServices = new Map<ProtocolVersion, AbstractUIService>();
}
+ public buildProtocolRequest(
+ id: string,
+ procedureName: ProcedureName,
+ requestPayload: RequestPayload,
+ ): ProtocolRequest {
+ return [id, procedureName, requestPayload];
+ }
+
+ public buildProtocolResponse(id: string, responsePayload: ResponsePayload): ProtocolResponse {
+ return [id, responsePayload];
+ }
+
+ public stop(): void {
+ this.chargingStations.clear();
+ }
+
+ public async sendInternalRequest(request: ProtocolRequest): Promise<ProtocolResponse> {
+ const protocolVersion = ProtocolVersion['0.0.1'];
+ this.registerProtocolVersionUIService(protocolVersion);
+ return this.uiServices
+ .get(protocolVersion)
+ ?.requestHandler(request) as Promise<ProtocolResponse>;
+ }
+
+ public hasResponseHandler(id: string): boolean {
+ return this.responseHandlers.has(id);
+ }
+
+ protected startHttpServer(): void {
+ if (this.httpServer.listening === false) {
+ this.httpServer.listen(this.uiServerConfiguration.options);
+ }
+ }
+
+ protected registerProtocolVersionUIService(version: ProtocolVersion): void {
+ if (this.uiServices.has(version) === false) {
+ this.uiServices.set(version, UIServiceFactory.getUIServiceImplementation(version, this));
+ }
+ }
+
+ protected authenticate(req: IncomingMessage, next: (err?: Error) => void): void {
+ if (this.isBasicAuthEnabled() === true) {
+ if (this.isValidBasicAuth(req) === false) {
+ next(new BaseError('Unauthorized'));
+ }
+ next();
+ }
+ next();
+ }
+
+ private isBasicAuthEnabled(): boolean {
+ return (
+ this.uiServerConfiguration.authentication?.enabled === true &&
+ this.uiServerConfiguration.authentication?.type === AuthenticationType.BASIC_AUTH
+ );
+ }
+
+ private isValidBasicAuth(req: IncomingMessage): boolean {
+ const authorizationHeader = req.headers.authorization ?? '';
+ const authorizationToken = authorizationHeader.split(/\s+/).pop() ?? '';
+ const authentication = Buffer.from(authorizationToken, 'base64').toString();
+ const authenticationParts = authentication.split(/:/);
+ const username = authenticationParts.shift();
+ const password = authenticationParts.join(':');
+ return (
+ this.uiServerConfiguration.authentication?.username === username &&
+ this.uiServerConfiguration.authentication?.password === password
+ );
+ }
+
public abstract start(): void;
- public abstract stop(): void;
- public abstract sendResponse(message: string): void;
- public abstract logPrefix(): string;
+ public abstract sendRequest(request: ProtocolRequest): void;
+ public abstract sendResponse(response: ProtocolResponse): void;
+ public abstract logPrefix(
+ moduleName?: string,
+ methodName?: string,
+ prefixSuffix?: string,
+ ): string;
}