Index: content/browser/wake_lock/wake_lock_context_host.cc |
diff --git a/content/browser/wake_lock/wake_lock_context_host.cc b/content/browser/wake_lock/wake_lock_context_host.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..7376d05f73becdf7de1b025368cc40b82695e511 |
--- /dev/null |
+++ b/content/browser/wake_lock/wake_lock_context_host.cc |
@@ -0,0 +1,60 @@ |
+// Copyright 2017 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/wake_lock/wake_lock_context_host.h" |
+ |
+#include "base/atomic_sequence_num.h" |
+#include "base/lazy_instance.h" |
+#include "content/public/common/service_manager_connection.h" |
+#include "device/wake_lock/public/interfaces/wake_lock_context_provider.mojom.h" |
+#include "mojo/public/cpp/bindings/strong_binding.h" |
+#include "services/device/public/interfaces/constants.mojom.h" |
+#include "services/service_manager/public/cpp/connector.h" |
+ |
+namespace content { |
+ |
+namespace { |
+ |
+base::StaticAtomicSequenceNumber g_unique_id; |
+ |
+base::LazyInstance<std::map<int, WakeLockContextHost*>>::Leaky |
+ g_id_to_context_host = LAZY_INSTANCE_INITIALIZER; |
+ |
+WakeLockContextHost* ContextHostFromId(int id) { |
+ auto it = g_id_to_context_host.Get().find(id); |
+ return it != g_id_to_context_host.Get().end() ? it->second : nullptr; |
+} |
+ |
+} // namespace |
+ |
+WakeLockContextHost::WakeLockContextHost(WebContents* web_contents) |
+ : id_(g_unique_id.GetNext()), web_contents_(web_contents) { |
+ g_id_to_context_host.Get()[id_] = this; |
+ |
+ // Connect to a WakeLockContext, associating it with |id_| (note that in some |
+ // testing contexts, the service manager connection isn't initialized). |
+ if (ServiceManagerConnection::GetForProcess()) { |
+ service_manager::Connector* connector = |
+ ServiceManagerConnection::GetForProcess()->GetConnector(); |
+ DCHECK(connector); |
+ device::mojom::WakeLockContextProviderPtr context_provider; |
+ connector->BindInterface(device::mojom::kServiceName, |
+ mojo::MakeRequest(&context_provider)); |
+ context_provider->GetContext(id_, mojo::MakeRequest(&wake_lock_context_)); |
+ } |
+} |
+ |
+WakeLockContextHost::~WakeLockContextHost() { |
+ g_id_to_context_host.Get().erase(id_); |
+} |
+ |
+// static |
+gfx::NativeView WakeLockContextHost::GetNativeViewForContext(int context_id) { |
+ WakeLockContextHost* context_host = ContextHostFromId(context_id); |
+ if (context_host) |
+ return context_host->web_contents_->GetNativeView(); |
+ return nullptr; |
+} |
+ |
+} // namespace content |