Save Page As MHTML

content/browser/download/save_package.cc

Index: content/browser/download/save_package.cc
diff --git a/content/browser/download/save_package.cc b/content/browser/download/save_package.cc
index 627215074055513a769f17aeb11642fdad45892c..3585821d28b51c94ab822e7ec2f683a404ae1064 100644
--- a/content/browser/download/save_package.cc
+++ b/content/browser/download/save_package.cc
@@ -291,6 +291,10 @@ bool SavePackage::Init() {
     // Get directory
     DCHECK(!saved_main_directory_path_.empty());
     GetAllSavableResourceLinksForCurrentPage();
+  } else if (save_type_ ==
+      content::SAVE_PAGE_TYPE_AS_COMPLETE_HTML_SINGLE_FILE_MHTML) {
+    web_contents()->GenerateMHTML(saved_main_file_path_, base::Bind(
+        &SavePackage::MHTMLGenerated, this));
   } else {
     wait_state_ = NET_FILES;
     SaveFileCreateInfo::SaveFileSource save_source = page_url_.SchemeIsFile() ?
@@ -311,6 +315,10 @@ bool SavePackage::Init() {
   return true;
 }
 
+void SavePackage::MHTMLGenerated(const FilePath& path, int64 size) {

[asanka, 2012/04/13 15:28:19]

I think the size will be -1 if there was an error during generation. If
unsuccessful we could call Cancel(false) instead of Finish().

Also on successful completion, we could set wrote_to_completed_file_ so that the
SAVE_PACKAGE_WRITE_TO_COMPLETED stat is recorded (unless you are considering a
separate set of stats for MHTML?)

[benjhayden, 2012/04/13 20:44:11]

Done.

+  Finish();
+}
+
 // On POSIX, the length of |pure_file_name| + |file_name_ext| is further
 // restricted by NAME_MAX. The maximum allowed path looks like:
 // '/path/to/save_dir' + '/' + NAME_MAX.