import { Map, List, fromJS } from 'immutable';
import {
  ENTRY_REQUEST, ENTRY_SUCCESS, ENTRIES_REQUEST, ENTRIES_SUCCESS
} from '../actions/entries';

export function entries(state = Map({entities: Map(), pages: Map()}), action) {
  switch (action.type) {
    case ENTRY_REQUEST:
      return state.setIn(['entities', `${action.payload.collection}.${action.payload.slug}`, 'isFetching'], true);
    case ENTRY_SUCCESS:
      return state.setIn(
        ['entities', `${action.payload.collection}.${action.payload.entry.slug}`],
        fromJS(action.payload.entry)
      );
    case ENTRIES_REQUEST:
      return state.setIn(['pages', action.payload.collection, 'isFetching'], true);
    case ENTRIES_SUCCESS:
      const { collection, entries, pages } = action.payload;
      return state.withMutations((map) => {
        entries.forEach((entry) => (
          map.setIn(['entities', `${collection}.${entry.slug}`], fromJS(entry).set('isFetching', false))
        ));
        map.setIn(['pages', collection], Map({
          ...pages,
          ids: List(entries.map((entry) => entry.slug))
        }));
      });
    default:
      return state;
  }
}

export function selectEntry(state, collection, slug) {
  return state.entries.getIn(['entities', `${collection}.${slug}`]);
}

export function selectEntries(state, collection) {
  const slugs = state.entries.getIn(['pages', collection, 'ids']);
  return slugs && slugs.map((slug) => selectEntry(state, collection, slug));
}