import { Map as ImmutableMap, List as ImmutableList, fromJS } from 'immutable'; import { blockAccountSuccess, muteAccountSuccess } from 'mastodon/actions/accounts'; import { NOTIFICATION_REQUESTS_EXPAND_REQUEST, NOTIFICATION_REQUESTS_EXPAND_SUCCESS, NOTIFICATION_REQUESTS_EXPAND_FAIL, NOTIFICATION_REQUESTS_FETCH_REQUEST, NOTIFICATION_REQUESTS_FETCH_SUCCESS, NOTIFICATION_REQUESTS_FETCH_FAIL, NOTIFICATION_REQUEST_FETCH_REQUEST, NOTIFICATION_REQUEST_FETCH_SUCCESS, NOTIFICATION_REQUEST_FETCH_FAIL, NOTIFICATION_REQUEST_ACCEPT_REQUEST, NOTIFICATION_REQUEST_DISMISS_REQUEST, NOTIFICATIONS_FOR_REQUEST_FETCH_REQUEST, NOTIFICATIONS_FOR_REQUEST_FETCH_SUCCESS, NOTIFICATIONS_FOR_REQUEST_FETCH_FAIL, NOTIFICATIONS_FOR_REQUEST_EXPAND_REQUEST, NOTIFICATIONS_FOR_REQUEST_EXPAND_SUCCESS, NOTIFICATIONS_FOR_REQUEST_EXPAND_FAIL, } from 'mastodon/actions/notifications'; import { notificationToMap } from './notifications'; const initialState = ImmutableMap({ items: ImmutableList(), isLoading: false, next: null, current: ImmutableMap({ isLoading: false, item: null, removed: false, notifications: ImmutableMap({ items: ImmutableList(), isLoading: false, next: null, }), }), }); const normalizeRequest = request => fromJS({ ...request, account: request.account.id, }); const removeRequest = (state, id) => { if (state.getIn(['current', 'item', 'id']) === id) { state = state.setIn(['current', 'removed'], true); } return state.update('items', list => list.filterNot(item => item.get('id') === id)); }; const removeRequestByAccount = (state, account_id) => { if (state.getIn(['current', 'item', 'account']) === account_id) { state = state.setIn(['current', 'removed'], true); } return state.update('items', list => list.filterNot(item => item.get('account') === account_id)); }; export const notificationRequestsReducer = (state = initialState, action) => { switch(action.type) { case NOTIFICATION_REQUESTS_FETCH_SUCCESS: return state.withMutations(map => { map.update('items', list => ImmutableList(action.requests.map(normalizeRequest)).concat(list)); map.set('isLoading', false); map.update('next', next => next ?? action.next); }); case NOTIFICATION_REQUESTS_EXPAND_SUCCESS: return state.withMutations(map => { map.update('items', list => list.concat(ImmutableList(action.requests.map(normalizeRequest)))); map.set('isLoading', false); map.set('next', action.next); }); case NOTIFICATION_REQUESTS_EXPAND_REQUEST: case NOTIFICATION_REQUESTS_FETCH_REQUEST: return state.set('isLoading', true); case NOTIFICATION_REQUESTS_EXPAND_FAIL: case NOTIFICATION_REQUESTS_FETCH_FAIL: return state.set('isLoading', false); case NOTIFICATION_REQUEST_ACCEPT_REQUEST: case NOTIFICATION_REQUEST_DISMISS_REQUEST: return removeRequest(state, action.id); case blockAccountSuccess.type: return removeRequestByAccount(state, action.payload.relationship.id); case muteAccountSuccess.type: return action.payload.relationship.muting_notifications ? removeRequestByAccount(state, action.payload.relationship.id) : state; case NOTIFICATION_REQUEST_FETCH_REQUEST: return state.set('current', initialState.get('current').set('isLoading', true)); case NOTIFICATION_REQUEST_FETCH_SUCCESS: return state.update('current', map => map.set('isLoading', false).set('item', normalizeRequest(action.request))); case NOTIFICATION_REQUEST_FETCH_FAIL: return state.update('current', map => map.set('isLoading', false)); case NOTIFICATIONS_FOR_REQUEST_FETCH_REQUEST: case NOTIFICATIONS_FOR_REQUEST_EXPAND_REQUEST: return state.setIn(['current', 'notifications', 'isLoading'], true); case NOTIFICATIONS_FOR_REQUEST_FETCH_SUCCESS: return state.updateIn(['current', 'notifications'], map => map.set('isLoading', false).update('items', list => ImmutableList(action.notifications.map(notificationToMap)).concat(list)).update('next', next => next ?? action.next)); case NOTIFICATIONS_FOR_REQUEST_EXPAND_SUCCESS: return state.updateIn(['current', 'notifications'], map => map.set('isLoading', false).update('items', list => list.concat(ImmutableList(action.notifications.map(notificationToMap)))).set('next', action.next)); case NOTIFICATIONS_FOR_REQUEST_FETCH_FAIL: case NOTIFICATIONS_FOR_REQUEST_EXPAND_FAIL: return state.setIn(['current', 'notifications', 'isLoading'], false); default: return state; } };