/**
 * @upsetjs/bundle
 * https://github.com/upsetjs/upsetjs
 *
 * Copyright (c) 2021 Samuel Gratzl <sam@sgratzl.com>
 */

import {
  categoricalAddon as categoricalAddonImpl,
  ICategory,
  ICategoricalStyleProps,
  ICategoryBins,
  categoricalAggregatedAddon as categoricalAggregatedAddonImpl,
} from '@upsetjs/addons';
import type { UpSetAddon } from '../react';

export type { ICategoricalStyleProps, ICategory } from '@upsetjs/addons';
export { categoricalHistogram, ICategoryBin, ICategories, ICategoryBins } from '@upsetjs/math';

/**
 * generates a categorical addon to render categorical distribution as UpSet.js addon for aggregated set data
 * @param prop accessor or name of the property within the element
 * @param elems list of elements or their minimum / maximum value for specifying the data domain
 * @param options additional options
 */
export function categoricalAddon<T>(
  prop: keyof T | ((v: T) => string),
  elems: readonly T[] | { categories: readonly (string | ICategory)[] },
  options: Partial<Pick<UpSetAddon<T>, 'size' | 'position' | 'name'>> & ICategoricalStyleProps = {}
): UpSetAddon<T> {
  return categoricalAddonImpl(prop, elems, options);
}

/**
 * generates a categorical addon to render categorical distribution as UpSet.js addon for aggregated set data
 * @param prop accessor or name of the property within the element
 * @param elems list of elements or their minimum / maximum value for specifying the data domain
 * @param options additional options
 */
export function categoricalAggregatedAddon<T>(
  acc: (v: readonly T[]) => ICategoryBins,
  options: Partial<Pick<UpSetAddon<T>, 'size' | 'position' | 'name'>> & ICategoricalStyleProps = {}
): UpSetAddon<T> {
  return categoricalAggregatedAddonImpl(acc, options);
}