Allow /drive files in "Report Issue" page.

chrome/browser/ui/webui/screenshot_source.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/screenshot_source.h"
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/file_util.h"
 #include "base/memory/ref_counted_memory.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/string16.h"
 #include "base/string_util.h"
 #include "chrome/browser/download/download_prefs.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/url_constants.h"
 #include "googleurl/src/url_canon.h"
 #include "googleurl/src/url_util.h"
 
 #if defined(OS_CHROMEOS)
 #include "ash/shell.h"
 #include "ash/shell_delegate.h"
+#include "chrome/browser/chromeos/gdata/gdata_file_system_interface.h"
+#include "chrome/browser/chromeos/gdata/gdata_system_service.h"
+#include "chrome/browser/chromeos/gdata/gdata_util.h"
 #include "content/public/browser/browser_thread.h"
 #endif
 
 static const char kCurrentScreenshotFilename[] = "current";
 #if defined(OS_CHROMEOS)
 static const char kSavedScreenshotsBasePath[] = "saved/";
 #endif
 
 ScreenshotSource::ScreenshotSource(
-    std::vector<unsigned char>* current_screenshot)
-    : DataSource(chrome::kChromeUIScreenshotPath, MessageLoop::current()) {
+    std::vector<unsigned char>* current_screenshot,
+    Profile* profile)
+    : DataSource(chrome::kChromeUIScreenshotPath, MessageLoop::current()),
+      profile_(profile) {
   // Setup the last screenshot taken.
   if (current_screenshot)
     current_screenshot_.reset(new ScreenshotData(*current_screenshot));
   else
     current_screenshot_.reset(new ScreenshotData());
 }
 
 ScreenshotSource::~ScreenshotSource() {}
 
 void ScreenshotSource::StartDataRequest(const std::string& path, bool,
@@ skipping 38 matching lines @@
 #endif
   } else {
     CacheAndSendScreenshot(
         path, request_id, ScreenshotDataPtr(new ScreenshotData()));
   }
 }
 
 #if defined(OS_CHROMEOS)
 void ScreenshotSource::SendSavedScreenshot(const std::string& screenshot_path,
                                            int request_id) {
-  ScreenshotDataPtr read_bytes(new ScreenshotData);
   std::string filename = screenshot_path.substr(
       strlen(kSavedScreenshotsBasePath));
 
   url_canon::RawCanonOutputT<char16> decoded;
   url_util::DecodeURLEscapeSequences(
       filename.data(), filename.size(), &decoded);
   // Screenshot filenames don't use non-ascii characters.
   std::string decoded_filename = UTF16ToASCII(string16(
       decoded.data(), decoded.length()));
 
-  int64 file_size = 0;
   DownloadPrefs* download_prefs = DownloadPrefs::FromBrowserContext(
       ash::Shell::GetInstance()->delegate()->GetCurrentBrowserContext());
-  FilePath file = download_prefs->DownloadPath().Append(decoded_filename);
-  if (!file_util::GetFileSize(file, &file_size)) {
+  FilePath download_path = download_prefs->DownloadPath();
+  if (gdata::util::IsUnderGDataMountPoint(download_path)) {
+    gdata::GDataFileSystemInterface* file_system =
+        gdata::GDataSystemServiceFactory::GetForProfile(
+            profile_)->file_system();
+    file_system->GetFileByResourceId(

[kinaba, 2012/08/15 08:19:57]

The method is DCHECKED to run on UI or IO thread, not FILE thread.

I think this file name resolution task must be done before posting
SendSavedScreenshot, and we can keep SendSavedScreenshot unchanged.

[Jun Mukai, 2012/08/15 08:30:42]

Aagh, right.  Modified to do that.

+        decoded_filename,
+        base::Bind(&ScreenshotSource::SendSavedScreenshotInternal,
+                   base::Unretained(this), screenshot_path, request_id),
+        gdata::GetContentCallback());
+  } else {
+    SendSavedScreenshotInternal(screenshot_path,
+                                request_id,
+                                gdata::GDATA_FILE_OK,
+                                download_path.Append(decoded_filename),
+                                "",
+                                gdata::REGULAR_FILE);
+  }
+
+}
+
+void ScreenshotSource::SendSavedScreenshotInternal(
+    const std::string& screenshot_path,
+    int request_id,
+    gdata::GDataFileError error,
+    const FilePath& file,
+    const std::string& mime_type,
+    gdata::GDataFileType file_type) {
+  ScreenshotDataPtr read_bytes(new ScreenshotData);
+  int64 file_size = 0;
+
+  if (error != gdata::GDATA_FILE_OK ||
+      file_type != gdata::REGULAR_FILE ||
+      !file_util::GetFileSize(file, &file_size)) {
     CacheAndSendScreenshot(screenshot_path, request_id, read_bytes);
     return;
   }
 
   read_bytes->resize(file_size);
   if (!file_util::ReadFile(file, reinterpret_cast<char*>(&read_bytes->front()),
                            static_cast<int>(file_size)))
     read_bytes->clear();
 
   CacheAndSendScreenshot(screenshot_path, request_id, read_bytes);
 }
 #endif
 
 void ScreenshotSource::CacheAndSendScreenshot(
     const std::string& screenshot_path,
     int request_id,
     ScreenshotDataPtr bytes) {
   cached_screenshots_[screenshot_path] = bytes;
   SendResponse(request_id, new base::RefCountedBytes(*bytes));
 }