Create CoreTabHelper, move remaining core TCW functionality into it.

chrome/browser/ui/tab_contents/core_tab_helper.cc

+// Copyright (c) 2011 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/tab_contents/core_tab_helper.h"
+
+#include "chrome/browser/google/google_util.h"
+#include "chrome/browser/pdf_unsupported_feature.h"
+#include "chrome/browser/prefs/pref_service.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/renderer_host/web_cache_manager.h"
+#include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
+#include "chrome/common/chrome_notification_types.h"
+#include "chrome/common/pref_names.h"
+#include "chrome/common/render_messages.h"
+#include "content/browser/renderer_host/render_view_host.h"
+#include "content/public/browser/notification_service.h"
+#include "grit/generated_resources.h"
+#include "ui/base/l10n/l10n_util.h"
+
+namespace {
+
+// The list of prefs we want to observe.
+const char* kPrefsToObserve[] = {
+  prefs::kAlternateErrorPagesEnabled,
+  prefs::kDefaultZoomLevel,
+};
+
+const int kPrefsToObserveLength = arraysize(kPrefsToObserve);
+
+}  // namespace
+
+CoreTabHelper::CoreTabHelper(TabContentsWrapper* wrapper)
+    : TabContentsObserver(wrapper->tab_contents()),
+      delegate_(NULL),
+      wrapper_(wrapper) {
+  PrefService* prefs = wrapper_->profile()->GetPrefs();
+  pref_change_registrar_.Init(prefs);
+  if (prefs) {
+    for (int i = 0; i < kPrefsToObserveLength; ++i)
+      pref_change_registrar_.Add(kPrefsToObserve[i], this);
+  }
+
+  registrar_.Add(this, chrome::NOTIFICATION_GOOGLE_URL_UPDATED,
+                 content::NotificationService::AllSources());
+}
+
+CoreTabHelper::~CoreTabHelper() {
+}
+
+// static
+void CoreTabHelper::RegisterUserPrefs(PrefService* prefs) {
+  prefs->RegisterBooleanPref(prefs::kAlternateErrorPagesEnabled,
+                             true,
+                             PrefService::SYNCABLE_PREF);
+}
+
+string16 CoreTabHelper::GetDefaultTitle() {
+  return l10n_util::GetStringUTF16(IDS_DEFAULT_TAB_TITLE);
+}
+
+string16 CoreTabHelper::GetStatusText() const {
+  if (!tab_contents()->IsLoading() ||
+      tab_contents()->load_state().state == net::LOAD_STATE_IDLE) {
+    return string16();
+  }
+
+  switch (tab_contents()->load_state().state) {
+    case net::LOAD_STATE_WAITING_FOR_DELEGATE:
+      return l10n_util::GetStringFUTF16(IDS_LOAD_STATE_WAITING_FOR_DELEGATE,
+                                        tab_contents()->load_state().param);
+    case net::LOAD_STATE_WAITING_FOR_CACHE:
+      return l10n_util::GetStringUTF16(IDS_LOAD_STATE_WAITING_FOR_CACHE);
+    case net::LOAD_STATE_WAITING_FOR_APPCACHE:
+      return l10n_util::GetStringUTF16(IDS_LOAD_STATE_WAITING_FOR_APPCACHE);
+    case net::LOAD_STATE_ESTABLISHING_PROXY_TUNNEL:
+      return
+          l10n_util::GetStringUTF16(IDS_LOAD_STATE_ESTABLISHING_PROXY_TUNNEL);
+    case net::LOAD_STATE_RESOLVING_PROXY_FOR_URL:
+      return l10n_util::GetStringUTF16(IDS_LOAD_STATE_RESOLVING_PROXY_FOR_URL);
+    case net::LOAD_STATE_RESOLVING_HOST_IN_PROXY_SCRIPT:
+      return l10n_util::GetStringUTF16(
+          IDS_LOAD_STATE_RESOLVING_HOST_IN_PROXY_SCRIPT);
+    case net::LOAD_STATE_RESOLVING_HOST:
+      return l10n_util::GetStringUTF16(IDS_LOAD_STATE_RESOLVING_HOST);
+    case net::LOAD_STATE_CONNECTING:
+      return l10n_util::GetStringUTF16(IDS_LOAD_STATE_CONNECTING);
+    case net::LOAD_STATE_SSL_HANDSHAKE:
+      return l10n_util::GetStringUTF16(IDS_LOAD_STATE_SSL_HANDSHAKE);
+    case net::LOAD_STATE_SENDING_REQUEST:
+      if (tab_contents()->upload_size())
+        return l10n_util::GetStringFUTF16Int(
+                    IDS_LOAD_STATE_SENDING_REQUEST_WITH_PROGRESS,
+                    static_cast<int>((100 * tab_contents()->upload_position()) /
+                        tab_contents()->upload_size()));
+      else
+        return l10n_util::GetStringUTF16(IDS_LOAD_STATE_SENDING_REQUEST);
+    case net::LOAD_STATE_WAITING_FOR_RESPONSE:
+      return l10n_util::GetStringFUTF16(IDS_LOAD_STATE_WAITING_FOR_RESPONSE,
+                                        tab_contents()->load_state_host());
+    // Ignore net::LOAD_STATE_READING_RESPONSE and net::LOAD_STATE_IDLE
+    case net::LOAD_STATE_IDLE:
+    case net::LOAD_STATE_READING_RESPONSE:
+      break;
+  }
+
+  return string16();
+}
+
+void CoreTabHelper::CaptureSnapshot() {
+  Send(new ChromeViewMsg_CaptureSnapshot(routing_id()));
+}
+
+void CoreTabHelper::ExitFullscreenMode() {
+  if (tab_contents() && tab_contents()->render_view_host())
+    tab_contents()->render_view_host()->ExitFullscreen();
+}
+
+////////////////////////////////////////////////////////////////////////////////
+// TabContentsObserver overrides
+
+void CoreTabHelper::RenderViewCreated(RenderViewHost* render_view_host) {
+  UpdateAlternateErrorPageURL(render_view_host);
+}
+
+void CoreTabHelper::DidBecomeSelected() {
+  // TODO(avi): Does this belong here or in some other tab helper?
+  WebCacheManager::GetInstance()->ObserveActivity(
+      tab_contents()->GetRenderProcessHost()->GetID());
+}
+
+bool CoreTabHelper::OnMessageReceived(const IPC::Message& message) {
+  bool handled = true;
+  IPC_BEGIN_MESSAGE_MAP(CoreTabHelper, message)
+    IPC_MESSAGE_HANDLER(ChromeViewHostMsg_Snapshot, OnSnapshot)
+    IPC_MESSAGE_HANDLER(ChromeViewHostMsg_PDFHasUnsupportedFeature,
+                        OnPDFHasUnsupportedFeature)
+    IPC_MESSAGE_UNHANDLED(handled = false)
+  IPC_END_MESSAGE_MAP()
+  return handled;
+}
+
+////////////////////////////////////////////////////////////////////////////////
+// content::NotificationObserver overrides
+
+void CoreTabHelper::Observe(int type,
+                            const content::NotificationSource& source,
+                            const content::NotificationDetails& details) {
+  switch (type) {
+    case chrome::NOTIFICATION_GOOGLE_URL_UPDATED:
+      UpdateAlternateErrorPageURL(tab_contents()->render_view_host());
+      break;
+    case chrome::NOTIFICATION_PREF_CHANGED: {
+      std::string* pref_name = content::Details<std::string>(details).ptr();
+      DCHECK(content::Source<PrefService>(source).ptr() ==
+             wrapper_->profile()->GetPrefs());
+      if (*pref_name == prefs::kAlternateErrorPagesEnabled) {
+        UpdateAlternateErrorPageURL(tab_contents()->render_view_host());
+      } else if (*pref_name == prefs::kDefaultZoomLevel) {
+        tab_contents()->render_view_host()->SetZoomLevel(
+            tab_contents()->GetZoomLevel());
+      } else {
+        NOTREACHED() << "unexpected pref change notification" << *pref_name;
+      }
+      break;
+    }
+    default:
+      NOTREACHED();
+  }
+}
+
+////////////////////////////////////////////////////////////////////////////////
+// Internal helpers
+
+void CoreTabHelper::OnSnapshot(const SkBitmap& bitmap) {
+  content::NotificationService::current()->Notify(
+      chrome::NOTIFICATION_TAB_SNAPSHOT_TAKEN,
+      content::Source<TabContentsWrapper>(wrapper_),
+      content::Details<const SkBitmap>(&bitmap));
+}
+
+void CoreTabHelper::OnPDFHasUnsupportedFeature() {
+  PDFHasUnsupportedFeature(wrapper_);
+}
+
+GURL CoreTabHelper::GetAlternateErrorPageURL() const {
+  GURL url;
+  // Disable alternate error pages when in Incognito mode.
+  if (wrapper_->profile()->IsOffTheRecord())
+    return url;
+
+  PrefService* prefs = wrapper_->profile()->GetPrefs();
+  if (prefs->GetBoolean(prefs::kAlternateErrorPagesEnabled)) {
+    url = google_util::AppendGoogleLocaleParam(
+        GURL(google_util::kLinkDoctorBaseURL));
+    url = google_util::AppendGoogleTLDParam(url);
+  }
+  return url;
+}
+
+void CoreTabHelper::UpdateAlternateErrorPageURL(RenderViewHost* rvh) {
+  rvh->SetAltErrorPageURL(GetAlternateErrorPageURL());
+}