Populate .current when navigator.serviceWorker is accessed

content/child/service_worker/service_worker_provider_context.cc

+// Copyright 2014 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/child/service_worker/service_worker_provider_context.h"
+
+#include "base/bind.h"
+#include "base/message_loop/message_loop_proxy.h"
+#include "base/stl_util.h"
+#include "content/child/child_thread.h"
+#include "content/child/service_worker/service_worker_dispatcher.h"
+#include "content/child/service_worker/service_worker_handle_reference.h"
+#include "content/child/thread_safe_sender.h"
+#include "content/child/worker_task_runner.h"
+#include "content/common/service_worker/service_worker_messages.h"
+
+namespace content {
+
+ServiceWorkerProviderContext::ServiceWorkerProviderContext(int provider_id)
+    : provider_id_(provider_id),
+      main_thread_loop_proxy_(base::MessageLoopProxy::current()) {
+  if (!ChildThread::current())
+    return;  // May be null in some tests.
+  thread_safe_sender_ = ChildThread::current()->thread_safe_sender();
+  ServiceWorkerDispatcher* dispatcher =
+      ServiceWorkerDispatcher::GetOrCreateThreadSpecificInstance(
+          thread_safe_sender_);
+  DCHECK(dispatcher);
+  dispatcher->AddProviderContext(this);
+}
+
+ServiceWorkerProviderContext::~ServiceWorkerProviderContext() {
+  if (ServiceWorkerDispatcher* dispatcher =
+          ServiceWorkerDispatcher::GetThreadSpecificInstance()) {
+    dispatcher->RemoveProviderContext(this);
+  }
+}
+
+scoped_ptr<ServiceWorkerHandleReference>
+ServiceWorkerProviderContext::GetCurrentServiceWorkerHandle() {
+  DCHECK(main_thread_loop_proxy_->RunsTasksOnCurrentThread());
+  if (!current_)
+    return scoped_ptr<ServiceWorkerHandleReference>();
+  return ServiceWorkerHandleReference::Create(
+      current_->info(), thread_safe_sender_);
+}
+
+void ServiceWorkerProviderContext::OnServiceWorkerStateChanged(
+    int handle_id,
+    blink::WebServiceWorkerState state) {
+  // Currently .current is the only ServiceWorker associated to this provider.
+  DCHECK_EQ(current_handle_id(), handle_id);
+  current_->set_state(state);
+
+  // TODO(kinuko): We can forward the message to other threads here
+  // when we support navigator.serviceWorker in dedicated workers.
+}
+
+void ServiceWorkerProviderContext::OnSetCurrentServiceWorker(
+    int provider_id,
+    const ServiceWorkerObjectInfo& info) {
+  DCHECK_EQ(provider_id_, provider_id);
+
+  // This context is is the primary owner of this handle, keeps the
+  // initial reference until it goes away.
+  current_ = ServiceWorkerHandleReference::CreateForDeleter(
+      info, thread_safe_sender_);
+
+  // TODO(kinuko): We can forward the message to other threads here
+  // when we support navigator.serviceWorker in dedicated workers.
+}
+
+int ServiceWorkerProviderContext::current_handle_id() const {
+  DCHECK(main_thread_loop_proxy_->RunsTasksOnCurrentThread());
+  return current_ ? current_->info().handle_id : kInvalidServiceWorkerHandleId;
+}
+
+}  // namespace content