Introduce ServiceWorkerHandle for tracking WebServiceWorkerImpl reference

content/browser/service_worker/service_worker_provider_host.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 "content/browser/service_worker/service_worker_provider_host.h"
 
 #include "base/stl_util.h"
 #include "content/browser/service_worker/service_worker_context_core.h"
+#include "content/browser/service_worker/service_worker_dispatcher_host.h"
+#include "content/browser/service_worker/service_worker_handle.h"
 #include "content/browser/service_worker/service_worker_utils.h"
 #include "content/browser/service_worker/service_worker_version.h"
 
 namespace content {
 
 ServiceWorkerProviderHost::ServiceWorkerProviderHost(
     int process_id, int provider_id,
-    base::WeakPtr<ServiceWorkerContextCore> context)
+    base::WeakPtr<ServiceWorkerContextCore> context,
+    ServiceWorkerDispatcherHost* dispatcher_host)
     : process_id_(process_id),
       provider_id_(provider_id),
-      context_(context) {
+      context_(context),
+      dispatcher_host_(dispatcher_host) {
 }
 
 ServiceWorkerProviderHost::~ServiceWorkerProviderHost() {
   if (active_version_)
     active_version_->RemoveControllee(this);
 }
 
 void ServiceWorkerProviderHost::AddScriptClient(int thread_id) {
   DCHECK(!ContainsKey(script_client_thread_ids_, thread_id));
   script_client_thread_ids_.insert(thread_id);
 }
 
 void ServiceWorkerProviderHost::RemoveScriptClient(int thread_id) {
   DCHECK(ContainsKey(script_client_thread_ids_, thread_id));
   script_client_thread_ids_.erase(thread_id);
 }
 
 void ServiceWorkerProviderHost::SetActiveVersion(
     ServiceWorkerVersion* version) {
   scoped_refptr<ServiceWorkerVersion> previous_version = active_version_;
   active_version_ = version;
   if (version)
     version->AddControllee(this);
   if (previous_version)
     previous_version->RemoveControllee(this);
+
+  dispatcher_host_->RegisterServiceWorkerHandle(
+      ServiceWorkerHandle::Create(context_, dispatcher_host_, version));
   // TODO(kinuko): dispatch activechange event to the script clients.
+  // dispatcher_host_->Send(ServiceWorkerMsg_ActiveChangeEvent(...));
 }
 
 void ServiceWorkerProviderHost::SetPendingVersion(
     ServiceWorkerVersion* version) {
   pending_version_ = version;
+
+  dispatcher_host_->RegisterServiceWorkerHandle(
+      ServiceWorkerHandle::Create(context_, dispatcher_host_, version));
   // TODO(kinuko): dispatch pendingchange event to the script clients.
+  // dispatcher_host_->Send(ServiceWorkerMsg_PendingChangeEvent(...));
 }
 
 bool ServiceWorkerProviderHost::SetHostedVersionId(int64 version_id) {
   if (!context_)
     return true;  // System is shutting down.
   if (active_version_)
     return false;  // Unexpected bad message.
 
   ServiceWorkerVersion* live_version = context_->GetLiveVersion(version_id);
   if (!live_version)
@@ skipping 20 matching lines @@
     return true;
 
   // TODO(kinuko): Handle ServiceWorker cases.
   // For now we always return false here, so that we don't handle
   // requests for ServiceWorker (either for the main or sub resources).
 
   return false;
 }
 
 }  // namespace content