add ability to stream statuses

This commit is contained in:
Nolan Lawson 2018-01-25 09:03:34 -08:00
parent 2b943fe0bf
commit 610e54469e
5 changed files with 97 additions and 16 deletions

21
package-lock.json generated
View file

@ -264,6 +264,14 @@
"js-tokens": "3.0.2"
}
},
"backoff": {
"version": "2.5.0",
"resolved": "https://registry.npmjs.org/backoff/-/backoff-2.5.0.tgz",
"integrity": "sha1-9hbtqdPktmuMp/ynn2lXIsX44m8=",
"requires": {
"precond": "0.2.3"
}
},
"balanced-match": {
"version": "0.4.2",
"resolved": "https://registry.npmjs.org/balanced-match/-/balanced-match-0.4.2.tgz",
@ -5473,6 +5481,11 @@
"uniqs": "2.0.0"
}
},
"precond": {
"version": "0.2.3",
"resolved": "https://registry.npmjs.org/precond/-/precond-0.2.3.tgz",
"integrity": "sha1-qpWRvKokkj8eD0hJ0kD0fvwQdaw="
},
"prepend-http": {
"version": "1.0.4",
"resolved": "https://registry.npmjs.org/prepend-http/-/prepend-http-1.0.4.tgz",
@ -7779,6 +7792,14 @@
}
}
},
"websocket.js": {
"version": "0.1.12",
"resolved": "https://registry.npmjs.org/websocket.js/-/websocket.js-0.1.12.tgz",
"integrity": "sha1-RsmAeHxX68jtz0SgJj5dY5NnuFs=",
"requires": {
"backoff": "2.5.0"
}
},
"whet.extend": {
"version": "0.9.9",
"resolved": "https://registry.npmjs.org/whet.extend/-/whet.extend-0.9.9.tgz",

View file

@ -52,6 +52,7 @@
"url-search-params": "^0.10.0",
"webpack": "^3.10.0",
"webpack-bundle-analyzer": "^2.9.2",
"websocket.js": "^0.1.12",
"workerize-loader": "^1.0.1",
"yargs": "^10.1.1"
},

View file

@ -29,6 +29,7 @@
import { timelines } from '../_static/timelines'
import { toast } from '../_utils/toast'
import { database } from '../_utils/database/database'
import { StatusStream } from '../_utils/mastodon/StatusStream'
const cachedTimelines = {}
@ -42,6 +43,7 @@
async oncreate() {
let timeline = this.get('timeline')
let instanceName = this.store.get('currentInstance')
let accessToken = this.store.get('accessToken')
let cachedStatusIds = cachedTimelines[timeline]
if (cachedStatusIds) {
this.set({statusIds: cachedStatusIds})
@ -50,8 +52,16 @@
}
/* no await */ getInstanceInfo(instanceName).then(instanceInfo => database.setInstanceInfo(instanceName, instanceInfo))
let instanceInfo = await database.getInstanceInfo(instanceName)
this._statusStream = new StatusStream(instanceInfo.urls.streaming_api, accessToken, timeline, {
onMessage(message) {
console.log('message', message)
}
})
},
ondestroy() {
if (this._statusStream) {
this._statusStream.close()
}
cachedTimelines[this.get('timeline')] = this.get('statusIds')
},
data: () => ({

View file

@ -1,6 +1,6 @@
import { loadCSS } from 'fg-loadcss';
const importURLSearchParams = () => import(
export const importURLSearchParams = () => import(
/* webpackChunkName: 'url-search-params' */ 'url-search-params'
).then(Params => {
window.URLSearchParams = Params
@ -11,23 +11,23 @@ const importURLSearchParams = () => import(
})
})
const importTimeline = () => import(
export const importTimeline = () => import(
/* webpackChunkName: 'Timeline' */ '../_components/Timeline.html'
).then(mod => mod.default)
const importIntersectionObserver = () => import(
export const importIntersectionObserver = () => import(
/* webpackChunkName: 'intersection-observer' */ 'intersection-observer'
)
const importRequestIdleCallback = () => import(
export const importRequestIdleCallback = () => import(
/* webpackChunkName: 'requestidlecallback' */ 'requestidlecallback'
)
const importIndexedDBGetAllShim = () => import(
export const importIndexedDBGetAllShim = () => import(
/* webpackChunkName: 'indexeddb-getall-shim' */ 'indexeddb-getall-shim'
)
const importDialogPolyfill = (() => {
export const importDialogPolyfill = (() => {
let cached
return () => {
if (cached) {
@ -40,12 +40,3 @@ const importDialogPolyfill = (() => {
})
}
})()
export {
importURLSearchParams,
importTimeline,
importIntersectionObserver,
importRequestIdleCallback,
importIndexedDBGetAllShim,
importDialogPolyfill
}

View file

@ -0,0 +1,58 @@
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()
}
}