add/remove audio renderer for local tracks

livekit · Sep 21, 2024 · ae0b528 · ae0b528
1 parent 471a1b3
commit ae0b528
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/Sources/LiveKit/Track/AudioTrack.swift b/Sources/LiveKit/Track/AudioTrack.swift
@@ -17,4 +17,10 @@
 import Foundation
 
 @objc
-public protocol AudioTrack where Self: Track {}
+public protocol AudioTrack where Self: Track {
+    @objc(addAudioRenderer:)
+    func add(audioRenderer: AudioRenderer)
+
+    @objc(removeAudioRenderer:)
+    func remove(audioRenderer: AudioRenderer)
+}
diff --git a/Sources/LiveKit/Track/Local/LocalAudioTrack.swift b/Sources/LiveKit/Track/Local/LocalAudioTrack.swift
@@ -94,3 +94,13 @@ public extension LocalAudioTrack {
     var publishOptions: TrackPublishOptions? { super._state.lastPublishOptions }
     var publishState: Track.PublishState { super._state.publishState }
 }
+
+public extension LocalAudioTrack {
+    func add(audioRenderer: AudioRenderer) {
+        AudioManager.shared.add(localAudioRenderer: audioRenderer)
+    }
+
+    func remove(audioRenderer: AudioRenderer) {
+        AudioManager.shared.remove(localAudioRenderer: audioRenderer)
+    }
+}