Part 1 of merging Extensions and DOMUI

chrome/browser/extensions/extension_function_dispatcher.cc

 // Copyright (c) 2009 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_function_dispatcher.h"
 
 #include "base/process_util.h"
 #include "base/singleton.h"
 #include "base/values.h"
 #include "chrome/browser/extensions/extension_bookmarks_module.h"
@@ skipping 150 matching lines @@
 
 bool ExtensionFunctionDispatcher::OverrideFunction(
     const std::string& name, ExtensionFunctionFactory factory) {
   return FactoryRegistry::instance()->OverrideFunction(name, factory);
 }
 
 void ExtensionFunctionDispatcher::ResetFunctions() {
   FactoryRegistry::instance()->ResetFunctions();
 }
 
+std::set<ExtensionFunctionDispatcher*>*
+    ExtensionFunctionDispatcher::all_instances() {
+  static std::set<ExtensionFunctionDispatcher*> instances;
+  return &instances;
+}
+
 ExtensionFunctionDispatcher::ExtensionFunctionDispatcher(
     RenderViewHost* render_view_host,
     Delegate* delegate,
-    const std::string& extension_id)
+    const GURL& url)
   : render_view_host_(render_view_host),
     delegate_(delegate),
-    extension_id_(extension_id),
+    url_(url),
     ALLOW_THIS_IN_INITIALIZER_LIST(peer_(new Peer(this))) {
+  all_instances()->insert(this);
   RenderProcessHost* process = render_view_host_->process();
   ExtensionMessageService* message_service =
       ExtensionMessageService::GetInstance(profile()->GetRequestContext());
   DCHECK(process);
   DCHECK(message_service);
-  message_service->RegisterExtension(extension_id, process->pid());
+  message_service->RegisterExtension(extension_id(), process->pid());
 }
 
 ExtensionFunctionDispatcher::~ExtensionFunctionDispatcher() {
+  all_instances()->erase(this);
   peer_->dispatcher_ = NULL;
 }
 
 Browser* ExtensionFunctionDispatcher::GetBrowser() {
   DCHECK(delegate_);
 
   Browser* retval = delegate_->GetBrowser();
   return retval;
 }
 
@@ skipping 25 matching lines @@
   } else {
     NOTREACHED();
     base::KillProcess(render_view_host_->process()->process().handle(),
                       ResultCodes::KILLED_BAD_MESSAGE, false);
   }
 }
 
 Profile* ExtensionFunctionDispatcher::profile() {
   return render_view_host_->process()->profile();
 }