import { combineReducers } from "redux";
import { ADD_BATCH_TO_STATES } from "../actions/states";

export const states = combineReducers({
  task: objectStates("task"),
  room: objectStates("room"),
  rack: objectStates("rack"),
  machine: objectStates("machine")
});

function objectStates(type) {
  return (state = {}, action) => {
    if (action.objectType !== type) {
      return state;
    }

    if (action.type === ADD_BATCH_TO_STATES) {
      const batch = {};
      for (let i in action.objects) {
        batch[action.objects[i].tick] = Object.assign(
          {},
          state[action.objects[i].tick],
          batch[action.objects[i].tick],
          { [action.objects[i][action.objectType + "Id"]]: action.objects[i] }
        );
      }

      return Object.assign({}, state, batch);
    }

    return state;
  };
}