import type { Spectrum } from '@zakodium/nmrium-core';
import { useMemo } from 'react';

import { useInsetOptions } from '../1d/inset/InsetProvider.js';

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

function getSpectrum(
  data: Spectrum[],
  spectrumKey: string,
  defaultValue?: any,
) {
  return data.find((datum) => datum.id === spectrumKey) || defaultValue;
}

export function useSpectrumWithDataSource(
  dataSource: any,
  defaultValue: any = null,
) {
  const activeSpectrum = useActiveSpectrum();
  const inset = useInsetOptions();

  return useMemo<Spectrum>(() => {
    if (!dataSource) return defaultValue;

    if (inset) {
      return getSpectrum(dataSource, inset.spectrumKey, defaultValue);
    }

    if (dataSource && activeSpectrum?.selected) {
      return getSpectrum(dataSource, activeSpectrum.id, defaultValue);
    }

    return defaultValue;
  }, [activeSpectrum, dataSource, defaultValue, inset]);
}