Implement DocuementHasImages

android_webview/renderer/aw_render_view_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/renderer/aw_render_view_ext.h"
+
+#include "android_webview/common/render_view_messages.h"
+#include "content/public/renderer/render_view.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebDocument.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebElement.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebFrame.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebView.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebVector.h"
+
+namespace android_webview {
+
+namespace {
+
+}  // namespace
+
+
+AwRenderViewExt::AwRenderViewExt(content::RenderView* render_view)
+    : content::RenderViewObserver(render_view) {
+}
+
+AwRenderViewExt::~AwRenderViewExt() {}
+
+bool AwRenderViewExt::OnMessageReceived(const IPC::Message& message) {
+  bool handled = true;
+  IPC_BEGIN_MESSAGE_MAP(AwRenderViewExt, message)
+    IPC_MESSAGE_HANDLER(AwViewMsg_DocumentHasImages, OnDocumentHasImagesRequest)
+    IPC_MESSAGE_UNHANDLED(handled = false)
+  IPC_END_MESSAGE_MAP()
+  return handled;
+}
+
+
+void AwRenderViewExt::OnDocumentHasImagesRequest(int id) {
+  bool hasImages = false;
+  if (render_view()) {
+    WebKit::WebView* webview = render_view()->GetWebView();
+    if (webview) {
+      WebKit::WebVector<WebKit::WebElement> images;
+      webview->mainFrame()->document().images(images);
+      hasImages = !images.isEmpty();
+    }
+  }
+  Send(new AwViewHostMsg_DocumentHasImagesResponse(routing_id(), id,
+                                                   hasImages));
+}
+
+}  // namespace android_webview