-
Notifications
You must be signed in to change notification settings - Fork 0
/
aria2ws.js
49 lines (49 loc) · 1.86 KB
/
aria2ws.js
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
class Aria2WebSocket {
constructor (...args) {
let path = args.join('#').match(/^(wss?:\/\/[^#]+)#?(.*)$/);
if (!path) { throw new Error('Invalid JSON-RPC entry: "' + args.join('", "') + '"'); }
this.jsonrpc = path[1];
this.secret = path[2];
this.params = this.secret ? ['token:' + this.secret] : [];
this.connect();
}
version = '0.7.0';
timeout = 10000;
connect () {
this.socket = new Promise((resolve, reject) => {
let ws = new WebSocket(this.jsonrpc);
ws.onopen = (event) => resolve(ws);
ws.onmessage = (event) => {
let response = JSON.parse(event.data);
if (!response.method) { ws.resolve(response); }
else if (typeof this._onmessage === 'function') { this._onmessage(response); }
};
ws.onclose = (event) => {
if (!event.wasClean) { setTimeout(() => this.connect(), this.timeout); }
if (typeof this._onclose === 'function') { this._onclose(event); }
};
});
}
disconnect () {
this.socket.then( (ws) => ws.close() );
}
set onmessage (callback) {
this._onmessage = typeof callback === 'function' ? callback : null;
}
get onmessage () {
return this._onmessage;
}
set onclose (callback) {
this._onclose = typeof callback === 'function' ? callback : null;
}
get onclose () {
return this._onclose;
}
send (...args) {
return this.socket.then((ws) => new Promise((resolve, reject) => {
ws.resolve = resolve;
ws.onerror = reject;
ws.send( JSON.stringify( args.map( ({ method, params = [] }) => ({id: '', jsonrpc: '2.0', method, params: [...this.params, ...params]}) ) ) );
}));
}
}