Adding the Previews infobar to pages that show a client LoFi image

chrome/browser/previews/previews_infobar_tab_helper.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/previews/previews_infobar_tab_helper.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "chrome/browser/loader/chrome_navigation_data.h"
 #include "chrome/browser/net/spdyproxy/data_reduction_proxy_chrome_settings.h"
@@ skipping 11 matching lines @@
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/navigation_handle.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/web_contents.h"
 #include "net/http/http_response_headers.h"
 #include "url/gurl.h"
 
 #if defined(OS_ANDROID)
 #include "chrome/browser/android/offline_pages/offline_page_tab_helper.h"
+#endif  // defined(OS_ANDROID)
 
 namespace {
 
-// Adds the preview navigation to the black list. This method is only used on
-// android currently.
+// Adds the preview navigation to the black list.
 void AddPreviewNavigationCallback(content::BrowserContext* browser_context,
                                   const GURL& url,
                                   previews::PreviewsType type,
                                   bool opt_out) {
   PreviewsService* previews_service = PreviewsServiceFactory::GetForProfile(
       Profile::FromBrowserContext(browser_context));
   if (previews_service && previews_service->previews_ui_service()) {
     previews_service->previews_ui_service()->AddPreviewNavigation(url, type,
                                                                   opt_out);
   }
 }
 
 }  // namespace
 
-#endif  // defined(OS_ANDROID)
-
 DEFINE_WEB_CONTENTS_USER_DATA_KEY(PreviewsInfoBarTabHelper);
 
 PreviewsInfoBarTabHelper::~PreviewsInfoBarTabHelper() {
   ClearLastNavigationAsync();
 }
 
 PreviewsInfoBarTabHelper::PreviewsInfoBarTabHelper(
     content::WebContents* web_contents)
     : content::WebContentsObserver(web_contents),
       browser_context_(web_contents->GetBrowserContext()),
@@ skipping 40 matching lines @@
     }
     data_reduction_proxy::DataReductionProxySettings*
         data_reduction_proxy_settings =
             DataReductionProxyChromeSettingsFactory::GetForBrowserContext(
                 web_contents()->GetBrowserContext());
     PreviewsInfoBarDelegate::Create(
         web_contents(), PreviewsInfoBarDelegate::OFFLINE,
         data_reduction_proxy_settings &&
             data_reduction_proxy_settings->IsDataReductionProxyEnabled(),
         base::Bind(&AddPreviewNavigationCallback, browser_context_,
-                   navigation_handle->GetURL(),
+                   navigation_handle->GetRedirectChain()[0],
                    previews::PreviewsType::OFFLINE));
     // Don't try to show other infobars if this is an offline preview.
     return;
   }
 #endif  // defined(OS_ANDROID)
 
   const net::HttpResponseHeaders* headers =
       navigation_handle->GetResponseHeaders();
   if (headers && data_reduction_proxy::IsLitePagePreview(*headers)) {
     PreviewsInfoBarDelegate::Create(
         web_contents(), PreviewsInfoBarDelegate::LITE_PAGE,
         true /* is_data_saver_user */,
-        PreviewsInfoBarDelegate::OnDismissPreviewsInfobarCallback());
+        base::Bind(&AddPreviewNavigationCallback, browser_context_,
+                   navigation_handle->GetRedirectChain()[0],
+                   previews::PreviewsType::LITE_PAGE));
   }
 }
 
 void PreviewsInfoBarTabHelper::ClearLastNavigationAsync() const {
   if (!committed_data_saver_navigation_id_)
     return;
   auto* data_reduction_proxy_settings =
       DataReductionProxyChromeSettingsFactory::GetForBrowserContext(
           browser_context_);
   if (!data_reduction_proxy_settings ||
       !data_reduction_proxy_settings->data_reduction_proxy_service()) {
     return;
   }
   data_reduction_proxy_settings->data_reduction_proxy_service()
       ->pingback_client()
       ->ClearNavigationKeyAsync(committed_data_saver_navigation_id_.value());
 }