[Media Galleries] Lazily initialize the storage monitor.

chrome/browser/ui/webui/options/media_galleries_handler.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.
 
 #include "chrome/browser/ui/webui/options/media_galleries_handler.h"
 
 #include "base/bind.h"
 #include "base/prefs/pref_service.h"
 #include "base/strings/string_number_conversions.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/media_galleries/media_file_system_registry.h"
 #include "chrome/browser/media_galleries/media_galleries_preferences.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/storage_monitor/storage_monitor.h"
 #include "chrome/browser/ui/chrome_select_file_policy.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/pref_names.h"
 #include "content/public/browser/notification_details.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_view.h"
 #include "content/public/browser/web_ui.h"
 #include "grit/generated_resources.h"
 
 namespace options {
 
 using chrome::MediaGalleriesPreferences;
 using chrome::MediaGalleriesPrefInfoMap;
 using chrome::MediaGalleryPrefInfo;
 
-MediaGalleriesHandler::MediaGalleriesHandler() {
+MediaGalleriesHandler::MediaGalleriesHandler()
+    : weak_ptr_factory_(this) {
 }
 
 MediaGalleriesHandler::~MediaGalleriesHandler() {
   if (select_file_dialog_.get())
     select_file_dialog_->ListenerDestroyed();
 }
 
 void MediaGalleriesHandler::GetLocalizedValues(DictionaryValue* values) {
   DCHECK(values);
 
   static OptionsStringResource resources[] = {
     { "mediaGalleriesSectionLabel", IDS_MEDIA_GALLERY_SECTION_LABEL },
     { "manageGalleriesButton", IDS_MEDIA_GALLERY_MANAGE_BUTTON },
     { "addNewGalleryButton", IDS_MEDIA_GALLERY_ADD_NEW_BUTTON },
   };
 
   RegisterStrings(values, resources, arraysize(resources));
   RegisterTitle(values, "manageMediaGalleries",
                 IDS_MEDIA_GALLERY_MANAGE_TITLE);
 }
 
 void MediaGalleriesHandler::InitializePage() {
+  chrome::StorageMonitor::GetInstance()->Initialize(
+      base::Bind(&MediaGalleriesHandler::InitializeOnStorageMonitorInit,
+                 weak_ptr_factory_.GetWeakPtr()));
+}
+
+void MediaGalleriesHandler::InitializeOnStorageMonitorInit() {
   Profile* profile = Profile::FromWebUI(web_ui());
   if (!chrome::MediaGalleriesPreferences::APIHasBeenUsed(profile))
     return;
 
   if (pref_change_registrar_.IsEmpty()) {
     pref_change_registrar_.Init(profile->GetPrefs());
     pref_change_registrar_.Add(
         prefs::kMediaGalleriesRememberedGalleries,
         base::Bind(&MediaGalleriesHandler::OnGalleriesChanged,
                    base::Unretained(this)));
   }
 
   OnGalleriesChanged();
 }
 
 void MediaGalleriesHandler::RegisterMessages() {
+  chrome::StorageMonitor::GetInstance()->Initialize(
+      base::Bind(&MediaGalleriesHandler::RegisterOnStorageMonitorInit,
+                 weak_ptr_factory_.GetWeakPtr()));
+}
+
+void MediaGalleriesHandler::RegisterOnStorageMonitorInit() {
   web_ui()->RegisterMessageCallback(
       "addNewGallery",
       base::Bind(&MediaGalleriesHandler::HandleAddNewGallery,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback(
       "forgetGallery",
       base::Bind(&MediaGalleriesHandler::HandleForgetGallery,
                  base::Unretained(this)));
 }
 
 void MediaGalleriesHandler::OnGalleriesChanged() {
+  DCHECK(chrome::StorageMonitor::GetInstance()->IsInitialized());
   Profile* profile = Profile::FromWebUI(web_ui());
   chrome::MediaGalleriesPreferences* preferences =
       g_browser_process->media_file_system_registry()->GetPreferences(profile);
 
   ListValue list;
   const MediaGalleriesPrefInfoMap& galleries = preferences->known_galleries();
   for (MediaGalleriesPrefInfoMap::const_iterator iter = galleries.begin();
        iter != galleries.end(); ++iter) {
     const MediaGalleryPrefInfo& gallery = iter->second;
     if (gallery.type == MediaGalleryPrefInfo::kBlackListed)
@@ skipping 27 matching lines @@
 
 void MediaGalleriesHandler::HandleForgetGallery(const base::ListValue* args) {
   std::string string_id;
   uint64 id = 0;
   if (!args->GetString(0, &string_id) ||
       !base::StringToUint64(string_id, &id)) {
     NOTREACHED();
     return;
   }
 
-  chrome::MediaGalleriesPreferences* prefs =
+  DCHECK(chrome::StorageMonitor::GetInstance()->IsInitialized());
+  chrome::MediaGalleriesPreferences* preferences =
       g_browser_process->media_file_system_registry()->GetPreferences(
           Profile::FromWebUI(web_ui()));
-  prefs->ForgetGalleryById(id);
+  preferences->ForgetGalleryById(id);
 }
 
 void MediaGalleriesHandler::FileSelected(const base::FilePath& path,
                                          int index,
                                          void* params) {
-  chrome::MediaGalleriesPreferences* prefs =
+  DCHECK(chrome::StorageMonitor::GetInstance()->IsInitialized());
+  chrome::MediaGalleriesPreferences* preferences =
       g_browser_process->media_file_system_registry()->GetPreferences(
           Profile::FromWebUI(web_ui()));
-  prefs->AddGalleryByPath(path);
+  preferences->AddGalleryByPath(path);
 }
 
 }  // namespace options