All files / scale-control/src useScaleControl.tsx

46.66% Statements 7/15
0% Branches 0/11
33.33% Functions 1/3
46.66% Lines 7/15

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39                1x 1x 1x 1x                                       1x 1x 1x          
import { useEffect, useState } from 'react';
import { useMapContext } from '@uiw/react-baidu-map-map';
import { useProperties, useVisiable } from '@uiw/react-baidu-map-utils';
import { ScaleControlProps } from './';
 
export interface UseScaleControl extends ScaleControlProps {}
 
export function useScaleControl(props = {} as UseScaleControl) {
  const [scaleControl, setScaleControl] = useState<BMap.ScaleControl>();
  const { anchor, offset } = props;
  const { map } = useMapContext();
  useEffect(() => {
    if (map && !scaleControl) {
      const instance = new BMap.ScaleControl({
        anchor: anchor || BMAP_ANCHOR_TOP_LEFT,
        offset,
      });
      map.addControl(instance);
      setScaleControl(instance);
    }
 
    return () => {
      if (map && scaleControl) {
        try {
          map.removeControl(scaleControl);
        } catch (error) {}
      }
    };
    // eslint-disable-next-line react-hooks/exhaustive-deps
  }, [map, scaleControl]);
 
  useVisiable(scaleControl!, props);
  useProperties<BMap.ScaleControl, UseScaleControl>(scaleControl!, props, ['Anchor', 'Offset', 'Unit']);
  return {
    scaleControl,
    setScaleControl,
  };
}