1 files changed, 7 insertions, 1 deletions
diff --git a/chrome/browser/ui/views/web_dialog_view.h b/chrome/browser/ui/views/web_dialog_view.h
index e6a718c4..915d2dc 100644
--- a/chrome/browser/ui/views/web_dialog_view.h
+++ b/chrome/browser/ui/views/web_dialog_view.h
@@ -18,6 +18,8 @@
 #include "ui/views/window/client_view.h"
 #include "ui/web_dialogs/web_dialog_delegate.h"
 
+class Browser;
+class WebDialogController;
 class Profile;
 
 namespace views {
@@ -37,7 +39,7 @@ class WebView;
 // TODO(akalin): Make WebDialogView contain an WebDialogWebContentsDelegate
 // instead of inheriting from it to avoid violating the "no multiple
 // inheritance" rule.
-// TODO(beng): This class should not depend on Profile, only
+// TODO(beng): This class should not depend on Browser or Profile, only
 //             content::BrowserContext.
 class WebDialogView
     : public views::ClientView,
@@ -47,6 +49,7 @@ class WebDialogView
       public TabRenderWatcher::Delegate {
  public:
   WebDialogView(Profile* profile,
+                Browser* browser,
                 ui::WebDialogDelegate* delegate);
   virtual ~WebDialogView();
 
@@ -136,6 +139,9 @@ class WebDialogView
   // using this variable.
   ui::WebDialogDelegate* delegate_;
 
+  // Controls lifetime of dialog.
+  scoped_ptr<WebDialogController> dialog_controller_;
+
   views::WebView* web_view_;
 
   DISALLOW_COPY_AND_ASSIGN(WebDialogView);