angelia.io 4.0.91 → 5.0.0

package/client/index.js

@@ -1,201 +1,524 @@
-import { ClientWebWorker } from './worker.js'
-
-const isFunction = v => typeof v === 'function'
+import { workerTransport } from './transport.js'
+import {
+	DEFAULT_PATH,
+	NO_RECONNECT,
+	PACKED_PARAM,
+	PROTOCOL_KEY,
+	assertCloseReason,
+} from '../shared/protocol.js'
 
 const error = console.error
 
+// Rejection messages for replies that can no longer arrive.
+const RECONNECTING =
+	'reconnecting: the connection dropped before the server replied'
+const DISCONNECTED =
+	'disconnected: disconnect(true) was called, so nothing more is sent or answered'
+const EMPTY_KEY = "[angelia.io] the empty key '' is reserved for replies"
+const stringify = JSON.stringify
+const parse = JSON.parse
+const isArray = Array.isArray
+
+/** @returns {v is Function} */
+const isFunction = (/** @type {any} */ v) => typeof v === 'function'
+
+// Pack queued messages into the connect URL while the handshake's
+// request line stays within nginx's default `large_client_header_buffers`
+// (8k): a longer line gets a 414.
+const REQUEST_LINE_LIMIT = 8192
+
+/** Length of `text` once form-encoded into a query param value. */
+const encodedLength = text =>
+	new URLSearchParams({ [PACKED_PARAM]: text }).toString().length -
+	PACKED_PARAM.length -
+	1
+
+// "Going away": a server shutting down, or the page unloading. Firefox
+// closes the page's sockets with it during a reload, before `pagehide`.
+const GOING_AWAY = 1001
+
+// Retries wait RETRY_MIN_MS plus up to RETRY_JITTER_MS, so a server
+// bounce doesn't get every client back at the same instant.
+const RETRY_MIN_MS = 2000
+const RETRY_JITTER_MS = 2000
+
+/** `request()`'s callback: resolve with the reply as it is. */
+const settle = reply => reply
+
+const noop = () => {}
+
+const newConnid = () =>
+	crypto.getRandomValues(new BigUint64Array(1))[0].toString(36)
+
+/** @import { ClientEvents, ClientReplies, EventKey, EventValue, ServerEvents, ServerReplies } from '../server/index.js' */
+
+/**
+ * @typedef {object} ClientOptions
+ * @property {string} url Server URL, `ws://` or `wss://`. Without a path
+ *   the client connects to the default one; give it to match the
+ *   server's `path` option.
+ * @property {Record<string, any> | (() => Record<string, any>)} [params]
+ *   Query params sent on connect; a function is called on every connect.
+ * @property {boolean} [noConnect] Don't connect until `connect()`.
+ */
+
+/**
+ * Events the client fires itself, with no payload.
+ *
+ * @typedef {'connect' | 'reconnect' | 'disconnect'} ClientPredefinedEvent
+ */
+
+/** @typedef {{ signal?: AbortSignal }} ListenOptions */
+
+/**
+ * A connection to an angelia.io server, run from a Web Worker. Connects
+ * on construction and reconnects on its own.
+ */
 export default class Client {
-	#callbacks
-	#listeners
-	#messages
-	#postMessage
+	#connected = false
 
 	/**
-	 * Creates a Socket Client Instance
+	 * Replies the server still owes, by callback id. `fire(args)` runs
+	 * the user's callback and resolves their promise; `cancel(err)`
+	 * rejects it (on reconnect, when the reply can no longer arrive).
 	 *
-	 * @param options {string | {url:string, params?:Function,
-	 *   noConnect?:boolean }}
+	 * @type {Map<
+	 * 	number,
+	 * 	{ fire: (args: any[]) => void; cancel: (err: Error) => void }
+	 * >}
 	 */
+	#callbacks = new Map()
+	/** Last callback id handed out; ids start at 1, as 0 means "none". */
+	#lastCallbackId = 0
+	// No prototype, so only registered keys are found ('constructor',
+	// '__proto__', … are ordinary event names).
+	/** @type {Record<string, Function[]>} */
+	#listeners = Object.create(null)
+	/** @type {any[]} */
+	#messages = []
+	/** @type {import('./transport.js').Connection | null} */
+	#io = null
+	/** @type {import('./transport.js').Transport} */
+	#transport
+	#connid = newConnid()
+	#isReconnect = false
+	#reconnect = true
+	#packFailed = false
+
+	/** @param {ClientOptions} options */
 	constructor(options) {
-		if (typeof options === 'string') {
-			options = {
-				url: options,
-			}
+		const { protocol } = new URL(options.url)
+		if (protocol !== 'ws:' && protocol !== 'wss:') {
+			throw new TypeError(
+				`Client url must be ws:// or wss://, got ${protocol}//. Give the server's WebSocket address, e.g. ws://localhost:3001.`,
+			)
+		}
+
+		if (typeof Worker === 'undefined') {
+			throw new TypeError(
+				'angelia.io/client runs its socket in a Web Worker; this environment has none. The client needs a browser.',
+			)
 		}
 
 		this.options = options
+		this.#transport = workerTransport()
 
-		this.#callbacks = [null]
-		this.#listeners = {
-			'': [this.#oncallback],
-		}
-		this.#messages = []
+		// Protocol messages (replies, commands) go through the same
+		// dispatch loop as events, under the reserved key.
+		this.#listeners[PROTOCOL_KEY] = [this.#onprotocol]
 
-		this.connected = false
+		// Deferred so emits made right after construction are packed
+		// into the connect URL.
+		if (!options.noConnect) queueMicrotask(() => this.connect())
+	}
 
-		// to force disconnect
-		this.on('disconnect', this.disconnect)
+	/** Open the socket; no-op while open or after `disconnect(true)`. */
+	connect() {
+		if (this.#io || !this.#reconnect) return
 
-		const workerURL = URL.createObjectURL(
-			new Blob(['(' + ClientWebWorker + ')()']),
-		)
-		const io = new Worker(workerURL)
-		URL.revokeObjectURL(workerURL)
+		const url = new URL(this.options.url)
+		if (url.pathname === '/') url.pathname = DEFAULT_PATH
 
-		io.onmessage = this.#onworkermessage
-		this.#postMessage = io.postMessage.bind(io)
+		const { searchParams } = url
+		const p = this.options.params
+		const params = (isFunction(p) ? p() : p) || {}
 
-		options.noConnect === undefined && this.connect()
-	}
+		for (const [k, v] of Object.entries(params)) {
+			searchParams.append(k, v)
+		}
+		searchParams.append('connid', this.#connid)
 
-	/** Connect to server */
-	connect() {
-		this.#postMessage([
-			'connect',
-			{
-				url: this.options.url,
-				params:
-					(isFunction(this.options.params)
-						? this.options.params()
-						: this.options.params) || {},
+		/** @type {any[]} */
+		let packed = []
+		if (this.#isReconnect) {
+			this.#messages = []
+		} else {
+			packed = this.#pack(url)
+		}
+
+		/** @type {number | null} When this connection opened. */
+		let openedAt = null
+		// Every handler checks it still belongs to the live connection,
+		// so a late event can't clobber a newer one.
+		const io = this.#transport.open(url, {
+			onopen: () => {
+				if (this.#io !== io) return
+				openedAt = Date.now()
+				this.#connected = true
+				if (this.#isReconnect) {
+					this.#dispatch([['reconnect', undefined]])
+				} else {
+					this.#isReconnect = true
+					this.#dispatch([['connect', undefined]])
+				}
+				this.#send()
+			},
+			onmessage: text => {
+				if (this.#io === io) this.#dispatch(parse(text))
 			},
-		])
+			onclose: code => {
+				if (this.#io !== io) return
+				// The URL never reached a server that accepted it: put the
+				// packed messages back in front, and stop packing, since the
+				// URL itself may be what got rejected.
+				if (openedAt === null && packed.length) {
+					this.#messages = [...packed, ...this.#messages]
+					this.#packFailed = true
+				}
+				this.#handleClosed(code, openedAt)
+			},
+			onerror: err =>
+				error('[angelia.io] client socket error. The client keeps going:', err),
+		})
+		this.#io = io
+	}
+
+	/**
+	 * @param {number} code Close code
+	 * @param {number | null} openedAt When the connection opened, if it did
+	 */
+	#handleClosed(code, openedAt) {
+		this.#io = null
+		const uptime = openedAt === null ? 0 : Date.now() - openedAt
+		// A reply only travels on the connection that asked, and a reconnect
+		// drops the queue, so none can come now. A connection that never
+		// opened sent nothing: its messages and callbacks wait for the next
+		// attempt.
+		if (openedAt !== null && this.#reconnect) this.#cancelPending(RECONNECTING)
+		if (this.#connected) {
+			this.#connected = false
+			this.#dispatch([['disconnect', undefined]])
+		}
+		if (!this.#reconnect) {
+			this.#shutdown()
+		} else if (uptime >= RETRY_MIN_MS && code !== GOING_AWAY) {
+			// A working connection dropped: get back right away.
+			this.connect()
+		} else {
+			// A failed attempt, a connection dropped as soon as it opened
+			// (which would otherwise loop as fast as it can), or "going
+			// away": an unloading page is gone before the delay ends, so it
+			// doesn't open a second connection, while a server that went
+			// away still gets a retry.
+			// `unref?.()` keeps a leaked Client from pinning a Node
+			// process alive; browsers return a number and skip it.
+			setTimeout(
+				() => this.connect(),
+				RETRY_MIN_MS + Math.random() * RETRY_JITTER_MS,
+			)?.unref?.()
+		}
+	}
+
+	/** Whether the socket is open. */
+	get connected() {
+		return this.#connected
+	}
+
+	/** `using client = new Client(…)` disconnects for good at scope end. */
+	[Symbol.dispose]() {
+		this.disconnect(true)
 	}
 
 	/**
-	 * Disconnect from server
+	 * Close the socket.
 	 *
-	 * @param {boolean} noReconnect - To prevent reconnections
+	 * @param {boolean} [noReconnect] `true` stops the auto-reconnect.
+	 * @param {string} [reason] Close reason sent to the server, up to 123
+	 *   bytes; longer throws a `RangeError` before anything closes.
 	 */
-	disconnect = noReconnect => {
-		this.#postMessage(['disconnect', noReconnect])
+	disconnect(noReconnect, reason = 'client disconnect') {
+		assertCloseReason(reason)
+		if (noReconnect) this.#reconnect = false
+		if (this.#io) this.#io.close(1000, reason)
+		// Between retries there's no connection whose close would clean up.
+		else if (noReconnect) this.#shutdown()
 	}
 
 	/**
-	 * Listen for data sent by the server
+	 * Emit to the server. With a callback, returns a promise for the
+	 * callback's result, rejected with a `reconnecting:` Error if the
+	 * connection drops before the reply arrives, or a `disconnected:` Error
+	 * after `disconnect(true)`.
 	 *
-	 * @param {string} key
-	 * @param {any | Function} [valueOrCallback]
+	 * @template {EventKey<ServerEvents>} K
+	 * @overload
+	 * @param {K} key
+	 * @returns {undefined}
+	 */
+	/**
+	 * @template {EventKey<ServerEvents>} K
+	 * @overload
+	 * @param {K} key
+	 * @param {EventValue<ServerEvents, K>} value
+	 * @returns {undefined}
+	 */
+	/**
+	 * @template {EventKey<ServerEvents>} K
+	 * @template R
+	 * @overload
+	 * @param {K} key
+	 * @param {EventValue<ServerEvents, K>} value
+	 * @param {(reply: EventValue<ServerReplies, K>) => R} callback
+	 * @returns {Promise<R>}
+	 */
+	/**
+	 * @param {any} key
+	 * @param {any} [value]
 	 * @param {Function} [callback]
-	 * @returns {Promise<any> | undefined} When callback is provided it
-	 *   returns a promise
+	 * @returns {Promise<any> | undefined}
 	 */
-	emit(key, valueOrCallback, callback) {
-		// empty `key` is reserved
-		if (key == '') {
-			error('`key` cannot be empty')
-		} else {
-			if (!this.#messages.length) {
-				queueMicrotask(() => queueMicrotask(this.#send))
-			}
-			if (callback) {
-				const [i, promise] = this.#callback(callback)
-				this.#messages.push([key, valueOrCallback, i])
-				return promise
-			} else if (isFunction(valueOrCallback)) {
-				const [i, promise] = this.#callback(valueOrCallback)
-				this.#messages.push([key, 0, i])
-				return promise
-			} else if (valueOrCallback !== undefined) {
-				this.#messages.push([key, valueOrCallback])
-			} else {
-				this.#messages.push([key])
-			}
+	emit(key, value, callback) {
+		if (key === PROTOCOL_KEY) {
+			error(`${EMPTY_KEY}; this message was not sent.`)
+			return
+		}
+		// After `disconnect(true)` nothing will be sent: don't queue it.
+		if (!this.#reconnect) {
+			return callback ? Promise.reject(new Error(DISCONNECTED)) : undefined
+		}
+
+		if (callback) {
+			const [id, promise] = this.#callback(callback)
+			this.#queue([key, value, id])
+			return promise
 		}
+		this.#queue(value === undefined ? [key] : [key, value])
+	}
+
+	/**
+	 * Emit and resolve with the server's reply.
+	 *
+	 * @template {EventKey<ServerEvents>} K
+	 * @overload
+	 * @param {K} key
+	 * @returns {Promise<EventValue<ServerReplies, K>>}
+	 */
+	/**
+	 * @template {EventKey<ServerEvents>} K
+	 * @overload
+	 * @param {K} key
+	 * @param {EventValue<ServerEvents, K>} value
+	 * @returns {Promise<EventValue<ServerReplies, K>>}
+	 */
+	/**
+	 * @param {any} key
+	 * @param {any} [value]
+	 * @returns {Promise<any>}
+	 */
+	request(key, value) {
+		return this.emit(key, value, settle)
 	}
 
 	/**
-	 * Listen for data sent by the server
+	 * Listen for a predefined event. Returns a function that stops
+	 * listening; aborting `options.signal` does the same.
 	 *
+	 * @overload
+	 * @param {ClientPredefinedEvent} key
+	 * @param {() => void} value
+	 * @param {ListenOptions} [options]
+	 * @returns {() => void}
+	 */
+	/**
+	 * Listen for a server event. Returns a function that stops
+	 * listening; aborting `options.signal` does the same.
+	 *
+	 * @template {EventKey<ClientEvents>} K
+	 * @overload
+	 * @param {K} key
+	 * @param {(
+	 * 	v: EventValue<ClientEvents, K>,
+	 * 	reply?: (v: EventValue<ClientReplies, K>) => void,
+	 * ) => void} value
+	 * @param {ListenOptions} [options]
+	 * @returns {() => void}
+	 */
+	/**
 	 * @param {string} key
 	 * @param {Function} value
-	 * @returns () => void | undefined - `off` function to remove the
-	 *   listener
+	 * @param {ListenOptions} [options]
+	 * @returns {() => void}
 	 */
-	on(key, value) {
-		// empty `key` is reserved
-		if (key == '') {
-			error('`key` cannot be empty')
-		} else {
-			this.#listeners[key] = this.#listeners[key] || []
-			this.#listeners[key].push(value)
-			return () => this.#off(key, value)
+	on(key, value, options) {
+		if (key === PROTOCOL_KEY) {
+			error(`${EMPTY_KEY}; this listener was not added.`)
+			return noop
+		}
+		const signal = options?.signal
+		if (signal?.aborted) return noop
+		// Copy on write: a dispatch already walking the old array is unaffected.
+		this.#listeners[key] = [...(this.#listeners[key] || []), value]
+		const off = () => {
+			this.#off(key, value)
+			signal?.removeEventListener('abort', off)
 		}
+		signal?.addEventListener('abort', off, { once: true })
+		return off
 	}
 
 	// PRIVATE API
 
-	#off(k, v) {
-		const i = this.#listeners[k].indexOf(v)
-		if (i !== -1) {
-			this.#listeners[k].splice(i, 1)
+	/**
+	 * Move the longest prefix of the queue that fits the request-line
+	 * budget into the connect URL, so it dispatches on connect; the rest
+	 * is sent once the socket opens. Stops at the first message that
+	 * doesn't fit, to keep order. Off after a failed attempt that carried
+	 * packed messages. Returns the packed messages.
+	 *
+	 * @param {URL} url
+	 */
+	#pack(url) {
+		if (this.#packFailed) return []
+		const messages = this.#messages
+		// Form-encoding works per character and yields ASCII, so encoded
+		// lengths add up and equal bytes. `search` always holds `connid`.
+		let length =
+			'GET '.length +
+			url.pathname.length +
+			url.search.length +
+			' HTTP/1.1\r\n'.length +
+			`&${PACKED_PARAM}=`.length +
+			encodedLength('[]')
+		const separator = encodedLength(',')
+		/** @type {string[]} */
+		const parts = []
+		for (const message of messages) {
+			const part = stringify(message)
+			const added =
+				encodedLength(part) + (parts.length ? separator : 0)
+			if (length + added > REQUEST_LINE_LIMIT) break
+			length += added
+			parts.push(part)
 		}
+		if (!parts.length) return []
+
+		url.searchParams.append(PACKED_PARAM, '[' + parts.join(',') + ']')
+		this.#messages = messages.slice(parts.length)
+		return messages.slice(0, parts.length)
+	}
+
+	/** After `disconnect(true)`: nothing will be sent or answered again. */
+	#shutdown() {
+		this.#transport.terminate()
+		this.#messages = []
+		this.#cancelPending(DISCONNECTED)
+	}
+
+	/** @param {string} message */
+	#cancelPending(message) {
+		if (!this.#callbacks.size) return
+		const error = new Error(message)
+		for (const pending of this.#callbacks.values()) pending.cancel(error)
+		this.#callbacks.clear()
+	}
+
+	/** Push a message, scheduling a flush when the queue was empty. */
+	#queue(message) {
+		// After `disconnect(true)` nothing will be sent (replies included).
+		if (!this.#reconnect) return
+		if (!this.#messages.length) queueMicrotask(this.#flushSoon)
+		this.#messages.push(message)
+	}
+
+	// Two microtasks deep, so emits made by code reacting to this one
+	// (e.g. in a `.then`) join the same batch. Built once, not per flush.
+	#flushSoon = () => queueMicrotask(this.#send)
+
+	#off(k, v) {
+		// Copy on write, like `on`: removing a listener mid-dispatch must not
+		// shift the array the dispatch loop is walking.
+		this.#listeners[k] = this.#listeners[k].filter(fn => fn !== v)
 	}
 
 	/** @returns {[number, Promise<any>]} */
 	#callback(c) {
-		let resolve
-		const promise = new Promise(r => (resolve = r))
-		const i = this.#callbacks.length
-		this.#callbacks[i] = args => resolve(c(...args))
-		return [i, promise]
+		const { promise, resolve, reject } = Promise.withResolvers()
+		const id = ++this.#lastCallbackId
+		this.#callbacks.set(id, {
+			fire: args => {
+				// A throwing callback rejects its promise instead of leaving it pending.
+				try {
+					resolve(c(...args))
+				} catch (err) {
+					reject(err)
+				}
+			},
+			cancel: reject,
+		})
+		return [id, promise]
 	}
-	#oncallback = d => {
-		this.#callbacks[d[0]](d[1])
-		this.#callbacks[d[0]] = null
+
+	/** A reply `[id, args]` or a command from the server. */
+	#onprotocol = d => {
+		if (d === NO_RECONNECT) {
+			this.disconnect(true)
+			return
+		}
+		const pending = isArray(d) && this.#callbacks.get(d[0])
+		if (pending) {
+			this.#callbacks.delete(d[0])
+			pending.fire(d[1])
+		}
 	}
 
 	#dispatch(d) {
 		for (const e of d) {
-			if (this.#listeners[e[0]]) {
-				for (const fn of this.#listeners[e[0]]) {
-					fn(
-						e[1],
-						e[2]
-							? (...d) => {
-									this.emit('', [e[2], d])
-								}
-							: undefined,
-					)
+			const listeners = this.#listeners[e[0]]
+			if (!listeners) continue
+			let replied = false
+			const reply = e[2]
+				? (...args) => {
+						if (replied) {
+							return error(
+								`[angelia.io] reply() already called for "${e[0]}": a message gets one reply, so this one was dropped.`,
+							)
+						}
+						replied = true
+
+						this.#queue([PROTOCOL_KEY, [e[2], args]])
+					}
+				: undefined
+			for (const fn of listeners) {
+				try {
+					fn(e[1], reply)
+				} catch (err) {
+					// Rethrown on its own so it still surfaces as uncaught,
+					// without skipping the rest of the batch.
+					queueMicrotask(() => {
+						throw err
+					})
 				}
 			}
 		}
 	}
 
 	#send = () => {
-		if (this.#messages.length) {
-			const messages = this.#messages
+		const io = this.#io
+		if (io && this.#connected && this.#messages.length) {
+			const batch = this.#messages
+			// A fresh array is cheaper for V8 than clearing this one.
 			this.#messages = []
-			this.#postMessage(['emit', messages])
-		}
-	}
-
-	#onworkermessage = e => {
-		switch (e.data[0]) {
-			case 'messages': {
-				this.#dispatch(e.data[1])
-				break
-			}
-			case 'dispatch': {
-				this.#dispatch([[e.data[1], e.data[2]]])
-				break
-			}
-			case 'connect': {
-				this.connect()
-				break
-			}
-			case 'connected': {
-				this.connected = e.data[1]
-				break
-			}
-		}
-	}
-
-	decode(s) {
-		if (s.b) {
-			return fetch(s.b).then(r => r.blob())
-		} else if (s.a) {
-			return fetch(s.a).then(r => r.arrayBuffer())
+			io.send(stringify(batch))
 		}
 	}
 }