[Offline Prefetch] Create a new JobScheduler task to wake up for net activity.

chrome/browser/ui/webui/offline/offline_internals_ui_message_handler.cc

Index: chrome/browser/ui/webui/offline/offline_internals_ui_message_handler.cc
diff --git a/chrome/browser/ui/webui/offline/offline_internals_ui_message_handler.cc b/chrome/browser/ui/webui/offline/offline_internals_ui_message_handler.cc
index f67e90377477922b8229b26f00ba013aed3323d6..d601fd2ff5495bbcc94a2ad65ee8bf7738d2e751 100644
--- a/chrome/browser/ui/webui/offline/offline_internals_ui_message_handler.cc
+++ b/chrome/browser/ui/webui/offline/offline_internals_ui_message_handler.cc
@@ -15,6 +15,7 @@
 #include "base/strings/string_number_conversions.h"
 #include "base/values.h"
 #include "chrome/browser/android/offline_pages/offline_page_model_factory.h"
+#include "chrome/browser/android/offline_pages/prefetch/prefetch_background_task.h"
 #include "chrome/browser/android/offline_pages/request_coordinator_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "components/offline_pages/core/client_namespace_constants.h"
@@ -227,6 +228,26 @@ void OfflineInternalsUIMessageHandler::HandleGetNetworkStatus(
                                                           : "Online"));
 }
 
+void OfflineInternalsUIMessageHandler::HandleScheduleNwake(
+    const base::ListValue* args) {
+  const base::Value* callback_id;
+  CHECK(args->Get(0, &callback_id));
+
+  offline_pages::PrefetchBackgroundTask::Schedule();
+
+  ResolveJavascriptCallback(*callback_id, base::Value("Scheduled."));
+}
+
+void OfflineInternalsUIMessageHandler::HandleCancelNwake(
+    const base::ListValue* args) {
+  const base::Value* callback_id;
+  CHECK(args->Get(0, &callback_id));
+
+  offline_pages::PrefetchBackgroundTask::Cancel();
+
+  ResolveJavascriptCallback(*callback_id, base::Value("Cancelled."));
+}
+
 void OfflineInternalsUIMessageHandler::HandleSetRecordRequestQueue(
     const base::ListValue* args) {
   bool should_record;
@@ -340,6 +361,15 @@ void OfflineInternalsUIMessageHandler::RegisterMessages() {
       "getNetworkStatus",
       base::Bind(&OfflineInternalsUIMessageHandler::HandleGetNetworkStatus,
                  weak_ptr_factory_.GetWeakPtr()));
+  web_ui()->RegisterMessageCallback(
+      "scheduleNwake",
+      base::Bind(&OfflineInternalsUIMessageHandler::HandleScheduleNwake,
+                 weak_ptr_factory_.GetWeakPtr()));
+
+  web_ui()->RegisterMessageCallback(
+      "cancelNwake",
+      base::Bind(&OfflineInternalsUIMessageHandler::HandleCancelNwake,
+                 weak_ptr_factory_.GetWeakPtr()));
 
   // Get the offline page model associated with this web ui.
   Profile* profile = Profile::FromWebUI(web_ui());