const state = {
  // The `defaultContext` of an extension's commandsModule
  DEFAULT_CONTEXT: 'CORNERSTONE',
  enabledElements: {},
};

/**
 * Sets the enabled element `dom` reference for an active viewport.
 * @param {HTMLElement} dom Active viewport element.
 * @return void
 */
const setEnabledElement = (
  viewportIndex: number,
  element: HTMLElement,
  context?: string
): void => {
  const targetContext = context || state.DEFAULT_CONTEXT;

  state.enabledElements[viewportIndex] = {
    element,
    context: targetContext,
  };
};

/**
 * Grabs the enabled element `dom` reference of an ative viewport.
 *
 * @return {HTMLElement} Active viewport element.
 */
const getEnabledElement = viewportIndex => {
  return state.enabledElements[viewportIndex];
};

const reset = () => {
  state.enabledElements = {};
};

export { setEnabledElement, getEnabledElement, reset };