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

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

Index: chrome/browser/ui/webui/chromeos/login/gaia_screen_handler.h
diff --git a/chrome/browser/ui/webui/chromeos/login/gaia_screen_handler.h b/chrome/browser/ui/webui/chromeos/login/gaia_screen_handler.h
index ac9f391b93e7e461b1ae20d605ff3142d0f9ddc1..c756bf0965963d0e7fef888f7920db352a5c63e7 100644
--- a/chrome/browser/ui/webui/chromeos/login/gaia_screen_handler.h
+++ b/chrome/browser/ui/webui/chromeos/login/gaia_screen_handler.h
@@ -5,12 +5,18 @@
 #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;
@@ -54,8 +60,9 @@ class GaiaScreenHandler : public BaseScreenHandler {
     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);
@@ -95,6 +102,15 @@ class GaiaScreenHandler : public BaseScreenHandler {
 
   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);
 
@@ -153,6 +169,9 @@ class GaiaScreenHandler : public BaseScreenHandler {
   // 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_;