Download offline DRM license from non-UI thread

PiperOrigin-RevId: 328904901

demos/main/src/main/java/com/google/android/exoplayer2/demo/DownloadTracker.java

@@ -16,10 +16,12 @@
 package com.google.android.exoplayer2.demo;
 
 import static com.google.android.exoplayer2.util.Assertions.checkNotNull;
+import static com.google.android.exoplayer2.util.Assertions.checkStateNotNull;
 
 import android.content.Context;
 import android.content.DialogInterface;
 import android.net.Uri;
+import android.os.AsyncTask;
 import android.widget.Toast;
 import androidx.annotation.NonNull;
 import androidx.annotation.Nullable;
@@ -170,6 +172,7 @@ public class DownloadTracker {
 
     private TrackSelectionDialog trackSelectionDialog;
     private MappedTrackInfo mappedTrackInfo;
+    private WidevineOfflineLicenseFetchTask widevineOfflineLicenseFetchTask;
     @Nullable private byte[] keySetId;
 
     public StartDownloadDialogHelper(
@@ -185,6 +188,9 @@ public class DownloadTracker {
       if (trackSelectionDialog != null) {
         trackSelectionDialog.dismiss();
       }
+      if (widevineOfflineLicenseFetchTask != null) {
+        widevineOfflineLicenseFetchTask.cancel(false);
+      }
     }
 
     // DownloadHelper.Callback implementation.
@@ -192,7 +198,12 @@ public class DownloadTracker {
     @Override
     public void onPrepared(@NonNull DownloadHelper helper) {
       @Nullable Format format = getFirstFormatWithDrmInitData(helper);
-      if (format != null) {
+      if (format == null) {
+        onDownloadPrepared(helper);
+        return;
+      }
+
+      // The content is DRM protected. We need to acquire an offline license.
       if (Util.SDK_INT < 18) {
         Toast.makeText(context, R.string.error_drm_unsupported_before_api_18, Toast.LENGTH_LONG)
             .show();
@@ -209,42 +220,10 @@ public class DownloadTracker {
                 + " supported");
         return;
       }
-        try {
+      widevineOfflineLicenseFetchTask =
-          // TODO(internal b/163107948): Download the license on another thread to keep the UI
+          new WidevineOfflineLicenseFetchTask(
-          //  thread unblocked.
+              format, mediaItem.playbackProperties.drmConfiguration.licenseUri, this, helper);
-          fetchOfflineLicense(format);
+      widevineOfflineLicenseFetchTask.execute();
-        } catch (DrmSession.DrmSessionException e) {
-          Toast.makeText(context, R.string.download_start_error_offline_license, Toast.LENGTH_LONG)
-              .show();
-          Log.e(TAG, "Failed to fetch offline DRM license", e);
-          return;
-        }
-      }
-
-      if (helper.getPeriodCount() == 0) {
-        Log.d(TAG, "No periods found. Downloading entire stream.");
-        startDownload();
-        downloadHelper.release();
-        return;
-      }
-
-      mappedTrackInfo = downloadHelper.getMappedTrackInfo(/* periodIndex= */ 0);
-      if (!TrackSelectionDialog.willHaveContent(mappedTrackInfo)) {
-        Log.d(TAG, "No dialog content. Downloading entire stream.");
-        startDownload();
-        downloadHelper.release();
-        return;
-      }
-      trackSelectionDialog =
-          TrackSelectionDialog.createForMappedTrackInfoAndParameters(
-              /* titleId= */ R.string.exo_download_description,
-              mappedTrackInfo,
-              trackSelectorParameters,
-              /* allowAdaptiveSelections =*/ false,
-              /* allowMultipleOverrides= */ true,
-              /* onClickListener= */ this,
-              /* onDismissListener= */ this);
-      trackSelectionDialog.show(fragmentManager, /* tag= */ null);
     }
 
     @Override
@@ -292,6 +271,44 @@ public class DownloadTracker {
 
     // Internal methods.
 
+    private void onOfflineLicenseFetched(DownloadHelper helper, byte[] keySetId) {
+      this.keySetId = keySetId;
+      onDownloadPrepared(helper);
+    }
+
+    private void onOfflineLicenseFetchedError(DrmSession.DrmSessionException e) {
+      Toast.makeText(context, R.string.download_start_error_offline_license, Toast.LENGTH_LONG)
+          .show();
+      Log.e(TAG, "Failed to fetch offline DRM license", e);
+    }
+
+    private void onDownloadPrepared(DownloadHelper helper) {
+      if (helper.getPeriodCount() == 0) {
+        Log.d(TAG, "No periods found. Downloading entire stream.");
+        startDownload();
+        downloadHelper.release();
+        return;
+      }
+
+      mappedTrackInfo = downloadHelper.getMappedTrackInfo(/* periodIndex= */ 0);
+      if (!TrackSelectionDialog.willHaveContent(mappedTrackInfo)) {
+        Log.d(TAG, "No dialog content. Downloading entire stream.");
+        startDownload();
+        downloadHelper.release();
+        return;
+      }
+      trackSelectionDialog =
+          TrackSelectionDialog.createForMappedTrackInfoAndParameters(
+              /* titleId= */ R.string.exo_download_description,
+              mappedTrackInfo,
+              trackSelectorParameters,
+              /* allowAdaptiveSelections =*/ false,
+              /* allowMultipleOverrides= */ true,
+              /* onClickListener= */ this,
+              /* onDismissListener= */ this);
+      trackSelectionDialog.show(fragmentManager, /* tag= */ null);
+    }
+
     private void startDownload() {
       startDownload(buildDownloadRequest());
     }
@@ -306,15 +323,54 @@ public class DownloadTracker {
           .getDownloadRequest(Util.getUtf8Bytes(checkNotNull(mediaItem.mediaMetadata.title)))
           .copyWithKeySetId(keySetId);
     }
+  }
 
+  /** Downloads a Widevine offline license in a background thread. */
   @RequiresApi(18)
-    private void fetchOfflineLicense(Format format) throws DrmSession.DrmSessionException {
+  private final class WidevineOfflineLicenseFetchTask extends AsyncTask<Void, Void, Void> {
+    private final Format format;
+    private final Uri licenseUri;
+    private final StartDownloadDialogHelper dialogHelper;
+    private final DownloadHelper downloadHelper;
+
+    @Nullable private byte[] keySetId;
+    @Nullable private DrmSession.DrmSessionException drmSessionException;
+
+    public WidevineOfflineLicenseFetchTask(
+        Format format,
+        Uri licenseUri,
+        StartDownloadDialogHelper dialogHelper,
+        DownloadHelper downloadHelper) {
+      this.format = format;
+      this.licenseUri = licenseUri;
+      this.dialogHelper = dialogHelper;
+      this.downloadHelper = downloadHelper;
+    }
+
+    @Override
+    protected Void doInBackground(Void... voids) {
       OfflineLicenseHelper offlineLicenseHelper =
           OfflineLicenseHelper.newWidevineInstance(
-              mediaItem.playbackProperties.drmConfiguration.licenseUri.toString(),
+              licenseUri.toString(),
               httpDataSourceFactory,
               new DrmSessionEventListener.EventDispatcher());
+      try {
         keySetId = offlineLicenseHelper.downloadLicense(format);
+      } catch (DrmSession.DrmSessionException e) {
+        drmSessionException = e;
+      } finally {
+        offlineLicenseHelper.release();
+      }
+      return null;
+    }
+
+    @Override
+    protected void onPostExecute(Void aVoid) {
+      if (drmSessionException != null) {
+        dialogHelper.onOfflineLicenseFetchedError(drmSessionException);
+      } else {
+        dialogHelper.onOfflineLicenseFetched(downloadHelper, checkStateNotNull(keySetId));
+      }
     }
   }