[Media Galleries] Lazily initialize the storage monitor.

chrome/browser/media_galleries/media_file_system_registry.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.
 
 // MediaFileSystemRegistry implementation.
 
 #include "chrome/browser/media_galleries/media_file_system_registry.h"
 
 #include <set>
 #include <vector>
@@ skipping 417 matching lines @@
  ******************/
 
 void MediaFileSystemRegistry::GetMediaFileSystemsForExtension(
     const content::RenderViewHost* rvh,
     const extensions::Extension* extension,
     const MediaFileSystemsCallback& callback) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   Profile* profile =
       Profile::FromBrowserContext(rvh->GetProcess()->GetBrowserContext());
-  MediaGalleriesPreferences* preferences = GetPreferences(profile);
+  GetPreferencesAsync(
+      profile, base::Bind(
+          &MediaFileSystemRegistry::GetMediaFileSystemsPostStorageMonitorInit,
+          weak_ptr_factory_.GetWeakPtr(),
+          rvh, extension, callback));
+}
+
+void MediaFileSystemRegistry::GetMediaFileSystemsPostStorageMonitorInit(
+    const content::RenderViewHost* rvh,
+    const extensions::Extension* extension,
+    const MediaFileSystemsCallback& callback,
+    MediaGalleriesPreferences* preferences) {
   MediaGalleryPrefIdSet galleries =
       preferences->GalleriesForExtension(*extension);
 
   if (galleries.empty()) {
     callback.Run(std::vector<MediaFileSystemInfo>());
     return;
   }
 
+  Profile* profile =
+      Profile::FromBrowserContext(rvh->GetProcess()->GetBrowserContext());
   if (!ContainsKey(pref_change_registrar_map_, profile)) {
     PrefChangeRegistrar* pref_registrar = new PrefChangeRegistrar;
     pref_registrar->Init(profile->GetPrefs());
     pref_registrar->Add(
         prefs::kMediaGalleriesRememberedGalleries,
         base::Bind(&MediaFileSystemRegistry::OnRememberedGalleriesChanged,
                    base::Unretained(this),
                    pref_registrar->prefs()));
     pref_change_registrar_map_[profile] = pref_registrar;
   }
 
   ExtensionGalleriesHost* extension_host =
       extension_hosts_map_[profile][extension->id()].get();
   if (!extension_host) {
     extension_host = new ExtensionGalleriesHost(
         file_system_context_.get(),
         base::Bind(&MediaFileSystemRegistry::OnExtensionGalleriesHostEmpty,
                    base::Unretained(this), profile, extension->id()));
     extension_hosts_map_[profile][extension->id()] = extension_host;
   }
   extension_host->ReferenceFromRVH(rvh);
 
   extension_host->GetMediaFileSystems(galleries, preferences->known_galleries(),
                                       callback);
 }
 
+void MediaFileSystemRegistry::GetPreferencesAsync(
+    Profile* profile,
+    base::Callback<void(MediaGalleriesPreferences*)> callback) {
+  StorageMonitor* monitor = StorageMonitor::GetInstance();
+  // TODO(gbillock): Make sure tests all have a test storage monitor
+  // and eliminate the checks here and elsewhere.
+  if (!monitor) {
+    callback.Run(MediaGalleriesPreferencesFactory::GetForProfile(profile));
+    return;
+  }
+
+  if (monitor->IsInitialized()) {
+    callback.Run(GetPreferences(profile));
+    return;
+  }
+
+  monitor->Initialize(
+      base::Bind(&MediaFileSystemRegistry::OnStorageMonitorInitialized,
+                 weak_ptr_factory_.GetWeakPtr(),
+                 profile, callback));
+
+  // TODO(gbillock): set a timeout here to bail out of init?
+}
+
+void MediaFileSystemRegistry::OnStorageMonitorInitialized(
+    Profile* profile,
+    base::Callback<void(MediaGalleriesPreferences*)> callback) {
+  callback.Run(GetPreferences(profile));
+}
+
 MediaGalleriesPreferences* MediaFileSystemRegistry::GetPreferences(
     Profile* profile) {
   MediaGalleriesPreferences* preferences =
       MediaGalleriesPreferencesFactory::GetForProfile(profile);
   if (ContainsKey(extension_hosts_map_, profile))
     return preferences;
 
   // Create an empty entry so the initialization code below only gets called
   // once per profile.
   extension_hosts_map_[profile] = ExtensionHostMap();
 
   // TODO(gbillock): Move this stanza to MediaGalleriesPreferences init code.
   // StorageMonitor may be NULL in unit tests.
   StorageMonitor* monitor = StorageMonitor::GetInstance();
   if (!monitor)
     return preferences;
+  DCHECK(monitor->IsInitialized());
   std::vector<StorageInfo> existing_devices = monitor->GetAttachedStorage();
   for (size_t i = 0; i < existing_devices.size(); i++) {
     if (!MediaStorageUtil::IsMediaDevice(existing_devices[i].device_id))
       continue;
     if (!existing_devices[i].name.empty()) {
       preferences->AddGalleryWithName(existing_devices[i].device_id,
                                       existing_devices[i].name,
                                       base::FilePath(),
                                       false /*not user added*/);
     } else {
@@ skipping 126 matching lines @@
     return registry_;
   }
 
  private:
   MediaFileSystemRegistry* registry_;
 
   DISALLOW_COPY_AND_ASSIGN(MediaFileSystemContextImpl);
 };
 
 MediaFileSystemRegistry::MediaFileSystemRegistry()
-    : file_system_context_(new MediaFileSystemContextImpl(this)) {
+    : file_system_context_(new MediaFileSystemContextImpl(this)),
+      weak_ptr_factory_(this) {
   // StorageMonitor may be NULL in unit tests.
   StorageMonitor* monitor = StorageMonitor::GetInstance();
   if (monitor)
     monitor->AddObserver(this);
 }
 
 MediaFileSystemRegistry::~MediaFileSystemRegistry() {
   // StorageMonitor may be NULL in unit tests.
   StorageMonitor* monitor = StorageMonitor::GetInstance();
   if (monitor)
@@ skipping 94 matching lines @@
     // previously used.
     PrefChangeRegistrarMap::iterator pref_it =
         pref_change_registrar_map_.find(profile);
     DCHECK(pref_it != pref_change_registrar_map_.end());
     delete pref_it->second;
     pref_change_registrar_map_.erase(pref_it);
   }
 }
 
 }  // namespace chrome