oauth_token_getter_proxy.cc 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. // Copyright 2019 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. #include "remoting/base/oauth_token_getter_proxy.h"
  5. #include "base/bind.h"
  6. #include "base/task/single_thread_task_runner.h"
  7. #include "base/threading/sequenced_task_runner_handle.h"
  8. #include "base/threading/thread_checker.h"
  9. namespace remoting {
  10. namespace {
  11. void ResolveCallback(
  12. OAuthTokenGetter::TokenCallback on_access_token,
  13. scoped_refptr<base::SequencedTaskRunner> original_task_runner,
  14. OAuthTokenGetter::Status status,
  15. const std::string& user_email,
  16. const std::string& access_token) {
  17. if (!original_task_runner->RunsTasksInCurrentSequence()) {
  18. original_task_runner->PostTask(
  19. FROM_HERE, base::BindOnce(std::move(on_access_token), status,
  20. user_email, access_token));
  21. return;
  22. }
  23. std::move(on_access_token).Run(status, user_email, access_token);
  24. }
  25. } // namespace
  26. OAuthTokenGetterProxy::OAuthTokenGetterProxy(
  27. base::WeakPtr<OAuthTokenGetter> token_getter,
  28. scoped_refptr<base::SequencedTaskRunner> task_runner)
  29. : token_getter_(token_getter), task_runner_(task_runner) {}
  30. OAuthTokenGetterProxy::OAuthTokenGetterProxy(
  31. base::WeakPtr<OAuthTokenGetter> token_getter)
  32. : OAuthTokenGetterProxy(token_getter,
  33. base::SequencedTaskRunnerHandle::Get()) {}
  34. OAuthTokenGetterProxy::~OAuthTokenGetterProxy() = default;
  35. void OAuthTokenGetterProxy::CallWithToken(
  36. OAuthTokenGetter::TokenCallback on_access_token) {
  37. if (!task_runner_->RunsTasksInCurrentSequence()) {
  38. auto task_runner_to_reply = base::SequencedTaskRunnerHandle::Get();
  39. auto reply_callback = base::BindOnce(
  40. &ResolveCallback, std::move(on_access_token), task_runner_to_reply);
  41. task_runner_->PostTask(
  42. FROM_HERE, base::BindOnce(&OAuthTokenGetter::CallWithToken,
  43. token_getter_, std::move(reply_callback)));
  44. return;
  45. }
  46. if (token_getter_) {
  47. token_getter_->CallWithToken(std::move(on_access_token));
  48. }
  49. }
  50. void OAuthTokenGetterProxy::InvalidateCache() {
  51. if (!task_runner_->RunsTasksInCurrentSequence()) {
  52. task_runner_->PostTask(
  53. FROM_HERE,
  54. base::BindOnce(&OAuthTokenGetter::InvalidateCache, token_getter_));
  55. return;
  56. }
  57. if (token_getter_) {
  58. token_getter_->InvalidateCache();
  59. }
  60. }
  61. } // namespace remoting