| OLD | NEW |
| (Empty) |
| 1 // Copyright 2016 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 #ifndef BLIMP_CLIENT_CORE_SESSION_ASSIGNMENT_SOURCE_H_ | |
| 6 #define BLIMP_CLIENT_CORE_SESSION_ASSIGNMENT_SOURCE_H_ | |
| 7 | |
| 8 #include <string> | |
| 9 | |
| 10 #include "base/callback.h" | |
| 11 #include "base/memory/ref_counted.h" | |
| 12 #include "base/memory/weak_ptr.h" | |
| 13 #include "blimp/client/public/session/assignment.h" | |
| 14 #include "net/url_request/url_fetcher_delegate.h" | |
| 15 #include "url/gurl.h" | |
| 16 | |
| 17 namespace base { | |
| 18 class SingleThreadTaskRunner; | |
| 19 class Value; | |
| 20 } | |
| 21 | |
| 22 namespace net { | |
| 23 class URLFetcher; | |
| 24 class URLRequestContextGetter; | |
| 25 } | |
| 26 | |
| 27 namespace blimp { | |
| 28 namespace client { | |
| 29 | |
| 30 // AssignmentSource provides functionality to find out how a client should | |
| 31 // connect to an engine. | |
| 32 class AssignmentSource : public net::URLFetcherDelegate { | |
| 33 public: | |
| 34 typedef base::Callback<void(AssignmentRequestResult, const Assignment&)> | |
| 35 AssignmentCallback; | |
| 36 | |
| 37 // |assigner_endpoint|: The URL of the Assigner service to query. | |
| 38 // |network_task_runner|: The task runner to use for querying the Assigner API | |
| 39 // over the network. | |
| 40 // |file_task_runner|: The task runner to use for reading cert files from disk | |
| 41 // (specified on the command line.) | |
| 42 AssignmentSource( | |
| 43 const GURL& assigner_endpoint, | |
| 44 const scoped_refptr<base::SingleThreadTaskRunner>& network_task_runner, | |
| 45 const scoped_refptr<base::SingleThreadTaskRunner>& file_task_runner); | |
| 46 | |
| 47 ~AssignmentSource() override; | |
| 48 | |
| 49 // Retrieves a valid assignment for the client and posts the result to the | |
| 50 // given callback. |client_auth_token| is the OAuth2 access token to send to | |
| 51 // the assigner when requesting an assignment. If this is called before a | |
| 52 // previous call has completed, the old callback will be called with | |
| 53 // RESULT_SERVER_INTERRUPTED and no Assignment. | |
| 54 void GetAssignment(const std::string& client_auth_token, | |
| 55 const AssignmentCallback& callback); | |
| 56 | |
| 57 private: | |
| 58 void OnGetAssignmentFromCommandLineDone(const std::string& client_auth_token, | |
| 59 Assignment parsed_assignment); | |
| 60 void QueryAssigner(const std::string& client_auth_token); | |
| 61 void ParseAssignerResponse(); | |
| 62 void OnJsonParsed(std::unique_ptr<base::Value> json); | |
| 63 void OnJsonParseError(const std::string& error); | |
| 64 | |
| 65 // net::URLFetcherDelegate implementation: | |
| 66 void OnURLFetchComplete(const net::URLFetcher* source) override; | |
| 67 | |
| 68 // GetAssignment() callback, invoked after URLFetcher completion. | |
| 69 AssignmentCallback callback_; | |
| 70 | |
| 71 const GURL assigner_endpoint_; | |
| 72 scoped_refptr<base::SingleThreadTaskRunner> file_task_runner_; | |
| 73 scoped_refptr<net::URLRequestContextGetter> url_request_context_; | |
| 74 std::unique_ptr<net::URLFetcher> url_fetcher_; | |
| 75 | |
| 76 base::WeakPtrFactory<AssignmentSource> weak_factory_; | |
| 77 | |
| 78 DISALLOW_COPY_AND_ASSIGN(AssignmentSource); | |
| 79 }; | |
| 80 | |
| 81 } // namespace client | |
| 82 } // namespace blimp | |
| 83 | |
| 84 #endif // BLIMP_CLIENT_CORE_SESSION_ASSIGNMENT_SOURCE_H_ | |
| OLD | NEW |