pinafore/routes/_utils/mastodon/StatusStream.js

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()
  }
}