Save Page As MHTML

chrome/browser/download/save_package_file_picker.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/browser/download/save_package_file_picker.h"
 
+#include "base/metrics/histogram.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/download/download_prefs.h"
 #include "chrome/browser/platform_util.h"
 #include "chrome/browser/prefs/pref_member.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/pref_names.h"
 #include "content/public/browser/download_manager.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/save_page_type.h"
 #include "content/public/browser/web_contents.h"
 #include "grit/generated_resources.h"
 #include "ui/base/l10n/l10n_util.h"
 
 using content::RenderProcessHost;
 using content::SavePageType;
 using content::WebContents;
 
 namespace {
 
 // If false, we don't prompt the user as to where to save the file.  This
 // exists only for testing.
 bool g_should_prompt_for_filename = true;
 
 // Used for mapping between SavePageType constants and the indexes above.
 const SavePageType kIndexToSaveType[] = {
   content::SAVE_PAGE_TYPE_UNKNOWN,
   content::SAVE_PAGE_TYPE_AS_ONLY_HTML,
   content::SAVE_PAGE_TYPE_AS_COMPLETE_HTML,
+  content::SAVE_PAGE_TYPE_AS_MHTML,
 };
 
 int SavePackageTypeToIndex(SavePageType type) {
   for (size_t i = 0; i < arraysize(kIndexToSaveType); ++i) {
     if (kIndexToSaveType[i] == type)
       return i;
   }
   NOTREACHED();
   return -1;
 }
 
 // Indexes used for specifying which element in the extensions dropdown
 // the user chooses when picking a save type.
 const int kSelectFileHtmlOnlyIndex = 1;
 const int kSelectFileCompleteIndex = 2;
+const int kSelectFileCompleteSingleFileIndex = 3;
 
 // Used for mapping between the IDS_ string identifiers and the indexes above.
 const int kIndexToIDS[] = {
-  0, IDS_SAVE_PAGE_DESC_HTML_ONLY, IDS_SAVE_PAGE_DESC_COMPLETE,
+  0,
+  IDS_SAVE_PAGE_DESC_HTML_ONLY,
+  IDS_SAVE_PAGE_DESC_COMPLETE,
+  IDS_SAVE_PAGE_DESC_COMPLETE_SINGLE_FILE,
 };
 
 }
 
 SavePackageFilePicker::SavePackageFilePicker(
     content::WebContents* web_contents,
     const FilePath& suggested_path,
     const FilePath::StringType& default_extension,
     bool can_save_as_complete,
     DownloadPrefs* download_prefs,
@@ skipping 11 matching lines @@
   if (can_save_as_complete) {
     bool add_extra_extension = false;
     FilePath::StringType extra_extension;
     if (!suggested_path.Extension().empty() &&
         suggested_path.Extension().compare(FILE_PATH_LITERAL("htm")) &&
         suggested_path.Extension().compare(FILE_PATH_LITERAL("html"))) {
       add_extra_extension = true;
       extra_extension = suggested_path.Extension().substr(1);
     }
 
-    file_type_info.extensions.resize(2);
+    file_type_info.extensions.resize(3);

[Randy Smith (Not in Mondays), 2012/04/16 17:45:47]

nit, suggestion (i.e. up to you), while you're here: Symbolic constant?

[benjhayden, 2012/04/16 19:08:06]

Done.

+

[Randy Smith (Not in Mondays), 2012/04/16 17:45:47]

nit, while you're here: Could you put in a comment explaining why we're
sometimes offsetting by 1?  (Just something indicating that the arguments to
SelectFileDialog is index origin 1, but the extension description arrays are
index origin 0).

[benjhayden, 2012/04/16 19:08:06]

Done.

+    file_type_info.extension_description_overrides.push_back(
+        l10n_util::GetStringUTF16(kIndexToIDS[kSelectFileHtmlOnlyIndex]));

[Randy Smith (Not in Mondays), 2012/04/16 17:45:47]

nit, suggestion, while you're here: We're dealing with these two arrays
(.extensions and .extension_description) differently, using push_back for one,
and resize() + direct index for the other.  I think the resize()+direct index
option is clearer; maybe switch over to using that for both?

[benjhayden, 2012/04/16 19:08:06]

Done.

     file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1].push_back(
         FILE_PATH_LITERAL("htm"));
     file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1].push_back(
         FILE_PATH_LITERAL("html"));
-
     if (add_extra_extension) {
       file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1].push_back(
           extra_extension);
     }
 
     file_type_info.extension_description_overrides.push_back(
-        l10n_util::GetStringUTF16(kIndexToIDS[kSelectFileCompleteIndex - 1]));
+        l10n_util::GetStringUTF16(kIndexToIDS[kSelectFileCompleteIndex]));
     file_type_info.extensions[kSelectFileCompleteIndex - 1].push_back(
         FILE_PATH_LITERAL("htm"));
     file_type_info.extensions[kSelectFileCompleteIndex - 1].push_back(
         FILE_PATH_LITERAL("html"));
-
     if (add_extra_extension) {
       file_type_info.extensions[kSelectFileCompleteIndex - 1].push_back(
           extra_extension);
     }
 
     file_type_info.extension_description_overrides.push_back(
-        l10n_util::GetStringUTF16(kIndexToIDS[kSelectFileCompleteIndex]));
+        l10n_util::GetStringUTF16(kIndexToIDS[
+          kSelectFileCompleteSingleFileIndex]));
+    file_type_info.extensions[kSelectFileCompleteSingleFileIndex - 1].push_back(
+        FILE_PATH_LITERAL("mht"));
+    file_type_info.extensions[kSelectFileCompleteSingleFileIndex - 1].push_back(
+        FILE_PATH_LITERAL("mhtml"));
+    if (add_extra_extension) {
+      file_type_info.extensions[
+        kSelectFileCompleteSingleFileIndex - 1].push_back(extra_extension);
+    }
+
     file_type_info.include_all_files = false;
   } else {
     file_type_info.extensions.resize(1);
     file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1].push_back(
         suggested_path.Extension());
 
     if (!file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1][0].empty()) {
       // Drop the .
       file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1][0].erase(0, 1);
     }
@@ skipping 25 matching lines @@
 
 void SavePackageFilePicker::SetShouldPromptUser(bool should_prompt) {
   g_should_prompt_for_filename = should_prompt;
 }
 
 void SavePackageFilePicker::FileSelected(const FilePath& path,
                                          int index,
                                          void* params) {
   // The option index is not zero-based.
   DCHECK(index >= kSelectFileHtmlOnlyIndex &&
-         index <= kSelectFileCompleteIndex);
+         index <= kSelectFileCompleteSingleFileIndex);
 
   RenderProcessHost* process = RenderProcessHost::FromID(render_process_id_);
   if (process) {
     SavePageType save_type = kIndexToSaveType[index];
+    UMA_HISTOGRAM_ENUMERATION(
+        "Download.SavePageType", save_type, content::SAVE_PAGE_TYPE_MAX);
     Profile* profile =
         Profile::FromBrowserContext(process->GetBrowserContext());
     PrefService* prefs = profile->GetPrefs();
     if (select_file_dialog_ &&
         select_file_dialog_->HasMultipleFileTypeChoices())
       prefs->SetInteger(prefs::kSaveFileType, save_type);
 
     StringPrefMember save_file_path;
     save_file_path.Init(prefs::kSaveFileDefaultDirectory, prefs, NULL);
 #if defined(OS_POSIX)
@@ skipping 10 matching lines @@
 
     callback_.Run(path, save_type);
   }
 
   delete this;
 }
 
 void SavePackageFilePicker::FileSelectionCanceled(void* params) {
   delete this;
 }