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() {
   AssociateVersion(NULL);
 }
 
 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::AssociateVersion(
     ServiceWorkerVersion* version) {
   if (associated_version())
-    associated_version_->RemoveProcessToWorker(process_id_);
+    associated_version_->RemoveProcessFromWorker(process_id_);
   associated_version_ = version;
   if (version)
     version->AddProcessToWorker(process_id_);
+
+  if (!dispatcher_host_)
+    return;  // Could be NULL in some tests.
+
+  for (std::set<int>::iterator it = script_client_thread_ids_.begin();
+       it != script_client_thread_ids_.end();
+       ++it) {
+    dispatcher_host_->RegisterServiceWorkerHandle(
+        ServiceWorkerHandle::Create(context_, dispatcher_host_, *it, version));
+  }
+  // TODO(kinuko): change this method into two for .active and .pending,
+  // and dispatch activechange/pendingchange event to the script clients here.
 }
 
 bool ServiceWorkerProviderHost::SetHostedVersionId(int64 version_id) {
   if (!context_)
     return true;  // System is shutting down.
   if (associated_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