Implement DocuementHasImages

android_webview/browser/renderer_host/aw_render_view_host_ext.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/browser/renderer_host/aw_render_view_host_ext.h"
+
+#include "android_webview/common/render_view_messages.h"
+#include "base/android/scoped_java_ref.h"
+#include "base/callback.h"
+#include "content/public/browser/user_metrics.h"
+#include "content/public/browser/web_contents.h"
+
+namespace android_webview {
+
+AwRenderViewHostExt::AwRenderViewHostExt(content::WebContents* contents)
+    : content::WebContentsObserver(contents) {
+}
+
+AwRenderViewHostExt::~AwRenderViewHostExt() {}
+
+void AwRenderViewHostExt::DocumentHasImages(DocumentHasImagesResult result) {
+  if (!web_contents()->GetRenderViewHost()) {
+    result.Run(false);
+    return;
+  }
+  static int next_id = 1;
+  int this_id = next_id++;
+  pending_document_has_images_requests_[this_id] = result;
+  Send(new AwViewMsg_DocumentHasImages(web_contents()->GetRoutingID(),
+                                       this_id));
+}
+
+void AwRenderViewHostExt::RenderViewGone(base::TerminationStatus status) {
+  for (std::map<int, DocumentHasImagesResult>::iterator pending_req =
+           pending_document_has_images_requests_.begin();
+       pending_req != pending_document_has_images_requests_.end();
+      ++pending_req) {
+    pending_req->second.Run(false);
+  }
+}
+
+bool AwRenderViewHostExt::OnMessageReceived(const IPC::Message& message) {
+  bool handled = true;
+  IPC_BEGIN_MESSAGE_MAP(AwRenderViewHostExt, message)
+    IPC_MESSAGE_HANDLER(AwViewHostMsg_DocumentHasImagesResponse,
+                        OnDocumentHasImagesResponse)
+    IPC_MESSAGE_UNHANDLED(handled = false)
+  IPC_END_MESSAGE_MAP()
+
+  if (!handled) handled = WebContentsObserver::OnMessageReceived(message);
+
+  return handled;
+}
+
+void AwRenderViewHostExt::OnDocumentHasImagesResponse(int msg_id,
+                                                      bool has_images) {
+  std::map<int, DocumentHasImagesResult>::iterator pending_req =
+      pending_document_has_images_requests_.find(msg_id);
+  if (pending_req != pending_document_has_images_requests_.end()) {
+    pending_req->second.Run(has_images);
+    pending_document_has_images_requests_.erase(pending_req);
+  }
+}
+
+}  // namespace android_webview