From cf5e1a0904c7c34d0d7ed0a1d40a341c5d8b1f99 Mon Sep 17 00:00:00 2001
From: gianlucaparadise <gianlucaparadise@gmail.com>
Date: Tue, 23 Nov 2021 06:11:32 +0100
Subject: [PATCH] RemoteMediaClient GetMediaInfo for iOS

---
 .../FlutterMediaLoadRequestDataHelper.swift   | 160 ++++++++++++++++++
 ...t => HostMediaLoadRequestDataHelper.swift} |   0
 .../SwiftFlutterCastFrameworkPlugin.swift     |  10 ++
 3 files changed, 170 insertions(+)
 create mode 100644 ios/Classes/FlutterMediaLoadRequestDataHelper.swift
 rename ios/Classes/{MediaLoadRequestDataHelper.swift => HostMediaLoadRequestDataHelper.swift} (100%)

diff --git a/ios/Classes/FlutterMediaLoadRequestDataHelper.swift b/ios/Classes/FlutterMediaLoadRequestDataHelper.swift
new file mode 100644
index 0000000..16fbd72
--- /dev/null
+++ b/ios/Classes/FlutterMediaLoadRequestDataHelper.swift
@@ -0,0 +1,160 @@
+//
+//  FlutterMediaLoadRequestDataHelper.swift
+//  flutter_cast_framework
+//
+//  Created by Gianluca Paradiso on 19/11/21.
+//
+
+import Foundation
+import GoogleCast
+
+func getFlutterMediaInfo(mediaInfo: GCKMediaInformation?) -> MediaInfo {
+    let result = MediaInfo()
+    
+    if (mediaInfo == nil) {
+        return result
+    }
+    
+    result.contentId = mediaInfo?.contentID
+    result.contentType = mediaInfo?.contentType
+    result.customDataAsJson = "\(mediaInfo?.customData ?? {})"
+    result.mediaMetadata = getFlutterMediaMetadata(mediaMetadata: mediaInfo?.metadata)
+    result.mediaTracks = getFlutterMediaTracks(mediaTracks: mediaInfo?.mediaTracks)
+    result.streamDuration = getStreamDuration(streamDuration: mediaInfo?.streamDuration)
+    result.streamType = getFlutterStreamType(streamType: mediaInfo?.streamType)
+    
+    return result
+}
+
+func getStreamDuration(streamDuration: TimeInterval?) -> NSNumber {
+    if (streamDuration == nil) {
+        return 0
+    }
+    
+    let durationRounded = streamDuration?.rounded() ?? 0
+    let durationInt = Int(durationRounded)
+    return NSNumber(value: durationInt)
+}
+
+func getFlutterStreamType(streamType: GCKMediaStreamType?) -> StreamType {
+    switch streamType {
+    case .buffered:
+        return StreamType.buffered
+    case .live:
+        return StreamType.live
+    case .unknown:
+        return StreamType.invalid
+    case nil:
+        return StreamType.invalid
+    case .some(.none):
+        return StreamType.invalid
+    @unknown default:
+        return StreamType.invalid
+    }
+}
+
+func getFlutterMediaTracks(mediaTracks: [GCKMediaTrack]?) -> [MediaTrack]? {
+    if (mediaTracks == nil) {
+        return nil
+    }
+    
+    var result = [MediaTrack]()
+    mediaTracks?.forEach({ (track: GCKMediaTrack) in
+        let resultTrack = getFlutterMediaTrack(mediaTrack: track)
+        result.append(resultTrack)
+    })
+    return result
+}x
+
+func getFlutterMediaTrack(mediaTrack: GCKMediaTrack) -> MediaTrack {
+    let result = MediaTrack()
+    
+    result.contentId = mediaTrack.contentIdentifier
+    result.id = NSNumber(value: mediaTrack.identifier)
+    result.language = mediaTrack.languageCode
+    result.name = mediaTrack.name
+    result.trackType = getFlutterTrackType(trackType: mediaTrack.type)
+    result.trackSubtype = getFlutterTrackSubtype(trackSubtype: mediaTrack.textSubtype)
+    
+    return result
+}
+
+func getFlutterTrackType(trackType: GCKMediaTrackType) -> TrackType {
+    switch trackType {
+    
+    case .unknown:
+        return TrackType.unknown
+    case .text:
+        return TrackType.text
+    case .audio:
+        return TrackType.audio
+    case .video:
+        return TrackType.video
+    @unknown default:
+        return TrackType.unknown
+    }
+}
+
+func getFlutterTrackSubtype(trackSubtype: GCKMediaTextTrackSubtype) -> TrackSubtype {
+    switch trackSubtype {
+    
+    case .unknown:
+        return TrackSubtype.unknown
+    case .subtitles:
+        return TrackSubtype.subtitles
+    case .captions:
+        return TrackSubtype.captions
+    case .descriptions:
+        return TrackSubtype.descriptions
+    case .chapters:
+        return TrackSubtype.chapters
+    case .metadata:
+        return TrackSubtype.metadata
+    @unknown default:
+        return TrackSubtype.unknown
+    }
+}
+
+func getFlutterMediaMetadata(mediaMetadata: GCKMediaMetadata?) -> MediaMetadata {
+    let result = MediaMetadata()
+    
+    result.mediaType = getFlutterMediaType(mediaType: mediaMetadata?.metadataType)
+    result.webImages = []
+    
+    mediaMetadata?.images().forEach({ (imageAny: Any) in
+        let image = imageAny as? GCKImage
+        if (image == nil) {
+            return
+        }
+        
+        let webImage = WebImage()
+        webImage.url = image?.url.absoluteString
+        result.webImages?.append(webImage)
+    })
+    
+    return result
+}
+
+func getFlutterMediaType(mediaType: GCKMediaMetadataType?) -> MediaType {
+    switch mediaType {
+    case .generic:
+        return MediaType.generic
+    case .movie:
+        return MediaType.movie
+    case .tvShow:
+        return MediaType.tvShow
+    case .musicTrack:
+        return MediaType.musicTrack
+    case .photo:
+        return MediaType.photo
+    case .audioBookChapter:
+        return MediaType.audiobookChapter
+    case .user:
+        return MediaType.user
+    case .none:
+        return MediaType.generic
+    @unknown default:
+        return MediaType.generic
+    }
+}
+
diff --git a/ios/Classes/MediaLoadRequestDataHelper.swift b/ios/Classes/HostMediaLoadRequestDataHelper.swift
similarity index 100%
rename from ios/Classes/MediaLoadRequestDataHelper.swift
rename to ios/Classes/HostMediaLoadRequestDataHelper.swift
diff --git a/ios/Classes/SwiftFlutterCastFrameworkPlugin.swift b/ios/Classes/SwiftFlutterCastFrameworkPlugin.swift
index 3211a66..03cc3a3 100644
--- a/ios/Classes/SwiftFlutterCastFrameworkPlugin.swift
+++ b/ios/Classes/SwiftFlutterCastFrameworkPlugin.swift
@@ -154,6 +154,16 @@ public class SwiftFlutterCastFrameworkPlugin: NSObject, FlutterPlugin, GCKSessio
         remoteMediaClient?.loadMedia(with: mediaLoadRequest)
     }
     
+    public func getMediaInfoWithError(_ error: AutoreleasingUnsafeMutablePointer<FlutterError?>) -> MediaInfo? {
+        let hostMediaInfo = remoteMediaClient?.mediaStatus?.mediaInformation
+        
+        if (hostMediaInfo == nil) {
+            return MediaInfo()
+        }
+        
+        return getFlutterMediaInfo(mediaInfo: hostMediaInfo)
+    }
+    
     // MARK: - GCKSessionManagerListener
     
     // onSessionSuspended