Index: chrome/browser/chromeos/arc/arc_auth_service.cc |
diff --git a/chrome/browser/chromeos/arc/arc_auth_service.cc b/chrome/browser/chromeos/arc/arc_auth_service.cc |
index 5755999c4e3467be926a25b7f62c11da1a602091..3614e03339040c4d8474c154f1060e6460bbd506 100644 |
--- a/chrome/browser/chromeos/arc/arc_auth_service.cc |
+++ b/chrome/browser/chromeos/arc/arc_auth_service.cc |
@@ -7,13 +7,24 @@ |
#include <utility> |
#include "base/command_line.h" |
-#include "chrome/browser/chromeos/arc/arc_auth_ui.h" |
+#include "base/strings/stringprintf.h" |
+#include "chrome/browser/extensions/extension_util.h" |
#include "chrome/browser/profiles/profile.h" |
+#include "chrome/browser/signin/profile_oauth2_token_service_factory.h" |
+#include "chrome/browser/signin/signin_manager_factory.h" |
+#include "chrome/browser/ui/extensions/app_launch_params.h" |
+#include "chrome/browser/ui/extensions/application_launch.h" |
#include "chrome/common/pref_names.h" |
#include "chromeos/chromeos_switches.h" |
#include "components/arc/arc_bridge_service.h" |
#include "components/pref_registry/pref_registry_syncable.h" |
#include "components/prefs/pref_service.h" |
+#include "components/signin/core/browser/profile_oauth2_token_service.h" |
+#include "components/signin/core/browser/signin_manager_base.h" |
+#include "content/public/browser/storage_partition.h" |
+#include "content/public/common/url_constants.h" |
+#include "extensions/browser/extension_registry.h" |
+#include "google_apis/gaia/gaia_constants.h" |
namespace arc { |
@@ -22,6 +33,9 @@ namespace { |
// Weak pointer. This class is owned by ArcServiceManager. |
ArcAuthService* arc_auth_service = nullptr; |
+const char kArcSupportExtensionId[] = "cnbgggchhmkkdmeppjobngjoejnihlei"; |
+const char kArcSupportStorageId[] = "arc_support"; |
+ |
// Skip creating UI in unit tests |
bool disable_ui_for_testing = false; |
@@ -40,7 +54,7 @@ ArcAuthService::ArcAuthService(ArcBridgeService* bridge_service) |
} |
ArcAuthService::~ArcAuthService() { |
- DCHECK(!auth_ui_ && !profile_); |
+ DCHECK(!profile_); |
arc_bridge_service()->RemoveObserver(this); |
DCHECK(arc_auth_service == this); |
@@ -112,6 +126,13 @@ void ArcAuthService::OnPrimaryUserProfilePrepared(Profile* profile) { |
Shutdown(); |
profile_ = profile; |
+ // Reuse storage used in ARC OptIn platform app. |
+ const std::string site_url = |
+ base::StringPrintf("%s://%s/persist?%s", content::kGuestScheme, |
+ kArcSupportExtensionId, kArcSupportStorageId); |
+ storage_partition_ = content::BrowserContext::GetStoragePartitionForSite( |
+ profile_, GURL(site_url)); |
+ CHECK(storage_partition_); |
// In case UI is disabled we assume that ARC is opted-in. |
if (!IsOptInVerificationDisabled()) { |
@@ -122,7 +143,9 @@ void ArcAuthService::OnPrimaryUserProfilePrepared(Profile* profile) { |
base::Unretained(this))); |
OnOptInPreferenceChanged(); |
} else { |
- SetAuthCodeAndStartArc(std::string()); |
+ auth_code_.clear(); |
+ ArcBridgeService::Get()->HandleStartup(); |
+ SetState(State::ENABLE); |
} |
} |
@@ -132,20 +155,51 @@ void ArcAuthService::Shutdown() { |
pref_change_registrar_.RemoveAll(); |
} |
+void ArcAuthService::OnMergeSessionSuccess(const std::string& data) { |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
+ |
+ const extensions::Extension* extension = |
+ extensions::ExtensionRegistry::Get(profile_)->GetInstalledExtension( |
+ kArcSupportExtensionId); |
+ CHECK(extension && |
+ extensions::util::IsAppLaunchable(kArcSupportExtensionId, profile_)); |
+ |
+ AppLaunchParams params(profile_, extension, SINGLETON_TAB, |
xiyuan
2016/02/12 17:06:59
nit: SINGLETON_TAB -> NEW_WINDOW
WindowOpenDispos
khmel
2016/02/12 18:15:14
Thanks for explanation. I could not understand the
|
+ extensions::SOURCE_CHROME_INTERNAL); |
+ OpenApplication(params); |
+} |
+ |
+void ArcAuthService::OnMergeSessionFailure( |
+ const GoogleServiceAuthError& error) { |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
+ VLOG(2) << "Failed to merge gaia session " << error.ToString() << "."; |
+ OnAuthCodeFailed(); |
+} |
+ |
+void ArcAuthService::OnUbertokenSuccess(const std::string& token) { |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
+ merger_fetcher_.reset( |
+ new GaiaAuthFetcher(this, GaiaConstants::kChromeOSSource, |
+ storage_partition_->GetURLRequestContext())); |
+ merger_fetcher_->StartMergeSession(token, std::string()); |
+} |
+ |
+void ArcAuthService::OnUbertokenFailure(const GoogleServiceAuthError& error) { |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
+ VLOG(2) << "Failed to get ubertoken " << error.ToString() << "."; |
+ OnAuthCodeFailed(); |
+} |
+ |
void ArcAuthService::OnOptInPreferenceChanged() { |
DCHECK(thread_checker_.CalledOnValidThread()); |
DCHECK(profile_); |
if (profile_->GetPrefs()->GetBoolean(prefs::kArcEnabled)) { |
- switch (state_) { |
- case State::DISABLE: |
- FetchAuthCode(); |
- break; |
- case State::NO_CODE: // Retry |
- FetchAuthCode(); |
- break; |
- default: |
- break; |
+ if (state_ != State::ENABLE) { |
+ CloseUI(); |
+ auth_code_.clear(); |
+ SetState(State::FETCHING_CODE); |
+ FetchAuthCode(); |
} |
} else { |
ShutdownBridgeAndCloseUI(); |
@@ -155,6 +209,8 @@ void ArcAuthService::OnOptInPreferenceChanged() { |
void ArcAuthService::ShutdownBridgeAndCloseUI() { |
CloseUI(); |
auth_fetcher_.reset(); |
+ ubertoken_fethcher_.reset(); |
+ merger_fetcher_.reset(); |
ArcBridgeService::Get()->Shutdown(); |
SetState(State::DISABLE); |
} |
@@ -170,35 +226,44 @@ void ArcAuthService::RemoveObserver(Observer* observer) { |
} |
void ArcAuthService::CloseUI() { |
- if (auth_ui_) { |
- auth_ui_->Close(); |
- DCHECK(!auth_ui_); |
- } |
+ FOR_EACH_OBSERVER(Observer, observer_list_, OnOptInUINeedToClose()); |
} |
void ArcAuthService::SetAuthCodeAndStartArc(const std::string& auth_code) { |
DCHECK(thread_checker_.CalledOnValidThread()); |
- DCHECK(!auth_code.empty() || IsOptInVerificationDisabled()); |
- DCHECK_NE(state_, State::ENABLE); |
+ DCHECK(!auth_code.empty()); |
+ |
+ State state = state_; |
ShutdownBridgeAndCloseUI(); |
+ if (state != State::FETCHING_CODE) |
+ return; |
+ |
auth_code_ = auth_code; |
ArcBridgeService::Get()->HandleStartup(); |
- |
SetState(State::ENABLE); |
} |
void ArcAuthService::FetchAuthCode() { |
DCHECK(thread_checker_.CalledOnValidThread()); |
- DCHECK(state_ == State::DISABLE || state_ == State::NO_CODE); |
- CloseUI(); |
- auth_code_.clear(); |
+ if (state_ != State::FETCHING_CODE) |
+ return; |
+ |
+ auth_fetcher_.reset( |
+ new ArcAuthFetcher(storage_partition_->GetURLRequestContext(), this)); |
+} |
+ |
+void ArcAuthService::CancelAuthCode() { |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
- SetState(State::FETCHING_CODE); |
+ if (state_ != State::FETCHING_CODE) |
+ return; |
- auth_fetcher_.reset(new ArcAuthFetcher(profile_->GetRequestContext(), this)); |
+ ShutdownBridgeAndCloseUI(); |
+ |
+ profile_->GetPrefs()->SetBoolean(prefs::kArcEnabled, false); |
} |
void ArcAuthService::OnAuthCodeFetched(const std::string& auth_code) { |
@@ -206,10 +271,29 @@ void ArcAuthService::OnAuthCodeFetched(const std::string& auth_code) { |
SetAuthCodeAndStartArc(auth_code); |
} |
+void ArcAuthService::ShowUI() { |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
+ |
+ // Get auth token to continue. |
+ ProfileOAuth2TokenService* token_service = |
+ ProfileOAuth2TokenServiceFactory::GetForProfile(profile_); |
+ SigninManagerBase* signin_manager = |
+ SigninManagerFactory::GetForProfile(profile_); |
+ CHECK(token_service && signin_manager); |
+ const std::string& account_id = signin_manager->GetAuthenticatedAccountId(); |
+ ubertoken_fethcher_.reset( |
+ new UbertokenFetcher(token_service, this, GaiaConstants::kChromeOSSource, |
+ storage_partition_->GetURLRequestContext())); |
+ ubertoken_fethcher_->StartFetchingToken(account_id); |
+} |
+ |
void ArcAuthService::OnAuthCodeNeedUI() { |
- CloseUI(); |
- if (!disable_ui_for_testing && !IsOptInVerificationDisabled()) |
- auth_ui_ = new ArcAuthUI(profile_, this); |
+ DCHECK(thread_checker_.CalledOnValidThread()); |
+ |
+ if (disable_ui_for_testing || IsOptInVerificationDisabled()) |
+ return; |
+ |
+ ShowUI(); |
} |
void ArcAuthService::OnAuthCodeFailed() { |
@@ -219,11 +303,6 @@ void ArcAuthService::OnAuthCodeFailed() { |
SetState(State::NO_CODE); |
} |
-void ArcAuthService::OnAuthUIClosed() { |
- DCHECK(auth_ui_); |
- auth_ui_ = nullptr; |
-} |
- |
std::ostream& operator<<(std::ostream& os, const ArcAuthService::State& state) { |
switch (state) { |
case ArcAuthService::State::DISABLE: |