import type { Draft } from 'immer';

import type { BrushAxis } from '../../EventsTrackers/BrushTracker.js';
import type { State } from '../Reducer.js';

export interface HistoryItem {
  xDomain: number[];
  yDomain: number[];
}

export type ZoomHistory = Record<string, HistoryItem[]>;
interface BaseZoomHistoryManager {
  historyStack: HistoryItem[];
  push: (value: HistoryItem) => void;
  setBase: (value: HistoryItem) => void;
  pop: () => HistoryItem;
  getLast: () => HistoryItem;
}
interface ZoomHistoryManager extends BaseZoomHistoryManager {
  clear: () => void;
}

export default function zoomHistoryManager(
  zoomHistory: ZoomHistory,
  nucleus: string,
): ZoomHistoryManager {
  if (!zoomHistory[nucleus] && !Array.isArray(zoomHistory[nucleus])) {
    zoomHistory[nucleus] = [];
  }
  const history = zoomHistory[nucleus];

  const setBase = prepareSetBase(history);
  const push = preparePush(history);
  const pop = preparePop(history);
  const getLast = prepareGetLast(history);
  const clear = () => (zoomHistory[nucleus] = []);

  return { historyStack: history, push, pop, getLast, clear, setBase };
}

function preparePush(historyStack: any) {
  return (val: HistoryItem) => {
    historyStack.push(val);
  };
}

function prepareSetBase(historyStack: any) {
  return (val: HistoryItem) => {
    if (historyStack.length === 0) {
      historyStack[0] = val;
    }
  };
}

function popZoomHistory(historyStack: any) {
  const val = historyStack.pop();

  if (historyStack.length === 0) {
    return null;
  }

  return val ? historyStack.at(-1) : null;
}

export function preparePop(historyStack: any) {
  return () => popZoomHistory(historyStack);
}

function prepareGetLast(historyStack: any) {
  return () => {
    if (historyStack.length === 0) return null;
    return historyStack.at(-1);
  };
}

export function addToBrushHistory(
  draft: Draft<State>,
  options: { axis?: BrushAxis | null; xDomain: number[]; yDomain: number[] },
) {
  const { displayerMode } = draft;
  const { axis } = options;
  let { xDomain, yDomain } = options;

  const brushHistory = zoomHistoryManager(
    draft.zoom.history,
    draft.view.spectra.activeTab,
  );

  brushHistory.setBase({ xDomain: draft.xDomain, yDomain: draft.yDomain });

  switch (axis) {
    case 'XY':
      draft.xDomain = xDomain;
      draft.yDomain = yDomain;
      break;
    case 'X':
      draft.xDomain = xDomain;
      yDomain = draft.yDomain;
      break;
    case 'Y':
      draft.yDomain = yDomain;
      xDomain = draft.xDomain;

      break;
    default:
      break;
  }

  if (displayerMode === '1D' && axis && ['XY', 'Y'].includes(axis)) {
    const ids = Object.keys(draft.yDomains);
    for (const id of ids) {
      draft.yDomains[id] = yDomain;
    }
  }

  if (brushHistory) {
    brushHistory.push({ xDomain, yDomain });
  }
}