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
|
// 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.
#ifndef CHROME_BROWSER_ASH_POLICY_REMOTE_COMMANDS_CRD_CRD_OAUTH_TOKEN_FETCHER_H_
#define CHROME_BROWSER_ASH_POLICY_REMOTE_COMMANDS_CRD_CRD_OAUTH_TOKEN_FETCHER_H_
#include <memory>
#include <optional>
#include <string>
#include "base/functional/callback.h"
#include "base/memory/raw_ref.h"
#include "google_apis/gaia/oauth2_access_token_manager.h"
class DeviceOAuth2TokenService;
namespace policy {
// Helper class that asynchronously fetches an OAuth token that can be used to
// start and accept an incoming CRD session.
class CrdOAuthTokenFetcher {
public:
using OAuthTokenCallback =
base::OnceCallback<void(std::optional<std::string>)>;
virtual ~CrdOAuthTokenFetcher() = default;
virtual void Start(OAuthTokenCallback done_callback) = 0;
};
// Real implementation of the OAuth token fetcher that will use the
// given `DeviceOAuth2TokenService` to fetch an OAuth token valid for an
// incoming CRD session.
class RealCrdOAuthTokenFetcher : public CrdOAuthTokenFetcher,
public OAuth2AccessTokenManager::Consumer {
public:
using OAuthTokenCallback =
base::OnceCallback<void(std::optional<std::string>)>;
explicit RealCrdOAuthTokenFetcher(DeviceOAuth2TokenService& oauth_service);
RealCrdOAuthTokenFetcher(const RealCrdOAuthTokenFetcher&) = delete;
RealCrdOAuthTokenFetcher& operator=(const RealCrdOAuthTokenFetcher&) = delete;
~RealCrdOAuthTokenFetcher() override;
void Start(OAuthTokenCallback done_callback) override;
private:
// `OAuth2AccessTokenManager::Consumer` implementation:
void OnGetTokenSuccess(
const OAuth2AccessTokenManager::Request* request,
const OAuth2AccessTokenConsumer::TokenResponse& token_response) override;
void OnGetTokenFailure(const OAuth2AccessTokenManager::Request* request,
const GoogleServiceAuthError& error) override;
const raw_ref<DeviceOAuth2TokenService> oauth_service_;
OAuthTokenCallback done_callback_;
// Handle for the OAuth access token request.
std::unique_ptr<OAuth2AccessTokenManager::Request> oauth_request_;
};
// Fake implementation of the OAuth token fetcher that will simply return
// the given `oauth_token`.
// Used during unittesting.
class FakeCrdOAuthTokenFetcher : public CrdOAuthTokenFetcher {
public:
explicit FakeCrdOAuthTokenFetcher(std::optional<std::string> oauth_token);
~FakeCrdOAuthTokenFetcher() override;
void Start(OAuthTokenCallback done_callback) override;
private:
std::optional<std::string> oauth_token_;
};
} // namespace policy
#endif // CHROME_BROWSER_ASH_POLICY_REMOTE_COMMANDS_CRD_CRD_OAUTH_TOKEN_FETCHER_H_
|
