vn-verdnaturachat/app/lib/ddp.js

127 lines
3.1 KiB
JavaScript
Raw Normal View History

import EJSON from 'ejson';
class EventEmitter {
constructor() {
this.events = {};
}
on(event, listener) {
if (typeof this.events[event] !== 'object') {
this.events[event] = [];
}
this.events[event].push(listener);
}
removeListener(event, listener) {
if (typeof this.events[event] === 'object') {
const idx = this.events[event].indexOf(listener);
if (idx > -1) {
this.events[event].splice(idx, 1);
}
}
}
emit(event, ...args) {
if (typeof this.events[event] === 'object') {
this.events[event].forEach((listener) => {
try {
listener.apply(this, args);
} catch (e) {
console.log(e);
}
});
}
}
once(event, listener) {
this.on(event, function g(...args) {
this.removeListener(event, g);
listener.apply(this, args);
});
}
}
export default class Socket extends EventEmitter {
constructor(url) {
super();
this.url = url.replace(/^http/, 'ws');
this.id = 0;
this.subscriptions = {};
this._connect();
this.ddp = new EventEmitter();
this.on('ping', () => this.send({ msg: 'pong' }));
this.on('result', data => this.ddp.emit(data.id, { result: data.result, error: data.error }));
this.on('ready', data => this.ddp.emit(data.subs[0], data));
}
send(obj) {
return new Promise((resolve, reject) => {
this.id += 1;
const id = obj.id || `${ this.id }`;
this.connection.send(EJSON.stringify({ ...obj, id }));
this.ddp.once(id, data => (data.error ? reject(data.error) : resolve(data.result || data.subs)));
});
}
_connect() {
const connection = new WebSocket(`${ this.url }/websocket`);
connection.onopen = () => {
this.emit('open');
this.send({ msg: 'connect', version: '1', support: ['1', 'pre2', 'pre1'] });
};
connection.onclose = e => this.emit('disconnected', e);
// connection.onerror = () => {
// // alert(error.type);
// // console.log(error);
// // console.log(`WebSocket Error ${ JSON.stringify({...error}) }`);
// };
connection.onmessage = (e) => {
const data = EJSON.parse(e.data);
this.emit(data.msg, data);
return data.collection && this.emit(data.collection, data);
};
// this.on('disconnected', e => alert(JSON.stringify(e)));
this.connection = connection;
}
logout() {
return this.call('logout').then(() => this.subscriptions = {});
}
disconnect() {
this.emit('disconnected_by_user');
this.connection.close();
}
reconnect() {
this.disconnect();
this.once('connected', () => {
Object.keys(this.subscriptions).forEach((key) => {
const { name, params } = this.subscriptions[key];
this.subscriptions[key].unsubscribe();
this.subscribe(name, params);
});
});
this._connect();
}
call(method, ...params) {
return this.send({
msg: 'method', method, params
});
}
unsubscribe(id) {
if (!this.subscriptions[id]) {
return Promise.reject();
}
delete this.subscriptions[id];
return this.send({
msg: 'unsub',
id
});
}
subscribe(name, ...params) {
return this.send({
msg: 'sub', name, params
}).then((data) => {
this.subscriptions[data.id] = {
name,
params,
unsubscribe: () => this.unsubscribe(data.id)
};
return this.subscriptions[data.id];
});
}
}