Move ownership of PowerSaveBlocker from WakeLockServiceContext to WakeLockServiceImpl

content/browser/web_contents/web_contents_impl.cc

 // Copyright (c) 2012 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/web_contents/web_contents_impl.h"
 
 #include <stddef.h>
 
 #include <cmath>
 #include <utility>
@@ skipping 2582 matching lines @@
     return nullptr;
 
   return guest->GetMainFrame();
 }
 
 device::GeolocationServiceContext*
 WebContentsImpl::GetGeolocationServiceContext() {
   return geolocation_service_context_.get();
 }
 
-device::mojom::WakeLockContext* WebContentsImpl::GetWakeLockServiceContext() {
+device::mojom::WakeLockContext* WebContentsImpl::GetWakeLockContext() {
   if (!wake_lock_context_host_)
     wake_lock_context_host_.reset(new WakeLockContextHost(this));
   return wake_lock_context_host_->GetWakeLockContext();
 }
 
+void WebContentsImpl::ConnectToBlinkWakeLock(
+    device::mojom::WakeLockServiceRequest request) {
+  // WebContents creates a long-lived connection to one WakeLockServiceImpl.
+  // All the frames' requests are added into the BindingSet of
+  // WakeLockServiceImpl via this connection.
+  if (!blink_wake_lock_) {
+    device::mojom::WakeLockContext* wake_lock_context = GetWakeLockContext();
+    if (!wake_lock_context) {
+      return;
+    }
+    wake_lock_context->GetWakeLock(
+        device::PowerSaveBlocker::PowerSaveBlockerType::
+            kPowerSaveBlockPreventDisplaySleep,
+        device::PowerSaveBlocker::Reason::kReasonOther, "Wake Lock API",
+        mojo::MakeRequest(&blink_wake_lock_));
+  }
+  blink_wake_lock_->AddClient(std::move(request));
+}
+
 void WebContentsImpl::OnShowValidationMessage(
     RenderViewHostImpl* source,
     const gfx::Rect& anchor_in_root_view,
     const base::string16& main_text,
     const base::string16& sub_text) {
   // TODO(nick): Should we consider |source| here or pass it to the delegate?
   if (delegate_)
     delegate_->ShowValidationMessage(
         this, anchor_in_root_view, main_text, sub_text);
 }
@@ skipping 2955 matching lines @@
     RenderViewHost* render_view_host = RenderViewHost::From(render_widget_host);
     if (!render_view_host)
       continue;
     render_view_host_set.insert(render_view_host);
   }
   for (RenderViewHost* render_view_host : render_view_host_set)
     render_view_host->OnWebkitPreferencesChanged();
 }
 
 }  // namespace content