Fetch OAuth2 tokens prior to profile creation.

chrome/browser/chromeos/login/session/user_session_manager.h

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_BROWSER_CHROMEOS_LOGIN_SESSION_USER_SESSION_MANAGER_H_
 #define CHROME_BROWSER_CHROMEOS_LOGIN_SESSION_USER_SESSION_MANAGER_H_
 
 #include <string>
 
 #include "base/basictypes.h"
 #include "base/callback.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/singleton.h"
 #include "base/memory/weak_ptr.h"
 #include "base/observer_list.h"
 #include "chrome/browser/chromeos/base/locale_util.h"
 #include "chrome/browser/chromeos/login/signin/oauth2_login_manager.h"
+#include "chrome/browser/chromeos/login/signin/oauth2_token_fetcher.h"
 #include "chromeos/dbus/session_manager_client.h"
 #include "chromeos/login/auth/authenticator.h"
+#include "chromeos/login/auth/login_performer.h"
 #include "chromeos/login/auth/user_context.h"
 #include "components/user_manager/user.h"
 #include "components/user_manager/user_manager.h"
 #include "net/base/network_change_notifier.h"
 #include "ui/base/ime/chromeos/input_method_manager.h"
 
 class GURL;
 class PrefRegistrySimple;
 class PrefService;
 class Profile;
@@ skipping 42 matching lines @@
 // * mark user as logged in and notify observers,
 // * initialize OAuth2 authentication session,
 // * initialize and launch user session based on the user type.
 // Also supports restoring active user sessions after browser crash:
 // load profile, restore OAuth authentication session etc.
 class UserSessionManager
     : public OAuth2LoginManager::Observer,
       public net::NetworkChangeNotifier::ConnectionTypeObserver,
       public base::SupportsWeakPtr<UserSessionManager>,
       public UserSessionManagerDelegate,
-      public user_manager::UserManager::UserSessionStateObserver {
+      public user_manager::UserManager::UserSessionStateObserver,
+      public OAuth2TokenFetcher::Delegate {
  public:
   // Context of StartSession calls.
   typedef enum {
     // Starting primary user session, through login UI.
     PRIMARY_USER_SESSION,
 
     // Starting secondary user session, through multi-profiles login UI.
     SECONDARY_USER_SESSION,
 
     // Starting primary user session after browser crash.
     PRIMARY_USER_SESSION_AFTER_CRASH,
 
     // Starting secondary user session after browser crash.
     SECONDARY_USER_SESSION_AFTER_CRASH,
   } StartSessionType;
 
+  using FetchOAuth2TokensCallback =
+      base::Callback<void(const UserContext& user_context,
+                          LoginPerformer::AuthorizationMode auth_mode)>;
+
   // Returns UserSessionManager instance.
   static UserSessionManager* GetInstance();
 
   // Called when user is logged in to override base::DIR_HOME path.
   static void OverrideHomedir();
 
   // Registers session related preferences.
   static void RegisterPrefs(PrefRegistrySimple* registry);
 
   // Invoked after the tmpfs is successfully mounted.
   // Asks session_manager to restart Chrome in Guest session mode.
   // |start_url| is an optional URL to be opened in Guest session browser.
   void CompleteGuestSessionLogin(const GURL& start_url);
 
   // Creates and returns the authenticator to use.
   // Single Authenticator instance is used for entire login process,
   // even for multiple retries. Authenticator instance holds reference to
   // login profile and is later used during fetching of OAuth tokens.
   scoped_refptr<Authenticator> CreateAuthenticator(
       AuthStatusConsumer* consumer);
 
+  void FetchOAuth2Tokens(const UserContext& user_context,
+                         const FetchOAuth2TokensCallback& callback);
+
   // Start user session given |user_context|.
   // OnProfilePrepared() will be called on |delegate| once Profile is ready.
   void StartSession(const UserContext& user_context,
                     StartSessionType start_session_type,
                     bool has_auth_cookies,
                     bool has_active_session,
                     UserSessionManagerDelegate* delegate);
 
   // Invalidates |delegate|, which was passed to StartSession method call.
   void DelegateDeleted(UserSessionManagerDelegate* delegate);
@@ skipping 117 matching lines @@
       OAuth2LoginManager::SessionRestoreState state) override;
 
   // net::NetworkChangeNotifier::ConnectionTypeObserver overrides:
   void OnConnectionTypeChanged(
       net::NetworkChangeNotifier::ConnectionType type) override;
 
   // UserSessionManagerDelegate overrides:
   // Used when restoring user sessions after crash.
   void OnProfilePrepared(Profile* profile, bool browser_launched) override;
 
+  // OAuth2TokenFetcher::Delegate overrides.
+  void OnOAuth2TokensAvailable(
+      const GaiaAuthConsumer::ClientOAuthResult& oauth2_tokens) override;
+  void OnOAuth2TokensFetchFailed() override;
+
   void ChildAccountStatusReceivedCallback();
 
   void StopChildStatusObserving();
 
   void CreateUserSession(const UserContext& user_context,
                          bool has_auth_cookies);
   void PreStartSession();
 
   // Store any useful UserContext data early on when profile has not been
   // created yet and user services were not yet initialized. Can store
@@ skipping 160 matching lines @@
 
   // Set of user_id for those users that we should restore authentication
   // session when notified about online state change.
   SigninSessionRestoreStateSet pending_signin_restore_sessions_;
 
   // Kiosk mode related members.
   // Chrome oauth client id and secret - override values for kiosk mode.
   std::string chrome_client_id_;
   std::string chrome_client_secret_;
 
+  std::string oauthlogin_access_token_;

[xiyuan, 2015/04/20 22:02:41]

nit: Consider add this to UserContext.

[achuithb, 2015/04/21 07:10:21]

Done.

+  std::string refresh_token_;

[xiyuan, 2015/04/20 22:02:41]

nit: Use refresh_token field in user_context_.

[achuithb, 2015/04/21 07:10:21]

Done.

+  FetchOAuth2TokensCallback login_callback_;
+
   // Per-user-session Input Methods states.
   std::map<Profile*, scoped_refptr<input_method::InputMethodManager::State>,
       ProfileCompare> default_ime_states_;
 
   // Manages Easy unlock cryptohome keys.
   scoped_ptr<EasyUnlockKeyManager> easy_unlock_key_manager_;
   bool running_easy_unlock_key_ops_;
   base::Closure easy_unlock_key_ops_finished_callback_;
 
   // Whether should launch browser, tests may override this value.
   bool should_launch_browser_;
 
   // Child account status is necessary for InitializeStartUrls call.
   bool waiting_for_child_account_status_;
 
+  scoped_ptr<OAuth2TokenFetcher> oauth2_token_fetcher_;
+
   base::WeakPtrFactory<UserSessionManager> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(UserSessionManager);
 };
 
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_CHROMEOS_LOGIN_SESSION_USER_SESSION_MANAGER_H_