|
|
- import createStream from '../stream';
- import {
- updateTimeline,
- deleteFromTimelines,
- refreshHomeTimeline,
- connectTimeline,
- disconnectTimeline,
- } from './timelines';
- import { updateNotifications, refreshNotifications } from './notifications';
- import { getLocale } from '../locales';
-
- const { messages } = getLocale();
-
- export function connectTimelineStream (timelineId, path, pollingRefresh = null) {
- return (dispatch, getState) => {
- const streamingAPIBaseURL = getState().getIn(['meta', 'streaming_api_base_url']);
- const accessToken = getState().getIn(['meta', 'access_token']);
- const locale = getState().getIn(['meta', 'locale']);
- let polling = null;
-
- const setupPolling = () => {
- polling = setInterval(() => {
- pollingRefresh(dispatch);
- }, 20000);
- };
-
- const clearPolling = () => {
- if (polling) {
- clearInterval(polling);
- polling = null;
- }
- };
-
- const subscription = createStream(streamingAPIBaseURL, accessToken, path, {
-
- connected () {
- if (pollingRefresh) {
- clearPolling();
- }
- dispatch(connectTimeline(timelineId));
- },
-
- disconnected () {
- if (pollingRefresh) {
- setupPolling();
- }
- dispatch(disconnectTimeline(timelineId));
- },
-
- received (data) {
- switch(data.event) {
- case 'update':
- dispatch(updateTimeline(timelineId, JSON.parse(data.payload)));
- break;
- case 'delete':
- dispatch(deleteFromTimelines(data.payload));
- break;
- case 'notification':
- dispatch(updateNotifications(JSON.parse(data.payload), messages, locale));
- break;
- }
- },
-
- reconnected () {
- if (pollingRefresh) {
- clearPolling();
- pollingRefresh(dispatch);
- }
- dispatch(connectTimeline(timelineId));
- },
-
- });
-
- const disconnect = () => {
- if (subscription) {
- subscription.close();
- }
- clearPolling();
- };
-
- return disconnect;
- };
- }
-
- function refreshHomeTimelineAndNotification (dispatch) {
- dispatch(refreshHomeTimeline());
- dispatch(refreshNotifications());
- }
-
- export const connectUserStream = () => connectTimelineStream('home', 'user', refreshHomeTimelineAndNotification);
- export const connectCommunityStream = () => connectTimelineStream('community', 'public:local');
- export const connectMediaStream = () => connectTimelineStream('community', 'public:local');
- export const connectPublicStream = () => connectTimelineStream('public', 'public');
- export const connectHashtagStream = (tag) => connectTimelineStream(`hashtag:${tag}`, `hashtag&tag=${tag}`);
|