[Offline Pages] Make prerenderer notify us of network progress.

chrome/browser/android/offline_pages/prerendering_offliner.cc

Index: chrome/browser/android/offline_pages/prerendering_offliner.cc
diff --git a/chrome/browser/android/offline_pages/prerendering_offliner.cc b/chrome/browser/android/offline_pages/prerendering_offliner.cc
index 31c0813bdb86721bd1922114781bf64ee9d8547a..5ceaf65abb73b74ad687d0f6a7122477991bb41d 100644
--- a/chrome/browser/android/offline_pages/prerendering_offliner.cc
+++ b/chrome/browser/android/offline_pages/prerendering_offliner.cc
@@ -12,11 +12,15 @@
 #include "chrome/browser/profiles/profile.h"
 #include "components/offline_pages/core/background/save_page_request.h"
 #include "components/offline_pages/core/client_namespace_constants.h"
+#include "components/offline_pages/core/downloads/download_ui_adapter.h"
 #include "components/offline_pages/core/offline_page_model.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/web_contents.h"
 
 namespace offline_pages {
+namespace {
+const char kDownloadUIAdapterKey[] = "download-ui-adapter";

[Dmitry Titov, 2017/02/25 00:17:19]

Apparently, this doesn't work since the GetUserData does not use the value of
those strings, it only is interested in an address passed in. So even if you
have 2 equal strings but located in different files, using them as 'keys' does
not work...
I'll fix it in follow-up.

+}
 
 PrerenderingOffliner::PrerenderingOffliner(
     content::BrowserContext* browser_context,
@@ -31,6 +35,20 @@ PrerenderingOffliner::PrerenderingOffliner(
 
 PrerenderingOffliner::~PrerenderingOffliner() {}
 
+void PrerenderingOffliner::OnNetworkProgress(const SavePageRequest& request,
+                                             int64_t bytes) {
+  if (!pending_request_)
+    return;
+
+  DownloadUIAdapter* ui_adapter = static_cast<DownloadUIAdapter*>(
+      offline_page_model_->GetUserData(kDownloadUIAdapterKey));
+
+  if (!ui_adapter)
+    return;
+
+  ui_adapter->UpdateProgress(request.request_id(), bytes);
+}
+
 void PrerenderingOffliner::OnLoadPageDone(
     const SavePageRequest& request,
     Offliner::RequestStatus load_status,
@@ -178,7 +196,9 @@ bool PrerenderingOffliner::LoadAndSave(const SavePageRequest& request,
   // Kick off load page attempt.
   bool accepted = GetOrCreateLoader()->LoadPage(
       request.url(), base::Bind(&PrerenderingOffliner::OnLoadPageDone,
-                                weak_ptr_factory_.GetWeakPtr(), request));
+                                weak_ptr_factory_.GetWeakPtr(), request),
+      base::Bind(&PrerenderingOffliner::OnNetworkProgress,
+                 weak_ptr_factory_.GetWeakPtr(), request));
   if (!accepted) {
     pending_request_.reset(nullptr);
   } else {