Add PreviewsInfoBarTabHelper

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 "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"
 #include "chrome/browser/net/spdyproxy/data_reduction_proxy_chrome_settings_factory.h"
+#include "chrome/browser/previews/previews_infobar_tab_helper.h"
 #include "chrome/grit/generated_resources.h"
 #include "components/infobars/core/infobar.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/web_contents.h"
 #include "ui/base/l10n/l10n_util.h"
 
 PreviewsInfoBarDelegate::~PreviewsInfoBarDelegate() {}
 
 // static
 void PreviewsInfoBarDelegate::Create(content::WebContents* web_contents,
                                      PreviewsInfoBarType infobar_type) {
-  // TODO(megjablon): Check that the infobar was not already shown.
+  PreviewsInfoBarTabHelper* infobar_tab_helper =
+      PreviewsInfoBarTabHelper::FromWebContents(web_contents);
+  if (infobar_tab_helper->DisplayedPreviewInfoBar())

[bengr, 2016/08/31 23:27:04]

Can infobar_tab_helper be null?

[megjablon, 2016/09/08 00:25:15]

The PreviewsInfoBarTabHelper is created in AttachTabHelpers().
AttachTabHelpers() is called whenever a WebContents is created for use as a
browser tab, so we should always have the helper. 

+    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))));
 
   if (infobar && (infobar_type == LITE_PAGE || infobar_type == LOFI)) {
     auto* data_reduction_proxy_settings =
         DataReductionProxyChromeSettingsFactory::GetForBrowserContext(
             web_contents->GetBrowserContext());
     data_reduction_proxy_settings->IncrementLoFiUIShown();
   }
+
+  infobar_tab_helper->SetDisplayedPreviewInfoBar(true);
 }
 
 PreviewsInfoBarDelegate::PreviewsInfoBarDelegate(
     content::WebContents* web_contents,
     PreviewsInfoBarType infobar_type)
     : ConfirmInfoBarDelegate(),
       infobar_type_(infobar_type) {}
 
 infobars::InfoBarDelegate::InfoBarIdentifier
 PreviewsInfoBarDelegate::GetIdentifier() const {
@@ skipping 40 matching lines @@
       web_contents->ReloadLoFiImages();
 
     auto* data_reduction_proxy_settings =
         DataReductionProxyChromeSettingsFactory::GetForBrowserContext(
             web_contents->GetBrowserContext());
     data_reduction_proxy_settings->IncrementLoFiUserRequestsForImages();
   }
 
   return true;
 }