gdata: Fix "save as pdf" to work on Google Drive.

chrome/browser/ui/webui/print_preview/print_preview_handler.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/ui/webui/print_preview/print_preview_handler.h"
 
 #include <ctype.h>
 
 #include <string>
 #include <vector>
@@ skipping 41 matching lines @@
 #include "content/public/browser/web_contents_delegate.h"
 #include "content/public/browser/web_ui.h"
 #include "printing/backend/print_backend.h"
 #include "printing/metafile.h"
 #include "printing/metafile_impl.h"
 #include "printing/page_range.h"
 #include "printing/page_size_margins.h"
 #include "printing/print_settings.h"
 #include "unicode/ulocdata.h"
 
+#ifdef OS_CHROMEOS
+#include "chrome/browser/chromeos/gdata/gdata_util.h"
+#endif
+
 #if !defined(OS_MACOSX)
 #include "base/command_line.h"
 #include "chrome/common/chrome_switches.h"
 #endif
 
 using content::BrowserThread;
 using content::NavigationEntry;
 using content::OpenURLParams;
 using content::RenderViewHost;
 using content::Referrer;
@@ skipping 134 matching lines @@
 
 // Callback that stores a PDF file on disk.
 void PrintToPdfCallback(Metafile* metafile, const FilePath& path) {
   metafile->SaveTo(path);
   // |metafile| must be deleted on the UI thread.
   BrowserThread::PostTask(
       BrowserThread::UI, FROM_HERE,
       base::Bind(&base::DeletePointer<Metafile>, metafile));
 }
 
+#ifdef OS_CHROMEOS
+void PrintToPdfCallbackWithCheck(Metafile* metafile,
+                                 gdata::GDataFileError error,
+                                 const FilePath& path) {
+  if (error != gdata::GDATA_FILE_OK) {
+    LOG(ERROR) << "Save to pdf failed to write: " << error;
+  } else {
+    metafile->SaveTo(path);
+  }
+  // |metafile| must be deleted on the UI thread.
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      base::Bind(&base::DeletePointer<Metafile>, metafile));
+}
+#endif
+
 static base::LazyInstance<printing::StickySettings> sticky_settings =
     LAZY_INSTANCE_INITIALIZER;
 
 }  // namespace
 
 // static
 printing::StickySettings* PrintPreviewHandler::GetStickySettings() {
   return sticky_settings.Pointer();
 }
 
@@ skipping 678 matching lines @@
 }
 
 void PrintPreviewHandler::PostPrintToPdfTask(base::RefCountedBytes* data) {
   if (!data) {
     NOTREACHED();
     return;
   }
   printing::PreviewMetafile* metafile = new printing::PreviewMetafile;
   metafile->InitFromData(static_cast<const void*>(data->front()), data->size());
   // PrintToPdfCallback takes ownership of |metafile|.
+#ifdef OS_CHROMEOS

[satorux1, 2012/08/02 06:59:16]

on what thread this function is called? can you add a DCHECK?

[kinaba, 2012/08/02 07:22:39]

Done.

+  gdata::util::PrepareWritableFileAndRun(
+      Profile::FromBrowserContext(preview_web_contents()->GetBrowserContext()),
+      *print_to_pdf_path_,
+      base::Bind(&PrintToPdfCallbackWithCheck, metafile));
+#else
   BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE,
                           base::Bind(&PrintToPdfCallback, metafile,
                                      *print_to_pdf_path_));
+#endif
+
   print_to_pdf_path_.reset();
   ActivateInitiatorTabAndClosePreviewTab();
 }
 
 void PrintPreviewHandler::FileSelectionCanceled(void* params) {
   PrintPreviewUI* print_preview_ui = static_cast<PrintPreviewUI*>(
       web_ui()->GetController());
   print_preview_ui->OnFileSelectionCancelled();
 }
 
 void PrintPreviewHandler::ClearInitiatorTabDetails() {
   TabContents* initiator_tab = GetInitiatorTab();
   if (!initiator_tab)
     return;
 
   // We no longer require the initiator tab details. Remove those details
   // associated with the preview tab to allow the initiator tab to create
   // another preview tab.
   printing::PrintPreviewTabController* tab_controller =
       printing::PrintPreviewTabController::GetInstance();
   if (tab_controller)
     tab_controller->EraseInitiatorTabInfo(preview_tab_contents());
 }