1c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher// Use of this source code is governed by a BSD-style license that can be
3c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher// found in the LICENSE file.
4c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher
5c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher#include "chrome/browser/ui/navigation_correction_tab_observer.h"
6c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher
7c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher#include "base/prefs/pref_service.h"
8c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher#include "chrome/browser/browser_process.h"
9c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher#include "chrome/browser/chrome_notification_types.h"
10c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher#include "chrome/browser/google/google_profile_helper.h"
11c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher#include "chrome/browser/google/google_url_tracker_factory.h"
12c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher#include "chrome/browser/profiles/profile.h"
13c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher#include "chrome/common/pref_names.h"
14c7fb74806190220ba46bc175568f5c0edd49b810Eric Christopher#include "chrome/common/render_messages.h"
15#include "components/google/core/browser/google_util.h"
16#include "components/pref_registry/pref_registry_syncable.h"
17#include "content/public/browser/render_frame_host.h"
18#include "content/public/browser/render_view_host.h"
19#include "content/public/browser/web_contents.h"
20#include "google_apis/google_api_keys.h"
21
22using content::RenderFrameHost;
23using content::RenderViewHost;
24using content::WebContents;
25
26DEFINE_WEB_CONTENTS_USER_DATA_KEY(NavigationCorrectionTabObserver);
27
28NavigationCorrectionTabObserver::NavigationCorrectionTabObserver(
29    WebContents* web_contents)
30    : content::WebContentsObserver(web_contents),
31      profile_(Profile::FromBrowserContext(web_contents->GetBrowserContext())) {
32  PrefService* prefs = profile_->GetPrefs();
33  if (prefs) {
34    pref_change_registrar_.Init(prefs);
35    pref_change_registrar_.Add(
36        prefs::kAlternateErrorPagesEnabled,
37        base::Bind(&NavigationCorrectionTabObserver::OnEnabledChanged,
38                   base::Unretained(this)));
39  }
40
41  GoogleURLTracker* google_url_tracker =
42      GoogleURLTrackerFactory::GetForProfile(profile_);
43  if (google_url_tracker) {
44    google_url_updated_subscription_ = google_url_tracker->RegisterCallback(
45        base::Bind(&NavigationCorrectionTabObserver::OnGoogleURLUpdated,
46                   base::Unretained(this)));
47  }
48}
49
50NavigationCorrectionTabObserver::~NavigationCorrectionTabObserver() {
51}
52
53// static
54void NavigationCorrectionTabObserver::RegisterProfilePrefs(
55    user_prefs::PrefRegistrySyncable* prefs) {
56  prefs->RegisterBooleanPref(prefs::kAlternateErrorPagesEnabled,
57                             true,
58                             user_prefs::PrefRegistrySyncable::SYNCABLE_PREF);
59}
60
61////////////////////////////////////////////////////////////////////////////////
62// WebContentsObserver overrides
63
64void NavigationCorrectionTabObserver::RenderViewCreated(
65    RenderViewHost* render_view_host) {
66  UpdateNavigationCorrectionInfo(render_view_host);
67}
68
69////////////////////////////////////////////////////////////////////////////////
70// Internal helpers
71
72void NavigationCorrectionTabObserver::OnGoogleURLUpdated() {
73  UpdateNavigationCorrectionInfo(web_contents()->GetRenderViewHost());
74}
75
76GURL NavigationCorrectionTabObserver::GetNavigationCorrectionURL() const {
77  // Disable navigation corrections when the preference is disabled or when in
78  // Incognito mode.
79  if (!profile_->GetPrefs()->GetBoolean(prefs::kAlternateErrorPagesEnabled) ||
80      profile_->IsOffTheRecord()) {
81    return GURL();
82  }
83
84  return google_util::LinkDoctorBaseURL();
85}
86
87void NavigationCorrectionTabObserver::OnEnabledChanged() {
88  UpdateNavigationCorrectionInfo(web_contents()->GetRenderViewHost());
89}
90
91void NavigationCorrectionTabObserver::UpdateNavigationCorrectionInfo(
92    RenderViewHost* rvh) {
93  RenderFrameHost* rfh = rvh->GetMainFrame();
94  rfh->Send(new ChromeViewMsg_SetNavigationCorrectionInfo(
95      rfh->GetRoutingID(),
96      GetNavigationCorrectionURL(),
97      google_util::GetGoogleLocale(g_browser_process->GetApplicationLocale()),
98      google_util::GetGoogleCountryCode(
99          google_profile_helper::GetGoogleHomePageURL(profile_)),
100      google_apis::GetAPIKey(),
101      google_util::GetGoogleSearchURL(
102          google_profile_helper::GetGoogleHomePageURL(profile_))));
103}
104