mastodon/app/assets/javascripts/components/reducers/timelines.jsx

import { TIMELINE_SET, TIMELINE_UPDATE, TIMELINE_DELETE } from '../actions/timelines';
import { REBLOG_SUCCESS, FAVOURITE_SUCCESS }              from '../actions/interactions';
import { ACCOUNT_SET_SELF }                               from '../actions/accounts';
import Immutable                                          from 'immutable';

const initialState = Immutable.Map({
  home: Immutable.List([]),
  mentions: Immutable.List([]),
  statuses: Immutable.Map(),
  accounts: Immutable.Map(),
  me: null
});

function statusToMaps(state, status) {
  // Separate account
  let account = status.get('account');
  status = status.set('account', account.get('id'));

  // Separate reblog, repeat for reblog
  let reblog = status.get('reblog');

  if (reblog !== null) {
    status = status.set('reblog', reblog.get('id'));
    state  = statusToMaps(state, reblog);
  }

  return state.withMutations(map => {
    map.setIn(['accounts', account.get('id')], account);
    map.setIn(['statuses', status.get('id')], status);
  });
};

function timelineToMaps(state, timeline, statuses) {
  statuses.forEach((status, i) => {
    state = statusToMaps(state, status);
    state = state.setIn([timeline, i], status.get('id'));
  });

  return state;
};

function updateTimelineWithMaps(state, timeline, status) {
  state = statusToMaps(state, status);
  state = state.update(timeline, list => list.unshift(status.get('id')));

  return state;
};

function deleteStatus(state, id) {
  ['home', 'mentions'].forEach(function (timeline) {
    state = state.update(timeline, list => list.filterNot(item => item === id));
  });

  return state.deleteIn(['statuses', id]);
};

export default function timelines(state = initialState, action) {
  switch(action.type) {
    case TIMELINE_SET:
      return timelineToMaps(state, action.timeline, Immutable.fromJS(action.statuses));
    case TIMELINE_UPDATE:
      return updateTimelineWithMaps(state, action.timeline, Immutable.fromJS(action.status));
    case TIMELINE_DELETE:
      return deleteStatus(state, action.id);
    case REBLOG_SUCCESS:
    case FAVOURITE_SUCCESS:
      return statusToMaps(state, Immutable.fromJS(action.response));
    case ACCOUNT_SET_SELF:
      return state.withMutations(map => {
        map.setIn(['accounts', action.account.id], Immutable.fromJS(action.account));
        map.set('me', action.account.id);
      });
    default:
      return state;
  }
};