From 771aa080f643f9f315852ee83f453a852548843f Mon Sep 17 00:00:00 2001
From: eguven <eguven@google.com>
Date: Fri, 7 Dec 2018 14:39:11 +0000
Subject: [PATCH] Add totalBytes to TaskState

PiperOrigin-RevId: 224506700
---
 .../android/exoplayer2/offline/DownloadManager.java      | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/library/core/src/main/java/com/google/android/exoplayer2/offline/DownloadManager.java b/library/core/src/main/java/com/google/android/exoplayer2/offline/DownloadManager.java
index 35ad625cdb..969b8c2e11 100644
--- a/library/core/src/main/java/com/google/android/exoplayer2/offline/DownloadManager.java
+++ b/library/core/src/main/java/com/google/android/exoplayer2/offline/DownloadManager.java
@@ -523,6 +523,8 @@ public final class DownloadManager {
     public final float downloadPercentage;
     /** The total number of downloaded bytes. */
     public final long downloadedBytes;
+    /** The total size of the media, or {@link C#LENGTH_UNSET} if unknown. */
+    public final long totalBytes;
 
     /** If {@link #state} is {@link #STATE_FAILED} then this is the cause, otherwise null. */
     @Nullable public final Throwable error;
@@ -533,12 +535,14 @@ public final class DownloadManager {
         @State int state,
         float downloadPercentage,
         long downloadedBytes,
+        long totalBytes,
         @Nullable Throwable error) {
       this.taskId = taskId;
       this.action = action;
       this.state = state;
       this.downloadPercentage = downloadPercentage;
       this.downloadedBytes = downloadedBytes;
+      this.totalBytes = totalBytes;
       this.error = error;
     }
 
@@ -587,11 +591,14 @@ public final class DownloadManager {
     public TaskState getTaskState() {
       float downloadPercentage = C.PERCENTAGE_UNSET;
       long downloadedBytes = 0;
+      long totalBytes = C.LENGTH_UNSET;
       if (downloader != null) {
         downloadPercentage = downloader.getDownloadPercentage();
         downloadedBytes = downloader.getDownloadedBytes();
+        totalBytes = downloader.getTotalBytes();
       }
-      return new TaskState(id, action, state, downloadPercentage, downloadedBytes, error);
+      return new TaskState(
+          id, action, state, downloadPercentage, downloadedBytes, totalBytes, error);
     }
 
     /** Returns whether the task is finished. */