mandoline: Add find in page.

components/html_viewer/html_frame.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 "components/html_viewer/html_frame.h"
 
 #include <algorithm>
 #include <limits>
 
 #include "base/bind.h"
@@ skipping 32 matching lines @@
 #include "mojo/common/common_type_converters.h"
 #include "mojo/converters/geometry/geometry_type_converters.h"
 #include "skia/ext/refptr.h"
 #include "third_party/WebKit/public/platform/Platform.h"
 #include "third_party/WebKit/public/platform/WebHTTPHeaderVisitor.h"
 #include "third_party/WebKit/public/platform/WebSecurityOrigin.h"
 #include "third_party/WebKit/public/platform/WebSize.h"
 #include "third_party/WebKit/public/web/WebConsoleMessage.h"
 #include "third_party/WebKit/public/web/WebDocument.h"
 #include "third_party/WebKit/public/web/WebElement.h"
+#include "third_party/WebKit/public/web/WebFindOptions.h"
 #include "third_party/WebKit/public/web/WebInputEvent.h"
 #include "third_party/WebKit/public/web/WebKit.h"
 #include "third_party/WebKit/public/web/WebLocalFrame.h"
 #include "third_party/WebKit/public/web/WebNavigationPolicy.h"
 #include "third_party/WebKit/public/web/WebRemoteFrame.h"
 #include "third_party/WebKit/public/web/WebRemoteFrameClient.h"
 #include "third_party/WebKit/public/web/WebScriptSource.h"
 #include "third_party/WebKit/public/web/WebView.h"
 #include "third_party/mojo/src/mojo/public/cpp/system/data_pipe.h"
 #include "third_party/skia/include/core/SkCanvas.h"
@@ skipping 433 matching lines @@
                                 blink::WebTextDirection direction) {
   // TODO(beng): handle |direction|.
   mojo::String formatted;
   if (!title.isNull()) {
     formatted =
         mojo::String::From(base::string16(title).substr(0, kMaxTitleChars));
   }
   server_->TitleChanged(formatted);
 }
 
+void HTMLFrame::reportFindInFrameMatchCount(int identifier,
+                                            int count,
+                                            bool finalUpdate) {
+  server_->OnFindInFrameCountUpdated(identifier, count, finalUpdate);
+}
+
+void HTMLFrame::reportFindInPageSelection(int identifier,
+                                          int activeMatchOrdinal,
+                                          const blink::WebRect& selection) {
+  server_->OnFindInPageSelectionUpdated(identifier, activeMatchOrdinal);
+}
+
 void HTMLFrame::Bind(
     web_view::mojom::FramePtr frame,
     mojo::InterfaceRequest<web_view::mojom::FrameClient> frame_client_request) {
   DCHECK(IsLocal());
   server_ = frame.Pass();
   server_.set_connection_error_handler(
       base::Bind(&HTMLFrame::Close, base::Unretained(this)));
   frame_client_binding_.reset(new mojo::Binding<web_view::mojom::FrameClient>(
       this, frame_client_request.Pass()));
 }
@@ skipping 139 matching lines @@
   web_layer_.reset();
 }
 
 void HTMLFrame::SwapDelegate(HTMLFrameDelegate* delegate) {
   DCHECK(IsLocal());
   HTMLFrameDelegate* old_delegate = delegate_;
   delegate_ = delegate;
   delegate->OnSwap(this, old_delegate);
 }
 
+blink::WebElement HTMLFrame::GetFocusedElement() {
+  if (!web_view())
+    return blink::WebElement();
+
+  HTMLFrame* frame = this;
+  while (frame) {
+    if (frame->web_view()) {
+      if (frame->web_view()->focusedFrame() == web_frame_) {
+        blink::WebDocument doc = web_frame_->document();
+        if (!doc.isNull())
+          return doc.focusedElement();
+      }
+      return blink::WebElement();
+    }
+    frame = frame->parent();
+  }
+
+  return blink::WebElement();
+}
+
 HTMLFrame* HTMLFrame::FindFrameWithWebFrame(blink::WebFrame* web_frame) {
   if (web_frame_ == web_frame)
     return this;
   for (HTMLFrame* child_frame : children_) {
     HTMLFrame* result = child_frame->FindFrameWithWebFrame(web_frame);
     if (result)
       return result;
   }
   return nullptr;
 }
@@ skipping 175 matching lines @@
       frame->web_frame_->toWebRemoteFrame()->didStopLoading();
   }
 }
 
 void HTMLFrame::OnDispatchFrameLoadEvent(uint32_t frame_id) {
   HTMLFrame* frame = frame_tree_manager_->root_->FindFrame(frame_id);
   if (frame && !frame->IsLocal())
     frame->web_frame_->toWebRemoteFrame()->DispatchLoadEventForFrameOwner();
 }
 
+void HTMLFrame::Find(int32 request_id,
+                     const mojo::String& search_text,
+                     const FindCallback& callback) {
+  // TODO(erg): We need to synchronize whether we're a singleton frame here
+  // with the parent; we can't trust our state.
+  bool wrap_within_frame = false;
+
+  blink::WebFindOptions options;
+  blink::WebRect selection_rect;
+  bool result = web_frame_->find(request_id, search_text.To<blink::WebString>(),
+                                 options, wrap_within_frame, &selection_rect);
+  if (!result) {
+    // don't leave text selected as you move to the next frame.
+    web_frame_->executeCommand(blink::WebString::fromUTF8("Unselect"),
+                               GetFocusedElement());
+  }
+
+  callback.Run(result);
+}
+
+void HTMLFrame::StopFinding(bool clear_selection) {
+  // TODO(erg): |clear_selection| isn't correct; this should be a state enum
+  // that lets us STOP_FIND_ACTION_ACTIVATE_SELECTION, too.
+  if (clear_selection) {
+    blink::WebElement focused_element = GetFocusedElement();
+    if (!focused_element.isNull()) {
+      web_frame_->executeCommand(blink::WebString::fromUTF8("Unselect"),
+                                 focused_element);
+    }
+  }
+
+  web_frame_->stopFinding(clear_selection);
+}
+
+void HTMLFrame::HighlightFindResults(int32_t request_id,
+                                     const mojo::String& search_text,
+                                     bool reset) {
+  blink::WebFindOptions options;
+  web_frame_->scopeStringMatches(request_id, search_text.To<blink::WebString>(),
+                                 options, reset);
+}
+
+void HTMLFrame::StopHighlightingFindResults() {
+  web_frame_->resetMatchCount();
+  web_frame_->cancelPendingScopingEffort();
+}
+
 void HTMLFrame::frameDetached(blink::WebRemoteFrameClient::DetachType type) {
   if (type == blink::WebRemoteFrameClient::DetachType::Swap) {
     web_frame_->close();
     return;
   }
 
   DCHECK(type == blink::WebRemoteFrameClient::DetachType::Remove);
   FrameDetachedImpl(web_frame_);
 }
 
@@ skipping 60 matching lines @@
   if (!surface_layer_)
     return;
 
   surface_layer_->SetSurfaceId(
       cc::SurfaceId(owned_view_->view()->id()),
       global_state()->device_pixel_ratio(),
       owned_view_->view()->bounds().To<gfx::Rect>().size());
 }
 
 }  // namespace mojo