@cutls/megalodon
Version:
Mastodon, Pleroma, Misskey API client for node.js and browser
270 lines (269 loc) • 7.48 kB
JavaScript
var __extends =
(this && this.__extends) ||
(() => {
var extendStatics = (d, b) => {
extendStatics =
Object.setPrototypeOf ||
({ __proto__: [] } instanceof Array &&
((d, b) => {
d.__proto__ = b
})) ||
((d, b) => {
for (var p in b) if (Object.hasOwn(b, p)) d[p] = b[p]
})
return extendStatics(d, b)
}
return (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 __())
}
})()
import WS from 'isomorphic-ws'
import dayjs from 'dayjs'
import { EventEmitter } from 'events'
import PleromaAPI from './api_client'
import { UnknownNotificationTypeError } from '../notification'
import { isBrowser } from '../default'
var WebSocket = ((_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 = dayjs()
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.subscribe = (_name, _stream, _add) => {}
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 () {
setTimeout(() => {
if (this._client && this._client.readyState === WS.CONNECTING) {
return
}
if (this._reconnectCurrentAttempts < this._reconnectMaxAttempts) {
this._reconnectCurrentAttempts++
this._clearBinding()
if (this._client) {
if (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 = (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 (isBrowser()) {
var cli = new WS(requestURL)
return cli
} else {
var options = {
headers: headers
}
var cli = new WS(requestURL, options)
return cli
}
}
WebSocket.prototype._clearBinding = function () {
if (this._client && !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) {
client.onclose = (event) => {
if (event.code === 1000) {
this.emit('close', {})
} else {
console.log('Closed connection with '.concat(event.code))
if (!this._connectionClosed) {
this._reconnect()
}
}
}
client.onopen = (_event) => {
this.emit('connect', {})
if (!isBrowser()) {
setTimeout(() => {
client.ping('')
}, 10000)
}
}
client.onmessage = (event) => {
this.parser.parse(event)
}
client.onerror = (event) => {
this.emit('error', event.error)
}
if (!isBrowser()) {
client.on('pong', () => {
this._pongWaiting = false
this.emit('pong', {})
this._pongReceivedTimestamp = dayjs()
setTimeout(() => this._checkAlive(this._pongReceivedTimestamp), this._heartbeatInterval)
})
}
}
WebSocket.prototype._setupParser = function () {
this.parser.on('update', (status) => {
this.emit('update', PleromaAPI.Converter.status(status))
})
this.parser.on('notification', (notification) => {
var n = PleromaAPI.Converter.notification(notification)
if (n instanceof UnknownNotificationTypeError) {
console.warn('Unknown notification event has received: '.concat(notification))
} else {
this.emit('notification', n)
}
})
this.parser.on('delete', (id) => {
this.emit('delete', id)
})
this.parser.on('conversation', (conversation) => {
this.emit('conversation', PleromaAPI.Converter.conversation(conversation))
})
this.parser.on('status_update', (status) => {
this.emit('status_update', PleromaAPI.Converter.status(status))
})
this.parser.on('error', (err) => {
this.emit('parser-error', err)
})
this.parser.on('heartbeat', (_) => {
this.emit('heartbeat', 'heartbeat')
})
}
WebSocket.prototype._checkAlive = function (timestamp) {
var now = dayjs()
if (now.diff(timestamp) > this._heartbeatInterval - 1000 && !this._connectionClosed) {
if (this._client && this._client.readyState !== WS.CONNECTING) {
this._pongWaiting = true
this._client.ping('')
setTimeout(() => {
if (this._pongWaiting) {
this._pongWaiting = false
this._reconnect()
}
}, 10000)
}
}
}
return WebSocket
})(EventEmitter)
export default WebSocket
var Parser = ((_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
})(EventEmitter)
export { Parser }