[DevTools] Introduce Unregister/Start/Stop/Inspect buttons to ServiceWorkersView [1/2 chromium]

content/browser/devtools/protocol/service_worker_handler.cc

 // Copyright 2015 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 "content/browser/devtools/protocol/service_worker_handler.h"
 
 #include "base/bind.h"
 #include "base/containers/scoped_ptr_hash_map.h"
 #include "base/strings/string_number_conversions.h"
 #include "content/browser/devtools/service_worker_devtools_agent_host.h"
 #include "content/browser/devtools/service_worker_devtools_manager.h"
 #include "content/browser/frame_host/frame_tree.h"
 #include "content/browser/frame_host/frame_tree_node.h"
 #include "content/browser/frame_host/render_frame_host_impl.h"
+#include "content/browser/service_worker/service_worker_context_core.h"
 #include "content/browser/service_worker/service_worker_context_watcher.h"
 #include "content/browser/service_worker/service_worker_context_wrapper.h"
 #include "content/browser/service_worker/service_worker_version.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/devtools_agent_host.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/storage_partition.h"
 #include "url/gurl.h"
 
 // Windows headers will redefine SendMessage.
 #ifdef SendMessage
 #undef SendMessage
 #endif
 
 namespace content {
 namespace devtools {
 namespace service_worker {
 
+using Response = DevToolsProtocolClient::Response;
+
 namespace {
 
-const char kServiceWorkerVersionRunningStatusStopped[] = "stopped";
-const char kServiceWorkerVersionRunningStatusStarting[] = "starting";
-const char kServiceWorkerVersionRunningStatusRunning[] = "running";
-const char kServiceWorkerVersionRunningStatusStopping[] = "stopping";
-
-const char kServiceWorkerVersionStatusNew[] = "new";
-const char kServiceWorkerVersionStatusInstalling[] = "installing";
-const char kServiceWorkerVersionStatusInstalled[] = "installed";
-const char kServiceWorkerVersionStatusActivating[] = "activating";
-const char kServiceWorkerVersionStatusActivated[] = "activated";
-const char kServiceWorkerVersionStatusRedundant[] = "redundant";
+void ResultNoOp(bool success) {
+}
+void StatusNoOp(ServiceWorkerStatusCode status) {
+}
 
 const std::string GetVersionRunningStatusString(
     content::ServiceWorkerVersion::RunningStatus running_status) {
   switch (running_status) {
     case content::ServiceWorkerVersion::STOPPED:
       return kServiceWorkerVersionRunningStatusStopped;
     case content::ServiceWorkerVersion::STARTING:
       return kServiceWorkerVersionRunningStatusStarting;
     case content::ServiceWorkerVersion::RUNNING:
       return kServiceWorkerVersionRunningStatusRunning;
@@ skipping 80 matching lines @@
       result[host->GetId()] = host;
   }
   return result;
 }
 
 bool CollectURLs(std::set<GURL>* urls, FrameTreeNode* tree_node) {
   urls->insert(tree_node->current_url());
   return false;
 }
 
+void StopServiceWorkerOnIO(scoped_refptr<ServiceWorkerContextWrapper> context,
+                           int64 version_id) {
+  ServiceWorkerContextCore* context_core = context->context();
+  if (!context_core)
+    return;
+  if (content::ServiceWorkerVersion* version =
+          context_core->GetLiveVersion(version_id)) {
+    version->StopWorker(base::Bind(&StatusNoOp));
+  }
+}
+
+void GetDevToolsRouteInfoOnIO(
+    scoped_refptr<ServiceWorkerContextWrapper> context,
+    int64 version_id,
+    const base::Callback<void(int, int)>& callback) {
+  ServiceWorkerContextCore* context_core = context->context();
+  if (!context_core)
+    return;
+  if (content::ServiceWorkerVersion* version =
+          context_core->GetLiveVersion(version_id)) {
+    BrowserThread::PostTask(
+        BrowserThread::UI, FROM_HERE,
+        base::Bind(
+            callback, version->embedded_worker()->process_id(),
+            version->embedded_worker()->worker_devtools_agent_route_id()));
+  }
+}
+
+Response CreateContextErrorResoponse() {
+  return Response::InternalError("Could not connect to the context");
+}
+
+Response CreateInvalidVersionIdErrorResoponse() {
+  return Response::InternalError("Invalid version ID");
+}
+
 }  // namespace
 
-using Response = DevToolsProtocolClient::Response;
-
 ServiceWorkerHandler::ServiceWorkerHandler()
     : enabled_(false), weak_factory_(this) {
 }
 
 ServiceWorkerHandler::~ServiceWorkerHandler() {
   Disable();
 }
 
 void ServiceWorkerHandler::SetRenderFrameHost(
     RenderFrameHostImpl* render_frame_host) {
@@ skipping 91 matching lines @@
 
 Response ServiceWorkerHandler::Stop(
     const std::string& worker_id) {
   auto it = attached_hosts_.find(worker_id);
   if (it == attached_hosts_.end())
     return Response::InternalError("Not connected to the worker");
   it->second->UnregisterWorker();
   return Response::OK();
 }
 
+Response ServiceWorkerHandler::Unregister(const std::string& scope_url) {
+  if (!enabled_)
+    return Response::OK();
+  if (!context_)
+    return CreateContextErrorResoponse();
+  context_->UnregisterServiceWorker(GURL(scope_url), base::Bind(&ResultNoOp));
+  return Response::OK();
+}
+
+Response ServiceWorkerHandler::StartWorker(const std::string& scope_url) {
+  if (!enabled_)
+    return Response::OK();
+  if (!context_)
+    return CreateContextErrorResoponse();
+  context_->StartServiceWorker(GURL(scope_url), base::Bind(&StatusNoOp));
+  return Response::OK();
+}
+
+Response ServiceWorkerHandler::StopWorker(const std::string& version_id) {
+  if (!enabled_)
+    return Response::OK();
+  if (!context_)
+    return CreateContextErrorResoponse();
+  int64 id = 0;
+  if (!base::StringToInt64(version_id, &id))
+    return CreateInvalidVersionIdErrorResoponse();
+  BrowserThread::PostTask(BrowserThread::IO, FROM_HERE,
+                          base::Bind(&StopServiceWorkerOnIO, context_, id));
+  return Response::OK();
+}
+
+Response ServiceWorkerHandler::InspectWorker(const std::string& version_id) {
+  if (!enabled_)
+    return Response::OK();
+  if (!context_)
+    return CreateContextErrorResoponse();
+
+  int64 id = 0;
+  if (!base::StringToInt64(version_id, &id))
+    return CreateInvalidVersionIdErrorResoponse();
+  BrowserThread::PostTask(
+      BrowserThread::IO, FROM_HERE,
+      base::Bind(&GetDevToolsRouteInfoOnIO, context_, id,
+                 base::Bind(&ServiceWorkerHandler::OpenNewDevToolsWindow,
+                            weak_factory_.GetWeakPtr())));
+  return Response::OK();
+}
+
+void ServiceWorkerHandler::OpenNewDevToolsWindow(int process_id,
+                                                 int devtools_agent_route_id) {
+  scoped_refptr<DevToolsAgentHostImpl> agent_host(
+      ServiceWorkerDevToolsManager::GetInstance()
+          ->GetDevToolsAgentHostForWorker(process_id, devtools_agent_route_id));
+  if (!agent_host.get())
+    return;
+  agent_host->Inspect(render_frame_host_->GetProcess()->GetBrowserContext());
+}
+
 void ServiceWorkerHandler::OnWorkerRegistrationUpdated(
     const std::vector<ServiceWorkerRegistrationInfo>& registrations) {
   std::vector<scoped_refptr<ServiceWorkerRegistration>> registration_values;
   for (const auto& registration : registrations) {
     registration_values.push_back(
         CreateRegistrationDictionaryValue(registration));
   }
   client_->WorkerRegistrationUpdated(
       WorkerRegistrationUpdatedParams::Create()->set_registrations(
           registration_values));
@@ skipping 66 matching lines @@
     return;
   host->DetachClient();
   client_->WorkerTerminated(WorkerTerminatedParams::Create()->
       set_worker_id(host->GetId()));
   attached_hosts_.erase(it);
 }
 
 }  // namespace service_worker
 }  // namespace devtools
 }  // namespace content