/// <reference types="@uiw/react-baidu-map-types" />
import React from 'react';
import Map, { MapProps } from '@uiw/react-baidu-map-map';

export interface WithMapProps {
  BMap?: typeof BMap;
  container?: string | HTMLDivElement | null;
  map?: BMap.Map;
}

export default function withMap(Comp: React.ComponentClass<WithMapProps>) {
  return React.forwardRef((props: MapProps, ref?: React.Ref<any>) => (
    <Map {...props}>
      {(mapProps: WithMapProps) => {
        return <Comp ref={ref} {...props} {...mapProps} />;
      }}
    </Map>
  ));
}