Add the feature "Add as seach engine..." when right clicking on ...

chrome/browser/tab_contents/render_view_context_menu.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 <algorithm>
 #include <set>
 
 #include "chrome/browser/tab_contents/render_view_context_menu.h"
 
 #include "base/command_line.h"
@@ skipping 24 matching lines @@
 #include "chrome/browser/profiles/profile_io_data.h"
 #include "chrome/browser/search_engines/template_url.h"
 #include "chrome/browser/search_engines/template_url_service.h"
 #include "chrome/browser/search_engines/template_url_service_factory.h"
 #include "chrome/browser/spellcheck_host.h"
 #include "chrome/browser/spellcheck_host_metrics.h"
 #include "chrome/browser/spellchecker_platform_engine.h"
 #include "chrome/browser/translate/translate_manager.h"
 #include "chrome/browser/translate/translate_prefs.h"
 #include "chrome/browser/translate/translate_tab_helper.h"
+#include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/download/download_tab_helper.h"
+#include "chrome/browser/ui/search_engines/search_engine_tab_helper.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/print_messages.h"
 #include "chrome/common/spellcheck_messages.h"
 #include "chrome/common/url_constants.h"
 #include "content/browser/child_process_security_policy.h"
 #include "content/browser/renderer_host/render_view_host.h"
 #include "content/browser/renderer_host/render_widget_host_view.h"
@@ skipping 722 matching lines @@
   menu_model_.AddItemWithStringId(IDC_CONTENT_CONTEXT_CUT,
                                   IDS_CONTENT_CONTEXT_CUT);
   menu_model_.AddItemWithStringId(IDC_CONTENT_CONTEXT_COPY,
                                   IDS_CONTENT_CONTEXT_COPY);
   menu_model_.AddItemWithStringId(IDC_CONTENT_CONTEXT_PASTE,
                                   IDS_CONTENT_CONTEXT_PASTE);
   menu_model_.AddItemWithStringId(IDC_CONTENT_CONTEXT_DELETE,
                                   IDS_CONTENT_CONTEXT_DELETE);
   menu_model_.AddSeparator();
 
+  if (!params_.keyword_url.is_empty()) {
+    menu_model_.AddItemWithStringId(IDC_CONTENT_CONTEXT_ADDSEARCHENGINE,
+                                    IDS_CONTENT_CONTEXT_ADDSEARCHENGINE);
+    menu_model_.AddSeparator();
+  }
+
   AppendSpellcheckOptionsSubMenu();
 
 #if defined(OS_MACOSX)
   // OS X provides a contextual menu to set writing direction for BiDi
   // languages.
   // This functionality is exposed as a keyboard shortcut on Windows & Linux.
   AppendBidiSubMenu();
 #endif  // OS_MACOSX
 
   menu_model_.AddSeparator();
@@ skipping 372 matching lines @@
     case IDC_CONTENT_CONTEXT_LOOK_UP_IN_DICTIONARY:
       // This is OK because the menu is not shown when it isn't
       // appropriate.
       return true;
 #elif defined(OS_POSIX)
     // TODO(suzhe): this should not be enabled for password fields.
     case IDC_INPUT_METHODS_MENU:
       return true;
 #endif
 
+    case IDC_CONTENT_CONTEXT_ADDSEARCHENGINE:
+      return !params_.keyword_url.is_empty();
+
     case IDC_SPELLCHECK_MENU:
       return true;
 
     case IDC_CONTENT_CONTEXT_OPENLINKWITH:
       return true;
 
     case IDC_CONTENT_CONTEXT_PROTOCOL_HANDLER_SETTINGS:
       return true;
 
     default:
@@ skipping 425 matching lines @@
       LookUpInDictionary();
       break;
 #endif  // OS_MACOSX
     case IDC_CONTENT_CONTEXT_PROTOCOL_HANDLER_SETTINGS: {
       std::string url = std::string(chrome::kChromeUISettingsURL) +
           chrome::kHandlerSettingsSubPage;
       OpenURL(GURL(url), GURL(), NEW_FOREGROUND_TAB, PageTransition::LINK);
       break;
     }
 
+    case IDC_CONTENT_CONTEXT_ADDSEARCHENGINE: {
+      // Make sure the model is loaded.
+      TemplateURLService* model =
+          TemplateURLServiceFactory::GetForProfile(profile_);
+      if (!model)
+        return;
+      model->Load();
+
+      scoped_ptr<TemplateURL> template_url(new TemplateURL);
+      string16 keyword =
+          net::StripWWW(UTF8ToUTF16((params_.page_url.host())));
+      template_url->set_short_name(keyword);
+      template_url->set_keyword(keyword);
+      template_url->SetURL(params_.keyword_url.spec(), 0, 0);
+      template_url->SetFaviconURL(TemplateURL::GenerateFaviconURL(
+          params_.page_url.GetOrigin()));
+
+      TabContentsWrapper* tab_contents_wrapper =
+          TabContentsWrapper::GetCurrentWrapperForContents(
+              source_tab_contents_);
+      if (tab_contents_wrapper &&
+          tab_contents_wrapper->search_engine_tab_helper() &&
+          tab_contents_wrapper->search_engine_tab_helper()->delegate()) {
+        // Takes ownership of |template_url|.
+        tab_contents_wrapper->search_engine_tab_helper()->delegate()->
+            ConfirmAddSearchProvider(template_url.release(), profile_);
+      }
+      break;
+    }
+
     default:
       NOTREACHED();
       break;
   }
 }
 
 ProtocolHandlerRegistry::ProtocolHandlerList
 RenderViewContextMenu::GetHandlersForLinkUrl() {
   ProtocolHandlerRegistry::ProtocolHandlerList handlers =
       protocol_handler_registry_->GetHandlersFor(params_.link_url.scheme());
@@ skipping 78 matching lines @@
       g_browser_process->clipboard());
 }
 
 void RenderViewContextMenu::MediaPlayerActionAt(
     const gfx::Point& location,
     const WebMediaPlayerAction& action) {
   RenderViewHost* rvh = source_tab_contents_->render_view_host();
   rvh->Send(new ViewMsg_MediaPlayerActionAt(
       rvh->routing_id(), location, action));
 }