import { get } from 'lodash-es';
import type { ItemDatum, ParsedData } from '../types';

/**
 * 根据 indexesKey 获取数据项
 */
export function getDatumByIndexes(
  data: ParsedData | ParsedData['items'],
  indexes: number[],
): ItemDatum {
  if (indexes.length === 0) return {} as ItemDatum;
  const base = Array.isArray(data) ? data[indexes[0]] : data.items[indexes[0]];
  if (indexes.length === 1) return base;

  const path = indexes
    .slice(1)
    .map((i) => `children[${i}]`)
    .join('.');

  return get(base, path);
}