provide access to created mapboxgl.Marker

Adorkable · Feb 24, 2022 · c4f34bc · c4f34bc
1 parent 3432caf
commit c4f34bc
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/components/mapboxgl/Marker.tsx b/src/components/mapboxgl/Marker.tsx
@@ -1,5 +1,6 @@
 import mapboxgl from 'mapbox-gl'
 import React, {
+  RefObject,
   useCallback,
   useContext,
   useEffect,
@@ -19,6 +20,8 @@ export interface Props {
   popup?: React.ReactElement | undefined
   // eslint-disable-next-line no-unused-vars
   onClick?: (event: MouseEvent) => void
+  mapboxMarkerRef?: (marker: RefObject<mapboxgl.Marker | undefined>) => void
+
   children?: React.ReactElement | undefined
 }
 
@@ -28,7 +31,13 @@ export interface MarkerMethods {
   getMapboxMarker: () => React.MutableRefObject<mapboxgl.Marker>
 }
 
-export const Marker = ({ location, popup, onClick, children }: Props) => {
+export const Marker = ({
+  location,
+  popup,
+  onClick,
+  mapboxMarkerRef,
+  children
+}: Props) => {
   const context = useContext(MapContext)
   const { map } = context
 
@@ -81,6 +90,9 @@ export const Marker = ({ location, popup, onClick, children }: Props) => {
       }
     }
 
+    if (mapboxMarkerRef) {
+      mapboxMarkerRef(mapboxMarker)
+    }
     return () => {
       if (!mapboxMarker.current) {
         return