chrome.experimental.downloads

chrome/browser/extensions/extension_browser_event_router.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/extensions/extension_browser_event_router.h"
 
 #include "base/json/json_writer.h"
 #include "base/values.h"
 #include "chrome/browser/extensions/extension_event_names.h"
 #include "chrome/browser/extensions/extension_event_router.h"
 #include "chrome/browser/extensions/extension_page_actions_module_constants.h"
 #include "chrome/browser/extensions/extension_tabs_module_constants.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/tabs/tab_strip_model.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/extension_constants.h"
 #include "content/browser/tab_contents/navigation_entry.h"
 #include "content/browser/tab_contents/tab_contents.h"
 #include "content/common/notification_service.h"
+#include "chrome/browser/download/download_manager.h"
+#include "chrome/browser/download/download_query.h"
 
 namespace events = extension_event_names;
 namespace tab_keys = extension_tabs_module_constants;
 namespace page_action_keys = extension_page_actions_module_constants;
 
 ExtensionBrowserEventRouter::TabEntry::TabEntry()
     : complete_waiting_on_load_(false),
       url_() {
 }
 
@@ skipping 22 matching lines @@
       tab_keys::kStatusValueLoading);
 
   if (contents->GetURL() != url_) {
     url_ = contents->GetURL();
     changed_properties->SetString(tab_keys::kUrlKey, url_.spec());
   }
 
   return changed_properties;
 }
 
+class ExtensionBrowserEventRouter::DownloadManagerObserver
+  : public DownloadManager::Observer {
+ public:
+  DownloadManagerObserver(ExtensionBrowserEventRouter* router)
+    : router_(router) {
+    DVLOG(1) << __PRETTY_FUNCTION__ << " " << this << " " << router_;
+    DCHECK(router_);
+    manager_ = router_->profile_->GetDownloadManager();
+    DCHECK(manager_);
+    manager_->AddObserver(this);
+  }
+  virtual ~DownloadManagerObserver() {
+    DVLOG(1) << __PRETTY_FUNCTION__ << " " << this << " " << router_;
+    if (manager_) manager_->RemoveObserver(this);
+  }
+
+  virtual void ModelChanged() {
+    DCHECK(manager_);
+    DVLOG(1) << __FUNCTION__ << " " << manager_ << " " << router_;
+    std::vector<DownloadItem*> items;
+    ListValue json_items;
+    manager_->Search(download_util::DownloadQuery(), &items, true, NULL, &json_items);
+    DVLOG(1) << __FUNCTION__ << " " << items.size();
+    base::hash_map<int64, DownloadItem*> updated;
+    for (std::vector<DownloadItem*>::const_iterator iter = items.begin();
+         iter != items.end(); ++iter) {
+      updated[(*iter)->id()] = *iter;
+    }
+
+    // An item is new if it's in items but not downloads_.
+    for (ListValue::const_iterator iter = json_items.begin();
+         iter != json_items.end(); ++iter) {
+      int item_id = -1;
+      DCHECK(((DictionaryValue*)(*iter))->GetInteger("id", &item_id));
+      base::hash_map<int64, DownloadItem*>::const_iterator found = downloads_.find(item_id);
+      if (found == downloads_.end()) {
+        DispatchEvent(events::kOnDownloadCreated, (*iter));
+        downloads_[item_id] = updated[item_id];
+      }
+    }
+
+    // An item was erased if it's in downloads_ but not updated.
+    for (base::hash_map<int64, DownloadItem*>::iterator iter = downloads_.begin();
+         iter != downloads_.end(); ++iter) {
+      if (updated.find(iter->first) == updated.end()) {
+        DispatchEvent(events::kOnDownloadErased, Value::CreateIntegerValue(iter->first));
+        downloads_.erase(iter);
+      }
+    }
+
+    // TODO assert that updated == downloads_
+  }
+
+  virtual void ManagerGoingDown() {
+    DVLOG(1) << __PRETTY_FUNCTION__ << " " << this << " " << router_ << " " << manager_;
+    manager_ = NULL;
+  }
+
+ private:
+  void DispatchEvent(const char* const event, Value* arg) {
+    ListValue args;
+    args.Append(arg);
+    std::string json_args;
+    base::JSONWriter::Write(&args, false, &json_args);
+    router_->DispatchEvent(router_->profile_, event, json_args);
+  }
+
+  ExtensionBrowserEventRouter* router_;
+  DownloadManager* manager_;
+  base::hash_map<int64, DownloadItem*> downloads_;
+
+  DISALLOW_COPY_AND_ASSIGN(DownloadManagerObserver);
+};
+
 void ExtensionBrowserEventRouter::Init() {
   if (initialized_)
     return;
   BrowserList::AddObserver(this);
 #if defined(TOOLKIT_VIEWS)
   views::FocusManager::GetWidgetFocusManager()->AddFocusChangeListener(this);
 #elif defined(TOOLKIT_GTK)
   ui::ActiveWindowWatcherX::AddObserver(this);
 #elif defined(OS_MACOSX)
   // Needed for when no suitable window can be passed to an extension as the
@@ skipping 11 matching lines @@
     // Also catch up our internal bookkeeping of tab entries.
     Browser* browser = *iter;
     if (browser->tabstrip_model()) {
       for (int i = 0; i < browser->tabstrip_model()->count(); ++i) {
         TabContents* contents = browser->GetTabContentsAt(i);
         int tab_id = ExtensionTabUtil::GetTabId(contents);
         tab_entries_[tab_id] = TabEntry();
       }
     }
   }
+  download_manager_observer_.reset(new DownloadManagerObserver(this));
 
   initialized_ = true;
 }
 
 ExtensionBrowserEventRouter::ExtensionBrowserEventRouter(Profile* profile)
     : initialized_(false),
       profile_(profile),
       focused_profile_(NULL),
       focused_window_id_(extension_misc::kUnknownWindowId) {
   DCHECK(!profile->IsOffTheRecord());
@@ skipping 482 matching lines @@
 
 void ExtensionBrowserEventRouter::BrowserActionExecuted(
     Profile* profile, const std::string& extension_id, Browser* browser) {
   TabContentsWrapper* tab_contents = NULL;
   int tab_id = 0;
   if (!ExtensionTabUtil::GetDefaultTab(browser, &tab_contents, &tab_id))
     return;
   DispatchEventWithTab(profile, extension_id, "browserAction.onClicked",
                        tab_contents->tab_contents(), true);
 }