Replace PeekMessage for TSF awareness

base/win/text_services_bridge.cc

Index: base/win/text_services_bridge.cc
diff --git a/base/win/text_services_bridge.cc b/base/win/text_services_bridge.cc
new file mode 100755
index 0000000000000000000000000000000000000000..1fc6e5e0d1615bb9fc1861072eab35c7bb3c68e9
--- /dev/null
+++ b/base/win/text_services_bridge.cc
@@ -0,0 +1,75 @@
+// 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 "base/win/text_services_bridge.h"
+
+namespace base {
+namespace win {
+
+TextServicesBridge::TextServicesBridge() : client_id_(TF_CLIENTID_NULL) {
+  is_initialized_ = scoped_com_initialiser_.succeeded();
+  if (!is_initialized_) {
+      return;
+  }
+
+  is_initialized_ &= SUCCEEDED(thread_mgr_.CreateInstance(CLSID_TF_ThreadMgr));
+  if (!is_initialized_)
+    return;
+
+  is_initialized_ &= SUCCEEDED(message_pump_.QueryFrom(thread_mgr_));
+  is_initialized_ &= SUCCEEDED(keystroke_mgr_.QueryFrom(thread_mgr_));
+  // When activate succeeded, |client_id| is set valid value.
+  is_initialized_ &= SUCCEEDED(thread_mgr_->Activate(&client_id_));
+}
+
+TextServicesBridge::~TextServicesBridge() {
+  if (is_initialized_)
+    thread_mgr_->Deactivate();
+}
+
+bool TextServicesBridge::Init() {
+  return is_initialized_;
+}
+
+BOOL TextServicesBridge::DoPeekMessage(MSG* msg,
+                                           HWND hwnd,
+                                           UINT msg_filter_min,
+                                           UINT msg_filter_max,
+                                           UINT remove_msg) {
+    BOOL result = FALSE;
+    if (FAILED(message_pump_->PeekMessage(msg, hwnd, msg_filter_min,
+                                          msg_filter_max,
+                                          remove_msg, &result))) {
+        result = FALSE;
+    }
+
+    return result;
+}
+
+bool TextServicesBridge::ProcessMessage(const MSG& msg) {
+  if (msg.message == WM_KEYDOWN) {
+    BOOL eaten = FALSE;
+    HRESULT hr = keystroke_mgr_->TestKeyDown(msg.wParam, msg.lParam, &eaten);
+    if (FAILED(hr) && !eaten)
+      return false;
+    eaten = FALSE;
+    hr = keystroke_mgr_->KeyDown(msg.wParam, msg.lParam, &eaten);
+    return (SUCCEEDED(hr) && eaten);
+  }
+
+  if (msg.message == WM_KEYUP) {
+    BOOL eaten = FALSE;
+    HRESULT hr = keystroke_mgr_->TestKeyUp(msg.wParam, msg.lParam, &eaten);
+    if (FAILED(hr) && !eaten)
+      return false;
+    eaten = FALSE;
+    hr = keystroke_mgr_->KeyUp(msg.wParam, msg.lParam, &eaten);
+    return (SUCCEEDED(hr) && eaten);
+  }
+
+  return false;
+}
+
+}  // namespace win
+}  // namespace base