A simple version of search term replacement for testing .

chrome/browser/ui/toolbar/toolbar_model.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/ui/toolbar/toolbar_model.h"
 
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/autocomplete/autocomplete_input.h"
+#include "chrome/browser/google/google_util.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ssl/ssl_error_info.h"
+#include "chrome/browser/ui/search/search.h"
 #include "chrome/browser/ui/toolbar/toolbar_model_delegate.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "content/public/browser/cert_store.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_ui.h"
 #include "content/public/common/content_constants.h"
@@ skipping 11 matching lines @@
 
 ToolbarModel::ToolbarModel(ToolbarModelDelegate* delegate)
     : delegate_(delegate),
       input_in_progress_(false) {
 }
 
 ToolbarModel::~ToolbarModel() {
 }
 
 // ToolbarModel Implementation.
-string16 ToolbarModel::GetText() const {
-  GURL url(chrome::kAboutBlankURL);
+string16 ToolbarModel::GetText(bool display_search_urls_as_search_terms) const {
+  GURL url(GetURL());
+
+  if (display_search_urls_as_search_terms) {
+    string16 search_terms = TryToExtractSearchTermsFromURL(url);
+    if (!search_terms.empty())
+      return search_terms;
+  }
   std::string languages;  // Empty if we don't have a |navigation_controller|.
+  Profile* profile = GetProfile();
+  if (profile)
+    languages = profile->GetPrefs()->GetString(prefs::kAcceptLanguages);
 
-  NavigationController* navigation_controller = GetNavigationController();
-  if (navigation_controller) {
-    Profile* profile =
-        Profile::FromBrowserContext(navigation_controller->GetBrowserContext());
-    languages = profile->GetPrefs()->GetString(prefs::kAcceptLanguages);
-    NavigationEntry* entry = navigation_controller->GetVisibleEntry();
-    if (!ShouldDisplayURL()) {
-      url = GURL();
-    } else if (entry) {
-      url = entry->GetVirtualURL();
-    }
-  }
   if (url.spec().length() > content::kMaxURLDisplayChars)
     url = url.IsStandard() ? url.GetOrigin() : GURL(url.scheme() + ":");
   // Note that we can't unescape spaces here, because if the user copies this
   // and pastes it into another program, that program may think the URL ends at
   // the space.
   return AutocompleteInput::FormattedStringWithEquivalentMeaning(
       url, net::FormatUrl(url, languages, net::kFormatUrlOmitAll,
                           net::UnescapeRule::NORMAL, NULL, NULL, NULL));
 }
 
+GURL ToolbarModel::GetURL() const {
+  const NavigationController* navigation_controller = GetNavigationController();
+  if (navigation_controller) {
+    const NavigationEntry* entry = navigation_controller->GetVisibleEntry();
+    if (entry)
+      return ShouldDisplayURL() ? entry->GetVirtualURL() : GURL();
+  }
+
+  return GURL(chrome::kAboutBlankURL);
+}
+
+bool ToolbarModel::WouldReplaceSearchURLWithSearchTerms() const {
+  return !TryToExtractSearchTermsFromURL(GetURL()).empty();
+}
+
 bool ToolbarModel::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
   //   of view-source:chrome://newtab, which should display its URL despite what
   //   chrome://newtab's WebUI says.
   NavigationController* controller = GetNavigationController();
   NavigationEntry* entry = controller ? controller->GetVisibleEntry() : NULL;
@@ skipping 95 matching lines @@
       UTF8ToUTF16(cert.subject().country_name));
 }
 
 NavigationController* ToolbarModel::GetNavigationController() const {
   // This |current_tab| can be NULL during the initialization of the
   // toolbar during window creation (i.e. before any tabs have been added
   // to the window).
   WebContents* current_tab = delegate_->GetActiveWebContents();
   return current_tab ? &current_tab->GetController() : NULL;
 }
+
+string16 ToolbarModel::TryToExtractSearchTermsFromURL(const GURL& url) const {
+  Profile* profile = GetProfile();
+  if (profile &&
+      chrome::search::IsInstantExtendedAPIEnabled(profile) &&
+      google_util::IsInstantExtendedAPIGoogleSearchUrl(url.spec())) {
+    // TODO(dominich): http://crbug.com/135106 - Replace this with whatever the
+    // final solution is as per http://crbug.com/139176.
+    return google_util::GetSearchTermsFromGoogleSearchURL(url.spec());
+  }
+  return string16();
+}
+
+Profile* ToolbarModel::GetProfile() const {
+  NavigationController* navigation_controller = GetNavigationController();
+  return navigation_controller ?
+      Profile::FromBrowserContext(navigation_controller->GetBrowserContext()) :
+      NULL;
+}