feat: unpublish track (#21)

Author: theomonnom

client-sdk-rust

@@ -292,6 +292,7 @@ class FfiEvent(google.protobuf.message.Message):
     DISCONNECT_FIELD_NUMBER: builtins.int
     DISPOSE_FIELD_NUMBER: builtins.int
     PUBLISH_TRACK_FIELD_NUMBER: builtins.int
+    UNPUBLISH_TRACK_FIELD_NUMBER: builtins.int
     PUBLISH_DATA_FIELD_NUMBER: builtins.int
     @property
     def room_event(self) -> room_pb2.RoomEvent: ...
@@ -310,6 +311,8 @@ class FfiEvent(google.protobuf.message.Message):
     @property
     def publish_track(self) -> room_pb2.PublishTrackCallback: ...
     @property
+    def unpublish_track(self) -> room_pb2.UnpublishTrackCallback: ...
+    @property
     def publish_data(self) -> room_pb2.PublishDataCallback: ...
     def __init__(
         self,
@@ -322,11 +325,12 @@ class FfiEvent(google.protobuf.message.Message):
         disconnect: room_pb2.DisconnectCallback | None = ...,
         dispose: global___DisposeCallback | None = ...,
         publish_track: room_pb2.PublishTrackCallback | None = ...,
+        unpublish_track: room_pb2.UnpublishTrackCallback | None = ...,
         publish_data: room_pb2.PublishDataCallback | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["audio_stream_event", b"audio_stream_event", "connect", b"connect", "disconnect", b"disconnect", "dispose", b"dispose", "message", b"message", "publish_data", b"publish_data", "publish_track", b"publish_track", "room_event", b"room_event", "track_event", b"track_event", "video_stream_event", b"video_stream_event"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["audio_stream_event", b"audio_stream_event", "connect", b"connect", "disconnect", b"disconnect", "dispose", b"dispose", "message", b"message", "publish_data", b"publish_data", "publish_track", b"publish_track", "room_event", b"room_event", "track_event", b"track_event", "video_stream_event", b"video_stream_event"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["message", b"message"]) -> typing_extensions.Literal["room_event", "track_event", "video_stream_event", "audio_stream_event", "connect", "disconnect", "dispose", "publish_track", "publish_data"] | None: ...
+    def HasField(self, field_name: typing_extensions.Literal["audio_stream_event", b"audio_stream_event", "connect", b"connect", "disconnect", b"disconnect", "dispose", b"dispose", "message", b"message", "publish_data", b"publish_data", "publish_track", b"publish_track", "room_event", b"room_event", "track_event", b"track_event", "unpublish_track", b"unpublish_track", "video_stream_event", b"video_stream_event"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["audio_stream_event", b"audio_stream_event", "connect", b"connect", "disconnect", b"disconnect", "dispose", b"dispose", "message", b"message", "publish_data", b"publish_data", "publish_track", b"publish_track", "room_event", b"room_event", "track_event", b"track_event", "unpublish_track", b"unpublish_track", "video_stream_event", b"video_stream_event"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["message", b"message"]) -> typing_extensions.Literal["room_event", "track_event", "video_stream_event", "audio_stream_event", "connect", "disconnect", "dispose", "publish_track", "unpublish_track", "publish_data"] | None: ...
 
 global___FfiEvent = FfiEvent
 

livekit/_proto/ffi_pb2.py

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f98811d47068760e512e51ba9967b70597977059a41ff9c9959c4540827c0007
-size 15007512
+oid sha256:c78d6904e35e79e4c9d731cd36a387e5bedfa45adaee400ba39f5db257845918
+size 15024824

livekit/_proto/ffi_pb2.pyi

@@ -34,6 +34,11 @@ def __init__(self, message: str) -> None:
         self.message = message
 
 
+class UnpublishTrackError(Exception):
+    def __init__(self, message: str) -> None:
+        self.message = message
+
+
 class PublishDataError(Exception):
     def __init__(self, message: str) -> None:
         self.message = message
@@ -147,6 +152,30 @@ def on_publish_callback(cb: proto_room.PublishTrackCallback):
         self.tracks[track_publication.sid] = track_publication
         return track_publication
 
+    async def unpublish_track(self, track_sid: str) -> None:
+        req = proto_ffi.FfiRequest()
+        req.unpublish_track.local_participant_handle = self._ffi_handle.handle
+        req.unpublish_track.track_sid = track_sid
+
+        resp = ffi_client.request(req)
+
+        future: asyncio.Future[proto_room.UnpublishTrackCallback] = asyncio.Future(
+        )
+
+        @ffi_client.on('unpublish_track')
+        def on_unpublish_callback(cb: proto_room.UnpublishTrackCallback):
+            if cb.async_id == resp.unpublish_track.async_id:
+                future.set_result(cb)
+                ffi_client.remove_listener(
+                    'unpublish_track', on_unpublish_callback)
+
+        cb = await future
+        if cb.error:
+            raise UnpublishTrackError(cb.error)
+
+        publication = self.tracks.pop(track_sid)
+        publication.track = None
+
 
 class RemoteParticipant(Participant):
     def __init__(self, handle: FfiHandle, info: proto_participant.ParticipantInfo) \