Redesigned and refactored ScopedMTPDeviceMapEntry, MTPDeviceMapService & MTPDeviceDelegate classes.

chrome/browser/media_gallery/media_file_system_registry.h

 // 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.
 
 // MediaFileSystemRegistry registers pictures directories and media devices as
 // File API filesystems and keeps track of the path to filesystem ID mappings.
 
 #ifndef CHROME_BROWSER_MEDIA_GALLERY_MEDIA_FILE_SYSTEM_REGISTRY_H_
 #define CHROME_BROWSER_MEDIA_GALLERY_MEDIA_FILE_SYSTEM_REGISTRY_H_
 
 #include <map>
+#include <set>
 #include <string>
 #include <utility>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/lazy_instance.h"
 #include "base/file_path.h"
 #include "base/memory/ref_counted.h"
-#include "base/memory/weak_ptr.h"
 #include "base/prefs/public/pref_change_registrar.h"
 #include "base/prefs/public/pref_observer.h"
 #include "base/system_monitor/system_monitor.h"
 #include "webkit/fileapi/media/mtp_device_file_system_config.h"
 
 class Profile;
 
 namespace content {
 class RenderViewHost;
 }
@@ skipping 26 matching lines @@
 class MediaFileSystemContext {
  public:
   virtual ~MediaFileSystemContext() {}
 
   // Register a media file system (filtered to media files) for |path| and
   // return the new file system id.
   virtual std::string RegisterFileSystemForMassStorage(
       const std::string& device_id, const FilePath& path) = 0;
 
 #if defined(SUPPORT_MTP_DEVICE_FILESYSTEM)
-  // Registers and returns the file system id for the MTP or PTP device
-  // specified by |device_id| and |path|. Updates |entry| with the corresponding
-  // ScopedMTPDeviceMapEntry object.
+  // For the given |galleries_host|, this function returns the file system id
+  // of the registered MTP or PTP device specified by |device_id| and |path|.
   virtual std::string RegisterFileSystemForMTPDevice(
       const std::string& device_id, const FilePath& path,
-      scoped_refptr<ScopedMTPDeviceMapEntry>* entry) = 0;
+      const ExtensionGalleriesHost* galleries_host) = 0;
+
+  // Removes the MTP or PTP device reference for the given |galleries_host|.
+  virtual void RemoveMTPDeviceReferenceForHost(
+      const FilePath::StringType& device_location,
+      const ExtensionGalleriesHost* galleries_host) = 0;
 #endif
 
   // Revoke the passed |fsid|.
   virtual void RevokeFileSystem(const std::string& fsid) = 0;
 };
 
 typedef base::Callback<void(const std::vector<MediaFileSystemInfo>&)>
     MediaFileSystemsCallback;
 
 class MediaFileSystemRegistry
@@ skipping 30 matching lines @@
 
   // Map an extension to the ExtensionGalleriesHost.
   typedef std::map<std::string /*extension_id*/,
                    scoped_refptr<ExtensionGalleriesHost> > ExtensionHostMap;
   // Map a profile and extension to the ExtensionGalleriesHost.
   typedef std::map<Profile*, ExtensionHostMap> ExtensionGalleriesHostMap;
   // Map a profile to a PrefChangeRegistrar.
   typedef std::map<Profile*, PrefChangeRegistrar*> PrefChangeRegistrarMap;
 
 #if defined(SUPPORT_MTP_DEVICE_FILESYSTEM)
-  // Map a MTP or PTP device location to the weak pointer of
-  // ScopedMTPDeviceMapEntry.
-  typedef std::map<const FilePath::StringType,
-                   base::WeakPtr<ScopedMTPDeviceMapEntry> >
+  // Map a MTP or PTP device location to the ScopedMTPDeviceMapEntry.
+  // This map owns ScopedMTPDeviceMapEntry.
+  typedef std::map<const FilePath::StringType, ScopedMTPDeviceMapEntry*>
       MTPDeviceDelegateMap;
+
+  // Set of extension galleries host.
+  typedef std::set<const ExtensionGalleriesHost*> ExtensionGalleriesHostSet;
+
+  // Map a device location to a set of extension galleries host.
+  // This map keeps track of extension galleries that currently registered
+  // the MTP device gallery as a file system.
+  typedef std::map<const FilePath::StringType, ExtensionGalleriesHostSet>
+      MTPDeviceReferencesMap;
 #endif
 
   // Obtain an instance of this class via GetInstance().
   MediaFileSystemRegistry();
   virtual ~MediaFileSystemRegistry();
 
   // PrefObserver implementation.
   virtual void OnPreferenceChanged(PrefServiceBase* service,
                                    const std::string& pref_name) OVERRIDE;
 
 #if defined(SUPPORT_MTP_DEVICE_FILESYSTEM)
-  // Returns ScopedMTPDeviceMapEntry object for the given |device_location|.
-  ScopedMTPDeviceMapEntry* GetOrCreateScopedMTPDeviceMapEntry(
-      const FilePath::StringType& device_location);
+  // Adds the |galleries_host| reference for the MTP device specified by the
+  // |device_location|.
+  void AddGalleriesHostReferenceForMTPDevice(
+      const FilePath::StringType& device_location,
+      const ExtensionGalleriesHost* galleries_host);
 
-  // Removes the ScopedMTPDeviceMapEntry associated with the given
+  // Removes the |galleries_host| reference for the MTP device specified by the
   // |device_location|.
-  void RemoveScopedMTPDeviceMapEntry(
-      const FilePath::StringType& device_location);
+  void RemoveGalleriesHostReferenceForMTPDevice(
+      const FilePath::StringType& device_location,
+      const ExtensionGalleriesHost* galleries_host);
 #endif
 
   void OnExtensionGalleriesHostEmpty(Profile* profile,
                                      const std::string& extension_id);
 
   // Only accessed on the UI thread. This map owns all the
   // ExtensionGalleriesHost objects created.
   ExtensionGalleriesHostMap extension_hosts_map_;
 
   PrefChangeRegistrarMap pref_change_registrar_map_;
 
 #if defined(SUPPORT_MTP_DEVICE_FILESYSTEM)
   // Only accessed on the UI thread.
-  MTPDeviceDelegateMap mtp_delegate_map_;
+  MTPDeviceDelegateMap mtp_device_delegate_map_;
+  MTPDeviceReferencesMap mtp_device_references_map_;
 #endif
 
   scoped_ptr<MediaFileSystemContext> file_system_context_;
 
   DISALLOW_COPY_AND_ASSIGN(MediaFileSystemRegistry);
 };
 
 }  // namespace chrome
 
 #endif  // CHROME_BROWSER_MEDIA_GALLERY_MEDIA_FILE_SYSTEM_REGISTRY_H_