UNPKG

@vis.gl/react-mapbox

Version:

React components for Mapbox GL JS

109 lines 3.86 kB
import * as React from 'react'; import { useContext, useEffect, useMemo, useState, useRef, cloneElement } from 'react'; import { MapContext } from "./map.js"; import assert from "../utils/assert.js"; import { deepEqual } from "../utils/deep-equal.js"; let sourceCounter = 0; function createSource(map, id, props) { if (map.isStyleLoaded()) { const options = { ...props }; delete options.id; delete options.children; // @ts-ignore map.addSource(id, options); return map.getSource(id); } return null; } /* eslint-disable complexity */ function updateSource(source, props, prevProps) { assert(props.id === prevProps.id, 'source id changed'); assert(props.type === prevProps.type, 'source type changed'); let changedKey = ''; let changedKeyCount = 0; for (const key in props) { if (key !== 'children' && key !== 'id' && !deepEqual(prevProps[key], props[key])) { changedKey = key; changedKeyCount++; } } if (!changedKeyCount) { return; } const type = props.type; if (type === 'geojson') { source.setData(props.data); } else if (type === 'image') { source.updateImage({ url: props.url, coordinates: props.coordinates }); } else if ('setCoordinates' in source && changedKeyCount === 1 && changedKey === 'coordinates') { source.setCoordinates(props.coordinates); } else if ('setUrl' in source && changedKey === 'url') { source.setUrl(props.url); } else if ('setTiles' in source && changedKey === 'tiles') { source.setTiles(props.tiles); } else { // eslint-disable-next-line console.warn(`Unable to update <Source> prop: ${changedKey}`); } } /* eslint-enable complexity */ export function Source(props) { const map = useContext(MapContext).map.getMap(); const propsRef = useRef(props); const [, setStyleLoaded] = useState(0); const id = useMemo(() => props.id || `jsx-source-${sourceCounter++}`, []); useEffect(() => { if (map) { /* global setTimeout */ const forceUpdate = () => setTimeout(() => setStyleLoaded(version => version + 1), 0); map.on('load', forceUpdate); map.on('styledata', forceUpdate); forceUpdate(); return () => { map.off('load', forceUpdate); map.off('styledata', forceUpdate); // @ts-ignore if (map.style && map.style._loaded && map.getSource(id)) { // Parent effects are destroyed before child ones, see // https://github.com/facebook/react/issues/16728 // Source can only be removed after all child layers are removed const allLayers = map.getStyle()?.layers; if (allLayers) { for (const layer of allLayers) { // @ts-ignore (2339) source does not exist on all layer types if (layer.source === id) { map.removeLayer(layer.id); } } } map.removeSource(id); } }; } return undefined; }, [map]); // @ts-ignore let source = map && map.style && map.getSource(id); if (source) { updateSource(source, props, propsRef.current); } else { source = createSource(map, id, props); } propsRef.current = props; return ((source && React.Children.map(props.children, child => child && cloneElement(child, { source: id }))) || null); } //# sourceMappingURL=source.js.map