| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/basictypes.h" | 5 #include "base/basictypes.h" |
| 6 #include "base/compiler_specific.h" | 6 #include "base/compiler_specific.h" |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "chrome/browser/protector/protector.h" | 8 #include "chrome/browser/protector/protector.h" |
| 9 #include "chrome/browser/protector/setting_change.h" | 9 #include "chrome/browser/protector/setting_change.h" |
| 10 #include "chrome/browser/search_engines/template_url.h" | 10 #include "chrome/browser/search_engines/template_url.h" |
| 11 #include "chrome/browser/search_engines/template_url_service.h" | 11 #include "chrome/browser/search_engines/template_url_service.h" |
| 12 #include "chrome/browser/webdata/keyword_table.h" | 12 #include "chrome/browser/webdata/keyword_table.h" |
| 13 #include "chrome/common/url_constants.h" | 13 #include "chrome/common/url_constants.h" |
| 14 #include "grit/chromium_strings.h" | |
| 15 #include "grit/generated_resources.h" | |
| 16 #include "googleurl/src/gurl.h" | 14 #include "googleurl/src/gurl.h" |
| 17 #include "ui/base/l10n/l10n_util.h" | |
| 18 | 15 |
| 19 namespace protector { | 16 namespace protector { |
| 20 | 17 |
| 21 namespace { | |
| 22 | |
| 23 // Maximum length of the search engine name to be displayed. | |
| 24 const size_t kMaxDisplayedNameLength = 10; | |
| 25 | |
| 26 } // namespace | |
| 27 | |
| 28 class DefaultSearchProviderChange : public SettingChange { | 18 class DefaultSearchProviderChange : public SettingChange { |
| 29 public: | 19 public: |
| 30 DefaultSearchProviderChange(const TemplateURL* old_url, | 20 DefaultSearchProviderChange(const TemplateURL* old_url, |
| 31 const TemplateURL* new_url); | 21 const TemplateURL* new_url); |
| 32 | 22 |
| 33 // SettingChange overrides: | 23 // SettingChange overrides: |
| 34 virtual bool Init(Protector* protector) OVERRIDE; | 24 virtual string16 GetOldSetting() const OVERRIDE; |
| 35 virtual void Apply(Protector* protector) OVERRIDE; | 25 virtual string16 GetNewSetting() const OVERRIDE; |
| 36 virtual void Discard(Protector* protector) OVERRIDE; | 26 virtual void Accept(Protector* protector) OVERRIDE; |
| 37 virtual string16 GetTitle() const OVERRIDE; | 27 virtual void Revert(Protector* protector) OVERRIDE; |
| 38 virtual string16 GetMessage() const OVERRIDE; | 28 virtual void DoDefault(Protector* protector) OVERRIDE; |
| 39 virtual string16 GetApplyButtonText() const OVERRIDE; | |
| 40 virtual string16 GetDiscardButtonText() const OVERRIDE; | |
| 41 | 29 |
| 42 private: | 30 private: |
| 43 virtual ~DefaultSearchProviderChange(); | 31 virtual ~DefaultSearchProviderChange(); |
| 44 | 32 |
| 45 // Sets the given default search provider to profile that |protector| is | 33 // Sets the given default search provider to profile that |protector| is |
| 46 // guarding. Returns the |TemplateURL| instance the default search provider | 34 // guarding. |
| 47 // has been set to. If no search provider with |id| exists and | 35 void SetDefaultSearchProvider(Protector* protector, int64 id); |
| 48 // |allow_fallback| is true, sets one of the prepoluated search providers. | |
| 49 const TemplateURL* SetDefaultSearchProvider(Protector* protector, | |
| 50 int64 id, | |
| 51 bool allow_fallback); | |
| 52 | |
| 53 // Opens the Search engine settings page in a new tab. | |
| 54 void OpenSearchEngineSettings(Protector* protector); | |
| 55 | 36 |
| 56 int64 old_id_; | 37 int64 old_id_; |
| 57 int64 new_id_; | 38 int64 new_id_; |
| 58 // ID of the search engine that we fall back to if the backup is lost. | |
| 59 int64 fallback_id_; | |
| 60 string16 old_name_; | 39 string16 old_name_; |
| 61 string16 new_name_; | 40 string16 new_name_; |
| 62 // Name of the search engine that we fall back to if the backup is lost. | |
| 63 string16 fallback_name_; | |
| 64 string16 product_name_; | |
| 65 | 41 |
| 66 DISALLOW_COPY_AND_ASSIGN(DefaultSearchProviderChange); | 42 DISALLOW_COPY_AND_ASSIGN(DefaultSearchProviderChange); |
| 67 }; | 43 }; |
| 68 | 44 |
| 69 DefaultSearchProviderChange::DefaultSearchProviderChange( | 45 DefaultSearchProviderChange::DefaultSearchProviderChange( |
| 70 const TemplateURL* old_url, | 46 const TemplateURL* old_url, |
| 71 const TemplateURL* new_url) | 47 const TemplateURL* new_url) |
| 72 : old_id_(0), | 48 : SettingChange(kSearchEngineChanged), |
| 73 new_id_(0), | 49 old_id_(0), |
| 74 fallback_id_(0), | 50 new_id_(0) { |
| 75 product_name_(l10n_util::GetStringUTF16(IDS_PRODUCT_NAME)) { | 51 DCHECK(new_url); |
| 76 if (new_url) { | 52 new_id_ = new_url->id(); |
| 77 new_id_ = new_url->id(); | 53 new_name_ = new_url->short_name(); |
| 78 new_name_ = new_url->short_name(); | |
| 79 } | |
| 80 if (old_url) { | 54 if (old_url) { |
| 81 old_id_ = old_url->id(); | 55 old_id_ = old_url->id(); |
| 82 old_name_ = old_url->short_name(); | 56 old_name_ = old_url->short_name(); |
| 83 } | 57 } |
| 84 } | 58 } |
| 85 | 59 |
| 86 DefaultSearchProviderChange::~DefaultSearchProviderChange() { | 60 DefaultSearchProviderChange::~DefaultSearchProviderChange() { |
| 87 } | 61 } |
| 88 | 62 |
| 89 bool DefaultSearchProviderChange::Init(Protector* protector) { | 63 string16 DefaultSearchProviderChange::GetOldSetting() const { |
| 90 // Initially reset the search engine to its previous setting. | 64 return old_name_; |
| 91 const TemplateURL* current_url = | |
| 92 SetDefaultSearchProvider(protector, old_id_, true); | |
| 93 if (!current_url) | |
| 94 return false; | |
| 95 if (!old_id_ || current_url->id() != old_id_) { | |
| 96 // Old settings is lost or invalid, so we had to fall back to one of the | |
| 97 // prepopulated search engines. | |
| 98 fallback_id_ = current_url->id(); | |
| 99 fallback_name_ = current_url->short_name(); | |
| 100 VLOG(1) << "Fallback to " << fallback_name_; | |
| 101 } | |
| 102 return true; | |
| 103 } | 65 } |
| 104 | 66 |
| 105 void DefaultSearchProviderChange::Apply(Protector* protector) { | 67 string16 DefaultSearchProviderChange::GetNewSetting() const { |
| 106 // TODO(avayvod): Add histrogram. | 68 return new_name_; |
| 107 if (!new_id_) { | |
| 108 // Open settings page in case the new setting is invalid. | |
| 109 OpenSearchEngineSettings(protector); | |
| 110 } else { | |
| 111 SetDefaultSearchProvider(protector, new_id_, false); | |
| 112 } | |
| 113 } | 69 } |
| 114 | 70 |
| 115 void DefaultSearchProviderChange::Discard(Protector* protector) { | 71 void DefaultSearchProviderChange::Accept(Protector* protector) { |
| 72 SetDefaultSearchProvider(protector, new_id_); |
| 116 // TODO(avayvod): Add histrogram. | 73 // TODO(avayvod): Add histrogram. |
| 117 if (!old_id_) { | |
| 118 // Open settings page in case the old setting is invalid. | |
| 119 OpenSearchEngineSettings(protector); | |
| 120 } | |
| 121 // Nothing to do otherwise since we have already set the search engine | |
| 122 // to |old_id_| in |Init|. | |
| 123 } | 74 } |
| 124 | 75 |
| 125 string16 DefaultSearchProviderChange::GetTitle() const { | 76 void DefaultSearchProviderChange::Revert(Protector* protector) { |
| 126 return l10n_util::GetStringUTF16(IDS_SEARCH_ENGINE_CHANGE_TITLE); | 77 SetDefaultSearchProvider(protector, old_id_); |
| 78 if (!old_id_) { |
| 79 // Open settings page in case the original setting was lost. |
| 80 protector->OpenTab( |
| 81 GURL(std::string(chrome::kChromeUISettingsURL) + |
| 82 chrome::kSearchEnginesSubPage)); |
| 83 } |
| 84 // TODO(avayvod): Add histrogram. |
| 127 } | 85 } |
| 128 | 86 |
| 129 string16 DefaultSearchProviderChange::GetMessage() const { | 87 void DefaultSearchProviderChange::DoDefault(Protector* protector) { |
| 130 if (fallback_name_.empty()) | 88 SetDefaultSearchProvider(protector, old_id_); |
| 131 return l10n_util::GetStringFUTF16( | 89 // TODO(avayvod): Add histrogram. |
| 132 IDS_SEARCH_ENGINE_CHANGE_MESSAGE, product_name_); | |
| 133 else | |
| 134 return l10n_util::GetStringFUTF16( | |
| 135 IDS_SEARCH_ENGINE_CHANGE_NO_BACKUP_MESSAGE, | |
| 136 product_name_, fallback_name_); | |
| 137 } | 90 } |
| 138 | 91 |
| 139 string16 DefaultSearchProviderChange::GetApplyButtonText() const { | 92 void DefaultSearchProviderChange::SetDefaultSearchProvider( |
| 140 if (new_id_) { | |
| 141 if (new_id_ == fallback_id_) { | |
| 142 // Old search engine is lost, the fallback search engine is the same as | |
| 143 // the new one so no need to show this button. | |
| 144 return string16(); | |
| 145 } | |
| 146 if (new_name_.length() > kMaxDisplayedNameLength) | |
| 147 return l10n_util::GetStringUTF16(IDS_CHANGE_SEARCH_ENGINE_NO_NAME); | |
| 148 else | |
| 149 return l10n_util::GetStringFUTF16(IDS_CHANGE_SEARCH_ENGINE, new_name_); | |
| 150 } else if (old_id_) { | |
| 151 // New setting is lost, offer to go to settings. | |
| 152 return l10n_util::GetStringUTF16(IDS_SELECT_SEARCH_ENGINE); | |
| 153 } else { | |
| 154 // Both settings are lost: don't show this button. | |
| 155 return string16(); | |
| 156 } | |
| 157 } | |
| 158 | |
| 159 string16 DefaultSearchProviderChange::GetDiscardButtonText() const { | |
| 160 if (old_id_) { | |
| 161 if (new_name_.length() > kMaxDisplayedNameLength) | |
| 162 return l10n_util::GetStringUTF16(IDS_KEEP_SETTING); | |
| 163 else | |
| 164 return l10n_util::GetStringFUTF16(IDS_KEEP_SEARCH_ENGINE, old_name_); | |
| 165 } else { | |
| 166 // Old setting is lost, offer to go to settings. | |
| 167 return l10n_util::GetStringUTF16(IDS_SELECT_SEARCH_ENGINE); | |
| 168 } | |
| 169 } | |
| 170 | |
| 171 const TemplateURL* DefaultSearchProviderChange::SetDefaultSearchProvider( | |
| 172 Protector* protector, | 93 Protector* protector, |
| 173 int64 id, | 94 int64 id) { |
| 174 bool allow_fallback) { | 95 DCHECK(protector); |
| 175 TemplateURLService* url_service = protector->GetTemplateURLService(); | 96 TemplateURLService* url_service = protector->GetTemplateURLService(); |
| 176 if (!url_service) { | 97 if (!url_service) { |
| 177 NOTREACHED() << "Can't get TemplateURLService object."; | 98 LOG(WARNING) << "Can't get TemplateURLService object."; |
| 178 return NULL; | 99 return; |
| 179 } | 100 } |
| 180 const TemplateURL* url = NULL; | 101 const TemplateURL* url = NULL; |
| 181 if (id) { | 102 const TemplateURLService::TemplateURLVector& urls = |
| 182 const TemplateURLService::TemplateURLVector& urls = | 103 url_service->GetTemplateURLs(); |
| 183 url_service->GetTemplateURLs(); | 104 for (size_t i = 0; i < urls.size(); ++i) |
| 184 for (size_t i = 0; i < urls.size(); ++i) { | 105 if (urls[i]->id() == id) { |
| 185 if (urls[i]->id() == id) { | 106 url = urls[i]; |
| 186 url = urls[i]; | 107 break; |
| 187 break; | |
| 188 } | |
| 189 } | 108 } |
| 190 } | 109 if (!url) |
| 191 if (!url && allow_fallback) { | |
| 192 url = url_service->FindNewDefaultSearchProvider(); | 110 url = url_service->FindNewDefaultSearchProvider(); |
| 193 DCHECK(url); | 111 url_service->SetDefaultSearchProvider(url); |
| 194 } | |
| 195 if (url) { | |
| 196 url_service->SetDefaultSearchProvider(url); | |
| 197 VLOG(1) << "Default search provider set to: " << url->short_name(); | |
| 198 } | |
| 199 return url; | |
| 200 } | |
| 201 | |
| 202 void DefaultSearchProviderChange::OpenSearchEngineSettings( | |
| 203 Protector* protector) { | |
| 204 protector->OpenTab( | |
| 205 GURL(std::string(chrome::kChromeUISettingsURL) + | |
| 206 chrome::kSearchEnginesSubPage)); | |
| 207 } | 112 } |
| 208 | 113 |
| 209 SettingChange* CreateDefaultSearchProviderChange( | 114 SettingChange* CreateDefaultSearchProviderChange( |
| 210 const TemplateURL* actual, | 115 const TemplateURL* actual, |
| 211 const TemplateURL* backup) { | 116 const TemplateURL* backup) { |
| 212 return new DefaultSearchProviderChange(backup, actual); | 117 return new DefaultSearchProviderChange(backup, actual); |
| 213 } | 118 } |
| 214 | 119 |
| 215 } // namespace protector | 120 } // namespace protector |
| OLD | NEW |