import { FAVOURITED_STATUSES_FETCH_SUCCESS, FAVOURITED_STATUSES_EXPAND_SUCCESS, } from 'themes/glitch/actions/favourites'; import { PINNED_STATUSES_FETCH_SUCCESS, } from 'themes/glitch/actions/pin_statuses'; import { Map as ImmutableMap, List as ImmutableList } from 'immutable'; import { FAVOURITE_SUCCESS, UNFAVOURITE_SUCCESS, PIN_SUCCESS, UNPIN_SUCCESS, } from 'themes/glitch/actions/interactions'; const initialState = ImmutableMap({ favourites: ImmutableMap({ next: null, loaded: false, items: ImmutableList(), }), pins: ImmutableMap({ next: null, loaded: false, items: ImmutableList(), }), }); const normalizeList = (state, listType, statuses, next) => { return state.update(listType, listMap => listMap.withMutations(map => { map.set('next', next); map.set('loaded', true); map.set('items', ImmutableList(statuses.map(item => item.id))); })); }; const appendToList = (state, listType, statuses, next) => { return state.update(listType, listMap => listMap.withMutations(map => { map.set('next', next); map.set('items', map.get('items').concat(statuses.map(item => item.id))); })); }; const prependOneToList = (state, listType, status) => { return state.update(listType, listMap => listMap.withMutations(map => { map.set('items', map.get('items').unshift(status.get('id'))); })); }; const removeOneFromList = (state, listType, status) => { return state.update(listType, listMap => listMap.withMutations(map => { map.set('items', map.get('items').filter(item => item !== status.get('id'))); })); }; export default function statusLists(state = initialState, action) { switch(action.type) { case FAVOURITED_STATUSES_FETCH_SUCCESS: return normalizeList(state, 'favourites', action.statuses, action.next); case FAVOURITED_STATUSES_EXPAND_SUCCESS: return appendToList(state, 'favourites', action.statuses, action.next); case FAVOURITE_SUCCESS: return prependOneToList(state, 'favourites', action.status); case UNFAVOURITE_SUCCESS: return removeOneFromList(state, 'favourites', action.status); case PINNED_STATUSES_FETCH_SUCCESS: return normalizeList(state, 'pins', action.statuses, action.next); case PIN_SUCCESS: return prependOneToList(state, 'pins', action.status); case UNPIN_SUCCESS: return removeOneFromList(state, 'pins', action.status); default: return state; } };