feat: exoplayer drm license.

oxyroid · Jan 20, 2024 · 15202b8 · 15202b8
1 parent 6337276
commit 15202b8
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 13 deletions.
diff --git a/data/src/main/java/com/m3u/data/database/model/Stream.kt b/data/src/main/java/com/m3u/data/database/model/Stream.kt
@@ -38,4 +38,10 @@ data class Stream(
         this.url == another.url && this.playlistUrl == another.playlistUrl && this.cover == another.cover
                 && this.group == another.group && this.title == another.title && this.licenseType == another.licenseType
                 && this.licenseKey == another.licenseKey
+
+    companion object {
+        const val LICENSE_TYPE_WIDEVINE = "com.widevine.alpha"
+        const val LICENSE_TYPE_CLEAR_KEY = "clearkey"
+        const val LICENSE_TYPE_PLAY_READY = "com.microsoft.playready"
+    }
 }
diff --git a/data/src/main/java/com/m3u/data/manager/impl/PlayerManagerImpl.kt b/data/src/main/java/com/m3u/data/manager/impl/PlayerManagerImpl.kt
@@ -20,6 +20,10 @@ import androidx.media3.datasource.DefaultDataSource
 import androidx.media3.datasource.okhttp.OkHttpDataSource
 import androidx.media3.exoplayer.DefaultRenderersFactory
 import androidx.media3.exoplayer.ExoPlayer
+import androidx.media3.exoplayer.drm.DefaultDrmSessionManager
+import androidx.media3.exoplayer.drm.DrmSessionManager
+import androidx.media3.exoplayer.drm.FrameworkMediaDrm
+import androidx.media3.exoplayer.drm.LocalMediaDrmCallback
 import androidx.media3.exoplayer.source.DefaultMediaSourceFactory
 import androidx.media3.exoplayer.trackselection.DefaultTrackSelector
 import androidx.media3.session.MediaSession
@@ -29,6 +33,7 @@ import com.m3u.core.architecture.pref.observeAsFlow
 import com.m3u.data.Certs
 import com.m3u.data.SSL
 import com.m3u.data.database.dao.StreamDao
+import com.m3u.data.database.model.Stream
 import com.m3u.data.manager.PlayerManager
 import dagger.hilt.android.qualifiers.ApplicationContext
 import kotlinx.coroutines.CoroutineScope
@@ -121,6 +126,11 @@ class PlayerManagerImpl @Inject constructor(
                     )
                 )
             )
+            .apply {
+                drmSessionManager?.let { manager ->
+                    setDrmSessionManagerProvider { manager }
+                }
+            }
         val ts = DefaultTrackSelector(context).apply {
             setParameters(buildUponParameters().setMaxVideoSizeSd())
         }
@@ -139,19 +149,40 @@ class PlayerManagerImpl @Inject constructor(
             }
     }
 
+    private var drmSessionManager: DrmSessionManager? = null
+
     override fun play(streamId: Int) {
-        _player.update { prev ->
-            if (prev != null) stop()
-            _streamId.value = streamId
-            createPlayer(isSSLVerification.value, connectTimeout.value).also {
-                it.addListener(this)
-                coroutineScope.launch {
-                    val stream = streamDao.get(streamId) ?: return@launch
-                    val mediaItem = MediaItem.fromUri(stream.url)
-                    withContext(Dispatchers.Main) {
-                        it.setMediaItem(mediaItem)
-                        it.prepare()
-                    }
+        val prev = _player.value
+        if (prev != null) stop()
+        _streamId.value = streamId
+
+        coroutineScope.launch {
+            val stream = streamDao.get(streamId) ?: return@launch
+            val useDrm = stream.licenseType != null && stream.licenseKey != null
+
+            drmSessionManager = if (useDrm) {
+                val licenseType = stream.licenseType!!
+                val licenseKey = stream.licenseKey!!
+                val callback = LocalMediaDrmCallback(licenseKey.toByteArray())
+                val uuid = when (stream.licenseType) {
+                    Stream.LICENSE_TYPE_WIDEVINE -> C.WIDEVINE_UUID
+                    Stream.LICENSE_TYPE_CLEAR_KEY -> C.CLEARKEY_UUID
+                    Stream.LICENSE_TYPE_PLAY_READY -> C.PLAYREADY_UUID
+                    else -> C.CLEARKEY_UUID
+                }
+                val mediaDrm = FrameworkMediaDrm.newInstance(uuid)
+                DefaultDrmSessionManager.Builder()
+                    .setUuidAndExoMediaDrmProvider(uuid) { mediaDrm }
+                    .build(callback)
+            } else null
+
+            withContext(Dispatchers.Main) {
+                _player.value = createPlayer(isSSLVerification.value, connectTimeout.value).also {
+                    it.addListener(this@PlayerManagerImpl)
+                    val url = stream.url
+                    val mediaItem = MediaItem.fromUri(url)
+                    it.setMediaItem(mediaItem)
+                    it.prepare()
                 }
             }
         }
@@ -228,7 +259,6 @@ class PlayerManagerImpl @Inject constructor(
             }
     }
 
-
     override fun chooseTrack(group: TrackGroup, trackIndex: Int) {
         val currentPlayer = _player.value ?: return
         val override = TrackSelectionOverride(group, trackIndex)

diff --git a/features/about/build.gradle.kts b/features/about/build.gradle.kts
@@ -54,3 +54,7 @@ dependencies {
 
     implementation(libs.aboutlibraries.core)
 }
+
+aboutLibraries {
+    offlineMode = true
+}