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
12namespace content {
13
14MHTMLGenerator::MHTMLGenerator(RenderViewImpl* render_view)
15    : RenderViewObserver(render_view) {
16}
17
18MHTMLGenerator::~MHTMLGenerator() {
19}
20
21// RenderViewObserver implementation:
22bool 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
31void 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
39void 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.
45int64 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
64