Chromium Code Reviews| 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..717959e437da9bcd9a85d407c04572353b2e9b57 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) { |
|
jam
2012/01/26 18:47:34
nit: here and below, CamelCase
jochen (gone - plz use gerrit)
2012/01/27 08:49:56
Done.
|
| + // 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 |