Merge pull request #220 from devonfw-forge/sp2_219-disabling-sharing-location-force-refresh_be

SP2 219 disabling sharing location must force refresh

Author: lpinon

wayat/backend/app/business/wayat_management/controllers/user.py

@@ -14,6 +14,7 @@
     UpdatePreferencesRequest, UserWithPhoneResponse, PendingFriendsRequestsResponse, HandleFriendRequestRequest,
     UserDTO,
 )
+from app.business.wayat_management.services.map import MapService
 from app.business.wayat_management.services.user import UserService
 from app.common import get_user
 from app.common.exceptions.http import HTTPError
@@ -87,9 +88,14 @@ async def add_contact(request: AddContactsRequest, user_service: UserService = D
 @router.post("/preferences", description="Update the preferences of a user")
 async def update_preferences(request: UpdatePreferencesRequest,
                              user_service: UserService = Depends(UserService),
+                             map_service: MapService = Depends(MapService),
                              user: FirebaseAuthenticatedUser = Depends(get_user())):
     logger.info(f"Updating preferences for user {user.uid} with values {request.dict(exclude_unset=True)}")
     await user_service.update_user(user.uid, **request.dict(exclude_unset=True))
+    if request.share_location is False:
+        # share_location was set to false in this request, so we must delete the user
+        # from all the maps in which he's present
+        await map_service.vanish_user(user.uid)
 
 
 @router.get("/contacts", description="Get the list of contacts for a user", response_model=ListUsersWithPhoneResponse)

wayat/backend/app/business/wayat_management/services/map.py

@@ -143,3 +143,8 @@ def _in_range(self, latitude: float, longitude: float, contact_location: Locatio
         distance = haversine_distance(latitude, longitude, location.latitude, location.longitude)
         logger.debug(f"Distance of {distance} calculated")
         return distance < self._distance_threshold
+
+    async def vanish_user(self, uid: str):
+        coroutines = [self.regenerate_map_status(uid=contact_uid)
+                      for contact_uid in await self._status_repository.find_maps_containing_user(uid)]
+        await asyncio.gather(*coroutines)

wayat/backend/app/domain/wayat_management/models/status.py

@@ -15,4 +15,5 @@ class ContactRefInfo(BaseModel):
 
 class AppStatusEntity(BaseFirebaseModel):
     active: bool = False
-    contact_refs: list[ContactRefInfo] = Field(default_factory=list)
+    contact_refs: list[ContactRefInfo] = Field(default_factory=list)
+    contact_refs_members: list[str] = Field(default_factory=list)

wayat/backend/app/domain/wayat_management/repositories/status.py

@@ -1,4 +1,7 @@
+from typing import AsyncIterable
+
 from fastapi import Depends
+from google.cloud.firestore import DocumentSnapshot
 
 from app.common.infra.gcp.base_firebase_repository import BaseFirestoreRepository, get_async_client
 from app.domain.wayat_management.models.status import AppStatusEntity, ContactRefInfo
@@ -14,7 +17,8 @@ async def initialize(self, uid: str):
 
     async def set_contact_refs(self, uid: str, contact_refs: list[ContactRefInfo]):
         await self.update(document_id=uid, data={
-            "contact_refs": [contact.dict() for contact in contact_refs]
+            "contact_refs": [contact.dict() for contact in contact_refs],
+            "contact_refs_members": [contact.uid for contact in contact_refs]
         })
 
     async def set_active(self, uid: str, value: bool, read_first=True):
@@ -34,3 +38,17 @@ async def set_active_batch(self, uid_list: list[str], value: bool):
                 ref = self._get_document_reference(uid)
                 batch.update(ref, {"active": value})
             await batch.commit()
+
+    async def find_maps_containing_user(self, uid: str) -> list[str]:
+        """
+        Given a User uid, returns a list of uids containing those Users whose map contains the initial User.
+        :param uid: The User to search in the maps
+        :return: A list of uids of Users that have the given uid in their map
+        """
+        result_stream = (
+            self
+            ._get_collection_reference()
+            .where("contact_refs_members", "array_contains", uid)
+            .stream()
+        )  # type: AsyncIterable[DocumentSnapshot]
+        return [document.id async for document in result_stream]