Media Galleries API: hookup the AllGalleries permission and add tests.

chrome/browser/media_gallery/media_file_system_registry.cc

Index: chrome/browser/media_gallery/media_file_system_registry.cc
diff --git a/chrome/browser/media_gallery/media_file_system_registry.cc b/chrome/browser/media_gallery/media_file_system_registry.cc
index 96bcf2aaa1812376165e9cd89dab991fa0f942e3..70cd3613f15ae13b5ca00230b3ece28cdfbf13f7 100644
--- a/chrome/browser/media_gallery/media_file_system_registry.cc
+++ b/chrome/browser/media_gallery/media_file_system_registry.cc
@@ -8,9 +8,12 @@
 
 #include <set>
 
+#include "base/file_path.h"
 #include "base/path_service.h"
+#include "base/system_monitor/system_monitor.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/common/chrome_paths.h"
+#include "chrome/common/extensions/extension.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/notification_source.h"
 #include "content/public/browser/notification_types.h"
@@ -27,6 +30,22 @@ using content::BrowserThread;
 using content::RenderProcessHost;
 using fileapi::IsolatedContext;
 
+namespace {
+
+bool IsGalleryPermittedForExtension(const extensions::Extension& extension,
+                                    SystemMonitor::MediaDeviceType type,
+                                    const FilePath::StringType& location) {
+  if (extension.HasAPIPermission(
+        extensions::APIPermission::kMediaGalleriesAllGalleries)) {
+    return true;
+  }
+  // TODO(vandebo) Check with prefs for permission to this gallery.
+  return false;
+}
+
+}  // namespace
+
+
 /******************
  * Public methods
  ******************/
@@ -38,7 +57,8 @@ MediaFileSystemRegistry* MediaFileSystemRegistry::GetInstance() {
 
 std::vector<MediaFileSystemRegistry::MediaFSIDAndPath>
 MediaFileSystemRegistry::GetMediaFileSystems(
-    const content::RenderProcessHost* rph) {
+    const content::RenderProcessHost* rph,
+    const extensions::Extension& extension) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   std::vector<MediaFSIDAndPath> results;
@@ -50,7 +70,10 @@ MediaFileSystemRegistry::GetMediaFileSystems(
     // file system mappings.
     RegisterForRPHGoneNotifications(rph);
     FilePath pictures_path;
-    if (PathService::Get(chrome::DIR_USER_PICTURES, &pictures_path)) {
+    // TODO(vandebo) file system galleries need a unique id as well.
+    if (PathService::Get(chrome::DIR_USER_PICTURES, &pictures_path) &&
+        IsGalleryPermittedForExtension(extension, SystemMonitor::TYPE_PATH,
+                                       pictures_path.value())) {
       std::string fsid = RegisterPathAsFileSystem(pictures_path);
       child_it->second.insert(std::make_pair(pictures_path, fsid));
     }
@@ -61,7 +84,9 @@ MediaFileSystemRegistry::GetMediaFileSystems(
   const std::vector<SystemMonitor::MediaDeviceInfo> media_devices =
       monitor->GetAttachedMediaDevices();
   for (size_t i = 0; i < media_devices.size(); ++i) {
-    if (media_devices[i].type == SystemMonitor::TYPE_PATH) {
+    if (media_devices[i].type == SystemMonitor::TYPE_PATH &&
+        IsGalleryPermittedForExtension(extension, media_devices[i].type,
+                                       media_devices[i].location)) {
       FilePath path(media_devices[i].location);
       device_id_map_.insert(std::make_pair(media_devices[i].unique_id, path));
       const std::string fsid = RegisterPathAsFileSystem(path);