• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright (c) 2012 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 REMOTING_HOST_SETUP_WIN_AUTH_CODE_GETTER_H
6 #define REMOTING_HOST_SETUP_WIN_AUTH_CODE_GETTER_H
7 
8 #include <ole2.h>
9 #include <exdisp.h>
10 
11 #include <string>
12 
13 #include "base/callback.h"
14 #include "base/threading/non_thread_safe.h"
15 #include "base/timer/timer.h"
16 #include "base/win/scoped_comptr.h"
17 
18 namespace remoting {
19 
20 // A class for getting an OAuth authorization code.
21 class AuthCodeGetter : public base::NonThreadSafe {
22  public:
23   AuthCodeGetter();
24   ~AuthCodeGetter();
25 
26   // Starts a browser and navigates it to a URL that starts an Installed
27   // Application OAuth flow. |on_auth_code| will be called with an
28   // authorization code, or an empty string on error.
29   void GetAuthCode(base::Callback<void(const std::string&)> on_auth_code);
30 
31  private:
32   // Starts a timer used to poll the browser's URL.
33   void StartTimer();
34   // Called when that timer fires.
35   void OnTimer();
36   // Returns whether to stop polling the browser's URL. If true, then
37   // |auth_code| is an authorization code, or the empty string on an error.
38   bool TestBrowserUrl(std::string* auth_code);
39   // Kills the browser.
40   void KillBrowser();
41 
42   // The authorization code callback.
43   base::Callback<void(const std::string&)> on_auth_code_;
44   // The browser through which the user requests an authorization code.
45   base::win::ScopedComPtr<IWebBrowser2, &IID_IWebBrowser2> browser_;
46   // A timer used to poll the browser's URL.
47   base::OneShotTimer<AuthCodeGetter> timer_;
48   // The interval at which the timer fires.
49   base::TimeDelta timer_interval_;
50 
51   DISALLOW_COPY_AND_ASSIGN(AuthCodeGetter);
52 };
53 
54 }  // namespace remoting
55 
56 #endif  // REMOTING_HOST_SETUP_WIN_AUTH_CODE_GETTER_H
57