Add option to find own location in map views

src/components/views/beacon/BeaconViewDialog.tsx

@@ -125,6 +125,9 @@ const BeaconViewDialog: React.FC<IProps> = ({ initialFocusedBeacon, roomId, matr
         setFocusedBeaconState({ beacon, ts: Date.now() });
     };
 
+    const hasOwnBeacon =
+        liveBeacons.filter((beacon) => beacon?.beaconInfoOwner === matrixClient.getUserId()).length > 0;
+
     return (
         <BaseDialog className="mx_BeaconViewDialog" onFinished={onFinished} fixedWidth={false}>
             <MatrixClientContext.Provider value={matrixClient}>
@@ -136,6 +139,7 @@ const BeaconViewDialog: React.FC<IProps> = ({ initialFocusedBeacon, roomId, matr
                         interactive
                         onError={setMapDisplayError}
                         className="mx_BeaconViewDialog_map"
+                        allowGeolocate={!hasOwnBeacon}
                     >
                         {({ map }: { map: maplibregl.Map }) => (
                             <>

src/components/views/location/LocationPicker.tsx

@@ -23,10 +23,9 @@ import { ClientEvent, IClientWellKnown } from "matrix-js-sdk/src/client";
 import { _t } from "../../../languageHandler";
 import MatrixClientContext from "../../../contexts/MatrixClientContext";
 import Modal from "../../../Modal";
-import SdkConfig from "../../../SdkConfig";
 import { tileServerFromWellKnown } from "../../../utils/WellKnownUtils";
 import { GenericPosition, genericPositionFromGeolocation, getGeoUri } from "../../../utils/beacon";
-import { LocationShareError, findMapStyleUrl } from "../../../utils/location";
+import { LocationShareError, findMapStyleUrl, positionFailureMessage } from "../../../utils/location";
 import ErrorDialog from "../dialogs/ErrorDialog";
 import AccessibleButton from "../elements/AccessibleButton";
 import { MapError } from "./MapError";
@@ -266,21 +265,3 @@ class LocationPicker extends React.Component<ILocationPickerProps, IState> {
 }
 
 export default LocationPicker;
-
-function positionFailureMessage(code: number): string {
-    const brand = SdkConfig.get().brand;
-    switch (code) {
-        case 1:
-            return _t(
-                "%(brand)s was denied permission to fetch your location. " +
-                    "Please allow location access in your browser settings.",
-                { brand },
-            );
-        case 2:
-            return _t("Failed to fetch your location. Please try again later.");
-        case 3:
-            return _t("Timed out trying to fetch your location. Please try again later.");
-        case 4:
-            return _t("Unknown error fetching location. Please try again later.");
-    }
-}

src/components/views/location/LocationViewDialog.tsx

@@ -68,6 +68,7 @@ export default class LocationViewDialog extends React.Component<IProps, IState>
                     onError={this.onError}
                     interactive
                     className="mx_LocationViewDialog_map"
+                    allowGeolocate={true}
                 >
                     {({ map }) => (
                         <>

src/components/views/location/Map.tsx

@@ -14,33 +14,38 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 
-import React, { ReactNode, useContext, useEffect } from "react";
+import React, { ReactNode, useContext, useEffect, useState } from "react";
 import classNames from "classnames";
 import * as maplibregl from "maplibre-gl";
 import { ClientEvent, IClientWellKnown } from "matrix-js-sdk/src/matrix";
 import { logger } from "matrix-js-sdk/src/logger";
 
 import MatrixClientContext from "../../../contexts/MatrixClientContext";
 import { useEventEmitterState } from "../../../hooks/useEventEmitter";
-import { parseGeoUri } from "../../../utils/location";
+import { parseGeoUri, positionFailureMessage } from "../../../utils/location";
 import { tileServerFromWellKnown } from "../../../utils/WellKnownUtils";
 import { useMap } from "../../../utils/location/useMap";
 import { Bounds } from "../../../utils/beacon/bounds";
+import Modal from "../../../Modal";
+import ErrorDialog from "../dialogs/ErrorDialog";
+import { _t } from "../../../languageHandler";
 
 const useMapWithStyle = ({
     id,
     centerGeoUri,
     onError,
     interactive,
     bounds,
+    allowGeolocate,
 }: {
     id: string;
     centerGeoUri?: string;
+    onError?(error: Error): void;
     interactive?: boolean;
     bounds?: Bounds;
-    onError(error: Error): void;
+    allowGeolocate?: boolean;
 }): {
-    map: maplibregl.Map;
+    map: maplibregl.Map | undefined;
     bodyId: string;
 } => {
     const bodyId = `mx_Map_${id}`;
@@ -86,12 +91,41 @@ const useMapWithStyle = ({
         }
     }, [map, bounds]);
 
+    const [geolocate] = useState(
+        allowGeolocate
+            ? new maplibregl.GeolocateControl({
+                  positionOptions: {
+                      enableHighAccuracy: true,
+                  },
+                  trackUserLocation: false,
+              })
+            : null,
+    );
+
+    useEffect(() => {
+        if (map && geolocate) {
+            map.addControl(geolocate);
+            geolocate.on("error", onGeolocateError);
+            return () => {
+                geolocate.off("error", onGeolocateError);
+            };
+        }
+    }, [map, geolocate]);
+
     return {
         map,
         bodyId,
     };
 };
 
+const onGeolocateError = (e: GeolocationPositionError): void => {
+    logger.error("Could not fetch location", e);
+    Modal.createDialog(ErrorDialog, {
+        title: _t("Could not fetch location"),
+        description: positionFailureMessage(e.code) ?? "",
+    });
+};
+
 interface MapProps {
     id: string;
     interactive?: boolean;
@@ -105,13 +139,24 @@ interface MapProps {
     centerGeoUri?: string;
     bounds?: Bounds;
     className?: string;
+    allowGeolocate?: boolean;
     onClick?: () => void;
     onError?: (error: Error) => void;
     children?: (renderProps: { map: maplibregl.Map }) => ReactNode;
 }
 
-const Map: React.FC<MapProps> = ({ bounds, centerGeoUri, children, className, id, interactive, onError, onClick }) => {
-    const { map, bodyId } = useMapWithStyle({ centerGeoUri, onError, id, interactive, bounds });
+const Map: React.FC<MapProps> = ({
+    bounds,
+    centerGeoUri,
+    children,
+    className,
+    allowGeolocate,
+    id,
+    interactive,
+    onError,
+    onClick,
+}) => {
+    const { map, bodyId } = useMapWithStyle({ centerGeoUri, onError, id, interactive, bounds, allowGeolocate });
 
     const onMapClick = (event: React.MouseEvent<HTMLDivElement, MouseEvent>): void => {
         // Eat click events when clicking the attribution button

src/i18n/strings/en_EN.json

@@ -787,6 +787,10 @@
     "Reset bearing to north": "Reset bearing to north",
     "Zoom in": "Zoom in",
     "Zoom out": "Zoom out",
+    "%(brand)s was denied permission to fetch your location. Please allow location access in your browser settings.": "%(brand)s was denied permission to fetch your location. Please allow location access in your browser settings.",
+    "Failed to fetch your location. Please try again later.": "Failed to fetch your location. Please try again later.",
+    "Timed out trying to fetch your location. Please try again later.": "Timed out trying to fetch your location. Please try again later.",
+    "Unknown error fetching location. Please try again later.": "Unknown error fetching location. Please try again later.",
     "Are you sure you want to exit during this export?": "Are you sure you want to exit during this export?",
     "Unnamed Room": "Unnamed Room",
     "Generating a ZIP": "Generating a ZIP",
@@ -2449,10 +2453,6 @@
     "Click to move the pin": "Click to move the pin",
     "Click to drop a pin": "Click to drop a pin",
     "Share location": "Share location",
-    "%(brand)s was denied permission to fetch your location. Please allow location access in your browser settings.": "%(brand)s was denied permission to fetch your location. Please allow location access in your browser settings.",
-    "Failed to fetch your location. Please try again later.": "Failed to fetch your location. Please try again later.",
-    "Timed out trying to fetch your location. Please try again later.": "Timed out trying to fetch your location. Please try again later.",
-    "Unknown error fetching location. Please try again later.": "Unknown error fetching location. Please try again later.",
     "You don't have permission to share locations": "You don't have permission to share locations",
     "You need to have the right permissions in order to share locations in this room.": "You need to have the right permissions in order to share locations in this room.",
     "We couldn't send your location": "We couldn't send your location",

src/utils/location/index.ts

@@ -20,3 +20,4 @@ export * from "./locationEventGeoUri";
 export * from "./LocationShareErrors";
 export * from "./map";
 export * from "./parseGeoUri";
+export * from "./positionFailureMessage";

src/utils/location/positionFailureMessage.ts

@@ -0,0 +1,36 @@
+/*
+Copyright 2023 The Matrix.org Foundation C.I.C.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+import { _t } from "../../languageHandler";
+import SdkConfig from "../../SdkConfig";
+
+export const positionFailureMessage = (code: number): string | undefined => {
+    const brand = SdkConfig.get().brand;
+    switch (code) {
+        case 1:
+            return _t(
+                "%(brand)s was denied permission to fetch your location. " +
+                    "Please allow location access in your browser settings.",
+                { brand },
+            );
+        case 2:
+            return _t("Failed to fetch your location. Please try again later.");
+        case 3:
+            return _t("Timed out trying to fetch your location. Please try again later.");
+        case 4:
+            return _t("Unknown error fetching location. Please try again later.");
+    }
+};

src/utils/location/useMap.ts

@@ -21,7 +21,7 @@ import { createMap } from "./map";
 
 interface UseMapProps {
     bodyId: string;
-    onError: (error: Error) => void;
+    onError?: (error: Error) => void;
     interactive?: boolean;
 }
 
@@ -39,7 +39,7 @@ export const useMap = ({ interactive, bodyId, onError }: UseMapProps): MapLibreM
             try {
                 setMap(createMap(interactive, bodyId, onError));
             } catch (error) {
-                onError(error);
+                onError?.(error);
             }
             return () => {
                 if (map) {

test/components/views/location/__snapshots__/LocationViewDialog-test.tsx.snap

@@ -2,6 +2,7 @@
 
 exports[`<LocationViewDialog /> renders map correctly 1`] = `
 <Map
+  allowGeolocate={true}
   centerGeoUri="geo:51.5076,-0.1276"
   className="mx_LocationViewDialog_map"
   id="mx_LocationViewDialog_$2"
@@ -29,12 +30,27 @@ exports[`<LocationViewDialog /> renders map correctly 1`] = `
                 MockAttributionControl {},
                 "top-right",
               ],
+              [
+                MockGeolocateControl {
+                  "_events": {
+                    "error": [Function],
+                  },
+                  "_eventsCount": 1,
+                  "_maxListeners": undefined,
+                  "trigger": [MockFunction],
+                  Symbol(kCapture): false,
+                },
+              ],
             ],
             "results": [
               {
                 "type": "return",
                 "value": undefined,
               },
+              {
+                "type": "return",
+                "value": undefined,
+              },
             ],
           },
           "fitBounds": [MockFunction],
@@ -97,12 +113,27 @@ exports[`<LocationViewDialog /> renders map correctly 1`] = `
                 MockAttributionControl {},
                 "top-right",
               ],
+              [
+                MockGeolocateControl {
+                  "_events": {
+                    "error": [Function],
+                  },
+                  "_eventsCount": 1,
+                  "_maxListeners": undefined,
+                  "trigger": [MockFunction],
+                  Symbol(kCapture): false,
+                },
+              ],
             ],
             "results": [
               {
                 "type": "return",
                 "value": undefined,
               },
+              {
+                "type": "return",
+                "value": undefined,
+              },
             ],
           },
           "fitBounds": [MockFunction],

test/components/views/messages/MBeaconBody-test.tsx

@@ -436,7 +436,7 @@ describe("<MBeaconBody />", () => {
         beforeEach(() => {
             // mock map utils to raise MapStyleUrlNotConfigured error
             jest.spyOn(mapUtilHooks, "useMap").mockImplementation(({ onError }) => {
-                onError(new Error(LocationShareError.MapStyleUrlNotConfigured));
+                onError?.(new Error(LocationShareError.MapStyleUrlNotConfigured));
                 return mockMap;
             });
         });