content/renderer/mhtml_generator.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/renderer/mhtml_generator.h"
   6
   7 #include "content/common/view_messages.h"
   8 #include "content/renderer/render_view_impl.h"
   9 #include "third_party/WebKit/public/platform/WebCString.h"
  10 #include "third_party/WebKit/public/web/WebPageSerializer.h"
  11
  12 namespace content {
  13
  14 MHTMLGenerator::MHTMLGenerator(RenderViewImpl* render_view)
  15     : RenderViewObserver(render_view) {
  16 }
  17
  18 MHTMLGenerator::~MHTMLGenerator() {
  19 }
  20
  21 // RenderViewObserver implementation:
  22 bool MHTMLGenerator::OnMessageReceived(const IPC::Message& message) {
  23   bool handled = true;
  24   IPC_BEGIN_MESSAGE_MAP(MHTMLGenerator, message)
  25       IPC_MESSAGE_HANDLER(ViewMsg_SavePageAsMHTML, OnSavePageAsMHTML)
  26       IPC_MESSAGE_UNHANDLED(handled = false)
  27   IPC_END_MESSAGE_MAP()
  28   return handled;
  29 }
  30
  31 void MHTMLGenerator::OnSavePageAsMHTML(
  32     int job_id, IPC::PlatformFileForTransit file_for_transit) {
  33   file_ = IPC::PlatformFileForTransitToFile(file_for_transit);
  34   int64 size = GenerateMHTML();
  35   file_.Close();
  36   NotifyBrowser(job_id, size);
  37 }
  38
  39 void MHTMLGenerator::NotifyBrowser(int job_id, int64 data_size) {
  40   render_view()->Send(new ViewHostMsg_SavedPageAsMHTML(job_id, data_size));
  41 }
  42
  43 // TODO(jcivelli): write the chunks in deferred tasks to give a chance to the
  44 //                 message loop to process other events.
  45 int64 MHTMLGenerator::GenerateMHTML() {
  46   blink::WebCString mhtml =
  47       blink::WebPageSerializer::serializeToMHTML(render_view()->GetWebView());
  48   const size_t chunk_size = 1024;
  49   const char* data = mhtml.data();
  50   size_t total_bytes_written = 0;
  51   while (total_bytes_written < mhtml.length()) {
  52     size_t copy_size =
  53         std::min(mhtml.length() - total_bytes_written, chunk_size);
  54     int bytes_written = file_.Write(total_bytes_written,
  55                                     data + total_bytes_written, copy_size);
  56     if (bytes_written == -1)
  57       return -1;
  58     total_bytes_written += bytes_written;
  59   }
  60   return total_bytes_written;
  61 }
  62
  63 }  // namespace content