megalodon
Version:
Fediverse API client for node.js and browser
276 lines (275 loc) • 10.4 kB
JavaScript
"use strict";
var __extends = (this && this.__extends) || (function () {
var extendStatics = function (d, b) {
extendStatics = Object.setPrototypeOf ||
({ __proto__: [] } instanceof Array && function (d, b) { d.__proto__ = b; }) ||
function (d, b) { for (var p in b) if (Object.prototype.hasOwnProperty.call(b, p)) d[p] = b[p]; };
return extendStatics(d, b);
};
return function (d, b) {
if (typeof b !== "function" && b !== null)
throw new TypeError("Class extends value " + String(b) + " is not a constructor or null");
extendStatics(d, b);
function __() { this.constructor = d; }
d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __());
};
})();
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.Parser = void 0;
var isomorphic_ws_1 = __importDefault(require("isomorphic-ws"));
var dayjs_1 = __importDefault(require("dayjs"));
var events_1 = require("events");
var api_client_js_1 = __importDefault(require("./api_client.js"));
var notification_js_1 = require("../notification.js");
var default_js_1 = require("../default.js");
var WebSocket = (function (_super) {
__extends(WebSocket, _super);
function WebSocket(url, stream, params, accessToken, userAgent) {
var _this = _super.call(this) || this;
_this._heartbeatInterval = 60000;
_this._pongWaiting = false;
_this.url = url;
_this.stream = stream;
if (params === undefined) {
_this.params = null;
}
else {
_this.params = params;
}
_this.parser = new Parser();
_this.headers = {
'User-Agent': userAgent
};
_this._accessToken = accessToken;
_this._reconnectInterval = 10000;
_this._reconnectMaxAttempts = Infinity;
_this._reconnectCurrentAttempts = 0;
_this._connectionClosed = false;
_this._client = null;
_this._pongReceivedTimestamp = (0, dayjs_1.default)();
return _this;
}
WebSocket.prototype.start = function () {
this._connectionClosed = false;
this._resetRetryParams();
this._startWebSocketConnection();
};
WebSocket.prototype._startWebSocketConnection = function () {
this._resetConnection();
this._setupParser();
this._client = this._connect(this.url, this.stream, this.params, this._accessToken, this.headers);
this._bindSocket(this._client);
};
WebSocket.prototype.stop = function () {
this._connectionClosed = true;
this._resetConnection();
this._resetRetryParams();
};
WebSocket.prototype._resetConnection = function () {
if (this._client) {
this._client.close(1000);
this._clearBinding();
this._client = null;
}
if (this.parser) {
this.parser.removeAllListeners();
}
};
WebSocket.prototype._resetRetryParams = function () {
this._reconnectCurrentAttempts = 0;
};
WebSocket.prototype._reconnect = function () {
var _this = this;
setTimeout(function () {
if (_this._client && _this._client.readyState === isomorphic_ws_1.default.CONNECTING) {
return;
}
if (_this._reconnectCurrentAttempts < _this._reconnectMaxAttempts) {
_this._reconnectCurrentAttempts++;
_this._clearBinding();
if (_this._client) {
if ((0, default_js_1.isBrowser)()) {
_this._client.close();
}
else {
_this._client.terminate();
}
}
console.log('Reconnecting');
_this._client = _this._connect(_this.url, _this.stream, _this.params, _this._accessToken, _this.headers);
_this._bindSocket(_this._client);
}
}, this._reconnectInterval);
};
WebSocket.prototype._connect = function (url, stream, params, accessToken, headers) {
var parameter = ["stream=".concat(stream)];
if (params) {
parameter.push(params);
}
if (accessToken !== null) {
parameter.push("access_token=".concat(accessToken));
}
var requestURL = "".concat(url, "?").concat(parameter.join('&'));
if ((0, default_js_1.isBrowser)()) {
var cli = new isomorphic_ws_1.default(requestURL);
return cli;
}
else {
var options = {
headers: headers
};
var cli = new isomorphic_ws_1.default(requestURL, options);
return cli;
}
};
WebSocket.prototype._clearBinding = function () {
if (this._client && !(0, default_js_1.isBrowser)()) {
this._client.removeAllListeners('close');
this._client.removeAllListeners('pong');
this._client.removeAllListeners('open');
this._client.removeAllListeners('message');
this._client.removeAllListeners('error');
}
};
WebSocket.prototype._bindSocket = function (client) {
var _this = this;
client.onclose = function (event) {
if (event.code === 1000) {
_this.emit('close', {});
}
else {
console.log("Closed connection with ".concat(event.code));
if (!_this._connectionClosed) {
_this._reconnect();
}
}
};
client.onopen = function (_event) {
_this.emit('connect', {});
if (!(0, default_js_1.isBrowser)()) {
setTimeout(function () {
client.ping('');
}, 10000);
}
};
client.onmessage = function (event) {
_this.parser.parse(event);
};
client.onerror = function (event) {
_this.emit('error', event.error);
};
if (!(0, default_js_1.isBrowser)()) {
client.on('pong', function () {
_this._pongWaiting = false;
_this.emit('pong', {});
_this._pongReceivedTimestamp = (0, dayjs_1.default)();
setTimeout(function () { return _this._checkAlive(_this._pongReceivedTimestamp); }, _this._heartbeatInterval);
});
}
};
WebSocket.prototype._setupParser = function () {
var _this = this;
this.parser.on('update', function (status) {
_this.emit('update', api_client_js_1.default.Converter.status(status));
});
this.parser.on('notification', function (notification) {
var n = api_client_js_1.default.Converter.notification(notification);
if (n instanceof notification_js_1.UnknownNotificationTypeError) {
console.warn("Unknown notification event has received: ".concat(notification));
}
else {
_this.emit('notification', n);
}
});
this.parser.on('delete', function (id) {
_this.emit('delete', id);
});
this.parser.on('conversation', function (conversation) {
_this.emit('conversation', api_client_js_1.default.Converter.conversation(conversation));
});
this.parser.on('status_update', function (status) {
_this.emit('status_update', api_client_js_1.default.Converter.status(status));
});
this.parser.on('error', function (err) {
_this.emit('parser-error', err);
});
this.parser.on('heartbeat', function (_) {
_this.emit('heartbeat', 'heartbeat');
});
};
WebSocket.prototype._checkAlive = function (timestamp) {
var _this = this;
var now = (0, dayjs_1.default)();
if (now.diff(timestamp) > this._heartbeatInterval - 1000 && !this._connectionClosed) {
if (this._client && this._client.readyState !== isomorphic_ws_1.default.CONNECTING) {
this._pongWaiting = true;
this._client.ping('');
setTimeout(function () {
if (_this._pongWaiting) {
_this._pongWaiting = false;
_this._reconnect();
}
}, 10000);
}
}
};
return WebSocket;
}(events_1.EventEmitter));
exports.default = WebSocket;
var Parser = (function (_super) {
__extends(Parser, _super);
function Parser() {
return _super !== null && _super.apply(this, arguments) || this;
}
Parser.prototype.parse = function (ev) {
var data = ev.data;
var message = data.toString();
if (typeof message !== 'string') {
this.emit('heartbeat', {});
return;
}
if (message === '') {
this.emit('heartbeat', {});
return;
}
var event = '';
var payload = '';
var mes = {};
try {
var obj = JSON.parse(message);
event = obj.event;
payload = obj.payload;
mes = JSON.parse(payload);
}
catch (err) {
if (event !== 'delete') {
this.emit('error', new Error("Error parsing websocket reply: ".concat(message, ", error message: ").concat(err)));
return;
}
}
switch (event) {
case 'update':
this.emit('update', mes);
break;
case 'notification':
this.emit('notification', mes);
break;
case 'conversation':
this.emit('conversation', mes);
break;
case 'delete':
this.emit('delete', payload);
break;
case 'status.update':
this.emit('status_update', mes);
break;
default:
this.emit('error', new Error("Unknown event has received: ".concat(message)));
}
};
return Parser;
}(events_1.EventEmitter));
exports.Parser = Parser;