buster/packages/ui-components/src/components/charts/BusterChartJS/hooks/useChartSpecificOptions/useChartSpecificOptions.ts

import type { ChartType as ChartJSChartType, PluginChartOptions } from 'chart.js';
import { useMemo } from 'react';
import type { DeepPartial } from 'utility-types';
import type { ChartProps } from '../../core';
import { barOptionsHandler, barPluginsHandler } from './barChartOptions';
import type { ChartSpecificOptionsProps, UseChartSpecificOptionsProps } from './interfaces';
import { pieOptionsHandler, piePluginsHandler } from './pieChartOptions';
import type { ChartType } from '@buster/server-shared/metrics';

export const useChartSpecificOptions = ({
  selectedChartType,
  ...props
}: UseChartSpecificOptionsProps): {
  chartPlugins: DeepPartial<PluginChartOptions<ChartJSChartType>>['plugins'];
  chartOptions: ChartProps<ChartJSChartType>['options'];
} => {
  const chartOptions = useMemo(() => {
    return chartTypeOptionsHandler[selectedChartType](props);
  }, [props]);

  const chartPlugins = useMemo(() => {
    return chartTypePluginsHandler[selectedChartType](props);
  }, [props]);

  return {
    chartPlugins,
    chartOptions,
  };
};

const DEFAULT_OPTIONS: ChartProps<ChartJSChartType>['options'] = {};

const defaultHandler = (): ChartProps<ChartJSChartType>['options'] => DEFAULT_OPTIONS;

const chartTypeOptionsHandler: Record<
  ChartType,
  (props: ChartSpecificOptionsProps) => ChartProps<ChartJSChartType>['options']
> = {
  ['pie']: pieOptionsHandler,
  ['line']: defaultHandler,
  ['scatter']: defaultHandler,
  ['bar']: barOptionsHandler,
  ['combo']: defaultHandler,
  ['metric']: defaultHandler,
  ['table']: defaultHandler,
};

//********** PLUGINS ************ */

const DEFAULT_PLUGINS: DeepPartial<PluginChartOptions<ChartJSChartType>>['plugins'] = {};

const defaultPluginsHandler = () => DEFAULT_PLUGINS;

const chartTypePluginsHandler: Record<
  ChartType,
  (props: ChartSpecificOptionsProps) => DeepPartial<PluginChartOptions<ChartJSChartType>>['plugins']
> = {
  ['pie']: piePluginsHandler,
  ['line']: defaultPluginsHandler,
  ['scatter']: defaultPluginsHandler,
  ['bar']: barPluginsHandler,
  ['combo']: defaultPluginsHandler,
  ['metric']: defaultPluginsHandler,
  ['table']: defaultPluginsHandler,
};