58 lines
1.3 KiB
JavaScript
58 lines
1.3 KiB
JavaScript
|
import { paramsString } from '../ajax'
|
||
|
|
import noop from 'lodash/noop'
|
||
|
|
import WebSocketClient from 'websocket.js'
|
||
|
|
|
||
|
|
function getStreamName(timeline) {
|
||
|
|
switch (timeline) {
|
||
|
|
case 'local':
|
||
|
|
return 'public:local'
|
||
|
|
case 'federated':
|
||
|
|
return 'public'
|
||
|
|
case 'home':
|
||
|
|
return 'user'
|
||
|
|
case 'notifications':
|
||
|
|
return 'user:notification'
|
||
|
|
}
|
||
|
|
if (timeline.startsWith('tag/')) {
|
||
|
|
return 'hashtag'
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
function getUrl(streamingApi, accessToken, timeline) {
|
||
|
|
let url = `${streamingApi}/api/v1/streaming`
|
||
|
|
let streamName = getStreamName(timeline)
|
||
|
|
|
||
|
|
let params = {
|
||
|
|
stream: streamName
|
||
|
|
}
|
||
|
|
|
||
|
|
if (timeline.startsWith('tag/')) {
|
||
|
|
params.tag = timeline.split('/').slice(-1)[0]
|
||
|
|
}
|
||
|
|
|
||
|
|
if (accessToken) {
|
||
|
|
params.access_token = accessToken
|
||
|
|
}
|
||
|
|
|
||
|
|
return url + '?' + paramsString(params)
|
||
|
|
}
|
||
|
|
|
||
|
|
export class StatusStream {
|
||
|
|
constructor(streamingApi, accessToken, timeline, opts) {
|
||
|
|
let url = getUrl(streamingApi, accessToken, timeline)
|
||
|
|
|
||
|
|
const ws = new WebSocketClient(url, null, { backoff: 'exponential' })
|
||
|
|
const onMessage = opts.onMessage || noop
|
||
|
|
|
||
|
|
ws.onopen = opts.onOpen || noop
|
||
|
|
ws.onmessage = e => onMessage(JSON.parse(e.data))
|
||
|
|
ws.onclose = opts.onClose || noop
|
||
|
|
ws.onreconnect = opts.onReconnect || noop
|
||
|
|
|
||
|
|
this._ws = ws
|
||
|
|
}
|
||
|
|
|
||
|
|
close() {
|
||
|
|
this._ws.close()
|
||
|
|
}
|
||
|
|
}
|