Hooking MHTML generation to the browser.

chrome/renderer/mhtml_generator.cc

+// Copyright (c) 2011 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 "chrome/renderer/mhtml_generator.h"
+
+#include "base/platform_file.h"
+#include "chrome/common/render_messages.h"
+#include "content/renderer/render_view.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebCString.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebPageSerializer.h"
+
+MHTMLGenerator::MHTMLGenerator(RenderView* render_view)
+    : RenderViewObserver(render_view),
+      file_(base::kInvalidPlatformFileValue) {
+}
+
+MHTMLGenerator::~MHTMLGenerator() {
+}
+
+// RenderViewObserver implementation:
+bool MHTMLGenerator::OnMessageReceived(const IPC::Message& message) {
+  bool handled = true;
+  IPC_BEGIN_MESSAGE_MAP(MHTMLGenerator, message)
+      IPC_MESSAGE_HANDLER(ViewMsg_SavePageAsMHTML, OnSavePageAsMHTML)
+      IPC_MESSAGE_UNHANDLED(handled = false)
+  IPC_END_MESSAGE_MAP()
+  return handled;
+}
+
+void MHTMLGenerator::OnSavePageAsMHTML(
+    int job_id, IPC::PlatformFileForTransit file_for_transit) {
+  base::PlatformFile file =
+      IPC::PlatformFileForTransitToPlatformFile(file_for_transit);
+  file_ = file;
+  bool success = GenerateMHTML();
+  NotifyBrowser(job_id, success);
+}
+
+void MHTMLGenerator::NotifyBrowser(int job_id, bool success) {
+  render_view()->Send(new ViewHostMsg_SavedPageAsMHTML(
+      render_view()->routing_id(), job_id, success));
+  file_ = base::kInvalidPlatformFileValue;
+}
+
+// TODO(jcivelli): write the chunks in deferred tasks to give a chance to the
+//                 message loop to process other events.
+bool MHTMLGenerator::GenerateMHTML() {
+  WebKit::WebCString mhtml =
+      WebKit::WebPageSerializer::serializeToMHTML(render_view()->webview());
+  const size_t chunk_size = 1024;
+  const char* data = mhtml.data();
+  size_t total_bytes_written = 0;
+  while (total_bytes_written < mhtml.length()) {
+    size_t copy_size =
+        std::min(mhtml.length() - total_bytes_written, chunk_size);
+    int bytes_written = base::WritePlatformFile(file_, total_bytes_written,
+                                                data + total_bytes_written,
+                                                copy_size);
+    if (bytes_written == -1)
+      return false;
+    total_bytes_written += bytes_written;
+  }
+  return true;
+}