Added a new function 'currentMediaTrack' to get the current track fro…

…m the VersaPlayerItem
josejuanqm · Mar 29, 2021 · 5bdf9fb · 5bdf9fb
1 parent 0f0174a
commit 5bdf9fb
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/VersaPlayer/Classes/Source/Core/Player/VersaPlayerItem.swift b/VersaPlayer/Classes/Source/Core/Player/VersaPlayerItem.swift
@@ -28,18 +28,31 @@ open class VersaPlayerItem: AVPlayerItem {
     deinit {
 
     }
+
+    private func convert(with mediaSelectionOption: AVMediaSelectionOption, group: AVMediaSelectionGroup) -> VersaPlayerMediaTrack {
+        let title = mediaSelectionOption.displayName
+        let language = mediaSelectionOption.extendedLanguageTag ?? "none"
+        return VersaPlayerMediaTrack(option: mediaSelectionOption, group: group, name: title, language: language)
+    }
 
     private func tracks(for characteristic: AVMediaCharacteristic) -> [VersaPlayerMediaTrack] {
         guard let group = asset.mediaSelectionGroup(forMediaCharacteristic: characteristic) else {
             return []
         }
         let options = group.options
         let tracks = options.map { (option) -> VersaPlayerMediaTrack in
-            let title = option.displayName
-            let language = option.extendedLanguageTag ?? "none"
-            return VersaPlayerMediaTrack(option: option, group: group, name: title, language: language)
+            return convert(with: option, group: group)
         }
         return tracks
     }
 
+    public func currentMediaTrack(for characteristic: AVMediaCharacteristic) -> VersaPlayerMediaTrack? {
+
+        if let tracks = asset.mediaSelectionGroup(forMediaCharacteristic: characteristic), let currentTrack = currentMediaSelection.selectedMediaOption(in: tracks) {
+            return convert(with: currentTrack, group: tracks)
+        }
+
+        return nil
+    }
+
 }