WebView context menu cleanup.

chrome/browser/extensions/api/web_view/chrome_web_view_internal_api.cc

 // Copyright 2014 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/extensions/api/web_view/chrome_web_view_internal_api.h"
 
 #include "chrome/browser/extensions/api/context_menus/context_menus_api.h"
 #include "chrome/browser/extensions/api/context_menus/context_menus_api_helpers.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/extensions/api/chrome_web_view_internal.h"
+#include "content/public/browser/render_process_host.h"
 #include "extensions/common/error_utils.h"
 
 namespace helpers = extensions::context_menus_api_helpers;
 namespace webview = extensions::api::chrome_web_view_internal;
 
 namespace extensions {
 
 // TODO(lazyboy): Add checks similar to
 // WebViewInternalExtensionFunction::RunAsyncSafe(WebViewGuest*).
 bool ChromeWebViewInternalContextMenusCreateFunction::RunAsync() {
   scoped_ptr<webview::ContextMenusCreate::Params> params(
       webview::ContextMenusCreate::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
 
   MenuItem::Id id(
       Profile::FromBrowserContext(browser_context())->IsOffTheRecord(),
-      MenuItem::ExtensionKey(extension_id(), params->instance_id));
+      MenuItem::ExtensionKey(
+          GetSenderWebContents()->GetRenderProcessHost()->GetID(),
+          params->instance_id));
 
   if (params->create_properties.id.get()) {
     id.string_uid = *params->create_properties.id;
   } else {
     // The Generated Id is added by web_view_internal_custom_bindings.js.
     base::DictionaryValue* properties = NULL;
     EXTENSION_FUNCTION_VALIDATE(args_->GetDictionary(1, &properties));
     EXTENSION_FUNCTION_VALIDATE(
         properties->GetInteger(helpers::kGeneratedIdKey, &id.uid));
   }
@@ skipping 10 matching lines @@
 }
 
 bool ChromeWebViewInternalContextMenusUpdateFunction::RunAsync() {
   scoped_ptr<webview::ContextMenusUpdate::Params> params(
       webview::ContextMenusUpdate::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
 
   Profile* profile = Profile::FromBrowserContext(browser_context());
   MenuItem::Id item_id(
       profile->IsOffTheRecord(),
-      MenuItem::ExtensionKey(extension_id(), params->instance_id));
+      MenuItem::ExtensionKey(
+          GetSenderWebContents()->GetRenderProcessHost()->GetID(),
+          params->instance_id));
 
   if (params->id.as_string)
     item_id.string_uid = *params->id.as_string;
   else if (params->id.as_integer)
     item_id.uid = *params->id.as_integer;
   else
     NOTREACHED();
 
   bool success = extensions::context_menus_api_helpers::UpdateMenuItem(
       params->update_properties, profile, extension(), item_id, &error_);
   SendResponse(success);
   return success;
 }
 
 bool ChromeWebViewInternalContextMenusRemoveFunction::RunAsync() {
   scoped_ptr<webview::ContextMenusRemove::Params> params(
       webview::ContextMenusRemove::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
 
   MenuManager* menu_manager =
       MenuManager::Get(Profile::FromBrowserContext(browser_context()));
 
   MenuItem::Id id(
       Profile::FromBrowserContext(browser_context())->IsOffTheRecord(),
-      MenuItem::ExtensionKey(extension_id(), params->instance_id));
+      MenuItem::ExtensionKey(
+          GetSenderWebContents()->GetRenderProcessHost()->GetID(),
+          params->instance_id));
 
   if (params->menu_item_id.as_string) {
     id.string_uid = *params->menu_item_id.as_string;
   } else if (params->menu_item_id.as_integer) {
     id.uid = *params->menu_item_id.as_integer;
   } else {
     NOTREACHED();
   }
 
   bool success = true;
@@ skipping 12 matching lines @@
   return success;
 }
 
 bool ChromeWebViewInternalContextMenusRemoveAllFunction::RunAsync() {
   scoped_ptr<webview::ContextMenusRemoveAll::Params> params(
       webview::ContextMenusRemoveAll::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
 
   MenuManager* menu_manager =
       MenuManager::Get(Profile::FromBrowserContext(browser_context()));
+  menu_manager->RemoveAllContextItems(MenuItem::ExtensionKey(
+      GetSenderWebContents()->GetRenderProcessHost()->GetID(),
+      params->instance_id));
 
-  int webview_instance_id = params->instance_id;
-  menu_manager->RemoveAllContextItems(
-      MenuItem::ExtensionKey(extension()->id(), webview_instance_id));
   SendResponse(true);
   return true;
 }
 
 ChromeWebViewInternalShowContextMenuFunction::
     ChromeWebViewInternalShowContextMenuFunction() {
 }
 
 ChromeWebViewInternalShowContextMenuFunction::
     ~ChromeWebViewInternalShowContextMenuFunction() {
 }
 
 bool ChromeWebViewInternalShowContextMenuFunction::RunAsyncSafe(
     WebViewGuest* guest) {
   scoped_ptr<webview::ShowContextMenu::Params> params(
       webview::ShowContextMenu::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
 
   // TODO(lazyboy): Actually implement filtering menu items, we pass NULL for
   // now.
   guest->ShowContextMenu(params->request_id, NULL);
 
   SendResponse(true);
   return true;
 }
 
 }  // namespace extensions