Upstream version 5.34.104.0

[platform/framework/web/crosswalk.git] / src / chrome / browser / media_galleries / media_scan_manager.h

diff --git a/src/chrome/browser/media_galleries/media_scan_manager.h b/src/chrome/browser/media_galleries/media_scan_manager.h
index 5b23c29..a04e7ea 100644 (file)
--- a/src/chrome/browser/media_galleries/media_scan_manager.h
+++ b/src/chrome/browser/media_galleries/media_scan_manager.h
@@ -11,15 +11,25 @@
 
 #include "base/basictypes.h"
 #include "base/compiler_specific.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/memory/weak_ptr.h"
+#include "chrome/browser/media_galleries/media_folder_finder.h"
+#include "chrome/browser/media_galleries/media_scan_types.h"
+#include "content/public/browser/notification_observer.h"
+#include "content/public/browser/notification_registrar.h"
 
 class Profile;
 class MediaScanManagerObserver;
 
+namespace extensions {
+class Extension;
+}  // namespace extensions
+
 // The MediaScanManager is owned by MediaFileSystemRegistry, which is global.
 // This class manages multiple 'virtual' media scans, up to one per extension
 // per profile, and also manages the one physical scan backing them.
 // This class lives and is called on the UI thread.
-class MediaScanManager {
+class MediaScanManager : public content::NotificationObserver {
  public:
   MediaScanManager();
   virtual ~MediaScanManager();
@@ -35,16 +45,55 @@ class MediaScanManager {
   // The results of the scan are reported to the registered
   // MediaScanManagerObserver via OnScanFinished. There must be an observer
   // registered for |profile| before the scan starts.
-  void StartScan(Profile* profile, const std::string& extension_id);
-  void CancelScan(Profile* profile, const std::string& extension_id);
+  void StartScan(Profile* profile, const extensions::Extension* extension,
+                 bool user_gesture);
+  void CancelScan(Profile* profile, const extensions::Extension* extension);
+
+ protected:
+  friend class MediaGalleriesPlatformAppBrowserTest;
+
+  typedef base::Callback<MediaFolderFinder*(
+      const MediaFolderFinder::MediaFolderFinderResultsCallback&)>
+          MediaFolderFinderFactory;
+
+  void SetMediaFolderFinderFactory(const MediaFolderFinderFactory& factory);
 
  private:
-  typedef std::map<Profile*, std::set<std::string> > ScanningExtensionsMap;
-  typedef std::map<Profile*, MediaScanManagerObserver*> ObserverMap;
+  struct ScanObservers {
+    ScanObservers();
+    ~ScanObservers();
+    MediaScanManagerObserver* observer;
+    std::set<std::string /*extension id*/> scanning_extensions;
+  };
+  typedef std::map<Profile*, ScanObservers> ScanObserverMap;
+
+  // content::NotificationObserver implementation.
+  virtual void Observe(int type,
+                       const content::NotificationSource& source,
+                       const content::NotificationDetails& details) OVERRIDE;
+
+  bool ScanInProgress() const;
+
+  void OnScanCompleted(
+      bool success,
+      const MediaFolderFinder::MediaFolderFinderResults& found_folders);
+
+  void OnFoundContainerDirectories(
+      const MediaFolderFinder::MediaFolderFinderResults& found_folders,
+      const MediaFolderFinder::MediaFolderFinderResults& container_folders);
+
+  scoped_ptr<MediaFolderFinder> folder_finder_;
+
+  // If not NULL, used to create |folder_finder_|. Used for testing.
+  MediaFolderFinderFactory testing_folder_finder_factory_;
 
   // Set of extensions (on all profiles) that have an in-progress scan.
-  ScanningExtensionsMap scanning_extensions_;
-  ObserverMap observers_;
+  ScanObserverMap observers_;
+
+  // Used to listen for NOTIFICATION_EXTENSION_UNLOADED events.
+  content::NotificationRegistrar registrar_;
+
+  base::WeakPtrFactory<MediaScanManager> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(MediaScanManager);
 };