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..2068e1bfc4dfa7a9df0e2a2f72b7bdb4d9cf868c 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* management_service);

[bartfab (slow), 2014/08/07 14:32:35]

Nit: I suggested renaming |consumer_management| to |management_service| inside
the |ConsumerManagementHandler| class, which inherently deals with consumer
management already and where consumer managemen" is implied. This class here
does not generally deal with consumer management, so the meaning of
|management_service| os rather ambiguous. I think |consumer_management| would be
the better name here.

[davidyu, 2014/08/08 03:56:24]

Done.

   virtual ~GaiaScreenHandler();
 
   void LoadGaia(const GaiaContext& context);
@@ -95,6 +102,15 @@ class GaiaScreenHandler : public BaseScreenHandler {
 
   void HandleGaiaUIReady();
 
+  void OnSetOwnerDone(const std::string& typed_email,
+                      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* management_service_;
+
   // Email to pre-populate with.
   std::string populated_email_;