forked from treehouse/mastodon
120 lines
3.3 KiB
JavaScript
120 lines
3.3 KiB
JavaScript
import {
|
|
NOTIFICATIONS_UPDATE,
|
|
NOTIFICATIONS_REFRESH_SUCCESS,
|
|
NOTIFICATIONS_EXPAND_SUCCESS,
|
|
NOTIFICATIONS_REFRESH_REQUEST,
|
|
NOTIFICATIONS_EXPAND_REQUEST,
|
|
NOTIFICATIONS_REFRESH_FAIL,
|
|
NOTIFICATIONS_EXPAND_FAIL,
|
|
NOTIFICATIONS_CLEAR,
|
|
NOTIFICATIONS_SCROLL_TOP,
|
|
} from '../actions/notifications';
|
|
import { ACCOUNT_BLOCK_SUCCESS } from '../actions/accounts';
|
|
import { TIMELINE_DELETE } from '../actions/timelines';
|
|
import { Map as ImmutableMap, List as ImmutableList } from 'immutable';
|
|
|
|
const initialState = ImmutableMap({
|
|
items: ImmutableList(),
|
|
next: null,
|
|
top: true,
|
|
unread: 0,
|
|
loaded: false,
|
|
isLoading: true,
|
|
});
|
|
|
|
const notificationToMap = notification => ImmutableMap({
|
|
id: notification.id,
|
|
type: notification.type,
|
|
account: notification.account.id,
|
|
status: notification.status ? notification.status.id : null,
|
|
});
|
|
|
|
const normalizeNotification = (state, notification) => {
|
|
const top = state.get('top');
|
|
|
|
if (!top) {
|
|
state = state.update('unread', unread => unread + 1);
|
|
}
|
|
|
|
return state.update('items', list => {
|
|
if (top && list.size > 40) {
|
|
list = list.take(20);
|
|
}
|
|
|
|
return list.unshift(notificationToMap(notification));
|
|
});
|
|
};
|
|
|
|
const normalizeNotifications = (state, notifications, next) => {
|
|
let items = ImmutableList();
|
|
const loaded = state.get('loaded');
|
|
|
|
notifications.forEach((n, i) => {
|
|
items = items.set(i, notificationToMap(n));
|
|
});
|
|
|
|
if (state.get('next') === null) {
|
|
state = state.set('next', next);
|
|
}
|
|
|
|
return state
|
|
.update('items', list => loaded ? items.concat(list) : list.concat(items))
|
|
.set('loaded', true)
|
|
.set('isLoading', false);
|
|
};
|
|
|
|
const appendNormalizedNotifications = (state, notifications, next) => {
|
|
let items = ImmutableList();
|
|
|
|
notifications.forEach((n, i) => {
|
|
items = items.set(i, notificationToMap(n));
|
|
});
|
|
|
|
return state
|
|
.update('items', list => list.concat(items))
|
|
.set('next', next)
|
|
.set('isLoading', false);
|
|
};
|
|
|
|
const filterNotifications = (state, relationship) => {
|
|
return state.update('items', list => list.filterNot(item => item.get('account') === relationship.id));
|
|
};
|
|
|
|
const updateTop = (state, top) => {
|
|
if (top) {
|
|
state = state.set('unread', 0);
|
|
}
|
|
|
|
return state.set('top', top);
|
|
};
|
|
|
|
const deleteByStatus = (state, statusId) => {
|
|
return state.update('items', list => list.filterNot(item => item.get('status') === statusId));
|
|
};
|
|
|
|
export default function notifications(state = initialState, action) {
|
|
switch(action.type) {
|
|
case NOTIFICATIONS_REFRESH_REQUEST:
|
|
case NOTIFICATIONS_EXPAND_REQUEST:
|
|
case NOTIFICATIONS_REFRESH_FAIL:
|
|
case NOTIFICATIONS_EXPAND_FAIL:
|
|
return state.set('isLoading', true);
|
|
case NOTIFICATIONS_SCROLL_TOP:
|
|
return updateTop(state, action.top);
|
|
case NOTIFICATIONS_UPDATE:
|
|
return normalizeNotification(state, action.notification);
|
|
case NOTIFICATIONS_REFRESH_SUCCESS:
|
|
return normalizeNotifications(state, action.notifications, action.next);
|
|
case NOTIFICATIONS_EXPAND_SUCCESS:
|
|
return appendNormalizedNotifications(state, action.notifications, action.next);
|
|
case ACCOUNT_BLOCK_SUCCESS:
|
|
return filterNotifications(state, action.relationship);
|
|
case NOTIFICATIONS_CLEAR:
|
|
return state.set('items', ImmutableList()).set('next', null);
|
|
case TIMELINE_DELETE:
|
|
return deleteByStatus(state, action.id);
|
|
default:
|
|
return state;
|
|
}
|
|
};
|