Introduces a search term extraction mechanism working for arbitrary search providers.

chrome/browser/ui/search/search_tab_helper.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/search/search_tab_helper.h"
 
 #include "chrome/browser/google/google_util.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/search_engines/template_url.h"
+#include "chrome/browser/search_engines/template_url_service.h"
+#include "chrome/browser/search_engines/template_url_service_factory.h"
 #include "chrome/browser/ui/search/search.h"
 #include "chrome/common/url_constants.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_details.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/render_widget_host_view.h"
 #include "content/public/browser/web_contents.h"
 
@@ skipping 133 matching lines @@
       NOTREACHED() << "Unexpected notification type.";
   }
 }
 
 void SearchTabHelper::UpdateModelBasedOnURL(const GURL& url,
                                             NTPLoadState load_state,
                                             bool animate) {
   Mode::Type type = Mode::MODE_DEFAULT;
   if (IsNTP(url))
     type = load_state == PAINTED ? Mode::MODE_NTP : Mode::MODE_NTP_LOADING;
-  else if (google_util::IsInstantExtendedAPIGoogleSearchUrl(url.spec()))
+  else if (CanExtractSearchTerms(url))
     type = Mode::MODE_SEARCH_RESULTS;
   model_.SetMode(Mode(type, animate));
 }
 
+bool SearchTabHelper::CanExtractSearchTerms(const GURL& url) {
+  Profile* profile =
+      Profile::FromBrowserContext(web_contents()->GetBrowserContext());
+
+  // Ensure instant extended API is enabled.
+  if (!profile || !chrome::search::IsInstantExtendedAPIEnabled(profile))
+    return false;
+
+  TemplateURLService* template_url_service =
+      TemplateURLServiceFactory::GetForProfile(profile);
+
+  if (!template_url_service)
+    return false;
+
+  TemplateURL *template_url = template_url_service->GetDefaultSearchProvider();
+  if (!template_url)
+    return false;
+
+  string16 dummy;
+  return template_url->ExtractSearchTermsFromInstantExtendedURL(url, &dummy);
+}
+
 const content::WebContents* SearchTabHelper::web_contents() const {
   return model_.web_contents();
 }
 
 content::RenderWidgetHost* SearchTabHelper::GetRenderWidgetHost() {
   content::RenderWidgetHostView* rwhv =
       web_contents()->GetRenderWidgetHostView();
   return rwhv ? rwhv->GetRenderWidgetHost() : NULL;
 }
 
 }  // namespace search
 }  // namespace chrome