Add a enterprise enrollment check screen to OOBE in ChromeOS.

chrome/browser/chromeos/login/enrollment/auto_enrollment_check_screen.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_ENROLLMENT_AUTO_ENROLLMENT_CHECK_STEP_H_
-#define CHROME_BROWSER_CHROMEOS_LOGIN_ENROLLMENT_AUTO_ENROLLMENT_CHECK_STEP_H_
+#ifndef CHROME_BROWSER_CHROMEOS_LOGIN_ENROLLMENT_AUTO_ENROLLMENT_CHECK_SCREEN_H_
+#define CHROME_BROWSER_CHROMEOS_LOGIN_ENROLLMENT_AUTO_ENROLLMENT_CHECK_SCREEN_H_
 
 #include "base/compiler_specific.h"
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
+#include "chrome/browser/chromeos/login/enrollment/auto_enrollment_check_screen_actor.h"
 #include "chrome/browser/chromeos/login/enrollment/auto_enrollment_controller.h"
 #include "chrome/browser/chromeos/login/screens/error_screen.h"
 #include "chrome/browser/chromeos/net/network_portal_detector.h"
 
 namespace chromeos {
 
 class ScreenObserver;
 
 // Handles the control flow after OOBE auto-update completes to wait for the
 // enterprise auto-enrollment check that happens as part of OOBE. This includes
 // keeping track of current auto-enrollment state and displaying and updating
 // the error screen upon failures. Similar to a screen controller, but it
 // doesn't actually drive a dedicated screen.
-class AutoEnrollmentCheckStep : NetworkPortalDetector::Observer {
+class AutoEnrollmentCheckScreen
+    : public AutoEnrollmentCheckScreenActor::Delegate,
+      public WizardScreen,
+      public NetworkPortalDetector::Observer {
  public:
-  AutoEnrollmentCheckStep(ScreenObserver* screen_observer,
-                          AutoEnrollmentController* auto_enrollment_controller);
-  virtual ~AutoEnrollmentCheckStep();
+  AutoEnrollmentCheckScreen(
+      ScreenObserver* observer,
+      AutoEnrollmentCheckScreenActor* actor);
+  virtual ~AutoEnrollmentCheckScreen();
 
   // Hands over OOBE control to this AutoEnrollmentCheckStep. It'll return the
   // flow back to the caller via the |screen_observer_|'s OnExit function.
   void Start();
 
+  void set_auto_enrollment_controller(
+      AutoEnrollmentController* auto_enrollment_controller) {
+    auto_enrollment_controller_ = auto_enrollment_controller;
+  }
+
+  // WizardScreen implementation:
+  virtual void PrepareToShow() OVERRIDE;
+  virtual void Show() OVERRIDE;
+  virtual void Hide() OVERRIDE;
+  virtual std::string GetName() const OVERRIDE;
+
+  // AutoEnrollmentCheckScreenActor::Delegate implementation:
+  virtual void OnExit() OVERRIDE;
+  virtual void OnActorDestroyed(AutoEnrollmentCheckScreenActor* actor) OVERRIDE;
+
   // NetworkPortalDetector::Observer implementation:
   virtual void OnPortalDetectionCompleted(
       const NetworkState* network,
       const NetworkPortalDetector::CaptivePortalState& state) OVERRIDE;
 
  private:
   // Handles update notifications regarding the auto-enrollment check.
   void OnAutoEnrollmentCheckProgressed(policy::AutoEnrollmentState state);
 
   // Handles a state update, updating the UI and saving the state.
@@ skipping 11 matching lines @@
   bool UpdateAutoEnrollmentState(
       policy::AutoEnrollmentState auto_enrollment_state);
 
   // Configures the error screen.
   void ShowErrorScreen(ErrorScreen::ErrorState error_state);
 
   // Signals completion. No further code should run after a call to this
   // function as the owner might destroy |this| in response.
   void SignalCompletion();
 
-  ScreenObserver* screen_observer_;
+  // Checks if the enrollment status check is needed. It can be disabled either
+  // by command line flags, build configuration or might have finished already.
+  bool IsStartNeeded();
+
+  AutoEnrollmentCheckScreenActor* actor_;
   AutoEnrollmentController* auto_enrollment_controller_;
 
   scoped_ptr<AutoEnrollmentController::ProgressCallbackList::Subscription>
       auto_enrollment_progress_subscription_;
 
   NetworkPortalDetector::CaptivePortalStatus captive_portal_status_;
   policy::AutoEnrollmentState auto_enrollment_state_;
 
-  DISALLOW_COPY_AND_ASSIGN(AutoEnrollmentCheckStep);
+  DISALLOW_COPY_AND_ASSIGN(AutoEnrollmentCheckScreen);
 };
 
 }  // namespace chromeos
 
-#endif  // CHROME_BROWSER_CHROMEOS_LOGIN_ENROLLMENT_AUTO_ENROLLMENT_CHECK_STEP_H_
+#endif  // CHROME_BROWSER_CHROMEOS_LOGIN_ENROLLMENT_AUTO_ENROLLMENT_CHECK_SCREEN_H_