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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// Copyright 2021 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/account_manager_core/chromeos/access_token_fetcher.h"
#include <string>
#include <utility>
#include <vector>
#include "base/functional/bind.h"
#include "base/task/sequenced_task_runner.h"
#include "chromeos/crosapi/mojom/account_manager.mojom.h"
#include "components/account_manager_core/account_manager_util.h"
#include "google_apis/gaia/gaia_urls.h"
#include "google_apis/gaia/google_service_auth_error.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
namespace crosapi {
AccessTokenFetcher::AccessTokenFetcher(
account_manager::AccountManager* account_manager,
mojom::AccountKeyPtr mojo_account_key,
const std::string& consumer_name,
base::OnceCallback<void(AccessTokenFetcher*)> done_callback,
mojo::PendingReceiver<mojom::AccessTokenFetcher> receiver)
: consumer_name_(consumer_name),
done_callback_(std::move(done_callback)),
receiver_(this, std::move(receiver)) {
receiver_.set_disconnect_handler(base::BindOnce(
&AccessTokenFetcher::OnMojoPipeError, base::Unretained(this)));
std::optional<account_manager::AccountKey> maybe_account_key =
account_manager::FromMojoAccountKey(mojo_account_key);
if (maybe_account_key.has_value()) {
access_token_fetcher_ = account_manager->CreateAccessTokenFetcher(
/*account_key=*/maybe_account_key.value(), /*consumer=*/this);
}
// else: access_token_fetcher_ will be `nullptr`. `Start` will handle this
// case.
}
AccessTokenFetcher::~AccessTokenFetcher() = default;
void AccessTokenFetcher::Start(const std::vector<std::string>& scopes,
StartCallback callback) {
DCHECK(callback_.is_null());
DCHECK(callback);
callback_ = std::move(callback);
if (!access_token_fetcher_) {
// `access_token_fetcher_` can be null only if `account_key` is invalid /
// unknown.
OnGetTokenFailure(GoogleServiceAuthError(
GoogleServiceAuthError::State::USER_NOT_SIGNED_UP));
return;
}
access_token_fetcher_->Start(
GaiaUrls::GetInstance()->oauth2_chrome_client_id(),
GaiaUrls::GetInstance()->oauth2_chrome_client_secret(), scopes);
}
void AccessTokenFetcher::OnGetTokenSuccess(
const TokenResponse& token_response) {
mojom::AccessTokenResultPtr result =
mojom::AccessTokenResult::NewAccessTokenInfo(mojom::AccessTokenInfo::New(
token_response.access_token, token_response.expiration_time,
token_response.id_token));
std::move(callback_).Run(std::move(result));
Finish();
}
void AccessTokenFetcher::OnGetTokenFailure(
const GoogleServiceAuthError& error) {
mojom::AccessTokenResultPtr result = mojom::AccessTokenResult::NewError(
account_manager::ToMojoGoogleServiceAuthError(error));
std::move(callback_).Run(std::move(result));
Finish();
}
std::string AccessTokenFetcher::GetConsumerName() const {
return consumer_name_;
}
void AccessTokenFetcher::OnMojoPipeError() {
if (access_token_fetcher_)
access_token_fetcher_->CancelRequest();
if (callback_) {
// We don't need to respond to callback. The Mojo pipe has been
// disconnected.
callback_.Reset();
}
Finish();
}
void AccessTokenFetcher::Finish() {
DCHECK(callback_.is_null())
<< "Finish called before responding to pending request";
// If `OnMojoPipeError` is called after `OnGetTokenSuccess` or
// `OnGetTokenFailure`, the `done_callback_` will be null.
if (done_callback_.is_null())
return;
// We cannot call `TriggerDeletion` directly because it will immediately start
// deleting `this`, before this method has had a chance to return.
base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
FROM_HERE, base::BindOnce(std::move(done_callback_), this));
}
} // namespace crosapi
|
