huridocs/uwazi

import Immutable from 'immutable';

import * as actions from 'app/ContextMenu/actions/actionTypes';
import * as ViewerActions from 'app/Viewer/actions/actionTypes';
import * as UploadActions from 'app/Uploads/actions/actionTypes';
import * as LibraryActions from 'app/Library/actions/actionTypes';

const initialState = { open: false, menu: null };

const panels = {
  referencePanel: 'ViewerSaveReferenceMenu',
  targetReferencePanel: 'ViewerSaveTargetReferenceMenu',
  viewMetadataPanel: 'MetadataPanelMenu',
};

export default function contextMenuReducer(state = initialState, action = {}) {
  if (action.type === actions.OPEN_MENU) {
    return state.set('open', true);
  }

  if (action.type === actions.CLOSE_MENU) {
    return state.set('open', false);
  }

  if (action.type === ViewerActions.SET_SELECTION) {
    return state.set('type', 'ViewerTextSelectedMenu');
  }

  if (action.type === LibraryActions.ENTER_LIBRARY) {
    return state.set('type', 'LibraryMenu');
  }

  if (action.type === UploadActions.ENTER_UPLOADS_SECTION) {
    return state.set('type', 'UploadsMenu');
  }

  if (action.type === ViewerActions.OPEN_PANEL) {
    return state.set('type', panels[action.panel]);
  }

  if (action.type === ViewerActions.SHOW_TEXT_SELECTION_MENU) {
    return state.set('type', 'ViewerTextSelectedMenu');
  }

  if (
    action.type === ViewerActions.UNSET_SELECTION ||
    action.type === ViewerActions.LOAD_DEFAULT_VIEWER_MENU ||
    action.type === ViewerActions.ADD_REFERENCE ||
    action.type === ViewerActions.CLOSE_PANEL ||
    action.type === 'viewer/targetDoc/SET'
  ) {
    return state.set('type', 'ViewerDefaultMenu');
  }

  return Immutable.fromJS(state);
}