Remove some chrome-specific methods from content::WebUI.

chrome/browser/ui/toolbar/toolbar_model_impl.cc

 // Copyright 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/ui/toolbar/toolbar_model_impl.h"
 
 #include "base/command_line.h"
 #include "base/prefs/pref_service.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/autocomplete/autocomplete_classifier.h"
@@ skipping 128 matching lines @@
 
   return GURL(chrome::kAboutBlankURL);
 }
 
 bool ToolbarModelImpl::WouldReplaceSearchURLWithSearchTerms() const {
   return !GetSearchTerms().empty();
 }
 
 bool ToolbarModelImpl::ShouldDisplayURL() const {
   // Note: The order here is important.
-  // - The WebUI test must come before the extension scheme test because there
-  //   can be WebUIs that have extension schemes (e.g. the bookmark manager). In
-  //   that case, we should prefer what the WebUI instance says.
-  // - The view-source test must come before the WebUI test because of the case
+  // - The NTP test must come before the extension scheme test because there
+  //   can be NTPs that have extension schemes (e.g. the bookmark manager). In

[Avi (use Gerrit), 2013/04/10 23:12:15]

"NTPs that have extension schemes"? I'm not sure all these instances of "WebUI"
should be altered this way.

[jam, 2013/04/10 23:45:35]

updated comment

+  //   that case, we should prefer what the NTP instance says.
+  // - The view-source test must come before the NTP test because of the case
   //   of view-source:chrome://newtab, which should display its URL despite what
-  //   chrome://newtab's WebUI says.
+  //   chrome://newtab says.
   NavigationController* controller = GetNavigationController();
   NavigationEntry* entry = controller ? controller->GetVisibleEntry() : NULL;
   if (entry) {
     if (entry->IsViewSourceMode() ||
         entry->GetPageType() == content::PAGE_TYPE_INTERSTITIAL) {
       return true;
     }
+
+    if (entry->GetURL().SchemeIs(chrome::kChromeUIScheme) &&
+        (entry->GetURL().host() == chrome::kChromeUINewTabHost ||
+         entry->GetURL().host() == chrome::kChromeUIExtensionActivityHost)) {
+      return false;
+    }
+
+    if (entry->GetURL().SchemeIs(extensions::kExtensionScheme))
+      return false;
+
+#if defined(OS_CHROMEOS)
+    if (entry->GetURL().SchemeIs(chrome::kDriveScheme))
+      return false;
+#endif
   }
 
-  WebContents* web_contents = delegate_->GetActiveWebContents();
-  if (web_contents && web_contents->GetWebUIForCurrentState())
-    return !web_contents->GetWebUIForCurrentState()->ShouldHideURL();
-
-  if (entry && entry->GetURL().SchemeIs(extensions::kExtensionScheme))
-    return false;
-
-#if defined(OS_CHROMEOS)
-  if (entry && entry->GetURL().SchemeIs(chrome::kDriveScheme))
-    return false;
-#endif
-
-  if (chrome::IsInstantNTP(web_contents))
+  if (chrome::IsInstantNTP(delegate_->GetActiveWebContents()))
     return false;
 
   return true;
 }
 
 ToolbarModel::SecurityLevel ToolbarModelImpl::GetSecurityLevel() const {
   if (input_in_progress_)  // When editing, assume no security style.
     return NONE;
 
   return GetSecurityLevelForWebContents(delegate_->GetActiveWebContents());
@@ skipping 74 matching lines @@
     Profile* profile =
         Profile::FromBrowserContext(contents->GetBrowserContext());
     AutocompleteClassifierFactory::GetForProfile(profile)->Classify(
         search_terms, false, false, &match, NULL);
     if (!AutocompleteMatch::IsSearchType(match.type))
       search_terms.clear();
   }
 
   return search_terms;
 }