Keyboard Lock Host implementation

content/browser/keyboard_lock/keyboard_lock_service_impl.cc

 // 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/keyboard_lock/keyboard_lock_service_impl.h"
 
 #include <memory>
 #include <utility>
 
+#include "base/bind.h"
+#include "base/location.h"
+#include "base/logging.h"
 #include "base/memory/ptr_util.h"
+#include "components/keyboard_lock/keyboard_lock_host.h"
+#include "content/public/browser/web_contents.h"
 #include "content/public/browser/render_frame_host.h"
 
 namespace content {
 
-KeyboardLockServiceImpl::KeyboardLockServiceImpl() = default;
+KeyboardLockServiceImpl::KeyboardLockServiceImpl(
+    RenderFrameHost* render_frame_host)
+    : web_contents_(WebContents::FromRenderFrameHost(render_frame_host)) {
+  DCHECK(web_contents_);
+}
 
 KeyboardLockServiceImpl::~KeyboardLockServiceImpl() = default;
 
 // static
 void KeyboardLockServiceImpl::CreateMojoService(
+    RenderFrameHost* render_frame_host,
     const service_manager::BindSourceInfo& source_info,
     blink::mojom::KeyboardLockServiceRequest request) {
   mojo::MakeStrongBinding(
-        base::MakeUnique<KeyboardLockServiceImpl>(),
+        base::MakeUnique<KeyboardLockServiceImpl>(render_frame_host),
         std::move(request));
 }
 
 void KeyboardLockServiceImpl::RequestKeyboardLock(
     const std::vector<std::string>& key_codes,
     const RequestKeyboardLockCallback& callback) {
-  // TODO(zijiehe): Implementation required.
-  callback.Run(blink::mojom::KeyboardLockRequestResult::SUCCESS);
+  keyboard_lock::KeyboardLockHost* host =
+      keyboard_lock::KeyboardLockHost::Find(web_contents_);
+  if (!host) {
+    // This should not happen in production.
+    callback.Run(blink::mojom::KeyboardLockRequestResult::SUCCESS);
+    return;
+  }
+
+  host->SetReservedKeyCodes(web_contents_, key_codes,
+      base::Bind([](RequestKeyboardLockCallback callback, bool result) {
+                   callback.Run(
+                       blink::mojom::KeyboardLockRequestResult::SUCCESS);
+                 },
+                 callback));
 }
 
 void KeyboardLockServiceImpl::CancelKeyboardLock() {
-  // TODO(zijiehe): Implementation required.
+  keyboard_lock::KeyboardLockHost* host =
+      keyboard_lock::KeyboardLockHost::Find(web_contents_);
+  if (!host) {
+    // This should not happen in production.
+    return;
+  }
+  host->ClearReservedKeys(web_contents_, base::Callback<void(bool)>());
 }
 
 }  // namespace content