12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697 |
- 'use strict'
-
- import http = require('http');
- import bsock = require('bsock');
-
- import * as T from './Types';
- import * as U from './Utils';
- import * as I from './Interfaces';
-
- /**
- * A Websocket-server-on-steroids with built-in RPC capabilities
- */
- export class RPCServer<
- SubResType = {},
- InterfaceT extends T.RPCInterface = T.RPCInterface,
- > implements I.Destroyable{
- private ws = http.createServer()
- private io = bsock.createServer()
- private visibility:T.Visibility
- private closeHandler:T.CloseHandler
- private errorHandler: T.ErrorHandler
- private connectionHandler: T.ConnectionHandler
- private sesame? : T.SesameFunction
-
- /**
- * @throws On RPC with no name
- * @param port The port to listen on
- * @param exporters A list of {@link RPCExporter} to publish
- * @param conf A {@link SocketConf} object with optional settings
- */
- constructor(
- private port:number,
- private exporters: I.RPCExporter<T.RPCInterface<InterfaceT>, keyof InterfaceT, SubResType>[] = [],
- conf: T.ServerConf = {}
- ){
-
- if(!conf.visibility) this.visibility = "127.0.0.1"
-
- if(conf.sesame){
- this.sesame = U.makeSesameFunction(conf.sesame)
- }
-
- this.errorHandler = (socket:I.Socket) => (error:any) => {
- if(conf.errorHandler) conf.errorHandler(socket, error)
- socket.destroy();
- console.error("Caught websocket error", String(error))
- }
-
- this.closeHandler = (socket:I.Socket) => {
- if(!conf.closeHandler) console.log("Connection on port "+socket.port+" closing")
- else conf.closeHandler(socket)
- }
-
- this.connectionHandler = (socket:I.Socket) => {
- if(!conf.connectionHandler) console.log("New connection on port "+socket.port)
- else conf.connectionHandler(socket)
- }
-
- let badRPC
- if(badRPC = exporters.flatMap(ex => ex.exportRPCs()).find(rpc => !rpc.name))
- throw new Error(`
- RPC did not provide a name.
- \nUse 'funtion name(..){ .. }' syntax instead.
- \n
- \n<------------OFFENDING RPC:
- \n`+badRPC.toString()+`
- \n>------------OFFENDING RPC`)
- this.startWebsocket()
- }
-
- private startWebsocket(){
- try{
- this.io.attach(this.ws)
- this.io.on('socket', (socket:I.Socket) => {
- socket.on('error', (err) => this.errorHandler(socket, err))
- socket.on('close', () => this.closeHandler(socket))
- this.connectionHandler(socket)
- this.initRPCs(socket)
- })
- this.ws.listen(this.port, this.visibility)
- }catch(e){
- this.errorHandler(this.io, e)
- }
- }
-
- protected initRPCs(socket:I.Socket){
- socket.hook('info', () => rpcInfos)
- const rpcInfos:T.ExtendedRpcInfo[] = [
- ...this.exporters.flatMap(exporter => U.rpcHooker(socket, exporter, this.sesame))
- ]
- }
-
- async destroy(): Promise<void> {
- this.io.close()
- this.ws.close()
- }
- }
|