ATG: fix start transaction requests counting

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

diff --git a/src/charging-station/ui-server/AbstractUIServer.ts b/src/charging-station/ui-server/AbstractUIServer.ts
index 9a19d39d9a6d29550c614b937f14fd18bcb514e3..a55d0070abb8fb6b4322203f608a8ea2d1a120cf 100644 (file)
--- a/src/charging-station/ui-server/AbstractUIServer.ts
+++ b/src/charging-station/ui-server/AbstractUIServer.ts
@@ -1,9 +1,9 @@
-import type { Server as HttpServer } from 'http';
-
-import type WebSocket from 'ws';
+import type { IncomingMessage, Server } from 'http';
 
 import type { ChargingStationData } from '../../types/ChargingStationWorker';
-import type {
+import type { UIServerConfiguration } from '../../types/ConfigurationData';
+import {
+  AuthenticationType,
   ProcedureName,
   ProtocolRequest,
   ProtocolResponse,
@@ -15,10 +15,10 @@ import type AbstractUIService from './ui-services/AbstractUIService';
 
 export abstract class AbstractUIServer {
   public readonly chargingStations: Map<string, ChargingStationData>;
-  protected server: WebSocket.Server | HttpServer;
+  protected httpServer: Server;
   protected readonly uiServices: Map<ProtocolVersion, AbstractUIService>;
 
-  public constructor() {
+  public constructor(protected readonly uiServerConfiguration: UIServerConfiguration) {
     this.chargingStations = new Map<string, ChargingStationData>();
     this.uiServices = new Map<ProtocolVersion, AbstractUIService>();
   }
@@ -35,6 +35,26 @@ export abstract class AbstractUIServer {
     return [id, responsePayload];
   }
 
+  protected isBasicAuthEnabled(): boolean {
+    return (
+      this.uiServerConfiguration.authentication?.enabled === true &&
+      this.uiServerConfiguration.authentication?.type === AuthenticationType.BASIC_AUTH
+    );
+  }
+
+  protected 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 sendRequest(request: ProtocolRequest): void;