Prototype version of the first-run dialog for Windows 8

chrome/browser/ui/startup/default_browser_prompt.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/startup/default_browser_prompt.h"
 
 #include "base/memory/weak_ptr.h"
 #include "base/message_loop.h"
 #include "base/metrics/histogram.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/first_run/first_run.h"
 #include "chrome/browser/infobars/infobar_tab_helper.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/shell_integration.h"
 #include "chrome/browser/tab_contents/confirm_infobar_delegate.h"
 #include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/browser_finder.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
+#include "chrome/browser/ui/webui/metroizer_ui.h"
 #include "chrome/common/pref_names.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/navigation_details.h"
+#include "content/public/browser/notification_service.h"
+#include "content/public/browser/notification_types.h"
 #include "grit/chromium_strings.h"
 #include "grit/generated_resources.h"
 #include "grit/theme_resources.h"
 #include "grit/theme_resources_standard.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/resource/resource_bundle.h"
 
 using content::BrowserThread;
 
 namespace {
 
 // Calls the appropriate function for setting Chrome as the default browser.
 // This requires IO access (registry) and may result in interaction with a
 // modal system UI.
-void SetChromeAsDefaultBrowser(bool interactive_flow) {
+void SetChromeAsDefaultBrowser(bool interactive_flow, PrefService* prefs) {
   if (interactive_flow) {
     UMA_HISTOGRAM_COUNTS("DefaultBrowserWarning.SetAsDefaultUI", 1);
-    if (!ShellIntegration::SetAsDefaultBrowserInteractive())
+    if (!ShellIntegration::SetAsDefaultBrowserInteractive()) {
       UMA_HISTOGRAM_COUNTS("DefaultBrowserWarning.SetAsDefaultUIFailed", 1);
+    } else if (!ShellIntegration::IsDefaultBrowser()) {
+      // If the interaction succeeded but we are still not the default browser
+      // it likely means the user simply selected another browser from the
+      // panel. We will respect this choice and write it down as 'no, thanks'.
+      UMA_HISTOGRAM_COUNTS("DefaultBrowserWarning.DontSetAsDefault", 1);
+      // User clicked "Don't ask me again", remember that.
+      if (prefs)
+        prefs->SetBoolean(prefs::kCheckDefaultBrowser, false);
+    }
   } else {
     UMA_HISTOGRAM_COUNTS("DefaultBrowserWarning.SetAsDefault", 1);
     ShellIntegration::SetAsDefaultBrowser();
   }
 }
 
 // The delegate for the infobar shown when Chrome is not the default browser.
 class DefaultBrowserInfoBarDelegate : public ConfirmInfoBarDelegate {
  public:
   DefaultBrowserInfoBarDelegate(InfoBarTabHelper* infobar_helper,
@@ skipping 80 matching lines @@
 
 bool DefaultBrowserInfoBarDelegate::NeedElevation(InfoBarButton button) const {
   return button == BUTTON_OK;
 }
 
 bool DefaultBrowserInfoBarDelegate::Accept() {
   action_taken_ = true;
   BrowserThread::PostTask(
       BrowserThread::FILE,
       FROM_HERE,
-      base::Bind(&SetChromeAsDefaultBrowser, interactive_flow_required_));
+      base::Bind(&SetChromeAsDefaultBrowser,

[grt (UTC plus 2), 2012/06/18 19:19:13]

condense

[motek., 2012/06/19 18:06:44]

Done.

+                 interactive_flow_required_,
+                 prefs_));
 
   return true;
 }
 
 bool DefaultBrowserInfoBarDelegate::Cancel() {
   action_taken_ = true;
   UMA_HISTOGRAM_COUNTS("DefaultBrowserWarning.DontSetAsDefault", 1);
   // User clicked "Don't ask me again", remember that.
   prefs_->SetBoolean(prefs::kCheckDefaultBrowser, false);
   return true;
 }
 
 void CheckDefaultBrowserCallback() {
   if (!ShellIntegration::IsDefaultBrowser()) {
     ShellIntegration::DefaultWebClientSetPermission default_change_mode =
         ShellIntegration::CanSetAsDefaultBrowser();
 
     if (default_change_mode != ShellIntegration::SET_DEFAULT_NOT_ALLOWED) {
       BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
           base::Bind(&browser::internal::NotifyNotDefaultBrowserCallback));
     }
   }
 }
 
+static void CheckDefaultAndShowMetroizerCallback(
+    Profile* profile, Browser* browser, bool dialog) {
+  if (!ShellIntegration::IsDefaultBrowser() ||
+      (ShellIntegration::CanSetAsDefaultBrowser() ==
+           ShellIntegration::SET_DEFAULT_INTERACTIVE)) {
+      MetroizerUI::Show(profile, browser, dialog);

[grt (UTC plus 2), 2012/06/18 19:19:13]

de-indent two spaces

[motek., 2012/06/19 18:06:44]

Done.

+  }
+}
+
+class SetMetroBrowserFlowLauncher : public content::NotificationObserver {
+ public:
+  static void LaunchSoon(Profile* profile) {
+    // The instance will manage its own lifetime.
+    new SetMetroBrowserFlowLauncher(profile);
+  }
+
+ private:
+  explicit SetMetroBrowserFlowLauncher(Profile* profile)
+      : profile_(profile) {
+    registrar_.Add(this, content::NOTIFICATION_LOAD_COMPLETED_MAIN_FRAME,
+                   content::NotificationService::AllSources());
+  }
+
+  // content::NotificationObserver override:
+  virtual void Observe(int type,
+                       const content::NotificationSource& source,
+                       const content::NotificationDetails& details) OVERRIDE;
+
+  content::NotificationRegistrar registrar_;
+  Profile* profile_;
+
+  DISALLOW_COPY_AND_ASSIGN(SetMetroBrowserFlowLauncher);
+};
+
+void SetMetroBrowserFlowLauncher::Observe(
+    int type,
+    const content::NotificationSource& source,
+    const content::NotificationDetails& details) {
+  DCHECK_EQ(type, content::NOTIFICATION_LOAD_COMPLETED_MAIN_FRAME);
+  Browser* browser = browser::FindBrowserWithWebContents(
+    content::Source<content::WebContents>(source).ptr());

[grt (UTC plus 2), 2012/06/18 19:19:13]

nit: indent two more spaces

[motek., 2012/06/19 18:06:44]

Done.

+
+  if (!browser || !browser->is_type_tabbed())
+    return;
+
+  // Unregister and delete.

[grt (UTC plus 2), 2012/06/18 19:19:13]

i don't see the delete.  is it handled by the registrar?  if it were, the
profile_ access on line 237 would be bad.  i think you need an explicit delete
this; at the very end of this function.

[motek., 2012/06/19 18:06:44]

You are right, obviously. I must have clipped it in editing, it was there when I
lifted this code from FirstRunBubbleLauncher.

+  registrar_.Remove(this, content::NOTIFICATION_LOAD_COMPLETED_MAIN_FRAME,
+                    content::NotificationService::AllSources());
+
+  bool as_dialog =
+      profile_->GetPrefs()->GetBoolean(prefs::kDefaultBrowserFlowDialog);
+  BrowserThread::PostTask(

[grt (UTC plus 2), 2012/06/18 19:19:13]

save the whitespace!  (args can share lines)

[motek., 2012/06/19 18:06:44]

Done.

+      BrowserThread::FILE,
+      FROM_HERE,
+      base::Bind(&CheckDefaultAndShowMetroizerCallback,
+                 profile_, browser, as_dialog));
+}
+
 }  // namespace
 
 namespace browser {
 
 void ShowDefaultBrowserPrompt(Profile* profile) {
   // We do not check if we are the default browser if:
   // - the user said "don't ask me again" on the infobar earlier.
   // - this is the first launch after the first run flow.
   // - There is a policy in control of this setting.
   if (!profile->GetPrefs()->GetBoolean(prefs::kCheckDefaultBrowser) ||
@@ skipping 12 matching lines @@
       // TODO(pastarmovj): We can't really do anything meaningful here yet but
       // just prevent showing the infobar.
     }
     return;
   }
   BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE,
                           base::Bind(&CheckDefaultBrowserCallback));
 
 }
 
+void ShowFirstRunDefaultBrowserPrompt(Profile* profile) {
+  if (ShellIntegration::SET_DEFAULT_INTERACTIVE ==
+          ShellIntegration::CanSetAsDefaultBrowser()) {
+    // If the only available mode of setting the default browser requires
+    // user interaction, it means this couldn't have been done yet. Perform
+    // this action as a delayed task (including the check).
+    SetMetroBrowserFlowLauncher::LaunchSoon(profile);
+  }
+}
+
 namespace internal {
 
 void NotifyNotDefaultBrowserCallback() {
   Browser* browser = BrowserList::GetLastActive();
   if (!browser)
     return;  // Reached during ui tests.
 
   // In ChromeBot tests, there might be a race. This line appears to get
   // called during shutdown and |tab| can be NULL.
   TabContentsWrapper* tab = browser->GetSelectedTabContentsWrapper();
   if (!tab)
     return;
 
   // Don't show the info-bar if there are already info-bars showing.
   InfoBarTabHelper* infobar_helper = tab->infobar_tab_helper();
   if (infobar_helper->infobar_count() > 0)
     return;
 
   bool interactive_flow = ShellIntegration::CanSetAsDefaultBrowser() ==
       ShellIntegration::SET_DEFAULT_INTERACTIVE;
   infobar_helper->AddInfoBar(
       new DefaultBrowserInfoBarDelegate(infobar_helper,
                                         tab->profile()->GetPrefs(),
                                         interactive_flow));
 }
 
 }  // namespace internal
 }  // namespace browser