Add Multiple download support

testpress · Dec 30, 2022 · 39b73c4 · 39b73c4
1 parent 5eb7a48
commit 39b73c4
Show file tree

Hide file tree

Showing 2 changed files with 62 additions and 19 deletions.
diff --git a/player/src/main/java/com/tpstream/player/TpStreamDownloadManager.kt b/player/src/main/java/com/tpstream/player/TpStreamDownloadManager.kt
@@ -38,5 +38,13 @@ class TpStreamDownloadManager(val context: Context) {
 
     private fun getDownloadUrl(offlineVideoInfo: OfflineVideoInfo): String {
         return offlineVideoInfo.url
+    }
+
+    private fun startDownloads(paramsList: List<TpInitParams>,videoResolution:Int){
+
+
+
+
+
     }
 }
diff --git a/player/src/main/java/com/tpstream/player/VideoDownloadRequestCreationHandler.kt b/player/src/main/java/com/tpstream/player/VideoDownloadRequestCreationHandler.kt
@@ -11,59 +11,90 @@ import androidx.media3.exoplayer.drm.DefaultDrmSessionManager
 import androidx.media3.exoplayer.offline.DownloadHelper
 import androidx.media3.exoplayer.offline.DownloadRequest
 import androidx.media3.exoplayer.trackselection.DefaultTrackSelector
+import com.tpstream.player.models.OfflineVideoInfo
 import com.tpstream.player.models.VideoInfo
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.launch
 import java.io.IOException
 
+typealias OnDownloadRequestCreation = (DownloadRequest, VideoInfo) -> Unit
+
 class VideoDownloadRequestCreationHandler(
     val context: Context,
-    private val player: TpStreamPlayer
+    private val player: TpStreamPlayer? = null,
+    private var params: TpInitParams? = null
 ) :
     DownloadHelper.Callback, DRMLicenseFetchCallback {
-    private val downloadHelper: DownloadHelper
-    private val trackSelectionParameters: DefaultTrackSelector.Parameters
+    private lateinit var downloadHelper: DownloadHelper
+    private lateinit var trackSelectionParameters: DefaultTrackSelector.Parameters
     var listener: Listener? = null
-    private val mediaItem: MediaItem
     private var keySetId: ByteArray? = null
+    private lateinit var videoInfo:VideoInfo
+    private var url: String? = null
+    private var onDownloadRequestCreation :OnDownloadRequestCreation? = null
 
-    init {
-        val url = player.videoInfo?.getPlaybackURL()!!
+    fun init(): VideoDownloadRequestCreationHandler{
+        if (player == null){
+            CoroutineScope(Dispatchers.IO).launch {
+                getVideoInfo(params!!)
+            }
+        }
+        url = player?.videoInfo?.getPlaybackURL()!!
+        params = player.params
+        videoInfo = player.videoInfo!!
         trackSelectionParameters = DownloadHelper.getDefaultTrackSelectorParameters(context)
-        mediaItem = MediaItem.Builder()
-            .setUri(url)
-            .setDrmConfiguration(
-                DrmConfiguration.Builder(C.WIDEVINE_UUID)
-                    .setMultiSession(true)
-                    .build()
-            )
-            .build()
         downloadHelper = getDownloadHelper()
         downloadHelper.prepare(this)
+        return this
+    }
+
+    private fun getVideoInfo(params: TpInitParams){
+    val url =
+        "/api/v2.5/video_info/${params.videoId}/?access_token=${params.accessToken}"
+    Network<VideoInfo>(params.orgCode).get(url, object : Network.TPResponse<VideoInfo> {
+        override fun onSuccess(result: VideoInfo) {
+            videoInfo = result
+        }
+
+        override fun onFailure(exception: TPException) {
+            Toast.makeText(context,"Download Failed",Toast.LENGTH_SHORT).show()
+        }
+    })
     }
 
     private fun getDownloadHelper(): DownloadHelper {
         val sessionManager = DefaultDrmSessionManager.Builder()
-            .build(CustomHttpDrmMediaCallback(context, player.params))
+            .build(CustomHttpDrmMediaCallback(context, params!!))
         sessionManager.setMode(DefaultDrmSessionManager.MODE_DOWNLOAD, null)
-        val dataSourceFactory = VideoDownloadManager(context).build(player.params)
+        val dataSourceFactory = VideoDownloadManager(context).build(params!!)
         val renderersFactory = DefaultRenderersFactory(context)
         return DownloadHelper.forMediaItem(
-            mediaItem,
+            getMediaItem(url!!),
             trackSelectionParameters,
             renderersFactory,
             dataSourceFactory,
             sessionManager
         )
     }
 
+    private fun getMediaItem(url: String):MediaItem{
+        return MediaItem.Builder()
+            .setUri(url)
+            .setDrmConfiguration(
+                DrmConfiguration.Builder(C.WIDEVINE_UUID)
+                    .setMultiSession(true)
+                    .build()
+            )
+            .build()
+    }
+
     override fun onPrepared(helper: DownloadHelper) {
         val videoOrAudioData = VideoPlayerUtil.getAudioOrVideoInfoWithDrmInitData(helper)
         val isDRMProtectedVideo = videoOrAudioData != null
         if (isDRMProtectedVideo) {
             if (hasDRMSchemaData(videoOrAudioData!!.drmInitData!!)) {
-                OfflineDRMLicenseHelper.fetchLicense(context, player.params, downloadHelper, this)
+                OfflineDRMLicenseHelper.fetchLicense(context, params!!, downloadHelper, this)
             } else {
                 Toast.makeText(
                     context,
@@ -92,7 +123,7 @@ class VideoDownloadRequestCreationHandler(
 
     fun buildDownloadRequest(overrides: MutableMap<TrackGroup, TrackSelectionOverride>): DownloadRequest {
         setSelectedTracks(overrides)
-        val name = player.videoInfo?.title!!
+        val name = videoInfo.title!!
         return downloadHelper.getDownloadRequest(Util.getUtf8Bytes(name)).copyWithKeySetId(keySetId)
     }
 
@@ -130,4 +161,8 @@ class VideoDownloadRequestCreationHandler(
 
         fun onDownloadRequestHandlerPrepareError(downloadHelper: DownloadHelper, e: IOException)
     }
+
+    fun setOnDownloadRequestCreation(listener: OnDownloadRequestCreation) {
+        onDownloadRequestCreation = listener
+    }
 }