[Android] Upstream the WebView find-in-page API implementation.

android_webview/native/aw_contents.cc

 // 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 "android_webview/native/aw_contents.h"
 
 #include "android_webview/browser/renderer_host/aw_render_view_host_ext.h"
 #include "android_webview/native/aw_browser_dependency_factory.h"
 #include "android_webview/native/aw_contents_container.h"
 #include "android_webview/native/aw_web_contents_delegate.h"
 #include "base/android/jni_android.h"
 #include "base/android/jni_string.h"
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/supports_user_data.h"
 #include "content/public/browser/android/content_view_core.h"
 #include "content/public/browser/web_contents.h"
 #include "jni/AwContents_jni.h"
 
 using base::android::AttachCurrentThread;
+using base::android::ConvertJavaStringToUTF16;
 using base::android::ConvertUTF8ToJavaString;
 using base::android::ConvertUTF16ToJavaString;
 using base::android::ScopedJavaGlobalRef;
 using base::android::ScopedJavaLocalRef;
 using content::ContentViewCore;
 using content::WebContents;
 
 namespace android_webview {
 
 namespace {
 
 const void* kAwContentsUserDataKey = &kAwContentsUserDataKey;
 
 class AwContentsUserData : public base::SupportsUserData::Data {
  public:
   AwContentsUserData(AwContents* ptr) : contents_(ptr) {}
-  AwContents* get() { return contents_; }
+
+  static AwContents* GetContents(WebContents* web_contents) {
+    if (!web_contents)
+      return NULL;
+    AwContentsUserData* data = reinterpret_cast<AwContentsUserData*>(
+        web_contents->GetUserData(kAwContentsUserDataKey));
+    return data ? data->contents_ : NULL;
+  }
 
  private:
   AwContents* contents_;
 };
 
 }  // namespace
 
 // static
-AwContents* AwContents::FromWebContents(content::WebContents* web_contents) {
-  if (web_contents) {
-    AwContentsUserData* data = reinterpret_cast<AwContentsUserData*>(
-              web_contents->GetUserData(kAwContentsUserDataKey));
-    if (data) return data->get();
-  }
-  return NULL;
+AwContents* AwContents::FromWebContents(WebContents* web_contents) {
+  return AwContentsUserData::GetContents(web_contents);
 }
 
 AwContents::AwContents(JNIEnv* env,
                        jobject obj,
                        jobject web_contents_delegate,
                        bool private_browsing)
     : java_ref_(env, obj),
       web_contents_delegate_(
           new AwWebContentsDelegate(env, web_contents_delegate)) {
   android_webview::AwBrowserDependencyFactory* dependency_factory =
       android_webview::AwBrowserDependencyFactory::GetInstance();
-  content::WebContents* web_contents =
+
+  WebContents* web_contents =
       dependency_factory->CreateWebContents(private_browsing);
+
+  DCHECK(!AwContents::FromWebContents(web_contents));
+  web_contents->SetUserData(kAwContentsUserDataKey,
+                            new AwContentsUserData(this));
+
   contents_container_.reset(dependency_factory->CreateContentsContainer(
       web_contents));
   web_contents->SetDelegate(web_contents_delegate_.get());
   web_contents_delegate_->SetJavaScriptDialogCreator(
       dependency_factory->GetJavaScriptDialogCreator());
-  web_contents->SetUserData(kAwContentsUserDataKey,
-                            new AwContentsUserData(this));
   render_view_host_ext_.reset(new AwRenderViewHostExt(web_contents));
 }
 
 AwContents::~AwContents() {
-  content::WebContents* web_contents = contents_container_->GetWebContents();
+  WebContents* web_contents = contents_container_->GetWebContents();
   DCHECK(AwContents::FromWebContents(web_contents) == this);
   web_contents->RemoveUserData(kAwContentsUserDataKey);
+  if (find_helper_.get())
+    find_helper_->SetListener(NULL);
 }
 
 jint AwContents::GetWebContents(JNIEnv* env, jobject obj) {
   return reinterpret_cast<jint>(contents_container_->GetWebContents());
 }
 
 void AwContents::Destroy(JNIEnv* env, jobject obj) {
   delete this;
 }
 
@@ skipping 33 matching lines @@
                  jboolean private_browsing) {
   AwContents* tab = new AwContents(env, obj, web_contents_delegate,
                                    private_browsing);
   return reinterpret_cast<jint>(tab);
 }
 
 bool RegisterAwContents(JNIEnv* env) {
   return RegisterNativesImpl(env) >= 0;
 }
 
+jint AwContents::FindAllSync(JNIEnv* env, jobject obj, jstring search_string) {
+  return GetFindHelper()->FindAllSync(
+      ConvertJavaStringToUTF16(env, search_string));
+}
+
+void AwContents::FindAllAsync(JNIEnv* env, jobject obj, jstring search_string) {
+  GetFindHelper()->FindAllAsync(ConvertJavaStringToUTF16(env, search_string));
+}
+
+void AwContents::FindNext(JNIEnv* env, jobject obj, jboolean forward) {
+  GetFindHelper()->FindNext(forward);
+}
+
+void AwContents::ClearMatches(JNIEnv* env, jobject obj) {
+  GetFindHelper()->ClearMatches();
+}
+
+FindHelper* AwContents::GetFindHelper() {
+  if (!find_helper_.get()) {
+    WebContents* web_contents = contents_container_->GetWebContents();
+    find_helper_.reset(new FindHelper(web_contents));
+    find_helper_->SetListener(this);
+  }
+  return find_helper_.get();
+}
+
+void AwContents::OnFindResultReceived(int active_ordinal,
+                                      int match_count,
+                                      bool finished) {
+  JNIEnv* env = AttachCurrentThread();
+  ScopedJavaLocalRef<jobject> obj = java_ref_.get(env);
+  if (obj.is_null())
+    return;
+
+  Java_AwContents_onFindResultReceived(
+      env, obj.obj(), active_ordinal, match_count, finished);
+}
 
 }  // namespace android_webview