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

chrome/browser/android/offline_pages/prerendering_offliner.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/android/offline_pages/prerendering_offliner.h"
 
 #include "base/bind.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/sys_info.h"
 #include "chrome/browser/android/offline_pages/offline_page_mhtml_archiver.h"
 #include "chrome/browser/android/offline_pages/offliner_helper.h"
 #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,
     const OfflinerPolicy* policy,
     OfflinePageModel* offline_page_model)
     : browser_context_(browser_context),
       offline_page_model_(offline_page_model),
       pending_request_(nullptr),
       is_low_end_device_(base::SysInfo::IsLowEndDevice()),
       app_listener_(nullptr),
       weak_ptr_factory_(this) {}
 
 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,
     content::WebContents* web_contents) {
   // Check if request is still pending receiving a callback.
   // Note: it is possible to get a loaded page, start the save operation,
   // and then get another callback from the Loader (eg, if its loaded
   // WebContents is being destroyed for some resource reclamation).
   if (!pending_request_)
     return;
@@ skipping 127 matching lines @@
     return false;
   }
 
   // Track copy of pending request for callback handling.
   pending_request_.reset(new SavePageRequest(request));
   completion_callback_ = callback;
 
   // 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 {
     // Create app listener for the pending request.
     app_listener_.reset(new base::android::ApplicationStatusListener(
         base::Bind(&PrerenderingOffliner::OnApplicationStateChange,
                    weak_ptr_factory_.GetWeakPtr())));
   }
 
   return accepted;
@@ skipping 46 matching lines @@
           base::android::APPLICATION_STATE_HAS_RUNNING_ACTIVITIES) {
     DVLOG(1) << "App became active, canceling current offlining request";
     SavePageRequest* request = pending_request_.get();
     Cancel();
     completion_callback_.Run(*request,
                              Offliner::RequestStatus::FOREGROUND_CANCELED);
   }
 }
 
 }  // namespace offline_pages