neeboo
6 years ago
parent
8dac30d2a1
commit
88d004d3cb
@ -0,0 +1,88 @@ |
|||||||
|
import { isWs } from '@harmony/utils'; |
||||||
|
import mitt from 'mitt'; |
||||||
|
import { BaseProvider } from './baseProvider'; |
||||||
|
|
||||||
|
export const enum SocketConnection { |
||||||
|
READY = 'ready', |
||||||
|
CONNECT = 'connect', |
||||||
|
ERROR = 'error', |
||||||
|
CLOSE = 'close', |
||||||
|
} |
||||||
|
|
||||||
|
export const enum SocketState { |
||||||
|
SOCKET_MESSAGE = 'socket_message', |
||||||
|
SOCKET_READY = 'socket_ready', |
||||||
|
SOCKET_CLOSE = 'socket_close', |
||||||
|
SOCKET_ERROR = 'socket_error', |
||||||
|
SOCKET_CONNECT = 'socket_connect', |
||||||
|
SOCKET_NETWORK_CHANGED = 'socket_networkChanged', |
||||||
|
SOCKET_ACCOUNTS_CHANGED = 'socket_accountsChanged', |
||||||
|
} |
||||||
|
|
||||||
|
class BaseSocket extends BaseProvider { |
||||||
|
url: string; |
||||||
|
emitter: mitt.Emitter; |
||||||
|
subscriptions: any = {}; |
||||||
|
handlers: any = {}; |
||||||
|
constructor(url: string) { |
||||||
|
super(url); |
||||||
|
if (!isWs(url)) { |
||||||
|
throw new Error(`${url} is not websocket`); |
||||||
|
} |
||||||
|
this.url = url; |
||||||
|
this.emitter = new mitt(this.handlers); |
||||||
|
} |
||||||
|
resetHandlers() { |
||||||
|
// tslint:disable-next-line: forin
|
||||||
|
for (const i in this.handlers) { |
||||||
|
delete this.handlers[i]; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
once(type: string, handler: mitt.Handler) { |
||||||
|
this.emitter.on(type, handler); |
||||||
|
this.removeListener(type); |
||||||
|
} |
||||||
|
|
||||||
|
removeListener(type?: string, handler?: mitt.Handler) { |
||||||
|
if (!type) { |
||||||
|
this.handlers = {}; |
||||||
|
return; |
||||||
|
} |
||||||
|
if (!handler) { |
||||||
|
delete this.handlers[type]; |
||||||
|
} else { |
||||||
|
return this.emitter.off(type, handler); |
||||||
|
} |
||||||
|
} |
||||||
|
reset() { |
||||||
|
this.removeListener('*'); |
||||||
|
// this.registerEventListeners();
|
||||||
|
} |
||||||
|
removeAllSocketListeners() { |
||||||
|
this.removeListener(SocketState.SOCKET_MESSAGE); |
||||||
|
this.removeListener(SocketState.SOCKET_READY); |
||||||
|
this.removeListener(SocketState.SOCKET_CLOSE); |
||||||
|
this.removeListener(SocketState.SOCKET_ERROR); |
||||||
|
this.removeListener(SocketState.SOCKET_CONNECT); |
||||||
|
} |
||||||
|
|
||||||
|
onReady(event: any) { |
||||||
|
this.emitter.on(SocketConnection.READY, () => event); |
||||||
|
this.emitter.on(SocketState.SOCKET_READY, () => event); |
||||||
|
} |
||||||
|
onError(error: any) { |
||||||
|
this.emitter.on(SocketConnection.ERROR, () => error); |
||||||
|
this.emitter.on(SocketState.SOCKET_ERROR, () => error); |
||||||
|
this.removeAllSocketListeners(); |
||||||
|
this.removeListener('*'); |
||||||
|
} |
||||||
|
onClose(error = null) { |
||||||
|
this.emitter.on(SocketConnection.CLOSE, () => error); |
||||||
|
this.emitter.on(SocketState.SOCKET_CLOSE, () => error); |
||||||
|
this.removeAllSocketListeners(); |
||||||
|
this.removeListener('*'); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
export { BaseSocket }; |
||||||
@ -0,0 +1,86 @@ |
|||||||
|
// TODO: implement Websocket Provider
|
||||||
|
import { w3cwebsocket as W3CWebsocket } from 'websocket'; |
||||||
|
import { BaseSocket } from './baseSocket'; |
||||||
|
import { isWs } from '@harmony/utils'; |
||||||
|
import { JsonRpc } from '../rpcMethod/rpcbuilder'; |
||||||
|
import { composeMiddleware } from '../rpcMethod/net'; |
||||||
|
import { RPCRequestPayload } from '../types'; |
||||||
|
|
||||||
|
class WSProvider extends BaseSocket { |
||||||
|
url: string; |
||||||
|
subscriptions: any = {}; |
||||||
|
options: any = {}; |
||||||
|
connection: W3CWebsocket | WebSocket; |
||||||
|
jsonRpc: JsonRpc; |
||||||
|
// ws: w3cwebsocket;
|
||||||
|
constructor(url: string, options: any = {}) { |
||||||
|
super(url); |
||||||
|
if (!isWs(url)) { |
||||||
|
throw new Error(`${url} is not websocket`); |
||||||
|
} |
||||||
|
this.url = url; |
||||||
|
this.options = options; |
||||||
|
this.connection = this.createWebsocketProvider(this.url, this.options); |
||||||
|
this.jsonRpc = new JsonRpc(); |
||||||
|
} |
||||||
|
|
||||||
|
createWebsocketProvider(url: string, options: any = {}) { |
||||||
|
// tslint:disable-next-line: no-string-literal
|
||||||
|
if (typeof window !== 'undefined' && (<any>window).WebSocket) { |
||||||
|
// tslint:disable-next-line: no-string-literal
|
||||||
|
return new WebSocket(url, options.protocol); |
||||||
|
} else { |
||||||
|
const headers = options.headers || {}; |
||||||
|
const urlObject = new URL(url); |
||||||
|
|
||||||
|
if (!headers.authorization && urlObject.username && urlObject.password) { |
||||||
|
const authToken = Buffer.from( |
||||||
|
`${urlObject.username}:${urlObject.password}`, |
||||||
|
).toString('base64'); |
||||||
|
headers.authorization = `Basic ${authToken}`; |
||||||
|
} |
||||||
|
|
||||||
|
return new W3CWebsocket( |
||||||
|
url, |
||||||
|
options.protocol, |
||||||
|
undefined, |
||||||
|
headers, |
||||||
|
undefined, |
||||||
|
options.clientConfig, |
||||||
|
); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
get connected() { |
||||||
|
return this.connection.readyState === this.connection.OPEN; |
||||||
|
} |
||||||
|
|
||||||
|
isConnecting() { |
||||||
|
return this.connection.readyState === this.connection.CONNECTING; |
||||||
|
} |
||||||
|
|
||||||
|
send(payload: RPCRequestPayload<object>): Promise<any> { |
||||||
|
const [tReq, tRes] = this.getMiddleware(payload.method); |
||||||
|
const reqMiddleware = composeMiddleware(...tReq); |
||||||
|
const resMiddleware = composeMiddleware(...tRes); |
||||||
|
|
||||||
|
return new Promise((resolve, reject) => { |
||||||
|
this.connection.send(reqMiddleware(JSON.stringify(payload))); |
||||||
|
this.connection.onmessage = (msg: MessageEvent) => { |
||||||
|
if (msg && msg.data) { |
||||||
|
let result; |
||||||
|
try { |
||||||
|
result = JSON.parse(msg.data); |
||||||
|
resolve(resMiddleware(result)); |
||||||
|
} catch (error) { |
||||||
|
reject(error); |
||||||
|
} |
||||||
|
} else { |
||||||
|
reject('provider error'); |
||||||
|
} |
||||||
|
}; |
||||||
|
}); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
export { WSProvider }; |
||||||
@ -0,0 +1,4 @@ |
|||||||
|
export interface Window { |
||||||
|
WebSocket: any; |
||||||
|
MozWebSocket: any; |
||||||
|
} |
||||||
Loading…
Reference in new issue