1 files changed, 313 insertions, 0 deletions

diff --git a/chrome/browser/ui/webui/options2/search_engine_manager_handler.cc b/chrome/browser/ui/webui/options2/search_engine_manager_handler.cc
new file mode 100644
index 0000000..1e289d8
--- /dev/null
+++ b/chrome/browser/ui/webui/options2/search_engine_manager_handler.cc
@@ -0,0 +1,313 @@
+// 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/webui/options2/search_engine_manager_handler.h"
+
+#include "base/bind.h"
+#include "base/string_number_conversions.h"
+#include "base/utf_string_conversions.h"
+#include "base/values.h"
+#include "chrome/browser/extensions/extension_service.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/ui/search_engines/keyword_editor_controller.h"
+#include "chrome/browser/ui/search_engines/template_url_table_model.h"
+#include "chrome/common/extensions/extension.h"
+#include "chrome/common/url_constants.h"
+#include "grit/generated_resources.h"
+#include "grit/locale_settings.h"
+#include "ui/base/l10n/l10n_util.h"
+
+namespace {
+
+enum EngineInfoIndexes {
+  ENGINE_NAME,
+  ENGINE_KEYWORD,
+  ENGINE_URL,
+};
+
+};  // namespace
+
+SearchEngineManagerHandler::SearchEngineManagerHandler() {
+}
+
+SearchEngineManagerHandler::~SearchEngineManagerHandler() {
+  if (list_controller_.get() && list_controller_->table_model())
+    list_controller_->table_model()->SetObserver(NULL);
+}
+
+void SearchEngineManagerHandler::Initialize() {
+  list_controller_.reset(
+      new KeywordEditorController(Profile::FromWebUI(web_ui_)));
+  if (list_controller_.get()) {
+    list_controller_->table_model()->SetObserver(this);
+    OnModelChanged();
+  }
+}
+
+void SearchEngineManagerHandler::GetLocalizedValues(
+    base::DictionaryValue* localized_strings) {
+  DCHECK(localized_strings);
+
+  RegisterTitle(localized_strings, "searchEngineManagerPage",
+                IDS_SEARCH_ENGINES_EDITOR_WINDOW_TITLE);
+  localized_strings->SetString("defaultSearchEngineListTitle",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINES_EDITOR_MAIN_SEPARATOR));
+  localized_strings->SetString("otherSearchEngineListTitle",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINES_EDITOR_OTHER_SEPARATOR));
+  localized_strings->SetString("extensionKeywordsListTitle",
+      l10n_util::GetStringUTF16(
+          IDS_SEARCH_ENGINES_EDITOR_EXTENSIONS_SEPARATOR));
+  localized_strings->SetString("manageExtensionsLinkText",
+      l10n_util::GetStringUTF16(IDS_MANAGE_EXTENSIONS));
+  localized_strings->SetString("searchEngineTableNameHeader",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINES_EDITOR_DESCRIPTION_COLUMN));
+  localized_strings->SetString("searchEngineTableKeywordHeader",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINES_EDITOR_KEYWORD_COLUMN));
+  localized_strings->SetString("searchEngineTableURLHeader",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINES_EDITOR_EDIT_BUTTON));
+  localized_strings->SetString("makeDefaultSearchEngineButton",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINES_EDITOR_MAKE_DEFAULT_BUTTON));
+  localized_strings->SetString("searchEngineTableNamePlaceholder",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINE_ADD_NEW_NAME_PLACEHOLDER));
+  localized_strings->SetString("searchEngineTableKeywordPlaceholder",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINE_ADD_NEW_KEYWORD_PLACEHOLDER));
+  localized_strings->SetString("searchEngineTableURLPlaceholder",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINE_ADD_NEW_URL_PLACEHOLDER));
+  localized_strings->SetString("editSearchEngineInvalidTitleToolTip",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINES_INVALID_TITLE_TT));
+  localized_strings->SetString("editSearchEngineInvalidKeywordToolTip",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINES_INVALID_KEYWORD_TT));
+  localized_strings->SetString("editSearchEngineInvalidURLToolTip",
+      l10n_util::GetStringUTF16(IDS_SEARCH_ENGINES_INVALID_URL_TT));
+}
+
+void SearchEngineManagerHandler::RegisterMessages() {
+  web_ui_->RegisterMessageCallback(
+      "managerSetDefaultSearchEngine",
+      base::Bind(&SearchEngineManagerHandler::SetDefaultSearchEngine,
+                 base::Unretained(this)));
+  web_ui_->RegisterMessageCallback(
+      "removeSearchEngine",
+      base::Bind(&SearchEngineManagerHandler::RemoveSearchEngine,
+                 base::Unretained(this)));
+  web_ui_->RegisterMessageCallback(
+      "editSearchEngine",
+      base::Bind(&SearchEngineManagerHandler::EditSearchEngine,
+                 base::Unretained(this)));
+  web_ui_->RegisterMessageCallback(
+      "checkSearchEngineInfoValidity",
+      base::Bind(&SearchEngineManagerHandler::CheckSearchEngineInfoValidity,
+                 base::Unretained(this)));
+  web_ui_->RegisterMessageCallback(
+      "searchEngineEditCancelled",
+      base::Bind(&SearchEngineManagerHandler::EditCancelled,
+                 base::Unretained(this)));
+  web_ui_->RegisterMessageCallback(
+      "searchEngineEditCompleted",
+      base::Bind(&SearchEngineManagerHandler::EditCompleted,
+                 base::Unretained(this)));
+}
+
+void SearchEngineManagerHandler::OnModelChanged() {
+  if (!list_controller_->loaded())
+    return;
+
+  // Find the default engine.
+  const TemplateURL* default_engine =
+      list_controller_->url_model()->GetDefaultSearchProvider();
+  int default_index = list_controller_->table_model()->IndexOfTemplateURL(
+      default_engine);
+
+  // Build the first list (default search engine options).
+  ListValue defaults_list;
+  int last_default_engine_index =
+      list_controller_->table_model()->last_search_engine_index();
+  for (int i = 0; i < last_default_engine_index; ++i) {
+    defaults_list.Append(CreateDictionaryForEngine(i, i == default_index));
+  }
+
+  // Build the second list (other search templates).
+  ListValue others_list;
+  if (last_default_engine_index < 0)
+    last_default_engine_index = 0;
+  int engine_count = list_controller_->table_model()->RowCount();
+  for (int i = last_default_engine_index; i < engine_count; ++i) {
+    others_list.Append(CreateDictionaryForEngine(i, i == default_index));
+  }
+
+  // Build the extension keywords list.
+  ListValue keyword_list;
+  ExtensionService* extension_service =
+      Profile::FromWebUI(web_ui_)->GetExtensionService();
+  if (extension_service) {
+    const ExtensionSet* extensions = extension_service->extensions();
+    for (ExtensionSet::const_iterator it = extensions->begin();
+         it != extensions->end(); ++it) {
+      if ((*it)->omnibox_keyword().size() > 0)
+        keyword_list.Append(CreateDictionaryForExtension(*(*it)));
+    }
+  }
+
+  web_ui_->CallJavascriptFunction("SearchEngineManager.updateSearchEngineList",
+                                  defaults_list, others_list, keyword_list);
+}
+
+void SearchEngineManagerHandler::OnItemsChanged(int start, int length) {
+  OnModelChanged();
+}
+
+void SearchEngineManagerHandler::OnItemsAdded(int start, int length) {
+  OnModelChanged();
+}
+
+void SearchEngineManagerHandler::OnItemsRemoved(int start, int length) {
+  OnModelChanged();
+}
+
+base::DictionaryValue* SearchEngineManagerHandler::CreateDictionaryForExtension(
+    const Extension& extension) {
+  base::DictionaryValue* dict = new base::DictionaryValue();
+  dict->SetString("name",  extension.name());
+  dict->SetString("displayName", extension.name());
+  dict->SetString("keyword", extension.omnibox_keyword());
+  GURL icon = extension.GetIconURL(16, ExtensionIconSet::MATCH_BIGGER);
+  dict->SetString("iconURL", icon.spec());
+  dict->SetString("url", string16());
+  return dict;
+}
+
+base::DictionaryValue* SearchEngineManagerHandler::CreateDictionaryForEngine(
+    int index, bool is_default) {
+  TemplateURLTableModel* table_model = list_controller_->table_model();
+  const TemplateURL* template_url = list_controller_->GetTemplateURL(index);
+
+  base::DictionaryValue* dict = new base::DictionaryValue();
+  dict->SetString("name",  template_url->short_name());
+  dict->SetString("displayName", table_model->GetText(
+    index, IDS_SEARCH_ENGINES_EDITOR_DESCRIPTION_COLUMN));
+  dict->SetString("keyword", table_model->GetText(
+    index, IDS_SEARCH_ENGINES_EDITOR_KEYWORD_COLUMN));
+  dict->SetString("url", template_url->url()->DisplayURL());
+  dict->SetBoolean("urlLocked", template_url->prepopulate_id() > 0);
+  GURL icon_url = template_url->GetFaviconURL();
+  if (icon_url.is_valid())
+    dict->SetString("iconURL", icon_url.spec());
+  dict->SetString("modelIndex", base::IntToString(index));
+
+  if (list_controller_->CanRemove(template_url))
+    dict->SetString("canBeRemoved", "1");
+  if (list_controller_->CanMakeDefault(template_url))
+    dict->SetString("canBeDefault", "1");
+  if (is_default)
+    dict->SetString("default", "1");
+  if (list_controller_->CanEdit(template_url))
+    dict->SetString("canBeEdited", "1");
+
+  return dict;
+}
+
+void SearchEngineManagerHandler::SetDefaultSearchEngine(const ListValue* args) {
+  int index;
+  if (!ExtractIntegerValue(args, &index)) {
+    NOTREACHED();
+    return;
+  }
+  if (index < 0 || index >= list_controller_->table_model()->RowCount())
+    return;
+
+  list_controller_->MakeDefaultTemplateURL(index);
+}
+
+void SearchEngineManagerHandler::RemoveSearchEngine(const ListValue* args) {
+  int index;
+  if (!ExtractIntegerValue(args, &index)) {
+    NOTREACHED();
+    return;
+  }
+  if (index < 0 || index >= list_controller_->table_model()->RowCount())
+    return;
+
+  if (list_controller_->CanRemove(list_controller_->GetTemplateURL(index)))
+    list_controller_->RemoveTemplateURL(index);
+}
+
+void SearchEngineManagerHandler::EditSearchEngine(const ListValue* args) {
+  int index;
+  if (!ExtractIntegerValue(args, &index)) {
+    NOTREACHED();
+    return;
+  }
+  // Allow -1, which means we are adding a new engine.
+  if (index < -1 || index >= list_controller_->table_model()->RowCount())
+    return;
+
+  const TemplateURL* edit_url = NULL;
+  if (index != -1)
+    edit_url = list_controller_->GetTemplateURL(index);
+  edit_controller_.reset(new EditSearchEngineController(
+      edit_url, this, Profile::FromWebUI(web_ui_)));
+}
+
+void SearchEngineManagerHandler::OnEditedKeyword(
+    const TemplateURL* template_url,
+    const string16& title,
+    const string16& keyword,
+    const std::string& url) {
+  if (template_url) {
+    list_controller_->ModifyTemplateURL(template_url, title, keyword, url);
+  } else {
+    list_controller_->AddTemplateURL(title, keyword, url);
+  }
+  edit_controller_.reset();
+}
+
+void SearchEngineManagerHandler::CheckSearchEngineInfoValidity(
+    const ListValue* args)
+{
+  if (!edit_controller_.get())
+    return;
+  string16 name;
+  string16 keyword;
+  std::string url;
+  std::string modelIndex;
+  if (!args->GetString(ENGINE_NAME, &name) ||
+      !args->GetString(ENGINE_KEYWORD, &keyword) ||
+      !args->GetString(ENGINE_URL, &url) ||
+      !args->GetString(3, &modelIndex)) {
+    NOTREACHED();
+    return;
+  }
+
+  base::DictionaryValue validity;
+  validity.SetBoolean("name", edit_controller_->IsTitleValid(name));
+  validity.SetBoolean("keyword", edit_controller_->IsKeywordValid(keyword));
+  validity.SetBoolean("url", edit_controller_->IsURLValid(url));
+  StringValue indexValue(modelIndex);
+  web_ui_->CallJavascriptFunction("SearchEngineManager.validityCheckCallback",
+                                  validity, indexValue);
+}
+
+void SearchEngineManagerHandler::EditCancelled(const ListValue* args) {
+  if (!edit_controller_.get())
+    return;
+  edit_controller_->CleanUpCancelledAdd();
+  edit_controller_.reset();
+}
+
+void SearchEngineManagerHandler::EditCompleted(const ListValue* args) {
+  if (!edit_controller_.get())
+    return;
+  string16 name;
+  string16 keyword;
+  std::string url;
+  if (!args->GetString(ENGINE_NAME, &name) ||
+      !args->GetString(ENGINE_KEYWORD, &keyword) ||
+      !args->GetString(ENGINE_URL, &url)) {
+    NOTREACHED();
+    return;
+  }
+  edit_controller_->AcceptAddOrEdit(name, keyword, url);
+}