Add device policy to disallow shutdown - ash UI modifications

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

 // Copyright (c) 2012 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.
 
 #include "chrome/browser/ui/webui/chromeos/login/oobe_ui.h"
 
+#include "base/bind.h"
+#include "base/bind_helpers.h"
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/strings/stringprintf.h"
 #include "base/values.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/browser_process_platform_part.h"
 #include "chrome/browser/chromeos/kiosk_mode/kiosk_mode_settings.h"
 #include "chrome/browser/chromeos/login/enrollment/auto_enrollment_check_screen_actor.h"
 #include "chrome/browser/chromeos/login/enrollment/enrollment_screen_actor.h"
 #include "chrome/browser/chromeos/login/screens/error_screen.h"
 #include "chrome/browser/chromeos/login/wizard_controller.h"
 #include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h"
 #include "chrome/browser/chromeos/policy/consumer_management_service.h"
+#include "chrome/browser/chromeos/settings/cros_settings.h"
 #include "chrome/browser/chromeos/system/input_device_settings.h"
 #include "chrome/browser/extensions/signin/gaia_auth_extension_loader.h"
 #include "chrome/browser/extensions/tab_helper.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/webui/about_ui.h"
 #include "chrome/browser/ui/webui/chromeos/login/app_launch_splash_screen_handler.h"
 #include "chrome/browser/ui/webui/chromeos/login/auto_enrollment_check_screen_handler.h"
 #include "chrome/browser/ui/webui/chromeos/login/base_screen_handler.h"
 #include "chrome/browser/ui/webui/chromeos/login/controller_pairing_screen_handler.h"
 #include "chrome/browser/ui/webui/chromeos/login/device_disabled_screen_handler.h"
@@ skipping 527 matching lines @@
     ready_callbacks_[i].Run();
   ready_callbacks_.clear();
 
   // Notify 'initialize' for synchronously loaded screens.
   for (size_t i = 0; i < handlers_.size(); ++i) {
     if (handlers_[i]->async_assets_load_id().empty())
       handlers_[i]->InitializeBase();
   }
 }
 
+void OobeUI::InitShutdownPolicyObserver() {
+  // Instantiate the ShutdownPolicyObserver.
+  shutdown_policy_observer_.reset(
+      new ShutdownPolicyObserver(CrosSettings::Get(), this));
+
+  // Trigger an initial update.
+  shutdown_policy_observer_->CheckIfRebootOnShutdown(
+      base::Bind(&OobeUI::OnShutdownPolicyChanged, base::Unretained(this)));
+}
+
+void OobeUI::OnShutdownPolicyChanged(bool reboot_on_shutdown) {
+  core_handler_->UpdateShutdownButtonVisibility(!reboot_on_shutdown);
+}
+
 void OobeUI::OnScreenAssetsLoaded(const std::string& async_assets_load_id) {
   DCHECK(!async_assets_load_id.empty());
 
   for (size_t i = 0; i < handlers_.size(); ++i) {
     if (handlers_[i]->async_assets_load_id() == async_assets_load_id)
       handlers_[i]->InitializeBase();
   }
 }
 
 bool OobeUI::IsJSReady(const base::Closure& display_is_ready_callback) {
@@ skipping 45 matching lines @@
   DCHECK(screen_ids_.count(screen))
       << "Screen should be registered in InitializeScreenMaps()";
   Screen new_screen = screen_ids_[screen];
   FOR_EACH_OBSERVER(Observer,
                     observer_list_,
                     OnCurrentScreenChanged(current_screen_, new_screen));
   current_screen_ = new_screen;
 }
 
 }  // namespace chromeos