GData save package support with MHTML.

chrome/browser/download/save_package_file_picker_chromeos.cc

+// Copyright (c) 2012 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_chromeos.h"
+
+#include "base/bind.h"
+#include "base/bind_helpers.h"
+#include "base/threading/sequenced_worker_pool.h"
+#include "chrome/browser/chromeos/gdata/gdata_download_observer.h"
+#include "chrome/browser/chromeos/gdata/gdata_file_system.h"
+#include "chrome/browser/chromeos/gdata/gdata_system_service.h"
+#include "chrome/browser/chromeos/gdata/gdata_util.h"
+#include "chrome/browser/platform_util.h"
+#include "chrome/browser/profiles/profile.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/web_contents.h"
+
+using content::BrowserThread;
+
+SavePackageFilePickerChromeOS::SavePackageFilePickerChromeOS(
+    content::WebContents* web_contents,
+    const FilePath& suggested_path)
+    : content::WebContentsObserver(web_contents) {
+  select_file_dialog_ = SelectFileDialog::Create(this);
+  select_file_dialog_->SelectFile(SelectFileDialog::SELECT_SAVEAS_FILE,
+                                  string16(),
+                                  suggested_path.ReplaceExtension("mhtml"),
+                                  NULL,
+                                  0,
+                                  "mhtml",
+                                  web_contents,
+                                  platform_util::GetTopLevel(
+                                      web_contents->GetNativeView()),
+                                  NULL);
+}
+
+SavePackageFilePickerChromeOS::~SavePackageFilePickerChromeOS() {
+}
+
+void SavePackageFilePickerChromeOS::FileSelected(const FilePath& path,
+                                                 int index,
+                                                 void* params) {
+  if (!web_contents()) {
+    delete this;
+    return;
+  }
+
+  gdata::GDataFileSystem* gdata_filesystem = GetGDataFileSystem();
+  if (gdata_filesystem && gdata::util::IsUnderGDataMountPoint(path)) {
+    FilePath gdata_tmp_download_dir =
+        gdata_filesystem->GetGDataTempDownloadFolderPath();
+
+    selected_path_ = path;
+    FilePath* gdata_tmp_download_path = new FilePath();
+    BrowserThread::GetBlockingPool()->PostTaskAndReply(FROM_HERE,
+        base::Bind(&gdata::GDataDownloadObserver::GetGDataTempDownloadPath,
+                   gdata_tmp_download_dir,
+                   gdata_tmp_download_path),
+        base::Bind(&SavePackageFilePickerChromeOS::GenerateMHTML,
+                   base::Unretained(this),
+                   base::Owned(gdata_tmp_download_path)));
+  } else {
+    DVLOG(1) << "SavePackageFilePickerChromeOS non-gdata file";
+    GenerateMHTML(&path);
+  }
+}
+
+void SavePackageFilePickerChromeOS::FileSelectionCanceled(void* params) {
+  delete this;
+}
+
+void SavePackageFilePickerChromeOS::GenerateMHTML(const FilePath* mhtml_path) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  if (!web_contents()) {
+    delete this;
+    return;
+  }
+
+  DVLOG(1) << "GenerateMHTML " << mhtml_path->value();
+  web_contents()->GenerateMHTML(*mhtml_path,
+      base::Bind(&SavePackageFilePickerChromeOS::OnMHTMLGenerated,
+                 base::Unretained(this)));
+}
+
+void SavePackageFilePickerChromeOS::OnMHTMLGenerated(const FilePath& src_path,
+                                                   int64 file_size) {

[asanka, 2012/03/27 01:54:33]

Nit: Indent

+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  if (!web_contents()) {
+    delete this;
+    return;
+  }
+
+  gdata::GDataFileSystem* gdata_filesystem = GetGDataFileSystem();
+  if (gdata_filesystem && !selected_path_.empty()) {
+    DVLOG(1) << "TransferFile from " << src_path.value()
+             << " to " << selected_path_.value();
+    gdata_filesystem->TransferFile(src_path,
+        gdata::util::ExtractGDataPath(selected_path_),
+        base::Bind(&SavePackageFilePickerChromeOS::OnTransferFile,
+                   base::Unretained(this)));
+  } else {
+    delete this;
+  }
+}
+
+void SavePackageFilePickerChromeOS::OnTransferFile(
+    base::PlatformFileError error) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  DCHECK_EQ(error, base::PLATFORM_FILE_OK);
+  delete this;
+}
+
+gdata::GDataFileSystem*
+SavePackageFilePickerChromeOS::GetGDataFileSystem() {
+  DCHECK(web_contents());
+  Profile* profile = Profile::FromBrowserContext(
+      web_contents()->GetBrowserContext());
+  DCHECK(profile);
+  gdata::GDataSystemService* system_service =
+      gdata::GDataSystemServiceFactory::GetForProfile(profile);
+  // system_service is NULL in incognito.
+  return system_service ? system_service->file_system() : NULL;
+}