Switch SearchTabHelper to use WebContentsUserData.

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/ui/tab_contents/tab_contents.h"
+#include "chrome/browser/profiles/profile.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"
 
+int chrome::search::SearchTabHelper::kUserDataKey;
+
 namespace {
 
 bool IsNTP(const GURL& url) {
   return url.SchemeIs(chrome::kChromeUIScheme) &&
          url.host() == chrome::kChromeUINewTabHost;
 }
 
+bool IsSearchEnabled(content::WebContents* web_contents) {
+  Profile* profile =
+      Profile::FromBrowserContext(web_contents->GetBrowserContext());
+  return chrome::search::IsInstantExtendedAPIEnabled(profile);
+}
+
 }  // namespace
 
 namespace chrome {
 namespace search {
 
-SearchTabHelper::SearchTabHelper(
-    TabContents* contents,
-    bool is_search_enabled)
-    : WebContentsObserver(contents->web_contents()),
-      is_search_enabled_(is_search_enabled),
+SearchTabHelper::SearchTabHelper(content::WebContents* web_contents)
+    : WebContentsObserver(web_contents),
+      is_search_enabled_(IsSearchEnabled(web_contents)),
       is_initial_navigation_commit_(true),
-      model_(contents),
+      model_(web_contents),
       ntp_load_state_(DEFAULT),
       main_frame_id_(0) {
-  if (!is_search_enabled)
+  if (!is_search_enabled_)
     return;
 
   registrar_.Add(
       this,
       content::NOTIFICATION_NAV_ENTRY_COMMITTED,
       content::Source<content::NavigationController>(
-          &contents->web_contents()->GetController()));
+          &web_contents->GetController()));
 }
 
 SearchTabHelper::~SearchTabHelper() {
 }
 
 void SearchTabHelper::OmniboxEditModelChanged(bool user_input_in_progress,
                                               bool cancelling) {
   if (!is_search_enabled_)
     return;
 
@@ skipping 93 matching lines @@
                                             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()))
     type = Mode::MODE_SEARCH_RESULTS;
   model_.SetMode(Mode(type, animate));
 }
 
-content::WebContents* SearchTabHelper::web_contents() {
-  return model_.tab_contents()->web_contents();
+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