Support mute/unmute for local tracks

livekit · Dec 4, 2024 · 28e64ef · 28e64ef
1 parent 99f79c3
commit 28e64ef
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/livekit-rtc/livekit/rtc/track.py b/livekit-rtc/livekit/rtc/track.py
@@ -80,6 +80,18 @@ def create_audio_track(name: str, source: "AudioSource") -> "LocalAudioTrack":
         resp = FfiClient.instance.request(req)
         return LocalAudioTrack(resp.create_audio_track.track)
 
+    def mute(self):
+        req = proto_ffi.FfiRequest()
+        req.local_track_mute.track_handle = self._ffi_handle.handle
+        req.local_track_mute.mute = True
+        FfiClient.instance.request(req)
+
+    def unmute(self):
+        req = proto_ffi.FfiRequest()
+        req.local_track_mute.track_handle = self._ffi_handle.handle
+        req.local_track_mute.mute = False
+        FfiClient.instance.request(req)
+
     def __repr__(self) -> str:
         return f"rtc.LocalAudioTrack(sid={self.sid}, name={self.name})"
 
@@ -97,6 +109,18 @@ def create_video_track(name: str, source: "VideoSource") -> "LocalVideoTrack":
         resp = FfiClient.instance.request(req)
         return LocalVideoTrack(resp.create_video_track.track)
 
+    def mute(self):
+        req = proto_ffi.FfiRequest()
+        req.local_track_mute.track_handle = self._ffi_handle.handle
+        req.local_track_mute.mute = True
+        FfiClient.instance.request(req)
+
+    def unmute(self):
+        req = proto_ffi.FfiRequest()
+        req.local_track_mute.track_handle = self._ffi_handle.handle
+        req.local_track_mute.mute = False
+        FfiClient.instance.request(req)
+
     def __repr__(self) -> str:
         return f"rtc.LocalVideoTrack(sid={self.sid}, name={self.name})"