Add a new app API to enable watchdog behavior restarts in kiosk apps

chrome/browser/extensions/api/runtime/chrome_runtime_api_delegate.cc

 // 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.
 
 #include "chrome/browser/extensions/api/runtime/chrome_runtime_api_delegate.h"
 
 #include <memory>
 #include <string>
 #include <utility>
 #include <vector>
 
 #include "base/lazy_instance.h"
 #include "base/location.h"
 #include "base/metrics/histogram.h"
 #include "base/single_thread_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "base/time/time.h"
 #include "build/build_config.h"
+#include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_tab_util.h"
 #include "chrome/browser/extensions/updater/extension_updater.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser_finder.h"
 #include "chrome/browser/ui/browser_navigator.h"
 #include "chrome/browser/ui/browser_navigator_params.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "components/update_client/update_query_params.h"
 #include "content/public/browser/notification_service.h"
@@ skipping 103 matching lines @@
 };
 
 ChromeRuntimeAPIDelegate::ChromeRuntimeAPIDelegate(
     content::BrowserContext* context)
     : browser_context_(context),
       registered_for_updates_(false),
       extension_registry_observer_(this) {
   registrar_.Add(this,
                  extensions::NOTIFICATION_EXTENSION_UPDATE_FOUND,
                  content::NotificationService::AllSources());
+  registrar_.Add(this,
+                 chrome::NOTIFICATION_APP_TERMINATING,
+                 content::NotificationService::AllSources());
   extension_registry_observer_.Add(
       extensions::ExtensionRegistry::Get(browser_context_));
 }
 
 ChromeRuntimeAPIDelegate::~ChromeRuntimeAPIDelegate() {
 }
 
 // static
 void ChromeRuntimeAPIDelegate::set_tick_clock_for_tests(
     base::TickClock* clock) {
@@ skipping 165 matching lines @@
     chromeos::DBusThreadManager::Get()
         ->GetPowerManagerClient()
         ->RequestRestart();
     return true;
   }
 #endif
   *error_message = "Function available only for ChromeOS kiosk mode.";
   return false;
 }
 
+void ChromeRuntimeAPIDelegate::SetOnDeviceShutdownCallback(
+    const base::Closure& callback) {
+  on_device_shutdown_callback_ = callback;
+}
+
 bool ChromeRuntimeAPIDelegate::OpenOptionsPage(const Extension* extension) {
   Profile* profile = Profile::FromBrowserContext(browser_context_);
   Browser* browser = chrome::FindLastActiveWithProfile(profile);
   if (!browser)
     return false;
   return extensions::ExtensionTabUtil::OpenOptionsPage(extension, browser);
 }
 
 void ChromeRuntimeAPIDelegate::Observe(
     int type,
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
-  DCHECK(type == extensions::NOTIFICATION_EXTENSION_UPDATE_FOUND);
-  typedef const std::pair<std::string, Version> UpdateDetails;
-  const std::string& id = content::Details<UpdateDetails>(details)->first;
-  const Version& version = content::Details<UpdateDetails>(details)->second;
-  if (version.IsValid()) {
-    CallUpdateCallbacks(
-        id, UpdateCheckResult(true, kUpdateFound, version.GetString()));
+  switch (type) {
+    case chrome::NOTIFICATION_APP_TERMINATING:
+      if (!on_device_shutdown_callback_.is_null())
+        on_device_shutdown_callback_.Run();
+      break;
+
+    case extensions::NOTIFICATION_EXTENSION_UPDATE_FOUND: {
+      typedef const std::pair<std::string, Version> UpdateDetails;
+      const std::string& id = content::Details<UpdateDetails>(details)->first;
+      const Version& version = content::Details<UpdateDetails>(details)->second;
+      if (version.IsValid()) {
+        CallUpdateCallbacks(
+            id, UpdateCheckResult(true, kUpdateFound, version.GetString()));
+      }
+      break;
+    }
+
+    default:
+      NOTREACHED() << "Unexpected notification.";
   }
 }
 
 void ChromeRuntimeAPIDelegate::OnExtensionInstalled(
     content::BrowserContext* browser_context,
     const Extension* extension,
     bool is_update) {
   if (!is_update)
     return;
   auto info = update_check_info_.find(extension->id());
@@ skipping 30 matching lines @@
     const UpdateCheckResult& result) {
   auto it = update_check_info_.find(extension_id);
   if (it == update_check_info_.end())
     return;
   std::vector<UpdateCheckCallback> callbacks;
   it->second.callbacks.swap(callbacks);
   for (const auto& callback : callbacks) {
     callback.Run(result);
   }
 }