import type { ILayer, IDataset, ILayerType } from '../typings';
import { generateUnRepeatValue, getRandomId } from '../utils/tools';
import { useCallback, useContext } from 'react';
import { cloneDeep } from 'lodash';
import {
  DEFAULT_LINE_LAYER_CONFIG,
  DEFAULT_POINT_LAYER_CONFIG,
  DEFAULT_TRIP_LAYER_CONFIG,
  DEFAULT_POLYGON_LAYER_CONFIG,
  DEFAULT_HEX_LAYER_CONFIG,
  DEFAULT_HEAT_LAYER_CONFIG,
} from '../constants';
import { getLayerTypes } from '../components/AppLayerConfig/common';
import { message } from 'antd';
import { LayerModelContext } from '../context/LayerContext';

const useLayer = () => {
  const { layerList, setLayerList } = useContext(LayerModelContext);

  const getDefaultConfig = useCallback((type: ILayerType) => {
    let config: any = {};
    if (type === 'point') {
      config = DEFAULT_POINT_LAYER_CONFIG;
    }
    if (type === 'line') {
      config = DEFAULT_LINE_LAYER_CONFIG;
    }
    if (type === 'trip') {
      config = DEFAULT_TRIP_LAYER_CONFIG;
    }
    if (type === 'polygon') {
      config = DEFAULT_POLYGON_LAYER_CONFIG;
    }
    if (type === 'hex') {
      config = DEFAULT_HEX_LAYER_CONFIG;
    }

    if (type === 'heat') {
      config = DEFAULT_HEAT_LAYER_CONFIG;
    }
    return cloneDeep(config);
  }, []);

  const addLayer: (
    dataset: IDataset,
    initType?: ILayerType,
  ) => ILayer | undefined = useCallback(
    (dataset, initType) => {
      const { id } = dataset;
      const layerTypes = getLayerTypes(dataset);
      if (!layerTypes.length) {
        message.success('当前无可创建的图层');
        return;
      }
      const type = initType ?? layerTypes[0].value;
      const newLayer: ILayer = {
        id: getRandomId('layer'),
        name: generateUnRepeatValue<ILayer, string>(layerList, 'name', '图层'),
        order: layerList.length + 1,
        datasetId: id,
        createTime: Date.now(),
        config: getDefaultConfig(type),
        type: type,
        visible: true,
        zIndex: 0,
      };
      const newlayerList = [...layerList, newLayer];
      setLayerList(newlayerList);
      return newLayer;
    },
    [getDefaultConfig, layerList, setLayerList],
  );

  const copyLayer = useCallback(
    (filter: ILayer) => {
      const newFilter: ILayer = {
        ...cloneDeep(filter),
        id: getRandomId('layer'),
        name: generateUnRepeatValue<ILayer, string>(layerList, 'name', '图层'),
        order: layerList.length + 1,
        createTime: Date.now(),
      };

      setLayerList([...layerList, newFilter]);

      return newFilter;
    },
    [layerList, setLayerList],
  );

  return {
    addLayer,
    copyLayer,
    getDefaultConfig,
  };
};

export default useLayer;