OLD | NEW |
| (Empty) |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/browser/extensions/api/identity/account_tracker.h" | |
6 | |
7 #include "base/logging.h" | |
8 #include "base/stl_util.h" | |
9 #include "chrome/browser/browser_process.h" | |
10 #include "chrome/browser/chrome_notification_types.h" | |
11 #include "chrome/browser/profiles/profile.h" | |
12 #include "chrome/browser/signin/profile_oauth2_token_service_factory.h" | |
13 #include "chrome/browser/signin/signin_manager_factory.h" | |
14 #include "components/signin/core/browser/profile_oauth2_token_service.h" | |
15 #include "components/signin/core/browser/signin_manager.h" | |
16 #include "content/public/browser/notification_details.h" | |
17 #include "extensions/browser/extension_system.h" | |
18 | |
19 namespace extensions { | |
20 | |
21 AccountTracker::AccountTracker(Profile* profile) : profile_(profile) { | |
22 ProfileOAuth2TokenService* service = | |
23 ProfileOAuth2TokenServiceFactory::GetForProfile(profile_); | |
24 service->AddObserver(this); | |
25 service->signin_error_controller()->AddProvider(this); | |
26 SigninManagerFactory::GetForProfile(profile_)->AddObserver(this); | |
27 } | |
28 | |
29 AccountTracker::~AccountTracker() {} | |
30 | |
31 void AccountTracker::ReportAuthError(const std::string& account_id, | |
32 const GoogleServiceAuthError& error) { | |
33 account_errors_.insert(make_pair(account_id, error)); | |
34 ProfileOAuth2TokenServiceFactory::GetForProfile(profile_)-> | |
35 signin_error_controller()->AuthStatusChanged(); | |
36 UpdateSignInState(account_id, false); | |
37 } | |
38 | |
39 void AccountTracker::Shutdown() { | |
40 STLDeleteValues(&user_info_requests_); | |
41 SigninManagerFactory::GetForProfile(profile_)->RemoveObserver(this); | |
42 ProfileOAuth2TokenService* service = | |
43 ProfileOAuth2TokenServiceFactory::GetForProfile(profile_); | |
44 service->signin_error_controller()->RemoveProvider(this); | |
45 service->RemoveObserver(this); | |
46 } | |
47 | |
48 void AccountTracker::AddObserver(Observer* observer) { | |
49 observer_list_.AddObserver(observer); | |
50 } | |
51 | |
52 void AccountTracker::RemoveObserver(Observer* observer) { | |
53 observer_list_.RemoveObserver(observer); | |
54 } | |
55 | |
56 std::vector<AccountIds> AccountTracker::GetAccounts() const { | |
57 const std::string primary_account_id = signin_manager_account_id(); | |
58 std::vector<AccountIds> accounts; | |
59 | |
60 for (std::map<std::string, AccountState>::const_iterator it = | |
61 accounts_.begin(); | |
62 it != accounts_.end(); | |
63 ++it) { | |
64 const AccountState& state = it->second; | |
65 bool is_visible = state.is_signed_in && !state.ids.gaia.empty(); | |
66 | |
67 if (it->first == primary_account_id) { | |
68 if (is_visible) | |
69 accounts.insert(accounts.begin(), state.ids); | |
70 else | |
71 return std::vector<AccountIds>(); | |
72 | |
73 } else if (is_visible) { | |
74 accounts.push_back(state.ids); | |
75 } | |
76 } | |
77 return accounts; | |
78 } | |
79 | |
80 std::string AccountTracker::FindAccountKeyByGaiaId(const std::string& gaia_id) { | |
81 for (std::map<std::string, AccountState>::const_iterator it = | |
82 accounts_.begin(); | |
83 it != accounts_.end(); | |
84 ++it) { | |
85 const AccountState& state = it->second; | |
86 if (state.ids.gaia == gaia_id) { | |
87 return state.ids.account_key; | |
88 } | |
89 } | |
90 | |
91 return std::string(); | |
92 } | |
93 | |
94 void AccountTracker::OnRefreshTokenAvailable(const std::string& account_id) { | |
95 // Ignore refresh tokens if there is no primary account ID at all. | |
96 if (signin_manager_account_id().empty()) | |
97 return; | |
98 | |
99 DVLOG(1) << "AVAILABLE " << account_id; | |
100 ClearAuthError(account_id); | |
101 UpdateSignInState(account_id, true); | |
102 } | |
103 | |
104 void AccountTracker::OnRefreshTokenRevoked(const std::string& account_id) { | |
105 DVLOG(1) << "REVOKED " << account_id; | |
106 UpdateSignInState(account_id, false); | |
107 } | |
108 | |
109 void AccountTracker::GoogleSigninSucceeded(const std::string& username, | |
110 const std::string& password) { | |
111 std::vector<std::string> accounts = | |
112 ProfileOAuth2TokenServiceFactory::GetForProfile(profile_)->GetAccounts(); | |
113 | |
114 for (std::vector<std::string>::const_iterator it = accounts.begin(); | |
115 it != accounts.end(); | |
116 ++it) { | |
117 OnRefreshTokenAvailable(*it); | |
118 } | |
119 } | |
120 | |
121 void AccountTracker::GoogleSignedOut(const std::string& username) { | |
122 if (username == signin_manager_account_id() || | |
123 signin_manager_account_id().empty()) { | |
124 StopTrackingAllAccounts(); | |
125 } else { | |
126 StopTrackingAccount(username); | |
127 } | |
128 } | |
129 | |
130 void AccountTracker::SetAccountStateForTest(AccountIds ids, bool is_signed_in) { | |
131 accounts_[ids.account_key].ids = ids; | |
132 accounts_[ids.account_key].is_signed_in = is_signed_in; | |
133 | |
134 DVLOG(1) << "SetAccountStateForTest " << ids.account_key << ":" | |
135 << is_signed_in; | |
136 | |
137 if (VLOG_IS_ON(1)) { | |
138 for (std::map<std::string, AccountState>::const_iterator it = | |
139 accounts_.begin(); | |
140 it != accounts_.end(); | |
141 ++it) { | |
142 DVLOG(1) << it->first << ":" << it->second.is_signed_in; | |
143 } | |
144 } | |
145 } | |
146 | |
147 const std::string AccountTracker::signin_manager_account_id() const { | |
148 return SigninManagerFactory::GetForProfile(profile_) | |
149 ->GetAuthenticatedAccountId(); | |
150 } | |
151 | |
152 void AccountTracker::NotifyAccountAdded(const AccountState& account) { | |
153 DCHECK(!account.ids.gaia.empty()); | |
154 FOR_EACH_OBSERVER( | |
155 Observer, observer_list_, OnAccountAdded(account.ids)); | |
156 } | |
157 | |
158 void AccountTracker::NotifyAccountRemoved(const AccountState& account) { | |
159 DCHECK(!account.ids.gaia.empty()); | |
160 FOR_EACH_OBSERVER( | |
161 Observer, observer_list_, OnAccountRemoved(account.ids)); | |
162 } | |
163 | |
164 void AccountTracker::NotifySignInChanged(const AccountState& account) { | |
165 DCHECK(!account.ids.gaia.empty()); | |
166 FOR_EACH_OBSERVER(Observer, | |
167 observer_list_, | |
168 OnAccountSignInChanged(account.ids, account.is_signed_in)); | |
169 } | |
170 | |
171 void AccountTracker::ClearAuthError(const std::string& account_key) { | |
172 account_errors_.erase(account_key); | |
173 ProfileOAuth2TokenServiceFactory::GetForProfile(profile_)-> | |
174 signin_error_controller()->AuthStatusChanged(); | |
175 } | |
176 | |
177 void AccountTracker::UpdateSignInState(const std::string& account_key, | |
178 bool is_signed_in) { | |
179 StartTrackingAccount(account_key); | |
180 AccountState& account = accounts_[account_key]; | |
181 bool needs_gaia_id = account.ids.gaia.empty(); | |
182 bool was_signed_in = account.is_signed_in; | |
183 account.is_signed_in = is_signed_in; | |
184 | |
185 if (needs_gaia_id && is_signed_in) | |
186 StartFetchingUserInfo(account_key); | |
187 | |
188 if (!needs_gaia_id && (was_signed_in != is_signed_in)) | |
189 NotifySignInChanged(account); | |
190 } | |
191 | |
192 void AccountTracker::StartTrackingAccount(const std::string& account_key) { | |
193 if (!ContainsKey(accounts_, account_key)) { | |
194 DVLOG(1) << "StartTracking " << account_key; | |
195 AccountState account_state; | |
196 account_state.ids.account_key = account_key; | |
197 account_state.ids.email = account_key; | |
198 account_state.is_signed_in = false; | |
199 accounts_.insert(make_pair(account_key, account_state)); | |
200 } | |
201 } | |
202 | |
203 void AccountTracker::StopTrackingAccount(const std::string& account_key) { | |
204 if (ContainsKey(accounts_, account_key)) { | |
205 AccountState& account = accounts_[account_key]; | |
206 if (!account.ids.gaia.empty()) { | |
207 UpdateSignInState(account_key, false); | |
208 NotifyAccountRemoved(account); | |
209 } | |
210 accounts_.erase(account_key); | |
211 } | |
212 | |
213 ClearAuthError(account_key); | |
214 | |
215 if (ContainsKey(user_info_requests_, account_key)) | |
216 DeleteFetcher(user_info_requests_[account_key]); | |
217 } | |
218 | |
219 void AccountTracker::StopTrackingAllAccounts() { | |
220 while (!accounts_.empty()) | |
221 StopTrackingAccount(accounts_.begin()->first); | |
222 } | |
223 | |
224 void AccountTracker::StartFetchingUserInfo(const std::string& account_key) { | |
225 if (ContainsKey(user_info_requests_, account_key)) | |
226 DeleteFetcher(user_info_requests_[account_key]); | |
227 | |
228 DVLOG(1) << "StartFetching " << account_key; | |
229 AccountIdFetcher* fetcher = | |
230 new AccountIdFetcher(profile_, this, account_key); | |
231 user_info_requests_[account_key] = fetcher; | |
232 fetcher->Start(); | |
233 } | |
234 | |
235 void AccountTracker::OnUserInfoFetchSuccess(AccountIdFetcher* fetcher, | |
236 const std::string& gaia_id) { | |
237 const std::string& account_key = fetcher->account_key(); | |
238 DCHECK(ContainsKey(accounts_, account_key)); | |
239 AccountState& account = accounts_[account_key]; | |
240 | |
241 account.ids.gaia = gaia_id; | |
242 NotifyAccountAdded(account); | |
243 | |
244 if (account.is_signed_in) | |
245 NotifySignInChanged(account); | |
246 | |
247 DeleteFetcher(fetcher); | |
248 } | |
249 | |
250 void AccountTracker::OnUserInfoFetchFailure(AccountIdFetcher* fetcher) { | |
251 LOG(WARNING) << "Failed to get UserInfo for " << fetcher->account_key(); | |
252 std::string key = fetcher->account_key(); | |
253 DeleteFetcher(fetcher); | |
254 StopTrackingAccount(key); | |
255 } | |
256 | |
257 std::string AccountTracker::GetAccountId() const { | |
258 if (account_errors_.size() == 0) | |
259 return std::string(); | |
260 else | |
261 return account_errors_.begin()->first; | |
262 } | |
263 | |
264 std::string AccountTracker::GetUsername() const { | |
265 std::string id = GetAccountId(); | |
266 if (!id.empty()) { | |
267 std::map<std::string, AccountState>::const_iterator it = | |
268 accounts_.find(id); | |
269 if (it != accounts_.end()) | |
270 return it->second.ids.email; | |
271 } | |
272 return std::string(); | |
273 } | |
274 | |
275 GoogleServiceAuthError AccountTracker::GetAuthStatus() const { | |
276 if (account_errors_.size() == 0) | |
277 return GoogleServiceAuthError::AuthErrorNone(); | |
278 else | |
279 return account_errors_.begin()->second; | |
280 } | |
281 | |
282 void AccountTracker::DeleteFetcher(AccountIdFetcher* fetcher) { | |
283 const std::string& account_key = fetcher->account_key(); | |
284 DCHECK(ContainsKey(user_info_requests_, account_key)); | |
285 DCHECK_EQ(fetcher, user_info_requests_[account_key]); | |
286 user_info_requests_.erase(account_key); | |
287 delete fetcher; | |
288 } | |
289 | |
290 AccountIdFetcher::AccountIdFetcher(Profile* profile, | |
291 AccountTracker* tracker, | |
292 const std::string& account_key) | |
293 : OAuth2TokenService::Consumer("extensions_account_tracker"), | |
294 profile_(profile), | |
295 tracker_(tracker), | |
296 account_key_(account_key) {} | |
297 | |
298 AccountIdFetcher::~AccountIdFetcher() {} | |
299 | |
300 void AccountIdFetcher::Start() { | |
301 ProfileOAuth2TokenService* service = | |
302 ProfileOAuth2TokenServiceFactory::GetForProfile(profile_); | |
303 login_token_request_ = service->StartRequest( | |
304 account_key_, OAuth2TokenService::ScopeSet(), this); | |
305 } | |
306 | |
307 void AccountIdFetcher::OnGetTokenSuccess( | |
308 const OAuth2TokenService::Request* request, | |
309 const std::string& access_token, | |
310 const base::Time& expiration_time) { | |
311 DCHECK_EQ(request, login_token_request_.get()); | |
312 | |
313 gaia_oauth_client_.reset(new gaia::GaiaOAuthClient( | |
314 g_browser_process->system_request_context())); | |
315 | |
316 const int kMaxGetUserIdRetries = 3; | |
317 gaia_oauth_client_->GetUserId(access_token, kMaxGetUserIdRetries, this); | |
318 } | |
319 | |
320 void AccountIdFetcher::OnGetTokenFailure( | |
321 const OAuth2TokenService::Request* request, | |
322 const GoogleServiceAuthError& error) { | |
323 LOG(ERROR) << "OnGetTokenFailure: " << error.ToString(); | |
324 DCHECK_EQ(request, login_token_request_.get()); | |
325 tracker_->OnUserInfoFetchFailure(this); | |
326 } | |
327 | |
328 void AccountIdFetcher::OnGetUserIdResponse(const std::string& gaia_id) { | |
329 tracker_->OnUserInfoFetchSuccess(this, gaia_id); | |
330 } | |
331 | |
332 void AccountIdFetcher::OnOAuthError() { | |
333 LOG(ERROR) << "OnOAuthError"; | |
334 tracker_->OnUserInfoFetchFailure(this); | |
335 } | |
336 | |
337 void AccountIdFetcher::OnNetworkError(int response_code) { | |
338 LOG(ERROR) << "OnNetworkError " << response_code; | |
339 tracker_->OnUserInfoFetchFailure(this); | |
340 } | |
341 | |
342 } // namespace extensions | |
OLD | NEW |