1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/optimization_guide/core/access_token_helper.h"
#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "base/metrics/histogram_functions.h"
#include "components/optimization_guide/core/optimization_guide_enums.h"
#include "components/signin/public/identity_manager/access_token_fetcher.h"
#include "components/signin/public/identity_manager/access_token_info.h"
#include "components/signin/public/identity_manager/primary_account_access_token_fetcher.h"
#include "google_apis/gaia/gaia_constants.h"
#include "google_apis/gaia/google_service_auth_error.h"
namespace optimization_guide {
namespace {
constexpr char kOptGuideOAuthConsumerName[] = "optimization_guide";
void RecordAccessTokenResultHistogram(
OptimizationGuideAccessTokenResult result) {
base::UmaHistogramEnumeration("OptimizationGuide.AccessTokenHelper.Result",
result);
}
// Invoked when access token is ready.
void OnAccessTokenRequestCompleted(
std::unique_ptr<signin::PrimaryAccountAccessTokenFetcher>,
AccessTokenReceivedCallback callback,
GoogleServiceAuthError error,
signin::AccessTokenInfo access_token_info) {
DCHECK_EQ(access_token_info.token.empty(),
error.state() != GoogleServiceAuthError::NONE);
std::move(callback).Run(access_token_info.token);
if (error.state() == GoogleServiceAuthError::NONE) {
RecordAccessTokenResultHistogram(
OptimizationGuideAccessTokenResult::kSuccess);
} else if (error.IsTransientError()) {
RecordAccessTokenResultHistogram(
OptimizationGuideAccessTokenResult::kTransientError);
} else {
RecordAccessTokenResultHistogram(
OptimizationGuideAccessTokenResult::kPersistentError);
}
}
} // namespace
void HandleTokenRequestFlow(bool require_token,
signin::IdentityManager* identity_manager,
const std::set<std::string>& oauth_scopes,
AccessTokenReceivedCallback callback) {
DCHECK(!oauth_scopes.empty());
if (!require_token) {
std::move(callback).Run(std::string());
return;
}
if (!identity_manager) {
RecordAccessTokenResultHistogram(
OptimizationGuideAccessTokenResult::kUserNotSignedIn);
std::move(callback).Run(std::string());
return;
}
if (!identity_manager->HasPrimaryAccount(signin::ConsentLevel::kSignin)) {
RecordAccessTokenResultHistogram(
OptimizationGuideAccessTokenResult::kUserNotSignedIn);
std::move(callback).Run(std::string());
return;
}
auto access_token_fetcher =
std::make_unique<signin::PrimaryAccountAccessTokenFetcher>(
kOptGuideOAuthConsumerName, identity_manager, oauth_scopes,
signin::PrimaryAccountAccessTokenFetcher::Mode::kImmediate,
signin::ConsentLevel::kSignin);
auto* access_token_fetcher_ptr = access_token_fetcher.get();
access_token_fetcher_ptr->Start(
base::BindOnce(&OnAccessTokenRequestCompleted,
std::move(access_token_fetcher), std::move(callback)));
}
} // namespace optimization_guide
|
