Media Galleries API Picasa: Make PicasaFileUtil support album contents.

chrome/browser/media_galleries/fileapi/picasa/picasa_file_util.cc

 // Copyright 2013 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/media_galleries/fileapi/picasa/picasa_file_util.h"
 
 #include <string>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/bind_helpers.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/media_galleries/fileapi/media_file_system_backend.h"
 #include "chrome/browser/media_galleries/fileapi/picasa/picasa_data_provider.h"
 #include "chrome/browser/media_galleries/imported_media_gallery_registry.h"
 #include "chrome/common/media_galleries/picasa_types.h"
 #include "content/public/browser/browser_thread.h"
 #include "webkit/browser/fileapi/file_system_operation_context.h"
 #include "webkit/browser/fileapi/file_system_url.h"
+#include "webkit/browser/fileapi/native_file_util.h"
 #include "webkit/common/fileapi/file_system_util.h"
 
 using base::FilePath;
 using fileapi::DirectoryEntry;
 using fileapi::FileSystemOperationContext;
 using fileapi::FileSystemURL;
 
 namespace picasa {
 
 namespace {
 
-// |error| is only set when the method fails and the return is NULL.
 base::PlatformFileError FindAlbumInfo(const std::string& key,
                                       const AlbumMap* map,
                                       AlbumInfo* album_info) {
   if (!map)
     return base::PLATFORM_FILE_ERROR_FAILED;
 
   AlbumMap::const_iterator it = map->find(key);
 
   if (it == map->end())
     return base::PLATFORM_FILE_ERROR_NOT_FOUND;
 
   if (album_info != NULL)
     *album_info = it->second;
 
   return base::PLATFORM_FILE_OK;
 }
 
+// This method returns a pointer to the AlbumImages instead of a

[Greg Billock, 2013/08/21 17:13:17]

You could have const AlbumImages** as an outparam.

[tommycli, 2013/08/21 21:37:06]

I moved it to the data provider. Can't do AlbumImages** there because need to
make a copy of the struct.

+// base::PlatformFileError because copying an AlbumImages is expensive.
+const AlbumImages* FindAlbumImages(const std::string& key,

[Greg Billock, 2013/08/21 17:13:17]

Should this be a method in the data provider?

[tommycli, 2013/08/21 21:37:06]

Yes, that's actually much better. I put those changes into the ancestor CL:
https://codereview.chromium.org/18986012/

+                                   const AlbumImagesMap* map,
+                                   base::PlatformFileError* error) {
+  DCHECK(error);
+
+  if (!map) {

[Greg Billock, 2013/08/21 17:13:17]

should this be DCHECK?

[tommycli, 2013/08/21 21:37:06]

Done.

+    *error = base::PLATFORM_FILE_ERROR_FAILED;
+    return NULL;
+  }
+
+  AlbumImagesMap::const_iterator it = map->find(key);
+
+  if (it == map->end()) {
+    *error = base::PLATFORM_FILE_ERROR_NOT_FOUND;
+    return NULL;
+  }
+
+  *error = base::PLATFORM_FILE_OK;
+  return &(it->second);
+}
+
+PicasaDataProvider::DataType GetNeededDataType(
+    const fileapi::FileSystemURL& url) {
+  std::vector<std::string> components;
+  fileapi::VirtualPath::GetComponentsUTF8Unsafe(url.path(), &components);
+
+  if (components.size() >= 2 && components[0] == kPicasaDirAlbums)

[Greg Billock, 2013/08/21 17:13:17]

How about adding a comment to PicasaFileUtil explaining how the virtual FS is
laid out. I'm getting a /albums/album-name/filename impression, but I had to
work for it. :-)

[tommycli, 2013/08/21 21:37:06]

Done. Good idea.

+    return PicasaDataProvider::ALBUMS_IMAGES_DATA;
+
+  return PicasaDataProvider::LIST_OF_ALBUMS_AND_FOLDERS_DATA;
+}
+
 }  // namespace
 
 const char kPicasaDirAlbums[]  = "albums";
 const char kPicasaDirFolders[] = "folders";
 
 PicasaFileUtil::PicasaFileUtil(chrome::MediaPathFilter* media_path_filter)
     : chrome::NativeMediaFileUtil(media_path_filter),
       weak_factory_(this) {
 }
 
 PicasaFileUtil::~PicasaFileUtil() {}
 
 void PicasaFileUtil::GetFileInfoOnTaskRunnerThread(
     scoped_ptr<fileapi::FileSystemOperationContext> context,
     const fileapi::FileSystemURL& url,
     const GetFileInfoCallback& callback) {
   GetDataProvider()->RefreshData(
-      PicasaDataProvider::LIST_OF_ALBUMS_AND_FOLDERS_DATA,
+      GetNeededDataType(url),
       base::Bind(&PicasaFileUtil::GetFileInfoWithFreshDataProvider,
                  weak_factory_.GetWeakPtr(),
                  base::Passed(&context),
                  url,
                  callback));
 }
 
 void PicasaFileUtil::ReadDirectoryOnTaskRunnerThread(
     scoped_ptr<fileapi::FileSystemOperationContext> context,
     const fileapi::FileSystemURL& url,
     const ReadDirectoryCallback& callback) {
   GetDataProvider()->RefreshData(
-      PicasaDataProvider::LIST_OF_ALBUMS_AND_FOLDERS_DATA,
+      GetNeededDataType(url),

[Greg Billock, 2013/08/21 17:13:17]

GetDataTypeForUrl ?

[tommycli, 2013/08/21 21:37:06]

Done.

       base::Bind(&PicasaFileUtil::ReadDirectoryWithFreshDataProvider,
                  weak_factory_.GetWeakPtr(),
                  base::Passed(&context),
                  url,
                  callback));
 }
 
 base::PlatformFileError PicasaFileUtil::GetFileInfoSync(
     FileSystemOperationContext* context, const FileSystemURL& url,
     base::PlatformFileInfo* file_info, base::FilePath* platform_path) {
@@ skipping 106 matching lines @@
         for (AlbumMap::const_iterator it = folders->begin();
              it != folders->end(); ++it) {
           file_list->push_back(
               DirectoryEntry(it->first, DirectoryEntry::DIRECTORY, 0,
                              it->second.timestamp));
         }
       }
       break;
     case 2:
       if (components[0] == kPicasaDirAlbums) {
-        // TODO(tommycli): Implement album contents.
+        scoped_ptr<AlbumMap> album_map = GetDataProvider()->GetAlbums();
+        AlbumInfo album_info;
+        base::PlatformFileError error =
+            FindAlbumInfo(components[1], album_map.get(), &album_info);
+        if (error != base::PLATFORM_FILE_OK)
+          return error;
+
+        scoped_ptr<AlbumImagesMap> album_images_map =
+            GetDataProvider()->GetAlbumsImages();
+        const AlbumImages* album_images =
+            FindAlbumImages(album_info.uid, album_images_map.get(), &error);
+        if (error != base::PLATFORM_FILE_OK)
+          return error;
+
+        for (AlbumImages::const_iterator it = album_images->begin();
+             it != album_images->end();
+             ++it) {
+          fileapi::DirectoryEntry entry;
+          base::PlatformFileInfo info;
+
+          // Simply skip files that we can't get info on.
+          if (fileapi::NativeFileUtil::GetFileInfo(it->second, &info) !=
+              base::PLATFORM_FILE_OK) {
+            continue;
+          }
+
+          file_list->push_back(DirectoryEntry(
+              it->first, DirectoryEntry::FILE, info.size, info.last_modified));
+        }
       }
 
       if (components[0] == kPicasaDirFolders) {
         EntryList super_list;
         base::PlatformFileError error =
             NativeMediaFileUtil::ReadDirectorySync(context, url, &super_list);
         if (error != base::PLATFORM_FILE_OK)
           return error;
 
         for (EntryList::const_iterator it = super_list.begin();
@@ skipping 27 matching lines @@
         if (error != base::PLATFORM_FILE_OK)
           return error;
 
         *local_file_path = album_info.path;
         return base::PLATFORM_FILE_OK;
       }
       break;
     case 3:
       if (components[0] == kPicasaDirAlbums) {
         scoped_ptr<AlbumMap> album_map = GetDataProvider()->GetAlbums();
+        AlbumInfo album_info;
         base::PlatformFileError error =
-            FindAlbumInfo(components[1], album_map.get(), NULL);
+            FindAlbumInfo(components[1], album_map.get(), &album_info);
         if (error != base::PLATFORM_FILE_OK)
           return error;
 
-        // TODO(tommycli): Implement album contents.
-        return base::PLATFORM_FILE_ERROR_NOT_FOUND;
+        scoped_ptr<AlbumImagesMap> album_images_map =
+            GetDataProvider()->GetAlbumsImages();

[Greg Billock, 2013/08/21 17:13:17]

Is this file missing from the change? I don't see GetAlbumsImages in the class.

[tommycli, 2013/08/21 21:37:06]

Yes, this CL depends on https://codereview.chromium.org/18986012/. I updated the
description. Sorry.

+        const AlbumImages* album_images =
+            FindAlbumImages(album_info.uid, album_images_map.get(), &error);
+        if (error != base::PLATFORM_FILE_OK)
+          return error;
+
+        AlbumImages::const_iterator it = album_images->find(components[2]);
+        if (it == album_images->end())
+          return base::PLATFORM_FILE_ERROR_NOT_FOUND;
+
+        *local_file_path = it->second;
+        return base::PLATFORM_FILE_OK;
       }
 
       if (components[0] == kPicasaDirFolders) {
         scoped_ptr<AlbumMap> album_map = GetDataProvider()->GetFolders();
         AlbumInfo album_info;
         base::PlatformFileError error =
             FindAlbumInfo(components[1], album_map.get(), &album_info);
         if (error != base::PLATFORM_FILE_OK)
           return error;
 
@@ skipping 43 matching lines @@
   }
   NativeMediaFileUtil::ReadDirectoryOnTaskRunnerThread(
       context.Pass(), url, callback);
 }
 
 PicasaDataProvider* PicasaFileUtil::GetDataProvider() {
   return chrome::ImportedMediaGalleryRegistry::PicasaDataProvider();
 }
 
 }  // namespace picasa