import type { ChartType } from '@buster/server-shared/metrics';
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';

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,
};