import {
  TypeKeys,
  CustomNodeAction,
  AddCustomNodeAction,
  RemoveCustomNodeAction
} from 'actions/config';
import { CustomNodesState as State } from './types';

const addCustomNode = (state: State, { payload }: AddCustomNodeAction): State => ({
  ...state,
  [payload.id]: payload
});

function removeCustomNode(state: State, { payload }: RemoveCustomNodeAction): State {
  const stateCopy = { ...state };
  Reflect.deleteProperty(stateCopy, payload);
  return stateCopy;
}

export const customNodes = (state: State = {}, action: CustomNodeAction): State => {
  switch (action.type) {
    case TypeKeys.CONFIG_ADD_CUSTOM_NODE:
      return addCustomNode(state, action);
    case TypeKeys.CONFIG_REMOVE_CUSTOM_NODE:
      return removeCustomNode(state, action);
    default:
      return state;
  }
};