feat(cha-769): add support to live locations

Author: Rafael Marinho

stream_chat/async_chat/client.py

@@ -22,6 +22,7 @@
 from stream_chat.types.base import SortParam
 from stream_chat.types.campaign import CampaignData, QueryCampaignsOptions
 from stream_chat.types.draft import QueryDraftsFilter, QueryDraftsOptions
+from stream_chat.types.shared_locations import SharedLocationsOptions
 from stream_chat.types.segment import (
     QuerySegmentsOptions,
     QuerySegmentTargetsOptions,
@@ -884,3 +885,16 @@ async def __aexit__(
         exc_tb: Optional[TracebackType],
     ) -> None:
         await self.close()
+
+    async def get_user_locations(self, user_id: str, **options: Any) -> StreamResponse:
+        params = {"user_id": user_id, **options}
+        return await self.get("users/live_locations", params=params)
+
+    async def update_user_location(
+        self, message_id: str, options: Optional[SharedLocationsOptions] = None,
+    ) -> StreamResponse:
+        data = {"message_id": message_id, **options}
+        if options is not None:
+            data.update(cast(dict, options))
+        return await self.put("users/live_locations", data=data)
+    

stream_chat/base/client.py

@@ -10,6 +10,7 @@
 from stream_chat.types.base import SortParam
 from stream_chat.types.campaign import CampaignData, QueryCampaignsOptions
 from stream_chat.types.draft import QueryDraftsFilter, QueryDraftsOptions
+from stream_chat.types.shared_locations import SharedLocationsOptions
 from stream_chat.types.segment import (
     QuerySegmentsOptions,
     QuerySegmentTargetsOptions,
@@ -1439,6 +1440,20 @@ def query_drafts(
     ) -> Union[StreamResponse, Awaitable[StreamResponse]]:
         pass
 
+    @abc.abstractmethod
+    def get_user_locations(self, user_id: str, **options: Any) -> StreamResponse:
+        """
+        Get the locations of a user.
+        """
+        pass
+
+    @abc.abstractmethod
+    def update_user_location(self, message_id: str, options: Optional[SharedLocationsOptions] = None) -> StreamResponse:
+        """
+        Update the location of a user.
+        """
+        pass
+
     #####################
     #  Private methods  #
     #####################

stream_chat/types/shared_locations.py

@@ -0,0 +1,8 @@
+from datetime import datetime
+from typing import Optional, TypedDict
+
+
+class SharedLocationsOptions(TypedDict):
+    longitude: Optional[int]
+    latitude: Optional[int]
+    end_at: Optional[datetime]