|
1 | 1 | /* eslint-disable @typescript-eslint/no-non-null-assertion */ |
2 | | -import {ChannelId, NetworkAdapter, PeerId} from 'automerge-repo'; |
| 2 | +import {Message, NetworkAdapter, PeerId} from '@automerge/automerge-repo'; |
3 | 3 |
|
4 | | -type NetworkEvent = { |
5 | | - senderId: PeerId; |
6 | | - targetId?: PeerId; |
7 | | - type: string; |
8 | | - channelId?: ChannelId; |
9 | | - message?: any; |
10 | | - broadcast?: boolean; |
11 | | -}; |
| 4 | +type ArriveMessage = {type: 'arrive'; senderId: PeerId; targetId: never}; |
| 5 | +type WelcomeMessage = {type: 'welcome'; senderId: PeerId; targetId: PeerId}; |
| 6 | +type BroadcastChannelMessage = ArriveMessage | WelcomeMessage | Message; |
12 | 7 |
|
13 | 8 | const adaptors: Set<AutomergeTestNetworkAdapter> = new Set(); |
14 | 9 |
|
15 | | -const sendEvent = (event: NetworkEvent) => |
16 | | - adaptors.forEach((adaptor) => adaptor.receiveEvent(event)); |
| 10 | +const broadcast = (message: BroadcastChannelMessage): void => |
| 11 | + adaptors.forEach((adaptor) => adaptor.receiveMessage(message)); |
17 | 12 |
|
18 | 13 | export const resetNetwork = () => adaptors.clear(); |
19 | 14 |
|
20 | 15 | export class AutomergeTestNetworkAdapter extends NetworkAdapter { |
21 | 16 | connect(peerId: PeerId) { |
22 | 17 | this.peerId = peerId; |
23 | 18 | adaptors.add(this); |
| 19 | + broadcast({type: 'arrive', senderId: peerId} as ArriveMessage); |
| 20 | + this.emit('ready', {network: this}); |
24 | 21 | } |
25 | 22 |
|
26 | | - receiveEvent({ |
27 | | - senderId, |
28 | | - targetId, |
29 | | - type, |
30 | | - channelId, |
31 | | - message, |
32 | | - broadcast, |
33 | | - }: NetworkEvent) { |
34 | | - if (targetId && targetId !== this.peerId && !broadcast) { |
| 23 | + receiveMessage(message: BroadcastChannelMessage) { |
| 24 | + const peerId: PeerId = this.peerId!; |
| 25 | + const {targetId, senderId, type} = message; |
| 26 | + if (targetId && targetId !== peerId) { |
35 | 27 | return; |
36 | 28 | } |
37 | 29 | switch (type) { |
38 | 30 | case 'arrive': |
39 | | - sendEvent({ |
| 31 | + broadcast({ |
40 | 32 | senderId: this.peerId!, |
41 | 33 | targetId: senderId, |
42 | 34 | type: 'welcome', |
43 | 35 | }); |
44 | | - this.announceConnection(channelId!, senderId); |
| 36 | + this.emit('peer-candidate', {peerId: senderId}); |
45 | 37 | break; |
46 | 38 | case 'welcome': |
47 | | - this.announceConnection(channelId!, senderId); |
48 | | - break; |
49 | | - case 'message': |
50 | | - this.emit('message', { |
51 | | - senderId, |
52 | | - targetId: targetId!, |
53 | | - channelId: channelId!, |
54 | | - message: new Uint8Array(message), |
55 | | - broadcast: broadcast!, |
56 | | - }); |
| 39 | + this.emit('peer-candidate', {peerId: senderId}); |
57 | 40 | break; |
58 | 41 | default: |
59 | | - throw new Error('unhandled message from network'); |
| 42 | + if (!('data' in message)) { |
| 43 | + this.emit('message', message); |
| 44 | + } else { |
| 45 | + this.emit('message', { |
| 46 | + ...message, |
| 47 | + data: new Uint8Array(message.data), |
| 48 | + }); |
| 49 | + } |
| 50 | + break; |
60 | 51 | } |
61 | 52 | } |
62 | 53 |
|
63 | | - announceConnection(channelId: ChannelId, peerId: PeerId) { |
64 | | - this.emit('peer-candidate', {peerId, channelId}); |
65 | | - } |
66 | | - |
67 | | - sendMessage( |
68 | | - peerId: PeerId, |
69 | | - channelId: ChannelId, |
70 | | - uint8message: Uint8Array, |
71 | | - broadcast: boolean, |
72 | | - ) { |
73 | | - const message = uint8message.buffer.slice( |
74 | | - uint8message.byteOffset, |
75 | | - uint8message.byteOffset + uint8message.byteLength, |
76 | | - ); |
77 | | - sendEvent({ |
78 | | - senderId: this.peerId!, |
79 | | - targetId: peerId, |
80 | | - type: 'message', |
81 | | - channelId, |
82 | | - message, |
83 | | - broadcast, |
84 | | - }); |
85 | | - } |
86 | | - |
87 | | - join(joinChannelId: ChannelId) { |
88 | | - sendEvent({ |
89 | | - senderId: this.peerId!, |
90 | | - channelId: joinChannelId, |
91 | | - type: 'arrive', |
92 | | - broadcast: true, |
93 | | - }); |
| 54 | + send(message: BroadcastChannelMessage) { |
| 55 | + if ('data' in message) { |
| 56 | + broadcast({ |
| 57 | + ...message, |
| 58 | + data: message.data.buffer.slice( |
| 59 | + message.data.byteOffset, |
| 60 | + message.data.byteOffset + message.data.byteLength, |
| 61 | + ) as any, |
| 62 | + }); |
| 63 | + } else { |
| 64 | + broadcast(message); |
| 65 | + } |
94 | 66 | } |
95 | 67 |
|
96 | | - leave() { |
| 68 | + disconnect() { |
97 | 69 | adaptors.delete(this); |
98 | 70 | } |
99 | 71 | } |
0 commit comments