experimental.popup support for tab-content-viewed extensions

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/execute_code_in_tab_function.h"
 #include "chrome/browser/extensions/extension_bookmarks_module.h"
 #include "chrome/browser/extensions/extension_bookmarks_module_constants.h"
 #include "chrome/browser/extensions/extension_browser_actions_api.h"
+#include "chrome/browser/extensions/extension_dom_ui.h"
 #include "chrome/browser/extensions/extension_function.h"
 #include "chrome/browser/extensions/extension_history_api.h"
 #include "chrome/browser/extensions/extension_i18n_api.h"
 #include "chrome/browser/extensions/extension_message_service.h"
 #include "chrome/browser/extensions/extension_page_actions_module.h"
 #include "chrome/browser/extensions/extension_page_actions_module_constants.h"
 #include "chrome/browser/extensions/extension_popup_api.h"
 #include "chrome/browser/extensions/extension_process_manager.h"
 #include "chrome/browser/extensions/extension_tabs_module.h"
 #include "chrome/browser/extensions/extension_tabs_module_constants.h"
@@ skipping 208 matching lines @@
 
 ExtensionFunctionDispatcher::~ExtensionFunctionDispatcher() {
   all_instances()->erase(this);
   peer_->dispatcher_ = NULL;
 }
 
 Browser* ExtensionFunctionDispatcher::GetBrowser() {
   return delegate_->GetBrowser();
 }
 
+ExtensionPopupHost* ExtensionFunctionDispatcher::GetPopupHost() {
+  ExtensionHost* extension_host = GetExtensionHost();
+  if (extension_host) {
+    DCHECK(!GetExtensionDOMUI()) <<
+        "Function dispatcher registered in too many environments.";
+    return extension_host->popup_host();
+  } else {
+    ExtensionDOMUI* dom_ui = GetExtensionDOMUI();
+    return dom_ui->popup_host();
+  }
+}
+
 ExtensionHost* ExtensionFunctionDispatcher::GetExtensionHost() {
   return delegate_->GetExtensionHost();
 }
 
+ExtensionDOMUI* ExtensionFunctionDispatcher::GetExtensionDOMUI() {
+  return delegate_->GetExtensionDOMUI();
+}
+
 Extension* ExtensionFunctionDispatcher::GetExtension() {
   ExtensionsService* service = profile()->GetExtensionsService();
   DCHECK(service);
 
   Extension* extension = service->GetExtensionById(extension_id(), false);
   DCHECK(extension);
 
   return extension;
 }
 
@@ skipping 26 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();
 }