82 lines
2.7 KiB
JavaScript
82 lines
2.7 KiB
JavaScript
import {
|
|
ACCOUNT_BLOCK_SUCCESS,
|
|
ACCOUNT_MUTE_SUCCESS,
|
|
} from '../actions/accounts';
|
|
import { CONTEXT_FETCH_SUCCESS } from '../actions/statuses';
|
|
import { TIMELINE_DELETE, TIMELINE_CONTEXT_UPDATE } from '../actions/timelines';
|
|
import { Map as ImmutableMap, List as ImmutableList } from 'immutable';
|
|
|
|
const initialState = ImmutableMap({
|
|
ancestors: ImmutableMap(),
|
|
descendants: ImmutableMap(),
|
|
});
|
|
|
|
const normalizeContext = (state, id, ancestors, descendants) => {
|
|
const ancestorsIds = ImmutableList(ancestors.map(ancestor => ancestor.id));
|
|
const descendantsIds = ImmutableList(descendants.map(descendant => descendant.id));
|
|
|
|
return state.withMutations(map => {
|
|
map.setIn(['ancestors', id], ancestorsIds);
|
|
map.setIn(['descendants', id], descendantsIds);
|
|
});
|
|
};
|
|
|
|
const deleteFromContexts = (immutableState, ids) => immutableState.withMutations(state => {
|
|
state.update('ancestors', immutableAncestors => immutableAncestors.withMutations(ancestors => {
|
|
state.update('descendants', immutableDescendants => immutableDescendants.withMutations(descendants => {
|
|
ids.forEach(id => {
|
|
descendants.get(id, ImmutableList()).forEach(descendantId => {
|
|
ancestors.update(descendantId, ImmutableList(), list => list.filterNot(itemId => itemId === id));
|
|
});
|
|
|
|
ancestors.get(id, ImmutableList()).forEach(ancestorId => {
|
|
descendants.update(ancestorId, ImmutableList(), list => list.filterNot(itemId => itemId === id));
|
|
});
|
|
|
|
descendants.delete(id);
|
|
ancestors.delete(id);
|
|
});
|
|
}));
|
|
}));
|
|
});
|
|
|
|
const filterContexts = (state, relationship, statuses) => {
|
|
const ownedStatusIds = statuses
|
|
.filter(status => status.get('account') === relationship.id)
|
|
.map(status => status.get('id'));
|
|
|
|
return deleteFromContexts(state, ownedStatusIds);
|
|
};
|
|
|
|
const updateContext = (state, status, references) => {
|
|
return state.update('descendants', map => {
|
|
references.forEach(parentId => {
|
|
map = map.update(parentId, ImmutableList(), list => {
|
|
if (list.includes(status.id)) {
|
|
return list;
|
|
}
|
|
|
|
return list.push(status.id);
|
|
});
|
|
});
|
|
|
|
return map;
|
|
});
|
|
};
|
|
|
|
export default function contexts(state = initialState, action) {
|
|
switch(action.type) {
|
|
case ACCOUNT_BLOCK_SUCCESS:
|
|
case ACCOUNT_MUTE_SUCCESS:
|
|
return filterContexts(state, action.relationship, action.statuses);
|
|
case CONTEXT_FETCH_SUCCESS:
|
|
return normalizeContext(state, action.id, action.ancestors, action.descendants);
|
|
case TIMELINE_DELETE:
|
|
return deleteFromContexts(state, [action.id]);
|
|
case TIMELINE_CONTEXT_UPDATE:
|
|
return updateContext(state, action.status, action.references);
|
|
default:
|
|
return state;
|
|
}
|
|
};
|