import type { Spectrum1D } from '@zakodium/nmrium-core';
import type { Draft } from 'immer';

import { isSpectrum1D } from '../../../data/data1d/Spectrum1D/index.js';
import groupByInfoKey from '../../utility/groupByInfoKey.js';
import nucleusToString from '../../utility/nucleusToString.js';
import type { State, VerticalAlignment } from '../Reducer.js';
import { getActiveSpectrum } from '../helper/getActiveSpectrum.js';
import type { ActionType } from '../types/ActionType.js';

import { setDomain } from './DomainActions.js';

type KeyPreferencesAction = ActionType<
  'SET_KEY_PREFERENCES' | 'APPLY_KEY_PREFERENCES',
  { keyCode: string }
>;

export type PreferencesActions = KeyPreferencesAction;

interface AlignmentOptions {
  verticalAlign?: VerticalAlignment | 'auto-check';
  activeTab?: string;
}

function changeSpectrumVerticalAlignment(
  draft: Draft<State>,
  options: AlignmentOptions,
) {
  const { verticalAlign = 'bottom', activeTab } = options;
  const nucleus = activeTab || draft.view.spectra.activeTab;

  if (draft.data && draft.data.length > 0) {
    let dataPerNucleus: Spectrum1D[] = [];
    if (['auto-check', 'stack'].includes(options.verticalAlign || '')) {
      dataPerNucleus = draft.data.filter(
        (datum) => datum.info.nucleus === nucleus && isSpectrum1D(datum),
      ) as Spectrum1D[];
    }
    if (nucleus) {
      if (verticalAlign === 'auto-check') {
        const isFid =
          dataPerNucleus[0]?.info.isFid &&
          !dataPerNucleus.some((d) => !d.info.isFid);

        if (isFid) {
          draft.view.verticalAlign[nucleus] = 'center';
        } else {
          let hasMoreThanOnFt = false;
          let count = 1;
          for (const spectrum of dataPerNucleus) {
            if (count > 1) {
              hasMoreThanOnFt = true;
              break;
            }
            if (spectrum.info.isFt) {
              count++;
            }
          }
          if (hasMoreThanOnFt) {
            draft.view.verticalAlign[nucleus] = 'stack';
          } else {
            draft.view.verticalAlign[nucleus] = 'bottom';
          }
        }
      } else {
        draft.view.verticalAlign[nucleus] = verticalAlign;
      }
    }
  }
}

function handleSetKeyPreferences(
  draft: Draft<State>,
  action: KeyPreferencesAction,
) {
  const { keyCode } = action.payload;
  const {
    data,

    zoom,
    xDomain,
    xDomains,
    yDomain,
    yDomains,
    originDomain,
    margin,
    displayerMode,
    view,
    keysPreferences,
  } = draft;

  const {
    spectra: { activeTab },
  } = view;

  const activeSpectrum = getActiveSpectrum(draft);

  if (activeTab) {
    const groupByNucleus = groupByInfoKey('nucleus');
    const spectraGroupsList = groupByNucleus(data);
    const spectra = spectraGroupsList[activeTab];
    keysPreferences[keyCode] = {
      activeSpectrum,
      displayerMode,
      view,
      zoom,
      xDomain,
      xDomains,
      yDomain,
      yDomains,
      originDomain,
      margin,
      data: spectra,
    };
  }
}
function setSpectraDisplayPreferences(draft: Draft<State>, preferences: any) {
  for (const [index, datum] of draft.data.entries()) {
    if (nucleusToString(datum.info.nucleus) === preferences.activeTab) {
      draft.data[index].display = {
        ...datum.display,
        ...preferences.data[datum.id].display,
      };
    }
  }
}

function handleApplyKeyPreferences(
  draft: Draft<State>,
  action: KeyPreferencesAction,
) {
  const { keyCode } = action.payload;
  const preferences = draft.keysPreferences[keyCode];
  if (preferences) {
    setSpectraDisplayPreferences(draft, preferences);
    draft.view = preferences.view;
    draft.displayerMode = preferences.displayerMode;

    draft.margin = preferences.margin;

    setDomain(draft);

    draft.xDomain = preferences.xDomain;
    draft.xDomains = preferences.xDomains;
    draft.yDomain = preferences.yDomain;
    draft.originDomain = preferences.originDomain;
    draft.yDomains = preferences.yDomains;

    if (draft.displayerMode !== '2D') {
      draft.zoom = preferences.zoom;
    }
  }
}

export {
  changeSpectrumVerticalAlignment,
  handleApplyKeyPreferences,
  handleSetKeyPreferences,
};