booth/Dispatch.js at master · StreetStrider/booth · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

import is_plain from 'is-plain-obj'

import Events from './_/Events.js'
import Rooms  from './_/Rooms.js'

import Endpoint from './Endpoint.js'


/* * */
function noop () {}


/* * */
import { WebSocketServer } from 'ws'

function Wss (wss)
{
	if (is_plain(wss))
	{
		return new WebSocketServer(wss)
	}

	return wss
}


/* * */
export default function Dispatch (wss)
{
	wss = Wss(wss)

	var events = Events()

	var dispatch =
	{
		on,
		close,

		rooms: Rooms(),
	}

	wss.on('connection', (ws) => Endpoint(null, null, { ws, dispatch, events }))
	wss.on('listening',  ()   => events?.emit('@listening', void 0, { dispatch }))

	function on (...args)
	{
		if (events)
		{
			return events.on(...args)
		}
		else
		{
			return noop
		}
	}

	function close ()
	{
		if (! dispatch) return

		wss.close()
		wss = null

		events   = null
		dispatch = null
	}

	return dispatch
}