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

import { getDefaultRangesViewState } from '../../hooks/useActiveSpectrumRangesViewState.js';
import type { State } from '../Reducer.js';

import { getActiveSpectrum } from './getActiveSpectrum.js';

export function setRangesViewProperty<T extends keyof RangesViewState>(
  draft: Draft<State>,
  key: T,
  options: {
    spectrumKey?: string;
    value: (value: RangesViewState[T]) => RangesViewState[T];
  },
): void;
export function setRangesViewProperty<T extends keyof RangesViewState>(
  draft: Draft<State>,
  key: T,
  options: {
    spectrumKey?: string;
    value: RangesViewState[T];
  },
): void;

export function setRangesViewProperty<T extends keyof RangesViewState>(
  draft: Draft<State>,
  key: T,
  options: {
    spectrumKey?: string;
    value:
      | ((value: RangesViewState[T]) => RangesViewState[T])
      | RangesViewState[T];
  },
) {
  const rangesView = draft.view.ranges;
  const { value, spectrumKey: externalSpectrumKey } = options;

  let spectrumKey = externalSpectrumKey;

  if (!spectrumKey) {
    const activeSpectrum = getActiveSpectrum(draft);

    if (!activeSpectrum) return;

    spectrumKey = activeSpectrum.id;
  }

  initializeRangeViewObject(draft, spectrumKey);

  rangesView[spectrumKey][key] =
    typeof value === 'function' ? value(rangesView[spectrumKey][key]) : value;
}

export function initializeRangeViewObject(
  draft: Draft<State>,
  spectrumID: string,
) {
  const {
    ranges,
    spectra: { activeTab: nucleus },
  } = draft.view;

  if (spectrumID in ranges) return;

  const defaultRangesView = getDefaultRangesViewState(nucleus);
  ranges[spectrumID] = defaultRangesView;
}