96e9a96896be609ff4a09760c33900886e3aa473
1 import { Protocol
, ProtocolCommand
, ProtocolRequest
, ProtocolVersion
} from
'../types/UIProtocol';
2 import WebSocket
, { OPEN
, Server
, ServerOptions
} from
'ws';
4 import AbstractUIService from
'./ui-websocket-services/AbstractUIService';
5 import BaseError from
'../exception/BaseError';
6 import Configuration from
'../utils/Configuration';
7 import { IncomingMessage
} from
'http';
8 import UIServiceFactory from
'./ui-websocket-services/UIServiceFactory';
9 import Utils from
'../utils/Utils';
10 import logger from
'../utils/Logger';
12 export default class UIWebSocketServer
extends Server
{
13 public readonly chargingStations
: Set
<string>;
14 public readonly uiServices
: Map
<ProtocolVersion
, AbstractUIService
>;
16 public constructor(options
?: ServerOptions
, callback
?: () => void) {
17 // Create the WebSocket Server
18 super(options
?? Configuration
.getUIWebSocketServer().options
, callback
);
19 this.chargingStations
= new Set
<string>();
20 this.uiServices
= new Map
<ProtocolVersion
, AbstractUIService
>();
21 for (const version
of Object.values(ProtocolVersion
)) {
22 this.uiServices
.set(version
, UIServiceFactory
.getUIServiceImplementation(version
, this));
26 public broadcastToClients(message
: string): void {
27 for (const client
of this.clients
) {
28 if (client
?.readyState
=== OPEN
) {
34 public start(): void {
35 this.on('connection', (socket
: WebSocket
, request
: IncomingMessage
): void => {
36 const protocolIndex
= socket
.protocol
.indexOf(Protocol
.UI
);
37 const version
= socket
.protocol
.substring(protocolIndex
+ Protocol
.UI
.length
) as ProtocolVersion
;
38 if (!this.uiServices
.has(version
)) {
39 throw new BaseError(`Could not find a UI service implementation for UI protocol version ${version}`);
41 // FIXME: check connection validity
42 socket
.on('message', (messageData
) => {
43 let [command
, payload
]: ProtocolRequest
= [ProtocolCommand
.UNKNOWN
, {}];
44 const protocolRequest
= JSON
.parse(messageData
.toString()) as ProtocolRequest
;
45 if (Utils
.isIterable(protocolRequest
)) {
46 [command
, payload
] = protocolRequest
;
48 throw new BaseError('UI protocol request is not iterable');
50 this.uiServices
.get(version
).handleMessage(command
, payload
).catch(() => {
51 logger
.error(`${this.logPrefix()} Error while handling command %s message: %j`, command
, payload
);
54 socket
.on('error', (error
) => {
55 logger
.error(`${this.logPrefix()} Error on WebSocket: %j`, error
);
64 public logPrefix(): string {
65 return Utils
.logPrefix(' UI WebSocket Server:');