chrome/browser/extensions/extension_infobar_delegate.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/extensions/extension_infobar_delegate.h"
   6
   7 #include "chrome/browser/chrome_notification_types.h"
   8 #include "chrome/browser/extensions/extension_view_host.h"
   9 #include "chrome/browser/extensions/extension_view_host_factory.h"
  10 #include "chrome/browser/infobars/infobar.h"
  11 #include "chrome/browser/infobars/infobar_service.h"
  12 #include "chrome/browser/profiles/profile.h"
  13 #include "chrome/browser/ui/browser.h"
  14 #include "content/public/browser/notification_details.h"
  15 #include "content/public/browser/notification_source.h"
  16 #include "extensions/common/extension.h"
  17
  18 ExtensionInfoBarDelegate::~ExtensionInfoBarDelegate() {
  19 }
  20
  21 // static
  22 void ExtensionInfoBarDelegate::Create(content::WebContents* web_contents,
  23                                       Browser* browser,
  24                                       const extensions::Extension* extension,
  25                                       const GURL& url,
  26                                       int height) {
  27   InfoBarService::FromWebContents(web_contents)->AddInfoBar(
  28       ExtensionInfoBarDelegate::CreateInfoBar(
  29           scoped_ptr<ExtensionInfoBarDelegate>(new ExtensionInfoBarDelegate(
  30               browser, extension, url, web_contents, height))));
  31 }
  32
  33 ExtensionInfoBarDelegate::ExtensionInfoBarDelegate(
  34     Browser* browser,
  35     const extensions::Extension* extension,
  36     const GURL& url,
  37     content::WebContents* web_contents,
  38     int height)
  39     : InfoBarDelegate(),
  40 #if defined(TOOLKIT_VIEWS)
  41       browser_(browser),
  42 #endif
  43       extension_(extension),
  44       closing_(false) {
  45   extension_view_host_.reset(
  46       extensions::ExtensionViewHostFactory::CreateInfobarHost(url, browser));
  47   extension_view_host_->SetAssociatedWebContents(web_contents);
  48
  49   registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_HOST_VIEW_SHOULD_CLOSE,
  50                  content::Source<Profile>(browser->profile()));
  51   registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_UNLOADED,
  52                  content::Source<Profile>(browser->profile()));
  53
  54   height_ = std::max(0, height);
  55   height_ = std::min(2 * InfoBar::kDefaultBarTargetHeight, height_);
  56   if (height_ == 0)
  57     height_ = InfoBar::kDefaultBarTargetHeight;
  58 }
  59
  60 // ExtensionInfoBarDelegate::CreateInfoBar() is implemented in platform-specific
  61 // files.
  62
  63 bool ExtensionInfoBarDelegate::EqualsDelegate(InfoBarDelegate* delegate) const {
  64   ExtensionInfoBarDelegate* extension_delegate =
  65       delegate->AsExtensionInfoBarDelegate();
  66   // When an extension crashes, an InfoBar is shown (for the crashed extension).
  67   // That will result in a call to this function (to see if this InfoBarDelegate
  68   // is already showing the 'extension crashed InfoBar', which it never is), but
  69   // if it is our extension that crashes, the extension delegate is NULL so
  70   // we cannot check.
  71   if (!extension_delegate)
  72     return false;
  73
  74   // Only allow one InfoBar at a time per extension.
  75   return extension_delegate->extension_view_host()->extension() ==
  76          extension_view_host_->extension();
  77 }
  78
  79 void ExtensionInfoBarDelegate::InfoBarDismissed() {
  80   closing_ = true;
  81 }
  82
  83 InfoBarDelegate::Type ExtensionInfoBarDelegate::GetInfoBarType() const {
  84   return PAGE_ACTION_TYPE;
  85 }
  86
  87 ExtensionInfoBarDelegate*
  88     ExtensionInfoBarDelegate::AsExtensionInfoBarDelegate() {
  89   return this;
  90 }
  91
  92 void ExtensionInfoBarDelegate::Observe(
  93     int type,
  94     const content::NotificationSource& source,
  95     const content::NotificationDetails& details) {
  96   if (type == chrome::NOTIFICATION_EXTENSION_HOST_VIEW_SHOULD_CLOSE) {
  97     if (extension_view_host_.get() ==
  98         content::Details<extensions::ExtensionHost>(details).ptr())
  99       infobar()->RemoveSelf();
 100   } else {
 101     DCHECK(type == chrome::NOTIFICATION_EXTENSION_UNLOADED);
 102     if (extension_ == content::Details<extensions::UnloadedExtensionInfo>(
 103         details)->extension)
 104       infobar()->RemoveSelf();
 105   }
 106 }