Adding offline navigations to the previews blacklist

chrome/browser/previews/previews_infobar_delegate.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_delegate.h"
 
 #include "base/metrics/histogram_macros.h"
 #include "chrome/browser/android/android_theme_resources.h"
 #include "chrome/browser/infobars/infobar_service.h"
 #include "chrome/browser/net/spdyproxy/data_reduction_proxy_chrome_settings.h"
@@ skipping 28 matching lines @@
     UMA_HISTOGRAM_ENUMERATION(kUMAPreviewsInfoBarActionLitePage, action,
                               PreviewsInfoBarDelegate::INFOBAR_INDEX_BOUNDARY);
   } else if (infobar_type == PreviewsInfoBarDelegate::OFFLINE) {
     UMA_HISTOGRAM_ENUMERATION(kUMAPreviewsInfoBarActionOffline, action,
                               PreviewsInfoBarDelegate::INFOBAR_INDEX_BOUNDARY);
   }
 }
 
 }  // namespace
 
-PreviewsInfoBarDelegate::~PreviewsInfoBarDelegate() {}
+PreviewsInfoBarDelegate::~PreviewsInfoBarDelegate() {
+  if (!on_dismiss_callback_.is_null())
+    on_dismiss_callback_.Run(false);
+}
 
 // static
-void PreviewsInfoBarDelegate::Create(content::WebContents* web_contents,
-                                     PreviewsInfoBarType infobar_type) {
+void PreviewsInfoBarDelegate::Create(
+    content::WebContents* web_contents,
+    PreviewsInfoBarType infobar_type,
+    const OnDismissPreviewsInfobarCallback& on_dismiss_callback) {
   PreviewsInfoBarTabHelper* infobar_tab_helper =
       PreviewsInfoBarTabHelper::FromWebContents(web_contents);
   if (infobar_tab_helper->displayed_preview_infobar())
     return;
 
   InfoBarService* infobar_service =
       InfoBarService::FromWebContents(web_contents);
 
   infobars::InfoBar* infobar =
       infobar_service->AddInfoBar(infobar_service->CreateConfirmInfoBar(
-          std::unique_ptr<ConfirmInfoBarDelegate>(
-              new PreviewsInfoBarDelegate(web_contents, infobar_type))));
+          std::unique_ptr<ConfirmInfoBarDelegate>(new PreviewsInfoBarDelegate(
+              web_contents, infobar_type, on_dismiss_callback))));
 
   if (infobar && (infobar_type == LITE_PAGE || infobar_type == LOFI)) {
     auto* data_reduction_proxy_settings =
         DataReductionProxyChromeSettingsFactory::GetForBrowserContext(
             web_contents->GetBrowserContext());
     data_reduction_proxy_settings->IncrementLoFiUIShown();
   }
 
   RecordPreviewsInfoBarAction(infobar_type, INFOBAR_SHOWN);
   infobar_tab_helper->set_displayed_preview_infobar(true);
 }
 
 PreviewsInfoBarDelegate::PreviewsInfoBarDelegate(
     content::WebContents* web_contents,
-    PreviewsInfoBarType infobar_type)
+    PreviewsInfoBarType infobar_type,
+    const OnDismissPreviewsInfobarCallback& on_dismiss_callback)
     : ConfirmInfoBarDelegate(),
-      infobar_type_(infobar_type) {}
+      infobar_type_(infobar_type),
+      on_dismiss_callback_(on_dismiss_callback) {}
 
 infobars::InfoBarDelegate::InfoBarIdentifier
 PreviewsInfoBarDelegate::GetIdentifier() const {
   return DATA_REDUCTION_PROXY_PREVIEW_INFOBAR_DELEGATE;
 }
 
 int PreviewsInfoBarDelegate::GetIconId() const {
 #if defined(OS_ANDROID)
   return IDR_ANDROID_INFOBAR_PREVIEWS;
 #else
@@ skipping 20 matching lines @@
 int PreviewsInfoBarDelegate::GetButtons() const {
   return BUTTON_NONE;
 }
 
 base::string16 PreviewsInfoBarDelegate::GetLinkText() const {
   return l10n_util::GetStringUTF16(IDS_PREVIEWS_INFOBAR_LINK);
 }
 
 bool PreviewsInfoBarDelegate::LinkClicked(WindowOpenDisposition disposition) {
   RecordPreviewsInfoBarAction(infobar_type_, INFOBAR_LOAD_ORIGINAL_CLICKED);
+  if (!on_dismiss_callback_.is_null())
+    on_dismiss_callback_.Run(true);
+  on_dismiss_callback_.Reset();
+
   content::WebContents* web_contents =
       InfoBarService::WebContentsFromInfoBar(infobar());
   if (infobar_type_ == LITE_PAGE || infobar_type_ == LOFI) {
     if (infobar_type_ == LITE_PAGE)
       web_contents->GetController().ReloadDisableLoFi(true);
     else if (infobar_type_ == LOFI)
       web_contents->ReloadLoFiImages();
 
     auto* data_reduction_proxy_settings =
         DataReductionProxyChromeSettingsFactory::GetForBrowserContext(
             web_contents->GetBrowserContext());
     data_reduction_proxy_settings->IncrementLoFiUserRequestsForImages();
   } else if (infobar_type_ == OFFLINE) {
     web_contents->GetController().Reload(true);
   }
 
   return true;
 }