app_mode: Add runtime.onRestartRequired event.

chrome/browser/chromeos/app_mode/kiosk_app_update_service.cc

 // Copyright 2013 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/chromeos/app_mode/kiosk_app_update_service.h"
 
 #include "base/logging.h"
 #include "chrome/browser/app_mode/app_mode_utils.h"
+#include "chrome/browser/chromeos/system/automatic_reboot_manager.h"
+#include "chrome/browser/extensions/api/runtime/runtime_api.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_system.h"
 #include "chrome/browser/extensions/extension_system_factory.h"
 #include "chrome/browser/lifetime/application_lifetime.h"
 #include "chrome/browser/profiles/profile.h"
 #include "components/browser_context_keyed_service/browser_context_dependency_manager.h"
 
 namespace chromeos {
 
 namespace {
 
 // How low to wait after an update is available before we force a restart.
 const int kForceRestartWaitTimeMs = 24 * 3600 * 1000;  // 24 hours.
 
 }  // namespace
 
 KioskAppUpdateService::KioskAppUpdateService(Profile* profile)
     : profile_(profile) {
   ExtensionService* service =
       extensions::ExtensionSystem::Get(profile_)->extension_service();
   if (service)
     service->AddUpdateObserver(this);
+
+  system::AutomaticRebootManager::Get()->AddObserver(this);
 }
 
 KioskAppUpdateService::~KioskAppUpdateService() {
 }
 
-void KioskAppUpdateService::Shutdown() {
-  ExtensionService* service = profile_->GetExtensionService();
-  if (service)
-    service->RemoveUpdateObserver(this);
-}
-
-void KioskAppUpdateService::OnAppUpdateAvailable(const std::string& app_id) {
-  DCHECK(!app_id_.empty());
-  if (app_id != app_id_)
-    return;
-
-  StartRestartTimer();
-}
-
 void KioskAppUpdateService::StartRestartTimer() {
   if (restart_timer_.IsRunning())
     return;
 
   // Setup timer to force restart once the wait period expires.
   restart_timer_.Start(
       FROM_HERE, base::TimeDelta::FromMilliseconds(kForceRestartWaitTimeMs),
       this, &KioskAppUpdateService::ForceRestart);
 }
 
 void KioskAppUpdateService::ForceRestart() {
   // Force a chrome restart (not a logout or reboot) by closing all browsers.
   LOG(WARNING) << "Force closing all browsers to update kiosk app.";
   chrome::CloseAllBrowsers();
 }
 
+void KioskAppUpdateService::Shutdown() {
+  ExtensionService* service = profile_->GetExtensionService();
+  if (service)
+    service->RemoveUpdateObserver(this);
+
+  system::AutomaticRebootManager::Get()->RemoveObserver(this);
+}
+
+void KioskAppUpdateService::OnAppUpdateAvailable(const std::string& app_id) {
+  DCHECK(!app_id_.empty());
+  if (app_id != app_id_)
+    return;
+
+  extensions::RuntimeEventRouter::DispatchOnRestartRequiredEvent(
+      profile_,
+      app_id_,
+      extensions::RuntimeEventRouter::RESTART_REASON_APP_UPDATE);
+
+  StartRestartTimer();
+}
+
+void KioskAppUpdateService::OnRebootScheduled(Reason reason) {
+  extensions::RuntimeEventRouter::RestartReason restart_reason;
+  switch (reason) {
+    case REBOOT_REASON_OS_UPDATE:
+      restart_reason = extensions::RuntimeEventRouter::RESTART_REASON_OS_UPDATE;
+      break;
+    case REBOOT_REASON_PERIODIC:
+      restart_reason = extensions::RuntimeEventRouter::RESTART_REASON_PERIODIC;
+      break;
+    default:
+      NOTREACHED() << "Unknown reboot reason=" << reason;
+      return;
+  }
+
+  extensions::RuntimeEventRouter::DispatchOnRestartRequiredEvent(
+      profile_, app_id_, restart_reason);
+}
+
 KioskAppUpdateServiceFactory::KioskAppUpdateServiceFactory()
     : BrowserContextKeyedServiceFactory(
         "KioskAppUpdateService",
         BrowserContextDependencyManager::GetInstance()) {
   DependsOn(extensions::ExtensionSystemFactory::GetInstance());
 }
 
 KioskAppUpdateServiceFactory::~KioskAppUpdateServiceFactory() {
 }
 
@@ skipping 14 matching lines @@
   return Singleton<KioskAppUpdateServiceFactory>::get();
 }
 
 BrowserContextKeyedService*
 KioskAppUpdateServiceFactory::BuildServiceInstanceFor(
     content::BrowserContext* profile) const {
   return new KioskAppUpdateService(static_cast<Profile*>(profile));
 }
 
 }  // namespace chromeos