Hide the main window when Set-As-Default dialog is present.

chrome/browser/ui/webui/set_as_default_browser_ui.cc

 // Copyright (c) 2012 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/ui/webui/set_as_default_browser_ui.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/path_service.h"
 #include "chrome/browser/first_run/first_run.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/shell_integration.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_dialogs.h"
 #include "chrome/browser/ui/browser_finder.h"
 #include "chrome/browser/ui/browser_list.h"
+#include "chrome/browser/ui/browser_tabstrip.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "chrome/browser/ui/chrome_pages.h"
 #include "chrome/browser/ui/singleton_tabs.h"
 #include "chrome/browser/ui/webui/chrome_web_ui_data_source.h"
 #include "chrome/browser/ui/webui/sync_promo/sync_promo_ui.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/installer/util/install_util.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_delegate.h"
+#include "content/public/browser/web_contents_view.h"
 #include "content/public/browser/web_ui.h"
 #include "content/public/browser/web_ui_message_handler.h"
 #include "grit/browser_resources.h"
 #include "grit/generated_resources.h"
 #include "grit/locale_settings.h"
 #include "ui/base/l10n/l10n_font_util.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/gfx/font.h"
 #include "ui/web_dialogs/web_dialog_delegate.h"
 
@@ skipping 14 matching lines @@
                                   IDS_METRO_FLOW_SET_DEFAULT);
   data_source->AddLocalizedString("chromeLogoString",
                                   IDS_METRO_FLOW_LOGO_STRING_ALT);
   data_source->set_json_path("strings.js");
   data_source->add_resource_path("set_as_default_browser.js",
       IDR_SET_AS_DEFAULT_BROWSER_JS);
   data_source->set_default_resource(IDR_SET_AS_DEFAULT_BROWSER_HTML);
   return data_source;
 }
 
+// A simple class serving as a delegate for passing down the result of the
+// interaction.
+class ResponseDelegate {
+ public:
+  virtual void SetChromeShutdownRequired(bool shutdown_chrome) = 0;
+};

[sky, 2012/08/09 16:13:47]

Add a protected virtual destructor.

[motek., 2012/08/09 17:17:48]

Done.

+
 // Event handler for SetAsDefaultBrowserUI. Capable of setting Chrome as the
 // default browser on button click, closing itself and triggering Chrome
 // restart.
 class SetAsDefaultBrowserHandler
     : public WebUIMessageHandler,
       public base::SupportsWeakPtr<SetAsDefaultBrowserHandler>,
       public ShellIntegration::DefaultWebClientObserver {
  public:
-  SetAsDefaultBrowserHandler();
+  explicit SetAsDefaultBrowserHandler(ResponseDelegate* response_delegate);
   virtual ~SetAsDefaultBrowserHandler();
 
   // WebUIMessageHandler implementation.
   virtual void RegisterMessages() OVERRIDE;
 
   // ShellIntegration::DefaultWebClientObserver implementation.
   virtual void SetDefaultWebClientUIState(
       ShellIntegration::DefaultWebClientUIState state) OVERRIDE;
-  virtual void OnSetAsDefaultConcluded(bool call_result)  OVERRIDE;
+  virtual void OnSetAsDefaultConcluded(bool close_chrome)  OVERRIDE;
   virtual bool IsInteractiveSetDefaultPermitted() OVERRIDE;
 
  private:
   // Handler for the 'Next' (or 'make Chrome the Metro browser') button.
   void HandleLaunchSetDefaultBrowserFlow(const ListValue* args);
 
   // Close this web ui.
-  void ConcludeInteraction();
+  void ConcludeInteraction(bool mark_success);
 
   // If required and possible, spawns a new Chrome in Metro mode and closes the
   // current instance. Windows 8 only, on earlier systems it will simply close
   // this UI.
   void ActivateMetroChrome();
 
   scoped_refptr<ShellIntegration::DefaultBrowserWorker> default_browser_worker_;
   bool set_default_returned_;
   bool set_default_result_;
+  ResponseDelegate* response_delegate_;
 
   DISALLOW_COPY_AND_ASSIGN(SetAsDefaultBrowserHandler);
 };
 
-SetAsDefaultBrowserHandler::SetAsDefaultBrowserHandler()
+SetAsDefaultBrowserHandler::SetAsDefaultBrowserHandler(
+    ResponseDelegate* response_delegate)
     : ALLOW_THIS_IN_INITIALIZER_LIST(default_browser_worker_(
           new ShellIntegration::DefaultBrowserWorker(this))),
-      set_default_returned_(false), set_default_result_(false) {
+      set_default_returned_(false), set_default_result_(false),
+      response_delegate_(response_delegate) {
 }
 
 SetAsDefaultBrowserHandler::~SetAsDefaultBrowserHandler() {
   default_browser_worker_->ObserverDestroyed();
 }
 
 void SetAsDefaultBrowserHandler::RegisterMessages() {
   web_ui()->RegisterMessageCallback(
       "SetAsDefaultBrowser:LaunchSetDefaultBrowserFlow",
       base::Bind(&SetAsDefaultBrowserHandler::HandleLaunchSetDefaultBrowserFlow,
                  base::Unretained(this)));
 }
 
 void SetAsDefaultBrowserHandler::SetDefaultWebClientUIState(
     ShellIntegration::DefaultWebClientUIState state) {
   // The callback is expected to be invoked once the procedure has completed.
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   if (!set_default_returned_)
     return;
 
   if (state == ShellIntegration::STATE_NOT_DEFAULT && set_default_result_) {
     // The operation concluded, but Chrome is still not the default.
     // If the call has succeeded, this suggests user has decided not to make
     // chrome the default. We fold this UI and move on.
-    ConcludeInteraction();
+    ConcludeInteraction(false);
   } else if (state == ShellIntegration::STATE_IS_DEFAULT) {
     if (!Profile::FromWebUI(web_ui())->GetPrefs()->GetBoolean(
             prefs::kSuppressSwitchToMetroModeOnSetDefault)) {
       BrowserThread::PostTask(
           BrowserThread::FILE, FROM_HERE,
           base::Bind(&SetAsDefaultBrowserHandler::ActivateMetroChrome,
                      base::Unretained(this)));
     } else {
-      ConcludeInteraction();
+      ConcludeInteraction(false);
     }
   }
 }
 
 void SetAsDefaultBrowserHandler::OnSetAsDefaultConcluded(bool call_result) {
   set_default_returned_ = true;
   set_default_result_ = call_result;
 }
 
 bool SetAsDefaultBrowserHandler::IsInteractiveSetDefaultPermitted() {
   return true;
 }
 
 void SetAsDefaultBrowserHandler::HandleLaunchSetDefaultBrowserFlow(
     const ListValue* args) {
   set_default_returned_ = false;
   set_default_result_ = false;
   default_browser_worker_->StartSetAsDefault();
 }
 
-void SetAsDefaultBrowserHandler::ConcludeInteraction() {
+void SetAsDefaultBrowserHandler::ConcludeInteraction(bool close_chrome) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   WebContents* contents = web_ui()->GetWebContents();
+  if (response_delegate_)
+    response_delegate_->SetChromeShutdownRequired(close_chrome);
+
   if (contents) {
     content::WebContentsDelegate* delegate = contents->GetDelegate();
     if (delegate)
       delegate->CloseContents(contents);
   }
 }
 
 void SetAsDefaultBrowserHandler::ActivateMetroChrome() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
   FilePath cur_chrome_exe;
   bool sentinel_removed = false;
   if (PathService::Get(base::FILE_EXE, &cur_chrome_exe) &&
       first_run::IsChromeFirstRun() &&
       InstallUtil::IsPerUserInstall(cur_chrome_exe.value().c_str())) {
     // If this is per-user install, we will have to remove the sentinel file
     // to assure the user goes through the intended 'first-run flow'.
     sentinel_removed = first_run::RemoveSentinel();
   }
 
-  if (ShellIntegration::ActivateMetroChrome()) {
-    // If Metro Chrome has been activated, we should close this process.
-    // We are restarting as metro now.
-    BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
-        base::Bind(&BrowserList::CloseAllBrowsersWithProfile,
-                   Profile::FromWebUI(web_ui())));
-  } else {
-    // This will return false if the operation failed for any reason,
-    // including invocation under a Windows version earlier than 8.
-    // In such case we simply close the window and carry on.
-    if (sentinel_removed)
+  bool metro_chrome_activated = ShellIntegration::ActivateMetroChrome();
+  if (!metro_chrome_activated && sentinel_removed)
       first_run::CreateSentinel();
 
-    BrowserThread::PostTask(
+  BrowserThread::PostTask(
         BrowserThread::UI, FROM_HERE,
         base::Bind(&SetAsDefaultBrowserHandler::ConcludeInteraction,
-                    base::Unretained(this)));
-  }
+                   base::Unretained(this), metro_chrome_activated));
 }
 
 // A web dialog delegate implementation for when 'Make Chrome Metro' UI
 // is displayed on a dialog.
-class SetAsDefaultBrowserDialogImpl : public ui::WebDialogDelegate {
+class SetAsDefaultBrowserDialogImpl : public ui::WebDialogDelegate,
+                                      public ResponseDelegate {
  public:
   SetAsDefaultBrowserDialogImpl(Profile* profile, Browser* browser);
+  virtual ~SetAsDefaultBrowserDialogImpl();
   // Show a modal web dialog with kChromeUIMetroFlowURL page.
   void ShowDialog();
 
  protected:
   // Overridden from WebDialogDelegate:
   virtual ui::ModalType GetDialogModalType() const OVERRIDE;
   virtual string16 GetDialogTitle() const OVERRIDE;
   virtual GURL GetDialogContentURL() const OVERRIDE;
   virtual void GetWebUIMessageHandlers(
       std::vector<WebUIMessageHandler*>* handlers) const OVERRIDE;
   virtual void GetDialogSize(gfx::Size* size) const OVERRIDE;
   virtual std::string GetDialogArgs() const OVERRIDE;
   virtual void OnDialogClosed(const std::string& json_retval) OVERRIDE;
   virtual void OnCloseContents(WebContents* source,
                                bool* out_close_dialog) OVERRIDE;
   virtual bool ShouldShowDialogTitle() const OVERRIDE;
   virtual bool HandleContextMenu(
       const content::ContextMenuParams& params) OVERRIDE;
 
+  // Overridden from ResponseDelegate:
+  virtual void SetChromeShutdownRequired(bool shutdown_chrome);
+
  private:
   Profile* profile_;
   Browser* browser_;
+  mutable bool owns_handler_;
+  SetAsDefaultBrowserHandler* handler_;
+  bool response_is_close_chrome_;
 
   DISALLOW_COPY_AND_ASSIGN(SetAsDefaultBrowserDialogImpl);
 };
 
 SetAsDefaultBrowserDialogImpl::SetAsDefaultBrowserDialogImpl(Profile* profile,
                                                              Browser* browser)
-    : profile_(profile), browser_(browser) {
+    : profile_(profile),
+      browser_(browser),
+      owns_handler_(true),
+      handler_(new SetAsDefaultBrowserHandler(this)),
+      response_is_close_chrome_(false) {
+}
+
+SetAsDefaultBrowserDialogImpl::~SetAsDefaultBrowserDialogImpl() {
+  if (owns_handler_)
+    delete handler_;
 }
 
 void SetAsDefaultBrowserDialogImpl::ShowDialog() {
   chrome::ShowWebDialog(browser_->window()->GetNativeWindow(),
                         browser_->profile(), this);
 }
 
 ui::ModalType SetAsDefaultBrowserDialogImpl::GetDialogModalType() const {
   return ui::MODAL_TYPE_SYSTEM;
 }
 
 string16 SetAsDefaultBrowserDialogImpl::GetDialogTitle() const {
   return l10n_util::GetStringUTF16(IDS_METRO_FLOW_TAB_TITLE);
 }
 
 GURL SetAsDefaultBrowserDialogImpl::GetDialogContentURL() const {
   std::string url_string(chrome::kChromeUIMetroFlowURL);
   return GURL(url_string);
 }
 
 void SetAsDefaultBrowserDialogImpl::GetWebUIMessageHandlers(
     std::vector<WebUIMessageHandler*>* handlers) const {
+  handlers->push_back(handler_);
+  owns_handler_ = false;
 }
 
 void SetAsDefaultBrowserDialogImpl::GetDialogSize(gfx::Size* size) const {
   PrefService* prefs = profile_->GetPrefs();
   gfx::Font approximate_web_font(
       prefs->GetString(prefs::kWebKitSansSerifFontFamily),
       prefs->GetInteger(prefs::kWebKitDefaultFontSize));
 
   *size = ui::GetLocalizedContentsSizeForFont(
       IDS_METRO_FLOW_WIDTH_CHARS, IDS_METRO_FLOW_HEIGHT_LINES,
       approximate_web_font);
 }
 
 std::string SetAsDefaultBrowserDialogImpl::GetDialogArgs() const {
   return "[]";
 }
 
 void SetAsDefaultBrowserDialogImpl::OnDialogClosed(
     const std::string& json_retval) {
+  if (response_is_close_chrome_) {
+    // If Metro Chrome has been activated, we should close this process.
+    // We are restarting as metro now.
+    BrowserList::CloseAllBrowsersWithProfile(profile_);
+  } else {
+    // This will be false if the user closed the dialog without doing anything
+    // or if operation failed for any reason (including invocation under a
+    // Windows version earlier than 8).
+    // In such case we just carry on with a normal chrome session. However, for
+    // the purpose of surfacing this dialog the actual browser window had to
+    // remain hidden. Now it's the time to show it.
+    BrowserWindow* window = browser_->window();

[sky, 2012/08/09 16:13:47]

Do you need to worry about browser being deleted out from under you?

[motek., 2012/08/09 17:17:48]

I did, sort of. But if I am not mistaken, this code is IT (the popup that this
is a delegate of is associated with this instance of the browser). 
That said: would you think it would be better to revert to that iteration I had
here earlier?

+    WebContents* contents = chrome::GetActiveWebContents(browser_);
+    if (window)

[sky, 2012/08/09 16:13:47]

This should never be NULL.

[motek., 2012/08/09 17:17:48]

Done.

+      window->Show();
+    if (contents)
+      contents->GetView()->SetInitialFocus();
+  }
+
   delete this;
 }
 
 void SetAsDefaultBrowserDialogImpl::OnCloseContents(WebContents* source,
                                                     bool* out_close_dialog) {
   *out_close_dialog = true;
 }
 
 bool SetAsDefaultBrowserDialogImpl::ShouldShowDialogTitle() const {
   return true;
 }
 
 bool SetAsDefaultBrowserDialogImpl::HandleContextMenu(
     const content::ContextMenuParams& params) {
   return true;
 }
 
+void SetAsDefaultBrowserDialogImpl::SetChromeShutdownRequired(
+    bool shutdown_chrome) {
+  response_is_close_chrome_ = shutdown_chrome;
+}
+
 }  // namespace
 
 SetAsDefaultBrowserUI::SetAsDefaultBrowserUI(content::WebUI* web_ui)
-    : WebUIController(web_ui) {
-  web_ui->AddMessageHandler(new SetAsDefaultBrowserHandler());
+    : ui::WebDialogUI(web_ui) {
   ChromeURLDataManager::AddDataSource(Profile::FromWebUI(web_ui),
                                       CreateSetAsDefaultBrowserUIHTMLSource());
 }
 
 // static
 void SetAsDefaultBrowserUI::Show(Profile* profile, Browser* browser) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   SetAsDefaultBrowserDialogImpl* dialog =
       new SetAsDefaultBrowserDialogImpl(profile, browser);
   dialog->ShowDialog();
 }