1 //
2 // Copyright (C) 2019 The Android Open Source Project
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 //      http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 
16 #pragma once
17 
18 #include <chrono>
19 #include <memory>
20 
21 #include "curl_wrapper.h"
22 
23 namespace cuttlefish {
24 
25 class CredentialSource {
26 public:
27   virtual ~CredentialSource() = default;
28   virtual std::string Credential() = 0;
29 };
30 
31 class GceMetadataCredentialSource : public CredentialSource {
32   CurlWrapper curl;
33   std::string latest_credential;
34   std::chrono::steady_clock::time_point expiration;
35 
36   void RefreshCredential();
37 public:
38   GceMetadataCredentialSource();
39   GceMetadataCredentialSource(GceMetadataCredentialSource&&) = default;
40 
41   virtual std::string Credential();
42 
43   static std::unique_ptr<CredentialSource> make();
44 };
45 
46 class FixedCredentialSource : public CredentialSource {
47   std::string credential;
48 public:
49   FixedCredentialSource(const std::string& credential);
50 
51   virtual std::string Credential();
52 
53   static std::unique_ptr<CredentialSource> make(const std::string& credential);
54 };
55 
56 }
57