Implement online reauth UI for Locked Profiles on Mac

chrome/browser/profiles/profile_window.cc

 // Copyright 2013 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/profiles/profile_window.h"
 
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/prefs/pref_service.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/about_flags.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/lifetime/application_lifetime.h"
 #include "chrome/browser/pref_service_flags_storage.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_avatar_icon_util.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/signin/account_reconcilor_factory.h"
 #include "chrome/browser/signin/account_tracker_service_factory.h"
 #include "chrome/browser/signin/signin_manager_factory.h"
+#include "chrome/browser/signin/signin_promo.h"
 #include "chrome/browser/signin/signin_ui_util.h"
 #include "chrome/browser/sync/profile_sync_service.h"
 #include "chrome/browser/sync/profile_sync_service_factory.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_dialogs.h"
 #include "chrome/browser/ui/profile_chooser_constants.h"
 #include "chrome/browser/ui/user_manager.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
+#include "components/guest_view/browser/guest_view_manager.h"
 #include "components/signin/core/browser/account_reconcilor.h"
 #include "components/signin/core/browser/account_tracker_service.h"
 #include "components/signin/core/browser/signin_manager.h"
 #include "components/signin/core/common/profile_management_switches.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/user_metrics.h"
+#include "google_apis/gaia/gaia_urls.h"
 
 #if defined(ENABLE_EXTENSIONS)
 #include "chrome/browser/extensions/extension_service.h"
 #include "extensions/browser/extension_prefs.h"
 #include "extensions/browser/extension_registry.h"
 #include "extensions/browser/extension_registry_factory.h"
 #include "extensions/browser/extension_system.h"
 #endif  // defined(ENABLE_EXTENSIONS)
 
 #if !defined(OS_IOS)
@@ skipping 177 matching lines @@
 
 // Updates Chrome services that require notification when
 // the new_profile_management's status changes.
 void UpdateServicesWithNewProfileManagementFlag(Profile* profile,
                                                 bool new_flag_status) {
   AccountReconcilor* account_reconcilor =
       AccountReconcilorFactory::GetForProfile(profile);
   account_reconcilor->OnNewProfileManagementFlagChanged(new_flag_status);
 }
 
+bool AddToSet(std::set<content::WebContents*>* content_set,
+              content::WebContents* web_contents) {
+  content_set->insert(web_contents);
+  return false;
+}
+
 }  // namespace
 
 namespace profiles {
 
 // User Manager parameters are prefixed with hash.
 const char kUserManagerDisplayTutorial[] = "#tutorial";
 const char kUserManagerSelectProfileTaskManager[] = "#task-manager";
 const char kUserManagerSelectProfileAboutChrome[] = "#about-chrome";
 const char kUserManagerSelectProfileChromeSettings[] = "#chrome-settings";
 const char kUserManagerSelectProfileChromeMemory[] = "#chrome-memory";
 const char kUserManagerSelectProfileAppLauncher[] = "#app-launcher";
 
+ReauthDialogObserver::ReauthDialogObserver(content::WebContents* web_contents,
+                                           const std::string& email_address)
+    : web_contents_(web_contents),
+      email_address_(email_address) {}
+
+void ReauthDialogObserver::DidStopLoading() {
+  // If the sign in process reaches the termination URL, close the dialog.
+  // Make sure to remove any parts of the URL that gaia might append during
+  // signin.
+  GURL url = web_contents_->GetURL();

[Roger Tawa OOO till Jul 10th, 2015/08/05 13:36:05]

I don't believe this is correct.  It should be |web_contents()->GetURL()|.

Lines 286 and 288 below are correct, but could be replaced with web_contents()
too.  The if() check at like 278 essentially sees if we are still observing the
WebView's web_content. (In retrospect, I'm not sure if my comment at lines
275-277 is descriptive enough).

[anthonyvd, 2015/08/05 18:20:44]

Done.

+  url::Replacements<char> replacements;
+  replacements.ClearQuery();
+  replacements.ClearRef();
+  if (url.ReplaceComponents(replacements) ==
+        GaiaUrls::GetInstance()->signin_completed_continue_url()) {
+    CloseReauthDialog();
+    return;
+  }
+
+  // If still observing the top level web contents, try to find the embedded
+  // webview and observe it instead.  The webview may not be found in the
+  // initial page load since it loads asynchronously.
+  if (url.GetOrigin() !=
+        signin::GetReauthURLWithEmail(email_address_).GetOrigin()) {
+    return;
+  }
+
+  std::set<content::WebContents*> content_set;
+  guest_view::GuestViewManager* manager =
+      guest_view::GuestViewManager::FromBrowserContext(
+          web_contents_->GetBrowserContext());
+  if (manager)
+    manager->ForEachGuest(web_contents_, base::Bind(&AddToSet, &content_set));
+  DCHECK_LE(content_set.size(), 1U);
+  if (!content_set.empty())
+    Observe(*content_set.begin());
+}
+
 base::FilePath GetPathOfProfileWithEmail(ProfileManager* profile_manager,
                                          const std::string& email) {
   base::string16 profile_email = base::UTF8ToUTF16(email);
   std::vector<ProfileAttributesEntry*> entries =
       profile_manager->GetProfileInfoCache().GetAllProfilesAttributes();
   for (ProfileAttributesEntry* entry : entries) {
     if (entry->GetUserName() == profile_email)
       return entry->GetPath();
   }
   return base::FilePath();
@@ skipping 297 matching lines @@
   PrefService* local_state = g_browser_process->local_state();
   const bool dismissed = local_state->GetBoolean(
       prefs::kProfileAvatarRightClickTutorialDismissed);
 
   // Don't show the tutorial if it's already been dismissed or if right-clicking
   // wouldn't show any targets.
   return !dismissed && HasProfileSwitchTargets(profile);
 }
 
 }  // namespace profiles