Refactor downloads into a ProfileKeyedService.

chrome/browser/automation/testing_automation_provider.cc

 // Copyright (c) 2011 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/automation/testing_automation_provider.h"
 
 #include <map>
 #include <set>
 #include <string>
 #include <vector>
@@ skipping 27 matching lines @@
 #include "chrome/browser/automation/automation_util.h"
 #include "chrome/browser/automation/automation_window_tracker.h"
 #include "chrome/browser/automation/ui_controls.h"
 #include "chrome/browser/bookmarks/bookmark_model.h"
 #include "chrome/browser/bookmarks/bookmark_storage.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/browser_shutdown.h"
 #include "chrome/browser/content_settings/host_content_settings_map.h"
 #include "chrome/browser/debugger/devtools_window.h"
 #include "chrome/browser/download/download_prefs.h"
+#include "chrome/browser/download/download_service.h"
+#include "chrome/browser/download/download_service_factory.h"
 #include "chrome/browser/download/save_package_file_picker.h"
 #include "chrome/browser/extensions/extension_browser_event_router.h"
 #include "chrome/browser/extensions/extension_host.h"
 #include "chrome/browser/extensions/extension_process_manager.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_updater.h"
 #include "chrome/browser/history/top_sites.h"
 #include "chrome/browser/importer/importer_host.h"
 #include "chrome/browser/importer/importer_list.h"
 #include "chrome/browser/infobars/infobar_tab_helper.h"
@@ skipping 3013 matching lines @@
 
 // Sample json input: { "command": "GetDownloadsInfo" }
 // Refer chrome/test/pyautolib/download_info.py for sample json output.
 void TestingAutomationProvider::GetDownloadsInfo(Browser* browser,
                                                  DictionaryValue* args,
                                                  IPC::Message* reply_message) {
   AutomationJSONReply reply(this, reply_message);
   scoped_ptr<DictionaryValue> return_value(new DictionaryValue);
   ListValue* list_of_downloads = new ListValue;
 
-  if (browser->profile()->HasCreatedDownloadManager()) {
+  DownloadService* download_service(
+      DownloadServiceFactory::GetForProfile(profile()));
+
+  if (download_service->HasCreatedDownloadManager()) {
     std::vector<DownloadItem*> downloads;
-    browser->profile()->GetDownloadManager()->
+    download_service->GetDownloadManager()->
         GetAllDownloads(FilePath(), &downloads);
 
     for (std::vector<DownloadItem*>::iterator it = downloads.begin();
          it != downloads.end();
          it++) {  // Fill info about each download item.
       list_of_downloads->Append(GetDictionaryFromDownloadItem(*it));
     }
   }
   return_value->Set("downloads", list_of_downloads);
   reply.SendSuccess(return_value.get());
 }
 
 void TestingAutomationProvider::WaitForAllDownloadsToComplete(
     Browser* browser,
     DictionaryValue* args,
     IPC::Message* reply_message) {
   ListValue* pre_download_ids = NULL;
 
   if (!args->GetList("pre_download_ids", &pre_download_ids)) {
     AutomationJSONReply(this, reply_message)
         .SendError(StringPrintf("List of IDs of previous downloads required."));
     return;
   }
-  if (!browser->profile()->HasCreatedDownloadManager()) {
+
+  DownloadService* download_service =
+      DownloadServiceFactory::GetForProfile(profile());
+  if (!download_service->HasCreatedDownloadManager()) {
     // No download manager, so no downloads to wait for.
     AutomationJSONReply(this, reply_message).SendSuccess(NULL);
     return;
   }
 
   // This observer will delete itself.
   new AllDownloadsCompleteObserver(
-      this, reply_message, browser->profile()->GetDownloadManager(),
+      this, reply_message, download_service->GetDownloadManager(),
       pre_download_ids);
 }
 
 namespace {
 
 DownloadItem* GetDownloadItemFromId(int id, DownloadManager* download_manager) {
   std::vector<DownloadItem*> downloads;
   download_manager->GetAllDownloads(FilePath(), &downloads);
   DownloadItem* selected_item = NULL;
 
@@ skipping 13 matching lines @@
 
 // See PerformActionOnDownload() in chrome/test/pyautolib/pyauto.py for sample
 // json input and output.
 void TestingAutomationProvider::PerformActionOnDownload(
     Browser* browser,
     DictionaryValue* args,
     IPC::Message* reply_message) {
   int id;
   std::string action;
 
-  if (!browser->profile()->HasCreatedDownloadManager()) {
+  DownloadService* download_service =
+      DownloadServiceFactory::GetForProfile(profile());
+  if (!download_service->HasCreatedDownloadManager()) {
     AutomationJSONReply(this, reply_message).SendError("No download manager.");
     return;
   }
   if (!args->GetInteger("id", &id) || !args->GetString("action", &action)) {
     AutomationJSONReply(this, reply_message)
         .SendError("Must include int id and string action.");
     return;
   }
 
-  DownloadManager* download_manager = browser->profile()->GetDownloadManager();
+  DownloadManager* download_manager = download_service->GetDownloadManager();
   DownloadItem* selected_item = GetDownloadItemFromId(id, download_manager);
   if (!selected_item) {
     AutomationJSONReply(this, reply_message)
         .SendError(StringPrintf("No download with an id of %d\n", id));
     return;
   }
 
   if (action == "open") {
     selected_item->AddObserver(
         new AutomationProviderDownloadUpdatedObserver(
@@ skipping 508 matching lines @@
   if (!tab_contents->SavePage(
           FilePath(filename),
           FilePath(parent_directory),
           SavePackage::SAVE_AS_ONLY_HTML)) {
     AutomationJSONReply(this, reply_message).SendError(
         "Could not initiate SavePage");
     return;
   }
   // The observer will delete itself when done.
   new SavePackageNotificationObserver(
-      browser->profile()->GetDownloadManager(), this, reply_message);
+      DownloadServiceFactory::GetForProfile(profile())->GetDownloadManager(),
+      this, reply_message);
 }
 
 // Refer to ImportSettings() in chrome/test/pyautolib/pyauto.py for sample
 // json input.
 // Sample json output: "{}"
 void TestingAutomationProvider::ImportSettings(Browser* browser,
                                                DictionaryValue* args,
                                                IPC::Message* reply_message) {
   // Map from the json string passed over to the import item masks.
   std::map<std::string, importer::ImportItem> string_to_import_item;
@@ skipping 2792 matching lines @@
 
   Send(reply_message_);
   redirect_query_ = 0;
   reply_message_ = NULL;
 }
 
 void TestingAutomationProvider::OnRemoveProvider() {
   if (g_browser_process)
     g_browser_process->GetAutomationProviderList()->RemoveProvider(this);
 }