Service Worker: in Unregister, wait until after the active worker no longer controls a document

content/browser/service_worker/service_worker_registration.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_registration.h"
 
 #include "content/browser/service_worker/service_worker_context_core.h"
 #include "content/browser/service_worker/service_worker_info.h"
 #include "content/browser/service_worker/service_worker_register_job.h"
 #include "content/browser/service_worker/service_worker_utils.h"
@@ skipping 12 matching lines @@
 }  // namespace
 
 ServiceWorkerRegistration::ServiceWorkerRegistration(
     const GURL& pattern,
     const GURL& script_url,
     int64 registration_id,
     base::WeakPtr<ServiceWorkerContextCore> context)
     : pattern_(pattern),
       script_url_(script_url),
       registration_id_(registration_id),
+      is_uninstalling_(false),
       context_(context) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   DCHECK(context_);
   context_->AddLiveRegistration(this);
 }
 
 ServiceWorkerRegistration::~ServiceWorkerRegistration() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   if (context_)
     context_->RemoveLiveRegistration(registration_id_);
@@ skipping 41 matching lines @@
     return;
   ChangedVersionAttributesMask mask;
   UnsetVersionInternal(version, &mask);
   if (mask.changed()) {
     ServiceWorkerRegistrationInfo info = GetInfo();
     FOR_EACH_OBSERVER(Listener, listeners_,
                       OnVersionAttributesChanged(this, mask, info));
   }
 }
 
+void ServiceWorkerRegistration::OnNoControllees(ServiceWorkerVersion* version) {
+  DCHECK_EQ(active_version(), version);
+  version->RemoveListener(this);
+  if (is_uninstalling())
+    Uninstall();
+}
+
 void ServiceWorkerRegistration::SetVersionInternal(
     ServiceWorkerVersion* version,
     scoped_refptr<ServiceWorkerVersion>* data_member,
     int change_flag) {
   if (version == data_member->get())
     return;
   scoped_refptr<ServiceWorkerVersion> protect(version);
   ChangedVersionAttributesMask mask;
   if (version)
     UnsetVersionInternal(version, &mask);
@@ skipping 53 matching lines @@
   activating_version->SetStatus(ServiceWorkerVersion::ACTIVATING);
 
   // TODO(nhiroki): "8. Fire a simple event named controllerchange..."
 
   // "9. Queue a task to fire an event named activate..."
   activating_version->DispatchActivateEvent(
       base::Bind(&ServiceWorkerRegistration::OnActivateEventFinished,
                  this, activating_version, completion_callback));
 }
 
+void ServiceWorkerRegistration::StartUninstall() {

[michaeln, 2014/07/25 00:02:04]

let's harmonize on naming, i'm using

ActivateWaitingVersionWhenReady()
ActivateWaitingVersion()

sticking with that theme, something like

UninstallActiveVersionWhenReady()
UninstallActiveVersion()

[falken, 2014/07/25 12:47:25]

Sounds good.

+  DCHECK(context_);
+  set_uninstalling(true);
+  // "11. Wait until no document is using registration as their Service Worker
+  // registration."
+  if (active_version() && active_version()->HasControllee())
+    active_version()->AddListener(this);
+  else
+    Uninstall();
+}
+
+void ServiceWorkerRegistration::Uninstall() {
+  DCHECK(is_uninstalling());
+  if (!context_)
+    return;
+
+  // "12. If registration.activeWorker is not null, then:"
+  if (active_version()) {
+    // "1. Run the [[UpdateState]] algorithm passing registration.activeWorker
+    // and 'redundant' as the arguments."
+    active_version()->Doom();
+    // "2. Set registration.activeWorker to null."
+    UnsetVersion(active_version());
+  }
+
+  // "13. Delete a Record {[[key]], [[value]]} entry of its
+  // [[ScopeToRegistrationMap]] where scope matches entry.[[key]]."
+  context_->storage()->DeleteRegistration(
+      id(),
+      script_url().GetOrigin(),
+      base::Bind(&ServiceWorkerUtils::NoOpStatusCallback));
+}
+
 void ServiceWorkerRegistration::OnActivateEventFinished(
     ServiceWorkerVersion* activating_version,
     const StatusCallback& completion_callback,
     ServiceWorkerStatusCode status) {
   if (activating_version != active_version())
     return;
   // TODO(kinuko,falken): For some error cases (e.g. ServiceWorker is
   // unexpectedly terminated) we may want to retry sending the event again.
   if (status != SERVICE_WORKER_OK) {
     // "11. If activateFailed is true, then:..."
@@ skipping 15 matching lines @@
   activating_version->SetStatus(ServiceWorkerVersion::ACTIVATED);
   if (context_) {
     context_->storage()->UpdateToActiveState(
         this,
         base::Bind(&ServiceWorkerUtils::NoOpStatusCallback));
   }
   completion_callback.Run(SERVICE_WORKER_OK);
 }
 
 }  // namespace content