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

// @flow
import * as ACTIONS from 'constants/action_types';
import { parseURI, ACTIONS as LBRY_REDUX_ACTIONS } from 'lbry-redux';
import { handleActions } from 'util/redux-utils';

const defaultState: SubscriptionState = {
  subscriptions: [], // Deprecated
  following: [],
  loading: false,
  firstRunCompleted: false,
};

export default handleActions(
  {
    [ACTIONS.CHANNEL_SUBSCRIBE]: (state: SubscriptionState, action): SubscriptionState => {
      const newSubscription: { uri: string, channelName: string, notificationsDisabled: boolean } = action.data;
      const newSubscriptions: Array<Subscription> = state.subscriptions.slice();
      let newFollowing: Array<Following> = state.following.slice();
      // prevent duplicates in the sidebar
      if (!newSubscriptions.some((sub) => sub.uri === newSubscription.uri)) {
        //   $FlowFixMe
        newSubscriptions.unshift(newSubscription);
      }

      if (!newFollowing.some((sub) => sub.uri === newSubscription.uri)) {
        newFollowing.unshift({
          uri: newSubscription.uri,
          notificationsDisabled: newSubscription.notificationsDisabled,
        });
      } else {
        newFollowing = newFollowing.map((following) => {
          if (following.uri === newSubscription.uri) {
            return {
              uri: newSubscription.uri,
              notificationsDisabled: newSubscription.notificationsDisabled,
            };
          } else {
            return following;
          }
        });
      }

      return {
        ...state,
        subscriptions: newSubscriptions,
        following: newFollowing,
      };
    },
    [ACTIONS.CHANNEL_UNSUBSCRIBE]: (state: SubscriptionState, action): SubscriptionState => {
      const subscriptionToRemove: Subscription = action.data;
      const newSubscriptions = state.subscriptions
        .slice()
        .filter(
          (subscription) => subscription.channelName.toLowerCase() !== subscriptionToRemove.channelName.toLowerCase()
        );
      const newFollowing = state.following
        .slice()
        .filter((subscription) => subscription.uri !== subscriptionToRemove.uri);

      return {
        ...state,
        subscriptions: newSubscriptions,
        following: newFollowing,
      };
    },
    [ACTIONS.FETCH_SUBSCRIPTIONS_START]: (state: SubscriptionState): SubscriptionState => ({
      ...state,
      loading: true,
    }),
    [ACTIONS.FETCH_SUBSCRIPTIONS_FAIL]: (state: SubscriptionState): SubscriptionState => ({
      ...state,
      loading: false,
    }),
    [ACTIONS.FETCH_SUBSCRIPTIONS_SUCCESS]: (state: SubscriptionState, action): SubscriptionState => ({
      ...state,
      loading: false,
      subscriptions: action.data,
    }),
    [ACTIONS.SET_VIEW_MODE]: (state: SubscriptionState, action): SubscriptionState => ({
      ...state,
      viewMode: action.data,
    }),
    [LBRY_REDUX_ACTIONS.USER_STATE_POPULATE]: (
      state: SubscriptionState,
      action: { data: { subscriptions: ?Array<string>, following: ?Array<Subscription> } }
    ) => {
      const { subscriptions, following } = action.data;
      const incomingSubscriptions = Array.isArray(subscriptions) && subscriptions.length;
      if (!incomingSubscriptions) {
        return {
          ...state,
        };
      }
      let newSubscriptions;
      let newFollowing;

      if (!subscriptions) {
        newSubscriptions = state.subscriptions;
      } else {
        const parsedSubscriptions = subscriptions.map((uri) => {
          const { channelName } = parseURI(uri);

          return {
            uri,
            channelName: `@${channelName}`,
          };
        });
        newSubscriptions = parsedSubscriptions;
      }

      if (!following) {
        newFollowing = newSubscriptions.slice().map(({ uri }) => {
          return {
            uri,
            // Default first time movers to notifications on
            // This value is for email notifications too so we can't default off
            // New subscriptions after population will default off
            notificationsDisabled: false,
          };
        });
      } else {
        newFollowing = following;
      }

      return {
        ...state,
        subscriptions: newSubscriptions,
        following: newFollowing,
      };
    },
  },
  defaultState
);