Initial implementation of API WakeLock.

chrome/browser/wake_lock/chrome_wake_lock_permission_context.cc

+// Copyright (c) 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 "chrome/browser/wake_lock/chrome_wake_lock_permission_context.h"
+
+#include "base/bind.h"
+#include "chrome/browser/infobars/infobar_service.h"
+#include "chrome/browser/tab_contents/tab_util.h"
+#include "chrome/browser/wake_lock/wake_lock_infobar_delegate.h"
+#include "chrome/browser/wake_lock/wake_lock_preference.h"
+#include "content/public/browser/browser_thread.h"
+
+ChromeWakeLockPermissionContext::ChromeWakeLockPermissionContext(
+    Profile* profile)
+    : profile_(profile),
+      shutting_down_(false) {
+}
+
+ChromeWakeLockPermissionContext::~ChromeWakeLockPermissionContext() {
+}
+
+void ChromeWakeLockPermissionContext::RequestWakeLockPermission(
+    int render_process_id,
+    int render_view_id,
+    const GURL& requesting_frame,
+    WakeLockCallback callback) {
+  GURL requesting_frame_origin = requesting_frame.GetOrigin();
+  if (!content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)) {
+    content::BrowserThread::PostTask(
+        content::BrowserThread::UI, FROM_HERE,
+        base::Bind(
+            &ChromeWakeLockPermissionContext::RequestWakeLockPermission,
+            this, render_process_id, render_view_id,
+            requesting_frame_origin, callback));
+    return;
+  }
+
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+  if (shutting_down_)
+    return;
+
+  content::WebContents* web_contents =
+      tab_util::GetWebContentsByID(render_process_id, render_view_id);
+  GURL embedder = web_contents->GetLastCommittedURL().GetOrigin();
+  if (!embedder.is_valid()) {
+    callback.Run(false);
+    return;
+  }
+
+  scoped_ptr<WakeLockPreference> pref =
+      WakeLockPreference::createWakeLockPreference(
+          render_process_id, render_view_id);
+  if (pref->isSavedAllow(embedder.spec())) {
+    callback.Run(pref->isAllowed(embedder.spec()));
+    return;
+  }
+
+  WakeLockInfoBarDelegate::Create(
+      InfoBarService::FromWebContents(web_contents),
+      render_process_id,
+      render_view_id,
+      callback);
+}
+
+void ChromeWakeLockPermissionContext::ShutdownOnUIThread() {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+  shutting_down_ = true;
+}