Add device policy to disallow shutdown - WebUI modifications

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

 // Copyright (c) 2011 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_OOBE_UI_H_
 #define CHROME_BROWSER_UI_WEBUI_CHROMEOS_LOGIN_OOBE_UI_H_
 
 #include <map>
 #include <string>
 #include <vector>
 
 #include "base/compiler_specific.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/observer_list.h"
 #include "chrome/browser/chromeos/login/ui/oobe_display.h"
+#include "chrome/browser/chromeos/settings/shutdown_policy_handler.h"
 #include "chrome/browser/ui/webui/chromeos/login/core_oobe_handler.h"
 #include "content/public/browser/web_ui_controller.h"
 
 namespace base {
 class DictionaryValue;
 }  // namespace base
 
 namespace chromeos {
 class AppLaunchSplashScreenActor;
 class BaseScreenHandler;
@@ skipping 13 matching lines @@
 class SigninScreenHandlerDelegate;
 class UpdateScreenHandler;
 class UserBoardScreenHandler;
 
 // A custom WebUI that defines datasource for out-of-box-experience (OOBE) UI:
 // - welcome screen (setup language/keyboard/network).
 // - eula screen (CrOS (+ OEM) EULA content/TPM password/crash reporting).
 // - update screen.
 class OobeUI : public OobeDisplay,
                public content::WebUIController,
-               public CoreOobeHandler::Delegate {
+               public CoreOobeHandler::Delegate,
+               public ShutdownPolicyHandler::Delegate {
  public:
   class Observer {
    public:
     virtual ~Observer() {}
     virtual void OnCurrentScreenChanged(
         Screen current_screen, Screen new_screen) = 0;
   };
 
   // List of known types of OobeUI. Type added as path in chrome://oobe url, for
   // example chrome://oobe/user-adding.
@@ skipping 54 matching lines @@
       override;
   AppLaunchSplashScreenActor* GetAppLaunchSplashScreenActor() override;
   bool IsJSReady(const base::Closure& display_is_ready_callback) override;
   HIDDetectionScreenActor* GetHIDDetectionScreenActor() override;
   ControllerPairingScreenActor* GetControllerPairingScreenActor() override;
   HostPairingScreenActor* GetHostPairingScreenActor() override;
   DeviceDisabledScreenActor* GetDeviceDisabledScreenActor() override;
   GaiaScreenHandler* GetGaiaScreenActor() override;
   UserBoardView* GetUserBoardScreenActor() override;
 
+  // ShutdownPolicyObserver::Delegate
+  void OnShutdownPolicyChanged(bool reboot_on_shutdown) override;
+
   // Collects localized strings from the owned handlers.
   void GetLocalizedStrings(base::DictionaryValue* localized_strings);
 
   // Initializes the handlers.
   void InitializeHandlers();
 
   // Invoked after the async assets load. The screen handler that has the same
   // async assets load id will be initialized.
   void OnScreenAssetsLoaded(const std::string& async_assets_load_id);
 
@@ skipping 106 matching lines @@
   // Flag that indicates whether JS part is fully loaded and ready to accept
   // calls.
   bool ready_;
 
   // Callbacks to notify when JS part is fully loaded and ready to accept calls.
   std::vector<base::Closure> ready_callbacks_;
 
   // List of registered observers.
   ObserverList<Observer> observer_list_;
 
+  // Observer of CrosSettings watching the kRebootOnShutdown policy.
+  scoped_ptr<ShutdownPolicyHandler> shutdown_policy_handler_;
+
   DISALLOW_COPY_AND_ASSIGN(OobeUI);
 };
 
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_UI_WEBUI_CHROMEOS_LOGIN_OOBE_UI_H_