Add a Create method to DesktopNotificationHandler and stubs for the notification objects.

chrome/browser/task_manager/task_manager_resource_providers.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/task_manager/task_manager_resource_providers.h"
 
 #include "base/basictypes.h"
 #include "base/file_version_info.h"
 #include "base/i18n/rtl.h"
 #include "base/process_util.h"
 #include "base/stl_util-inl.h"
 #include "base/string_util.h"
 #include "base/threading/thread.h"
 #include "base/utf_string_conversions.h"
 #include "build/build_config.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "chrome/browser/background_contents_service.h"
 #include "chrome/browser/background_contents_service_factory.h"
 #include "chrome/browser/browser_process.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/favicon/favicon_tab_helper.h"
-#include "chrome/browser/notifications/balloon_collection.h"
-#include "chrome/browser/notifications/balloon_host.h"
-#include "chrome/browser/notifications/notification_ui_manager.h"
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/tab_contents/background_contents.h"
 #include "chrome/browser/tab_contents/tab_util.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/render_messages.h"
 #include "content/browser/browser_child_process_host.h"
@@ skipping 1138 matching lines @@
       pid_to_resources_.find(resource->process_id());
   DCHECK(pid_iter != pid_to_resources_.end());
   if (pid_iter != pid_to_resources_.end())
     pid_to_resources_.erase(pid_iter);
 
   // Finally, delete the resource.
   delete resource;
 }
 
 ////////////////////////////////////////////////////////////////////////////////
-// TaskManagerNotificationResource class
-////////////////////////////////////////////////////////////////////////////////
-
-SkBitmap* TaskManagerNotificationResource::default_icon_ = NULL;
-
-TaskManagerNotificationResource::TaskManagerNotificationResource(
-    BalloonHost* balloon_host)
-    : balloon_host_(balloon_host) {
-  if (!default_icon_) {
-    ResourceBundle& rb = ResourceBundle::GetSharedInstance();
-    default_icon_ = rb.GetBitmapNamed(IDR_PLUGIN);
-  }
-  process_handle_ = balloon_host_->render_view_host()->process()->GetHandle();
-  pid_ = base::GetProcId(process_handle_);
-  title_ = l10n_util::GetStringFUTF16(IDS_TASK_MANAGER_NOTIFICATION_PREFIX,
-                                      balloon_host_->GetSource());
-}
-
-TaskManagerNotificationResource::~TaskManagerNotificationResource() {
-}
-
-string16 TaskManagerNotificationResource::GetTitle() const {
-  return title_;
-}
-
-SkBitmap TaskManagerNotificationResource::GetIcon() const {
-  return *default_icon_;
-}
-
-base::ProcessHandle TaskManagerNotificationResource::GetProcess() const {
-  return process_handle_;
-}
-
-TaskManager::Resource::Type TaskManagerNotificationResource::GetType() const {
-  return NOTIFICATION;
-}
-
-bool TaskManagerNotificationResource::SupportNetworkUsage() const {
-  return false;
-}
-
-////////////////////////////////////////////////////////////////////////////////
-// TaskManagerNotificationResourceProvider class
-////////////////////////////////////////////////////////////////////////////////
-
-TaskManagerNotificationResourceProvider::
-    TaskManagerNotificationResourceProvider(TaskManager* task_manager)
-    : task_manager_(task_manager),
-      updating_(false) {
-}
-
-TaskManagerNotificationResourceProvider::
-    ~TaskManagerNotificationResourceProvider() {
-}
-
-TaskManager::Resource* TaskManagerNotificationResourceProvider::GetResource(
-    int origin_pid,
-    int render_process_host_id,
-    int routing_id) {
-  // TODO(johnnyg): provide resources by pid if necessary.
-  return NULL;
-}
-
-void TaskManagerNotificationResourceProvider::StartUpdating() {
-  DCHECK(!updating_);
-  updating_ = true;
-
-  // Add all the existing BalloonHosts.
-  BalloonCollection* collection =
-      g_browser_process->notification_ui_manager()->balloon_collection();
-  const BalloonCollection::Balloons& balloons = collection->GetActiveBalloons();
-  for (BalloonCollection::Balloons::const_iterator it = balloons.begin();
-       it != balloons.end(); ++it) {
-    AddToTaskManager((*it)->view()->GetHost());
-  }
-
-  // Register for notifications about extension process changes.
-  registrar_.Add(this, NotificationType::NOTIFY_BALLOON_CONNECTED,
-                 NotificationService::AllSources());
-  registrar_.Add(this, NotificationType::NOTIFY_BALLOON_DISCONNECTED,
-                 NotificationService::AllSources());
-}
-
-void TaskManagerNotificationResourceProvider::StopUpdating() {
-  DCHECK(updating_);
-  updating_ = false;
-
-  // Unregister for notifications about extension process changes.
-  registrar_.Remove(this, NotificationType::NOTIFY_BALLOON_CONNECTED,
-                    NotificationService::AllSources());
-  registrar_.Remove(this, NotificationType::NOTIFY_BALLOON_DISCONNECTED,
-                    NotificationService::AllSources());
-
-  // Delete all the resources.
-  STLDeleteContainerPairSecondPointers(resources_.begin(), resources_.end());
-  resources_.clear();
-}
-
-void TaskManagerNotificationResourceProvider::Observe(
-    NotificationType type,
-    const NotificationSource& source,
-    const NotificationDetails& details) {
-  switch (type.value) {
-    case NotificationType::NOTIFY_BALLOON_CONNECTED:
-      AddToTaskManager(Source<BalloonHost>(source).ptr());
-      break;
-    case NotificationType::NOTIFY_BALLOON_DISCONNECTED:
-      RemoveFromTaskManager(Source<BalloonHost>(source).ptr());
-      break;
-    default:
-      NOTREACHED() << "Unexpected notification.";
-      return;
-  }
-}
-
-void TaskManagerNotificationResourceProvider::AddToTaskManager(
-    BalloonHost* balloon_host) {
-  TaskManagerNotificationResource* resource =
-      new TaskManagerNotificationResource(balloon_host);
-  DCHECK(resources_.find(balloon_host) == resources_.end());
-  resources_[balloon_host] = resource;
-  task_manager_->AddResource(resource);
-}
-
-void TaskManagerNotificationResourceProvider::RemoveFromTaskManager(
-    BalloonHost* balloon_host) {
-  if (!updating_)
-    return;
-  std::map<BalloonHost*, TaskManagerNotificationResource*>::iterator iter =
-      resources_.find(balloon_host);
-  if (iter == resources_.end())
-    return;
-
-  // Remove the resource from the Task Manager.
-  TaskManagerNotificationResource* resource = iter->second;
-  task_manager_->RemoveResource(resource);
-
-  // Remove it from the map.
-  resources_.erase(iter);
-
-  // Finally, delete the resource.
-  delete resource;
-}
-
-////////////////////////////////////////////////////////////////////////////////
 // TaskManagerBrowserProcessResource class
 ////////////////////////////////////////////////////////////////////////////////
 
 SkBitmap* TaskManagerBrowserProcessResource::default_icon_ = NULL;
 
 TaskManagerBrowserProcessResource::TaskManagerBrowserProcessResource()
     : title_() {
   int pid = base::GetCurrentProcId();
   bool success = base::OpenPrivilegedProcessHandle(pid, &process_);
   DCHECK(success);
@@ skipping 114 matching lines @@
 
   return &resource_;
 }
 
 void TaskManagerBrowserProcessResourceProvider::StartUpdating() {
   task_manager_->AddResource(&resource_);
 }
 
 void TaskManagerBrowserProcessResourceProvider::StopUpdating() {
 }