Jon Staab
29 lines
641 B
TypeScript
29 lines
641 B
TypeScript
import {Emitter} from '@welshman/lib'
|
|
import type {Message} from '@welshman/util'
|
|
import type {Connection} from '../Connection'
|
|
|
|
export class Relay extends Emitter {
|
|
constructor(readonly connection: Connection) {
|
|
super()
|
|
|
|
this.connection.on('receive', this.onMessage)
|
|
}
|
|
|
|
get connections() {
|
|
return [this.connection]
|
|
}
|
|
|
|
send(...payload: Message) {
|
|
this.connection.send(payload)
|
|
}
|
|
|
|
onMessage = (connection: Connection, [verb, ...payload]: Message) => {
|
|
this.emit(verb, connection.url, ...payload)
|
|
}
|
|
|
|
cleanup = () => {
|
|
this.removeAllListeners()
|
|
this.connection.off('receive', this.onMessage)
|
|
}
|
|
}
|