Part 1 of merging Extensions and DOMUI

chrome/browser/dom_ui/dom_ui_factory.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/dom_ui/dom_ui_factory.h"
 
 #include "chrome/browser/dom_ui/downloads_ui.h"
 #include "chrome/browser/dom_ui/devtools_ui.h"
 #include "chrome/browser/dom_ui/history_ui.h"
 #include "chrome/browser/dom_ui/html_dialog_ui.h"
 #include "chrome/browser/dom_ui/new_tab_ui.h"
 #include "chrome/browser/extensions/extensions_ui.h"
+#include "chrome/browser/extensions/extension_dom_ui.h"
 #include "chrome/common/url_constants.h"
 #ifdef CHROME_PERSONALIZATION
 #include "chrome/personalization/personalization.h"
 #endif
 #include "googleurl/src/gurl.h"
 
 // Backend for both querying for and creating new DOMUI objects. If you're just
 // querying whether there's a DOM UI for the given URL, pass NULL for both the
 // web contents and the new_ui. The return value will indiacate whether a DOM UI
 // exists for the given URL.
 //
 // If you want to create a DOM UI, pass non-NULL pointers for both tab_contents
 // and new_ui. The *new_ui pointer will be filled with the created UI if it
 // succeeds (indicated by a return value of true). The caller owns the *new_ui
 // pointer.
 static bool CreateDOMUI(const GURL& url, TabContents* tab_contents,
                         DOMUI** new_ui) {
   // Currently, any gears: URL means an HTML dialog.
   if (url.SchemeIs(chrome::kGearsScheme)) {
     if (new_ui)
       *new_ui = new HtmlDialogUI(tab_contents);
     return true;
   }
 
+  if (url.SchemeIs(chrome::kExtensionScheme)) {
+    if (new_ui)
+      *new_ui = new ExtensionDOMUI(tab_contents);
+    return true;
+  }
+
 #ifdef CHROME_PERSONALIZATION
   if (Personalization::NeedsDOMUI(url)) {
     if (new_ui)
       *new_ui = new HtmlDialogUI(tab_contents);
     return true;
   }
 #endif
 
   // This will get called a lot to check all URLs, so do a quick check of other
   // schemes (gears was handled above) to filter out most URLs.
@@ skipping 37 matching lines @@
       *new_ui = new DevToolsUI(tab_contents);
     return true;
   }
 
   return false;
 }
 
 // static
 bool DOMUIFactory::HasDOMUIScheme(const GURL& url) {
   return url.SchemeIs(chrome::kChromeInternalScheme) ||
-         url.SchemeIs(chrome::kChromeUIScheme);
+         url.SchemeIs(chrome::kChromeUIScheme) ||
+         url.SchemeIs(chrome::kExtensionScheme);
 }
 
 // static
 bool DOMUIFactory::UseDOMUIForURL(const GURL& url) {
   return CreateDOMUI(url, NULL, NULL);
 }
 
 // static
 DOMUI* DOMUIFactory::CreateDOMUIForURL(TabContents* tab_contents,
                                        const GURL& url) {
   DOMUI* dom_ui;
   if (!CreateDOMUI(url, tab_contents, &dom_ui))
     return NULL;
   return dom_ui;
 }