43 lines
		
	
	
	
		
			1.5 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			43 lines
		
	
	
	
		
			1.5 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| import { CONTEXT_FETCH_SUCCESS } from '../actions/statuses';
 | |
| import { TIMELINE_DELETE } from '../actions/timelines';
 | |
| import { Map as ImmutableMap, List as ImmutableList, fromJS } from 'immutable';
 | |
| 
 | |
| const initialState = ImmutableMap({
 | |
|   ancestors: ImmutableMap(),
 | |
|   descendants: ImmutableMap(),
 | |
| });
 | |
| 
 | |
| const normalizeContext = (state, id, ancestors, descendants) => {
 | |
|   const ancestorsIds   = ancestors.map(ancestor => ancestor.get('id'));
 | |
|   const descendantsIds = descendants.map(descendant => descendant.get('id'));
 | |
| 
 | |
|   return state.withMutations(map => {
 | |
|     map.setIn(['ancestors', id], ancestorsIds);
 | |
|     map.setIn(['descendants', id], descendantsIds);
 | |
|   });
 | |
| };
 | |
| 
 | |
| const deleteFromContexts = (state, id) => {
 | |
|   state.getIn(['descendants', id], ImmutableList()).forEach(descendantId => {
 | |
|     state = state.updateIn(['ancestors', descendantId], ImmutableList(), list => list.filterNot(itemId => itemId === id));
 | |
|   });
 | |
| 
 | |
|   state.getIn(['ancestors', id], ImmutableList()).forEach(ancestorId => {
 | |
|     state = state.updateIn(['descendants', ancestorId], ImmutableList(), list => list.filterNot(itemId => itemId === id));
 | |
|   });
 | |
| 
 | |
|   state = state.deleteIn(['descendants', id]).deleteIn(['ancestors', id]);
 | |
| 
 | |
|   return state;
 | |
| };
 | |
| 
 | |
| export default function contexts(state = initialState, action) {
 | |
|   switch(action.type) {
 | |
|   case CONTEXT_FETCH_SUCCESS:
 | |
|     return normalizeContext(state, action.id, fromJS(action.ancestors), fromJS(action.descendants));
 | |
|   case TIMELINE_DELETE:
 | |
|     return deleteFromContexts(state, action.id);
 | |
|   default:
 | |
|     return state;
 | |
|   }
 | |
| };
 |