ServiceWorker: Support navigator.serviceWorker.active

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/message_port_message_filter.h"
 #include "content/browser/service_worker/service_worker_context_core.h"
 #include "content/browser/service_worker/service_worker_context_request_handler.h"
@@ skipping 12 matching lines @@
     int process_id, int provider_id,
     base::WeakPtr<ServiceWorkerContextCore> context,
     ServiceWorkerDispatcherHost* dispatcher_host)
     : process_id_(process_id),
       provider_id_(provider_id),
       context_(context),
       dispatcher_host_(dispatcher_host) {
 }
 
 ServiceWorkerProviderHost::~ServiceWorkerProviderHost() {
+  if (controlling_version_)
+    controlling_version_->RemoveControllee(this);
   if (active_version_)
-    active_version_->RemoveControllee(this);
+    active_version_->RemovePotentialControllee(this);
   if (waiting_version_)
     waiting_version_->RemovePotentialControllee(this);
   if (installing_version_)
     installing_version_->RemovePotentialControllee(this);
 }
 
 void ServiceWorkerProviderHost::SetDocumentUrl(const GURL& url) {
   DCHECK(!url.has_ref());
   document_url_ = url;
 }
 
+void ServiceWorkerProviderHost::SetControllerVersion(
+    ServiceWorkerVersion* version) {
+  DCHECK(CanAssociateVersion(version));
+  if (version == controlling_version_)
+    return;
+  scoped_refptr<ServiceWorkerVersion> previous_version = controlling_version_;
+  controlling_version_ = version;
+  if (version)
+    version->AddControllee(this);
+  if (previous_version)
+    previous_version->RemoveControllee(this);
+
+  if (!dispatcher_host_)
+    return;  // Could be NULL in some tests.
+
+  dispatcher_host_->Send(new ServiceWorkerMsg_SetControllerServiceWorker(
+      kDocumentMainThreadId, provider_id(), CreateHandleAndPass(version)));
+}
+
 void ServiceWorkerProviderHost::SetActiveVersion(
     ServiceWorkerVersion* version) {
   DCHECK(CanAssociateVersion(version));
   if (version == active_version_)
     return;
   scoped_refptr<ServiceWorkerVersion> previous_version = active_version_;
   active_version_ = version;
   if (version)
-    version->AddControllee(this);
+    version->AddPotentialControllee(this);
   if (previous_version)
-    previous_version->RemoveControllee(this);
+    previous_version->RemovePotentialControllee(this);
 
   if (!dispatcher_host_)
     return;  // Could be NULL in some tests.
 
-  dispatcher_host_->Send(new ServiceWorkerMsg_SetControllerServiceWorker(
+  dispatcher_host_->Send(new ServiceWorkerMsg_SetActiveServiceWorker(
       kDocumentMainThreadId, provider_id(), CreateHandleAndPass(version)));
 }
 
 void ServiceWorkerProviderHost::SetWaitingVersion(
     ServiceWorkerVersion* version) {
   DCHECK(CanAssociateVersion(version));
   if (version == waiting_version_)
     return;
   scoped_refptr<ServiceWorkerVersion> previous_version = waiting_version_;
   waiting_version_ = version;
@@ skipping 30 matching lines @@
 
 void ServiceWorkerProviderHost::UnsetVersion(ServiceWorkerVersion* version) {
   if (!version)
     return;
   if (installing_version_ == version)
     SetInstallingVersion(NULL);
   else if (waiting_version_  == version)
     SetWaitingVersion(NULL);
   else if (active_version_ == version)
     SetActiveVersion(NULL);
+  else if (controlling_version_ == version)
+    SetControllerVersion(NULL);
 }
 
 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 32 matching lines @@
 bool ServiceWorkerProviderHost::CanAssociateVersion(
     ServiceWorkerVersion* version) {
   if (!context_)
     return false;
   if (running_hosted_version_)
     return false;
   if (!version)
     return true;
 
   ServiceWorkerVersion* already_associated_version = NULL;
+  if (controlling_version_)
+    already_associated_version = controlling_version_;
   if (active_version_)
     already_associated_version = active_version_;
   else if (waiting_version_)
     already_associated_version = waiting_version_;
   else if (installing_version_)
     already_associated_version = installing_version_;
 
   return !already_associated_version ||
          already_associated_version->registration_id() ==
               version->registration_id();
@@ skipping 29 matching lines @@
     dispatcher_host_->RegisterServiceWorkerHandle(handle.Pass());
   }
   return info;
 }
 
 bool ServiceWorkerProviderHost::IsContextAlive() {
   return context_ != NULL;
 }
 
 }  // namespace content