Index: chrome/browser/extensions/extension_infobar_delegate.h |
diff --git a/chrome/browser/extensions/extension_infobar_delegate.h b/chrome/browser/extensions/extension_infobar_delegate.h |
index b9a337f7401e894867c89bd8db152edfc316a85f..b52ff8929f5bbf42c17c65b47992aa3dbbc2a602 100644 |
--- a/chrome/browser/extensions/extension_infobar_delegate.h |
+++ b/chrome/browser/extensions/extension_infobar_delegate.h |
@@ -6,13 +6,17 @@ |
#define CHROME_BROWSER_EXTENSIONS_EXTENSION_INFOBAR_DELEGATE_H_ |
#include "base/memory/scoped_ptr.h" |
-#include "chrome/browser/infobars/confirm_infobar_delegate.h" |
+#include "chrome/browser/infobars/infobar_delegate.h" |
#include "content/public/browser/notification_observer.h" |
#include "content/public/browser/notification_registrar.h" |
class Browser; |
class GURL; |
+namespace content { |
+class WebContents; |
+} |
+ |
namespace extensions { |
class Extension; |
class ExtensionViewHost; |
@@ -41,6 +45,8 @@ class ExtensionInfoBarDelegate : public InfoBarDelegate, |
bool closing() const { return closing_; } |
+ content::WebContents* web_contents() { return web_contents_; } |
+ |
private: |
ExtensionInfoBarDelegate(Browser* browser, |
const extensions::Extension* extension, |
@@ -57,6 +63,7 @@ class ExtensionInfoBarDelegate : public InfoBarDelegate, |
virtual void InfoBarDismissed() OVERRIDE; |
virtual Type GetInfoBarType() const OVERRIDE; |
virtual ExtensionInfoBarDelegate* AsExtensionInfoBarDelegate() OVERRIDE; |
+ virtual void CleanUp() OVERRIDE; |
// content::NotificationObserver: |
virtual void Observe(int type, |
@@ -67,6 +74,8 @@ class ExtensionInfoBarDelegate : public InfoBarDelegate, |
Browser* browser_; // We pass this to the ExtensionInfoBar. |
#endif |
+ content::WebContents* web_contents_; |
+ |
// The extension host we are showing the InfoBar for. |
// TODO(pkasting): Should this live on the InfoBar instead? |
scoped_ptr<extensions::ExtensionViewHost> extension_view_host_; |