Commit | Line | Data |
---|---|---|
17bc43d7 | 1 | import type { AbstractUIService } from './internal'; |
268a74bb JB |
2 | import { WorkerBroadcastChannel } from './WorkerBroadcastChannel'; |
3 | import { | |
4 | type BroadcastChannelResponse, | |
5 | type BroadcastChannelResponsePayload, | |
6 | type MessageEvent, | |
7 | type ResponsePayload, | |
8 | ResponseStatus, | |
9 | } from '../types'; | |
10 | import { logger } from '../utils/Logger'; | |
6c8f5d90 JB |
11 | |
12 | const moduleName = 'UIServiceWorkerBroadcastChannel'; | |
13 | ||
0d2cec76 JB |
14 | type Responses = { |
15 | responsesExpected: number; | |
16 | responsesReceived: number; | |
17 | responses: BroadcastChannelResponsePayload[]; | |
18 | }; | |
19 | ||
268a74bb | 20 | export class UIServiceWorkerBroadcastChannel extends WorkerBroadcastChannel { |
0d2cec76 | 21 | private readonly uiService: AbstractUIService; |
53870cff | 22 | private readonly responses: Map<string, Responses>; |
6c8f5d90 JB |
23 | |
24 | constructor(uiService: AbstractUIService) { | |
25 | super(); | |
26 | this.uiService = uiService; | |
27 | this.onmessage = this.responseHandler.bind(this) as (message: MessageEvent) => void; | |
28 | this.onmessageerror = this.messageErrorHandler.bind(this) as (message: MessageEvent) => void; | |
0d2cec76 | 29 | this.responses = new Map<string, Responses>(); |
6c8f5d90 JB |
30 | } |
31 | ||
32 | private responseHandler(messageEvent: MessageEvent): void { | |
5dea4c94 JB |
33 | const validatedMessageEvent = this.validateMessageEvent(messageEvent); |
34 | if (validatedMessageEvent === false) { | |
6c8f5d90 JB |
35 | return; |
36 | } | |
5dea4c94 JB |
37 | if (this.isRequest(validatedMessageEvent.data) === true) { |
38 | return; | |
39 | } | |
40 | const [uuid, responsePayload] = validatedMessageEvent.data as BroadcastChannelResponse; | |
0d2cec76 JB |
41 | if (this.responses.has(uuid) === false) { |
42 | this.responses.set(uuid, { | |
43 | responsesExpected: this.uiService.getBroadcastChannelExpectedResponses(uuid), | |
44 | responsesReceived: 1, | |
45 | responses: [responsePayload], | |
46 | }); | |
47 | } else if ( | |
6d9876e7 | 48 | this.responses.get(uuid)?.responsesReceived <= this.responses.get(uuid)?.responsesExpected |
0d2cec76 JB |
49 | ) { |
50 | this.responses.get(uuid).responsesReceived++; | |
72092cfc | 51 | this.responses.get(uuid)?.responses.push(responsePayload); |
6d9876e7 JB |
52 | } |
53 | if ( | |
54 | this.responses.get(uuid)?.responsesReceived === this.responses.get(uuid)?.responsesExpected | |
0d2cec76 | 55 | ) { |
0d2cec76 JB |
56 | this.uiService.sendResponse(uuid, this.buildResponsePayload(uuid)); |
57 | this.responses.delete(uuid); | |
58 | this.uiService.deleteBroadcastChannelRequest(uuid); | |
59 | } | |
60 | } | |
6c8f5d90 | 61 | |
0d2cec76 | 62 | private buildResponsePayload(uuid: string): ResponsePayload { |
c56450e3 JB |
63 | const responsesStatus = |
64 | this.responses | |
65 | .get(uuid) | |
66 | ?.responses.every(({ status }) => status === ResponseStatus.SUCCESS) === true | |
67 | ? ResponseStatus.SUCCESS | |
68 | : ResponseStatus.FAILURE; | |
0d2cec76 JB |
69 | return { |
70 | status: responsesStatus, | |
71 | hashIdsSucceeded: this.responses | |
72 | .get(uuid) | |
73 | ?.responses.map(({ status, hashId }) => { | |
74 | if (status === ResponseStatus.SUCCESS) { | |
75 | return hashId; | |
76 | } | |
77 | }) | |
72092cfc | 78 | .filter((hashId) => hashId !== undefined), |
0d2cec76 JB |
79 | ...(responsesStatus === ResponseStatus.FAILURE && { |
80 | hashIdsFailed: this.responses | |
81 | .get(uuid) | |
82 | ?.responses.map(({ status, hashId }) => { | |
83 | if (status === ResponseStatus.FAILURE) { | |
84 | return hashId; | |
85 | } | |
86 | }) | |
72092cfc | 87 | .filter((hashId) => hashId !== undefined), |
0d2cec76 | 88 | }), |
f53c88d0 JB |
89 | ...(responsesStatus === ResponseStatus.FAILURE && { |
90 | responsesFailed: this.responses | |
91 | .get(uuid) | |
72092cfc | 92 | ?.responses.map((response) => { |
f53c88d0 JB |
93 | if (response.status === ResponseStatus.FAILURE) { |
94 | return response; | |
95 | } | |
96 | }) | |
72092cfc | 97 | .filter((response) => response !== undefined), |
f53c88d0 | 98 | }), |
0d2cec76 | 99 | }; |
6c8f5d90 JB |
100 | } |
101 | ||
102 | private messageErrorHandler(messageEvent: MessageEvent): void { | |
103 | logger.error( | |
104 | `${this.uiService.logPrefix(moduleName, 'messageErrorHandler')} Error at handling message:`, | |
5083d31a | 105 | messageEvent |
6c8f5d90 JB |
106 | ); |
107 | } | |
108 | } |