1 files changed, 146 insertions, 0 deletions
diff --git a/chromium/chrome/browser/signin/process_dice_header_delegate_impl.cc b/chromium/chrome/browser/signin/process_dice_header_delegate_impl.cc
new file mode 100644
index 00000000000..5669e179b48
--- /dev/null
+++ b/chromium/chrome/browser/signin/process_dice_header_delegate_impl.cc
@@ -0,0 +1,146 @@
+// Copyright 2017 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/signin/process_dice_header_delegate_impl.h"
+
+#include <utility>
+
+#include "base/callback.h"
+#include "base/logging.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/signin/dice_tab_helper.h"
+#include "chrome/browser/signin/dice_web_signin_interceptor.h"
+#include "chrome/browser/signin/dice_web_signin_interceptor_factory.h"
+#include "chrome/browser/signin/identity_manager_factory.h"
+#include "chrome/browser/ui/webui/signin/signin_ui_error.h"
+#include "chrome/common/url_constants.h"
+#include "components/signin/public/identity_manager/identity_manager.h"
+#include "content/public/browser/navigation_controller.h"
+#include "content/public/browser/web_contents.h"
+#include "url/gurl.h"
+
+namespace {
+
+void RedirectToNtp(content::WebContents* contents) {
+  VLOG(1) << "RedirectToNtp";
+  contents->GetController().LoadURL(
+      GURL(chrome::kChromeUINewTabURL), content::Referrer(),
+      ui::PAGE_TRANSITION_AUTO_TOPLEVEL, std::string());
+}
+
+// Helper function similar to DiceTabHelper::FromWebContents(), but also handles
+// the case where |contents| is nullptr.
+DiceTabHelper* GetDiceTabHelperFromWebContents(content::WebContents* contents) {
+  if (!contents)
+    return nullptr;
+  return DiceTabHelper::FromWebContents(contents);
+}
+
+}  // namespace
+
+ProcessDiceHeaderDelegateImpl::ProcessDiceHeaderDelegateImpl(
+    content::WebContents* web_contents,
+    EnableSyncCallback enable_sync_callback,
+    ShowSigninErrorCallback show_signin_error_callback)
+    : web_contents_(web_contents->GetWeakPtr()),
+      profile_(Profile::FromBrowserContext(web_contents->GetBrowserContext())),
+      enable_sync_callback_(std::move(enable_sync_callback)),
+      show_signin_error_callback_(std::move(show_signin_error_callback)) {
+  DCHECK(profile_);
+
+  DiceTabHelper* tab_helper = DiceTabHelper::FromWebContents(web_contents);
+  if (tab_helper) {
+    is_sync_signin_tab_ = tab_helper->IsSyncSigninInProgress();
+    redirect_url_ = tab_helper->redirect_url();
+    access_point_ = tab_helper->signin_access_point();
+    promo_action_ = tab_helper->signin_promo_action();
+    reason_ = tab_helper->signin_reason();
+  }
+}
+
+ProcessDiceHeaderDelegateImpl::~ProcessDiceHeaderDelegateImpl() = default;
+
+bool ProcessDiceHeaderDelegateImpl::ShouldEnableSync() {
+  if (IdentityManagerFactory::GetForProfile(profile_)->HasPrimaryAccount(
+          signin::ConsentLevel::kSync)) {
+    VLOG(1) << "Do not start sync after web sign-in [already authenticated].";
+    return false;
+  }
+
+  if (!is_sync_signin_tab_) {
+    VLOG(1)
+        << "Do not start sync after web sign-in [not a Chrome sign-in tab].";
+    return false;
+  }
+
+  return true;
+}
+
+void ProcessDiceHeaderDelegateImpl::HandleTokenExchangeSuccess(
+    CoreAccountId account_id,
+    bool is_new_account) {
+  // is_sync_signin_tab_ tells whether the current signin is happening in a tab
+  // that was opened from a "Enable Sync" Chrome UI. Usually this is indeed a
+  // sync signin, but it is not always the case: the user may abandon the sync
+  // signin and do a simple web signin in the same tab instead.
+  DiceWebSigninInterceptorFactory::GetForProfile(profile_)
+      ->MaybeInterceptWebSignin(web_contents_.get(), account_id, is_new_account,
+                                is_sync_signin_tab_);
+}
+
+void ProcessDiceHeaderDelegateImpl::EnableSync(
+    const CoreAccountId& account_id) {
+  DiceTabHelper* tab_helper =
+      GetDiceTabHelperFromWebContents(web_contents_.get());
+  if (tab_helper)
+    tab_helper->OnSyncSigninFlowComplete();
+
+  if (!ShouldEnableSync()) {
+    // No special treatment is needed if the user is not enabling sync.
+    return;
+  }
+
+  content::WebContents* web_contents = web_contents_.get();
+  VLOG(1) << "Start sync after web sign-in.";
+  std::move(enable_sync_callback_)
+      .Run(profile_.get(), access_point_, promo_action_, reason_, web_contents,
+           account_id);
+
+  if (!web_contents)
+    return;
+
+  // After signing in to Chrome, the user should be redirected to the NTP,
+  // unless specified otherwise.
+  if (redirect_url_.is_empty()) {
+    RedirectToNtp(web_contents);
+    return;
+  }
+
+  DCHECK(redirect_url_.is_valid());
+  web_contents->GetController().LoadURL(redirect_url_, content::Referrer(),
+                                        ui::PAGE_TRANSITION_AUTO_TOPLEVEL,
+                                        std::string());
+}
+
+void ProcessDiceHeaderDelegateImpl::HandleTokenExchangeFailure(
+    const std::string& email,
+    const GoogleServiceAuthError& error) {
+  DCHECK_NE(GoogleServiceAuthError::NONE, error.state());
+  DiceTabHelper* tab_helper =
+      GetDiceTabHelperFromWebContents(web_contents_.get());
+  if (tab_helper)
+    tab_helper->OnSyncSigninFlowComplete();
+
+  bool should_enable_sync = ShouldEnableSync();
+
+  content::WebContents* web_contents = web_contents_.get();
+  if (should_enable_sync && web_contents)
+    RedirectToNtp(web_contents);
+
+  // Show the error even if the WebContents was closed, because the user may be
+  // signed out of the web.
+  std::move(show_signin_error_callback_)
+      .Run(profile_.get(), web_contents,
+           SigninUIError::FromGoogleServiceAuthError(email, error));
+}