lbry-desktop/ui/redux/reducers/livestream.js

// @flow

import * as ACTIONS from 'constants/action_types';
import { handleActions } from 'util/redux-utils';

const defaultState: LivestreamState = {
  fetchingById: {},
  viewersById: {},
  fetchingActiveLivestreams: 'pending',
  activeLivestreams: {},
  activeLivestreamsLastFetchedDate: 0,
  activeLivestreamsLastFetchedOptions: {},
  activeLivestreamsLastFetchedFailCount: 0,
  activeLivestreamInitialized: false,
  socketConnectionById: {},
};

/**
 * Update state.viewersById with the latest data
 * @param {object} activeLivestreams - streams with fetched data
 * @param {object} originalState - streams with only their view counts
 * @returns {*} - updated viewersById object if active streams passed, otherwise return old data
 */
function updateViewersById(activeLivestreams, originalState) {
  if (activeLivestreams) {
    const viewersById = Object.assign({}, originalState);
    Object.values(activeLivestreams).forEach((data) => {
      // $FlowFixMe: mixed
      if (data && data.claimId && data.viewCount) {
        // $FlowFixMe: mixed
        viewersById[data.claimId] = data.viewCount;
      }
    });
    return viewersById;
  }

  return originalState;
}

export default handleActions(
  {
    [ACTIONS.FETCH_NO_SOURCE_CLAIMS_STARTED]: (state: LivestreamState, action: any): LivestreamState => {
      const claimId = action.data;
      const newIdsFetching = Object.assign({}, state.fetchingById);
      newIdsFetching[claimId] = true;

      return { ...state, fetchingById: newIdsFetching };
    },
    [ACTIONS.FETCH_NO_SOURCE_CLAIMS_COMPLETED]: (state: LivestreamState, action: any): LivestreamState => {
      const claimId = action.data;
      const newIdsFetching = Object.assign({}, state.fetchingById);
      newIdsFetching[claimId] = false;

      return { ...state, fetchingById: newIdsFetching };
    },
    [ACTIONS.FETCH_NO_SOURCE_CLAIMS_FAILED]: (state: LivestreamState, action: any) => {
      const claimId = action.data;
      const newIdsFetching = Object.assign({}, state.fetchingById);
      newIdsFetching[claimId] = false;

      return { ...state, fetchingById: newIdsFetching };
    },
    [ACTIONS.VIEWERS_RECEIVED]: (state: LivestreamState, action: any) => {
      const { connected, claimId } = action.data;
      const newViewersById = Object.assign({}, state.viewersById);
      newViewersById[claimId] = connected;
      return { ...state, viewersById: newViewersById };
    },
    [ACTIONS.FETCH_ACTIVE_LIVESTREAMS_STARTED]: (state: LivestreamState) => {
      return { ...state, fetchingActiveLivestreams: true };
    },
    [ACTIONS.FETCH_ACTIVE_LIVESTREAMS_FAILED]: (state: LivestreamState, action: any) => {
      const { activeLivestreamsLastFetchedDate, activeLivestreamsLastFetchedOptions } = action.data;
      return {
        ...state,
        fetchingActiveLivestreams: false,
        activeLivestreamsLastFetchedDate,
        activeLivestreamsLastFetchedOptions,
        activeLivestreamsLastFetchedFailCount: state.activeLivestreamsLastFetchedFailCount + 1,
      };
    },
    [ACTIONS.FETCH_ACTIVE_LIVESTREAMS_COMPLETED]: (state: LivestreamState, action: any) => {
      const { activeLivestreams, activeLivestreamsLastFetchedDate, activeLivestreamsLastFetchedOptions } = action.data;
      return {
        ...state,
        fetchingActiveLivestreams: false,
        activeLivestreams,
        activeLivestreamsLastFetchedDate,
        activeLivestreamsLastFetchedOptions,
        activeLivestreamsLastFetchedFailCount: 0,
        viewersById: updateViewersById(activeLivestreams, state.viewersById),
      };
    },
    [ACTIONS.ADD_CHANNEL_TO_ACTIVE_LIVESTREAMS]: (state: LivestreamState, action: any) => {
      const activeLivestreams = Object.assign({}, state.activeLivestreams || {}, action.data);
      return {
        ...state,
        activeLivestreams,
        activeLivestreamInitialized: true,
        viewersById: updateViewersById(activeLivestreams, state.viewersById),
      };
    },
    [ACTIONS.REMOVE_CHANNEL_FROM_ACTIVE_LIVESTREAMS]: (state: LivestreamState, action: any) => {
      const activeLivestreams = Object.assign({}, state.activeLivestreams);
      activeLivestreams[action.data.channelId] = null;
      return { ...state, activeLivestreams, activeLivestreamInitialized: true };
    },
    [ACTIONS.SOCKET_CONNECTED_BY_ID]: (state: LivestreamState, action: any) => {
      const { connected, sub_category, id: claimId } = action.data;

      const socketConnectionById = Object.assign({}, state.socketConnectionById);
      socketConnectionById[claimId] = { connected, sub_category };

      return { ...state, socketConnectionById };
    },
  },
  defaultState
);