Added ConsumerManagementService class to handle enroll state and device owner info in boot lockbox.

chrome/browser/ui/webui/chromeos/login/gaia_screen_handler.h

 // Copyright 2013 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_UI_WEBUI_CHROMEOS_LOGIN_GAIA_SCREEN_HANDLER_H_
 #define CHROME_BROWSER_UI_WEBUI_CHROMEOS_LOGIN_GAIA_SCREEN_HANDLER_H_
 
+#include <string>
+
 #include "base/basictypes.h"
 #include "base/command_line.h"
 #include "base/memory/ref_counted.h"
 #include "chrome/browser/ui/webui/chromeos/login/network_state_informer.h"
 #include "net/base/net_errors.h"
 
+namespace policy {
+class ConsumerManagementService;
+}
+
 namespace chromeos {
 
 class SigninScreenHandler;
 
 // A class that's used to specify the way how Gaia should be loaded.
 struct GaiaContext {
   GaiaContext();
 
   // Forces Gaia to reload.
   bool force_reload;
@@ skipping 23 matching lines @@
 // A class that handles WebUI hooks in Gaia screen.
 class GaiaScreenHandler : public BaseScreenHandler {
  public:
   enum FrameState {
     FRAME_STATE_UNKNOWN = 0,
     FRAME_STATE_LOADING,
     FRAME_STATE_LOADED,
     FRAME_STATE_ERROR
   };
 
-  explicit GaiaScreenHandler(
-      const scoped_refptr<NetworkStateInformer>& network_state_informer);
+  GaiaScreenHandler(
+      const scoped_refptr<NetworkStateInformer>& network_state_informer,
+      policy::ConsumerManagementService* consumer_management);
   virtual ~GaiaScreenHandler();
 
   void LoadGaia(const GaiaContext& context);
   void UpdateGaia(const GaiaContext& context);
 
   // Sends request to reload Gaia. If |force_reload| is true, request
   // will be sent in any case, otherwise it will be sent only when Gaia is
   // not loading right now.
   void ReloadGaia(bool force_reload);
 
@@ skipping 19 matching lines @@
   void HandleCompleteLogin(const std::string& typed_email,
                            const std::string& password,
                            bool using_saml);
 
   void HandleUsingSAMLAPI();
   void HandleScrapedPasswordCount(int password_count);
   void HandleScrapedPasswordVerificationFailed();
 
   void HandleGaiaUIReady();
 
+  void OnSetOwnerDone(const std::string& typed_email,

[Nikita (slow), 2014/08/08 08:48:45]

nit: Please add comments for these 2 methods.

[davidyu, 2014/08/08 09:19:09]

Done.

+                      const std::string& password,
+                      bool using_saml,
+                      bool success);
+
+  void DoCompleteLogin(const std::string& typed_email,
+                       const std::string& password,
+                       bool using_saml);
+
   // Fill GAIA user name.
   void PopulateEmail(const std::string& user_id);
 
   // Mark user as having password changed:
   void PasswordChangedFor(const std::string& user_id);
 
   // Kick off cookie / local storage cleanup.
   void StartClearingCookies(const base::Closure& on_clear_callback);
   void OnCookiesCleared(const base::Closure& on_clear_callback);
 
@@ skipping 38 matching lines @@
 
   // Current state of Gaia frame.
   FrameState frame_state_;
 
   // Latest Gaia frame error.
   net::Error frame_error_;
 
   // Network state informer used to keep signin screen up.
   scoped_refptr<NetworkStateInformer> network_state_informer_;
 
+  // Consumer management service for checking if enrollment is in progress.
+  policy::ConsumerManagementService* consumer_management_;
+
   // Email to pre-populate with.
   std::string populated_email_;
 
   // Emails of the users, whose passwords have recently been changed.
   std::set<std::string> password_changed_for_;
 
   // True if dns cache cleanup is done.
   bool dns_cleared_;
 
   // True if DNS cache task is already running.
@@ skipping 30 matching lines @@
   SigninScreenHandler* signin_screen_handler_;
 
   base::WeakPtrFactory<GaiaScreenHandler> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(GaiaScreenHandler);
 };
 
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_UI_WEBUI_CHROMEOS_LOGIN_GAIA_SCREEN_HANDLER_H_