Pull shell window stuff out of ExtensionHost and put in ShellWindow

chrome/browser/task_manager/task_manager_resource_providers.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/task_manager/task_manager_resource_providers.h"
 
 #include <string>
 
 #include "base/basictypes.h"
 #include "base/bind.h"
@@ skipping 1241 matching lines @@
     if (process_manager) {
       const ExtensionProcessManager::ViewSet all_views =
           process_manager->GetAllViews();
       ExtensionProcessManager::ViewSet::const_iterator jt = all_views.begin();
       for (; jt != all_views.end(); ++jt) {
         content::RenderViewHost* rvh = *jt;
         // Don't add dead extension processes.
         if (!rvh->IsRenderViewLive())
           continue;
 
-        // Don't add WebContents (those are handled by
-        // TaskManagerTabContentsResourceProvider) or background contents
-        // (handled by TaskManagerBackgroundResourceProvider).
-        // TODO(benwells): create specific chrome::VIEW_TYPE_TAB_CONTENTS for
-        // tab contents, as VIEW_TYPE_WEB_CONTENTS is the default.
-        content::ViewType view_type = rvh->GetDelegate()->GetRenderViewType();
-        if (view_type == content::VIEW_TYPE_WEB_CONTENTS ||
-            view_type == chrome::VIEW_TYPE_BACKGROUND_CONTENTS) {
-          continue;
-        }
-
         AddToTaskManager(rvh);
       }
     }
   }
 
   // Register for notifications about extension process changes.
-  registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_HOST_CREATED,
+  registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_VIEW_REGISTERED,
                  content::NotificationService::AllBrowserContextsAndSources());
   registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_PROCESS_TERMINATED,
                  content::NotificationService::AllBrowserContextsAndSources());
-  registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_HOST_DESTROYED,
+  registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_VIEW_UNREGISTERED,
                  content::NotificationService::AllBrowserContextsAndSources());
 }
 
 void TaskManagerExtensionProcessResourceProvider::StopUpdating() {
   DCHECK(updating_);
   updating_ = false;
 
   // Unregister for notifications about extension process changes.
   registrar_.Remove(
-      this, chrome::NOTIFICATION_EXTENSION_HOST_CREATED,
+      this, chrome::NOTIFICATION_EXTENSION_VIEW_REGISTERED,
       content::NotificationService::AllBrowserContextsAndSources());
   registrar_.Remove(
       this, chrome::NOTIFICATION_EXTENSION_PROCESS_TERMINATED,
       content::NotificationService::AllBrowserContextsAndSources());
   registrar_.Remove(
-      this, chrome::NOTIFICATION_EXTENSION_HOST_DESTROYED,
+      this, chrome::NOTIFICATION_EXTENSION_VIEW_UNREGISTERED,
       content::NotificationService::AllBrowserContextsAndSources());
 
   // Delete all the resources.
   STLDeleteContainerPairSecondPointers(resources_.begin(), resources_.end());
 
   resources_.clear();
   pid_to_resources_.clear();
 }
 
 void TaskManagerExtensionProcessResourceProvider::Observe(
     int type,
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   switch (type) {
-    case chrome::NOTIFICATION_EXTENSION_HOST_CREATED:
+    case chrome::NOTIFICATION_EXTENSION_VIEW_REGISTERED:
       AddToTaskManager(
+          content::Details<content::RenderViewHost>(details).ptr());
+      break;
+    case chrome::NOTIFICATION_EXTENSION_PROCESS_TERMINATED:
+      RemoveFromTaskManager(
           content::Details<ExtensionHost>(details).ptr()->render_view_host());
       break;
-    case chrome::NOTIFICATION_EXTENSION_PROCESS_TERMINATED:
-    case chrome::NOTIFICATION_EXTENSION_HOST_DESTROYED:
+    case chrome::NOTIFICATION_EXTENSION_VIEW_UNREGISTERED:
       RemoveFromTaskManager(
-          content::Details<ExtensionHost>(details).ptr()->render_view_host());
+          content::Details<content::RenderViewHost>(details).ptr());
       break;
     default:
       NOTREACHED() << "Unexpected notification.";
       return;
   }
 }
 
+bool TaskManagerExtensionProcessResourceProvider::
+    IsHandledByThisProvider(content::RenderViewHost* render_view_host) {
+  // Don't add WebContents (those are handled by
+  // TaskManagerTabContentsResourceProvider) or background contents (handled
+  // by TaskManagerBackgroundResourceProvider).
+  // TODO(benwells): create specific chrome::VIEW_TYPE_TAB_CONTENTS for
+  // tab contents, as VIEW_TYPE_WEB_CONTENTS is the default.
+  content::ViewType view_type =
+      render_view_host->GetDelegate()->GetRenderViewType();
+  return (view_type != content::VIEW_TYPE_WEB_CONTENTS &&
+          view_type != chrome::VIEW_TYPE_BACKGROUND_CONTENTS);
+}
+
 void TaskManagerExtensionProcessResourceProvider::AddToTaskManager(
     content::RenderViewHost* render_view_host) {
+  if (!IsHandledByThisProvider(render_view_host))
+    return;
+
   TaskManagerExtensionProcessResource* resource =
       new TaskManagerExtensionProcessResource(render_view_host);
   DCHECK(resources_.find(render_view_host) == resources_.end());
   resources_[render_view_host] = resource;
   pid_to_resources_[resource->process_id()] = resource;
   task_manager_->AddResource(resource);
 }
 
 void TaskManagerExtensionProcessResourceProvider::RemoveFromTaskManager(
     content::RenderViewHost* render_view_host) {
@@ skipping 151 matching lines @@
 
   return &resource_;
 }
 
 void TaskManagerBrowserProcessResourceProvider::StartUpdating() {
   task_manager_->AddResource(&resource_);
 }
 
 void TaskManagerBrowserProcessResourceProvider::StopUpdating() {
 }