| Index: content/shell/shell_render_view_observer.cc
|
| diff --git a/content/shell/shell_render_view_observer.cc b/content/shell/shell_render_view_observer.cc
|
| index 942f6019464624eaafe0a8d50e3feb433d55c2b2..065551ed81c0f0c978f150ddcd50cb3547fddecb 100644
|
| --- a/content/shell/shell_render_view_observer.cc
|
| +++ b/content/shell/shell_render_view_observer.cc
|
| @@ -4,8 +4,54 @@
|
|
|
| #include "content/shell/shell_render_view_observer.h"
|
|
|
| +#include "content/public/renderer/render_view.h"
|
| +#include "content/shell/shell_messages.h"
|
| +#include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebCString.h"
|
| +#include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebString.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"
|
| +
|
| +using WebKit::WebFrame;
|
| +using WebKit::WebElement;
|
| +
|
| namespace content {
|
|
|
| +namespace {
|
| +
|
| +std::string DumpDocumentText(WebFrame* frame) {
|
| + // We use the document element's text instead of the body text here because
|
| + // not all documents have a body, such as XML documents.
|
| + WebElement documentElement = frame->document().documentElement();
|
| + if (documentElement.isNull())
|
| + return std::string();
|
| + return documentElement.innerText().utf8();
|
| +}
|
| +
|
| +std::string DumpFramesAsText(WebFrame* frame, bool recursive) {
|
| + std::string result;
|
| +
|
| + // Add header for all but the main frame. Skip emtpy frames.
|
| + if (frame->parent() && !frame->document().documentElement().isNull()) {
|
| + result.append("\n--------\nFrame: '");
|
| + result.append(frame->name().utf8().data());
|
| + result.append("'\n--------\n");
|
| + }
|
| +
|
| + result.append(DumpDocumentText(frame));
|
| + result.append("\n");
|
| +
|
| + if (recursive) {
|
| + for (WebFrame* child = frame->firstChild(); child;
|
| + child = child->nextSibling()) {
|
| + result.append(DumpFramesAsText(child, recursive));
|
| + }
|
| + }
|
| + return result;
|
| +}
|
| +
|
| +} // namespace
|
| ShellRenderViewObserver::ShellRenderViewObserver(RenderView* render_view)
|
| : RenderViewObserver(render_view) {
|
| }
|
| @@ -14,7 +60,19 @@ ShellRenderViewObserver::~ShellRenderViewObserver() {
|
| }
|
|
|
| bool ShellRenderViewObserver::OnMessageReceived(const IPC::Message& message) {
|
| - return false;
|
| + bool handled = true;
|
| + IPC_BEGIN_MESSAGE_MAP(ShellRenderViewObserver, message)
|
| + IPC_MESSAGE_HANDLER(ShellViewMsg_CaptureTextDump, OnCaptureTextDump)
|
| + IPC_MESSAGE_UNHANDLED(handled = false)
|
| + IPC_END_MESSAGE_MAP()
|
| +
|
| + return handled;
|
| +}
|
| +
|
| +void ShellRenderViewObserver::OnCaptureTextDump(bool recursive) {
|
| + std::string dump =
|
| + DumpFramesAsText(render_view()->GetWebView()->mainFrame(), recursive);
|
| + Send(new ShellViewHostMsg_TextDump(routing_id(), dump));
|
| }
|
|
|
| } // namespace content
|
|
|