Move IME related functions from WebFrame to WebLocalFrame

components/test_runner/text_input_controller.cc

 // Copyright 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 "components/test_runner/text_input_controller.h"
 
 #include "base/macros.h"
 #include "components/test_runner/web_test_proxy.h"
 #include "gin/arguments.h"
 #include "gin/handle.h"
@@ skipping 148 matching lines @@
 
 void TextInputController::InsertText(const std::string& text) {
   view()->confirmComposition(blink::WebString::fromUTF8(text));
 }
 
 void TextInputController::UnmarkText() {
   view()->confirmComposition();
 }
 
 void TextInputController::DoCommand(const std::string& text) {
-  if (view()->mainFrame())
-    view()->mainFrame()->executeCommand(blink::WebString::fromUTF8(text));
+  if (view()->mainFrame()) {
+    if (!view()->mainFrame()->toWebLocalFrame()) {
+      CHECK(false) << "This function cannot be called if the main frame is not"
+                      "a local frame.";
+    }
+    view()->mainFrame()->toWebLocalFrame()->executeCommand(
+        blink::WebString::fromUTF8(text));
+  }
 }
 
 void TextInputController::SetMarkedText(const std::string& text,
                                         int start,
                                         int length) {
   blink::WebString web_text(blink::WebString::fromUTF8(text));
 
   // Split underline into up to 3 elements (before, selection, and after).
   std::vector<blink::WebCompositionUnderline> underlines;
   blink::WebCompositionUnderline underline;
@@ skipping 11 matching lines @@
     underline.startOffset = underline.endOffset;
     underline.endOffset = web_text.length();
     underline.thick = false;
     underlines.push_back(underline);
   }
 
   view()->setComposition(web_text, underlines, start, start + length);
 }
 
 bool TextInputController::HasMarkedText() {
-  return view()->mainFrame() && view()->mainFrame()->hasMarkedText();
+  if (!view()->mainFrame())
+    return false;
+
+  if (!view()->mainFrame()->toWebLocalFrame()) {
+    CHECK(false) << "This function cannot be called if the main frame is not"
+                    "a local frame.";
+  }
+
+  return view()->mainFrame()->toWebLocalFrame()->hasMarkedText();
 }
 
 std::vector<int> TextInputController::MarkedRange() {
   if (!view()->mainFrame())
     return std::vector<int>();
 
-  blink::WebRange range = view()->mainFrame()->markedRange();
+  if (!view()->mainFrame()->toWebLocalFrame()) {
+    CHECK(false) << "This function cannot be called if the main frame is not"
+                    "a local frame.";
+  }
+
+  blink::WebRange range = view()->mainFrame()->toWebLocalFrame()->markedRange();
   std::vector<int> int_array(2);
   int_array[0] = range.startOffset();
   int_array[1] = range.endOffset();
 
   return int_array;
 }
 
 std::vector<int> TextInputController::SelectedRange() {
   if (!view()->mainFrame())
     return std::vector<int>();
 
-  blink::WebRange range = view()->mainFrame()->selectionRange();
+  if (!view()->mainFrame()->toWebLocalFrame()) {
+    CHECK(false) << "This function cannot be called if the main frame is not"
+                    "a local frame.";
+  }
+
+  blink::WebRange range =
+      view()->mainFrame()->toWebLocalFrame()->selectionRange();
   if (range.isNull())
     return std::vector<int>();
   std::vector<int> int_array(2);
   int_array[0] = range.startOffset();
   int_array[1] = range.endOffset();
 
   return int_array;
 }
 
 std::vector<int> TextInputController::FirstRectForCharacterRange(
@@ skipping 32 matching lines @@
       blink::WebString::fromUTF8(text),
       blink::WebVector<blink::WebCompositionUnderline>(underlines),
       text.length(), text.length());
 }
 
 blink::WebView* TextInputController::view() {
   return web_test_proxy_base_->web_view();
 }
 
 }  // namespace test_runner