Wake Lock API implementation (Chromium part)

content/browser/wake_lock/wake_lock_service_context.cc

+// Copyright 2015 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_service_context.h"
+
+#include "base/bind.h"
+#include "content/browser/power_save_blocker_impl.h"
+#include "content/browser/wake_lock/wake_lock_service_impl.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/power_save_blocker.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/common/service_registry.h"
+
+namespace content {
+
+WakeLockServiceContext::WakeLockServiceContext(WebContents* web_contents)
+    : WebContentsObserver(web_contents),
+      weak_factory_(this) {
+}
+
+WakeLockServiceContext::~WakeLockServiceContext() {
+}
+
+void WakeLockServiceContext::RenderFrameCreated(
+    RenderFrameHost* render_frame_host) {
+  if (auto registry = render_frame_host->GetServiceRegistry()) {
+    registry->AddService<WakeLockService>(

[mlamouri (slow - plz ping), 2015/08/31 14:59:15]

Could you do that in RenderFrameHostImpl::RegisterMojoServices instead?

[alogvinov, 2015/09/02 17:06:22]

Done.

+        base::Bind(&WakeLockServiceImpl::CreateService,
+                   weak_factory_.GetWeakPtr(),
+                   base::Unretained(render_frame_host)));
+  }
+}
+
+void WakeLockServiceContext::RenderFrameDeleted(
+    RenderFrameHost* render_frame_host) {
+  frames_requesting_lock_.erase(render_frame_host);

[mlamouri (slow - plz ping), 2015/08/31 14:59:15]

Call ::CancelWakeLock() instead.

[alogvinov, 2015/09/02 17:06:22]

Done.

+  UpdateWakeLock();
+}
+
+void WakeLockServiceContext::RenderFrameHostChanged(
+    RenderFrameHost* old_host,
+    RenderFrameHost* new_host) {
+  frames_requesting_lock_.erase(old_host);
+  UpdateWakeLock();

[mlamouri (slow - plz ping), 2015/08/31 14:59:15]

ditto

[alogvinov, 2015/09/02 17:06:22]

Done.

+}
+
+void WakeLockServiceContext::RequestWakeLock(
+    RenderFrameHost* render_frame_host) {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  frames_requesting_lock_.insert(render_frame_host);
+  UpdateWakeLock();
+}
+
+void WakeLockServiceContext::CancelWakeLock(
+    RenderFrameHost* render_frame_host) {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  frames_requesting_lock_.erase(render_frame_host);
+  UpdateWakeLock();
+}
+
+void WakeLockServiceContext::CreateWakeLock() {
+  DCHECK(!wake_lock_);
+  wake_lock_ = PowerSaveBlocker::Create(
+      PowerSaveBlocker::kPowerSaveBlockPreventDisplaySleep,
+      PowerSaveBlocker::kReasonOther,
+      "Wake Lock API");
+  // On Android, additionaly associate the blocker with this WebContents'

[mlamouri (slow - plz ping), 2015/08/31 14:59:15]

nit: add empty line before this comment and another after the DCHECK().

[alogvinov, 2015/09/02 17:06:22]

Done.

+  // NativeView.
+#if defined(OS_ANDROID)
+  DCHECK(web_contents());
+  static_cast<PowerSaveBlockerImpl*>(wake_lock_.get())->InitDisplaySleepBlocker(
+      web_contents());
+#endif
+}
+
+void WakeLockServiceContext::RemoveWakeLock() {
+  wake_lock_.reset();

[mlamouri (slow - plz ping), 2015/08/31 14:59:15]

nit: DCHECK(wake_lock_);

[alogvinov, 2015/09/02 17:06:22]

Done.

+}
+
+void WakeLockServiceContext::UpdateWakeLock() {
+  if (!frames_requesting_lock_.empty()) {
+    if (!wake_lock_) {
+      CreateWakeLock();
+    }
+  } else {
+    RemoveWakeLock();
+  }
+}
+
+}  // namespace content