refactor

testpress · Aug 20, 2024 · c7dd5cf · c7dd5cf
1 parent ec9320d
commit c7dd5cf
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/player/src/main/java/com/tpstream/player/data/Asset.kt b/player/src/main/java/com/tpstream/player/data/Asset.kt
@@ -82,7 +82,8 @@ data class Video(
     var bytesDownloaded: Long = 0,
     var totalSize: Long = 0,
     var downloadState: DownloadStatus? = null,
-    val tracks: List<Track>? = null
+    val tracks: List<Track>? = null,
+    val enableDRM: Boolean? = false
 ){
     val isTranscodingCompleted: Boolean
         get() = transcodingStatus == "Completed"

diff --git a/player/src/main/java/com/tpstream/player/data/AssetModelMapping.kt b/player/src/main/java/com/tpstream/player/data/AssetModelMapping.kt
@@ -21,6 +21,7 @@ internal fun LocalAsset.asDomainAsset(): Asset {
             downloadState = this.downloadState,
             width = this.videoWidth,
             height = this.videoHeight,
+            enableDRM = this.url.isDRMVideo()
         ),
         folderTree = this.folderTree,
         downloadStartTimeMs = this.downloadStartTimeMs,
@@ -52,7 +53,8 @@ internal fun NetworkAsset.asDomainAsset(): Asset {
             transcodingStatus = transcodingStatus ?: "",
             tracks = this.networkVideo?.tracks?.map {
                 it.asDomainTracks()
-            }
+            },
+            enableDRM = networkVideo?.enable_drm
         ),
         description = description ?: "",
         liveStream = getDomainLiveStream(this),
@@ -108,3 +110,5 @@ internal fun Asset.asLocalAsset(): LocalAsset {
         metadata = this.metadata
     )
 }
+
+internal fun String.isDRMVideo() = this.contains(".mpd")