Hook the activeTab permission up to the omnibox, page action, and context menu APIs.

chrome/browser/extensions/extension_menu_manager.cc

 // Copyright (c) 2012 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/extension_menu_manager.h"
 
 #include <algorithm>
 
 #include "base/json/json_writer.h"
 #include "base/logging.h"
 #include "base/stl_util.h"
 #include "base/string_util.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
 #include "chrome/browser/extensions/extension_event_names.h"
 #include "chrome/browser/extensions/extension_event_router.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_system.h"
+#include "chrome/browser/extensions/extension_tab_helper.h"
 #include "chrome/browser/extensions/extension_tab_util.h"
 #include "chrome/browser/extensions/state_store.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/ui/tab_contents/tab_contents.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/extensions/extension.h"
 #include "content/public/browser/notification_details.h"
 #include "content/public/browser/notification_source.h"
+#include "content/public/browser/web_contents.h"
 #include "content/public/common/context_menu_params.h"
 #include "ui/base/text/text_elider.h"
 #include "ui/gfx/favicon_size.h"
 
 using content::WebContents;
 
 namespace {
 
 // Keys for serialization to and from Value to store in the preferences.
 const char kContextMenusKey[] = "context_menus";
@@ skipping 525 matching lines @@
 static void AddURLProperty(DictionaryValue* dictionary,
                            const std::string& key, const GURL& url) {
   if (!url.is_empty())
     dictionary->SetString(key, url.possibly_invalid_spec());
 }
 
 void ExtensionMenuManager::ExecuteCommand(
     Profile* profile,
     WebContents* web_contents,
     const content::ContextMenuParams& params,
-    const ExtensionMenuItem::Id& menuItemId) {
+    const ExtensionMenuItem::Id& menu_item_id) {
   ExtensionEventRouter* event_router = profile->GetExtensionEventRouter();
   if (!event_router)
     return;
 
-  ExtensionMenuItem* item = GetItemById(menuItemId);
+  ExtensionMenuItem* item = GetItemById(menu_item_id);
   if (!item)
     return;
 
+  // ExtensionService/Extension can be NULL in unit tests :(
+  ExtensionService* service =
+      ExtensionSystem::Get(profile_)->extension_service();
+  const extensions::Extension* extension = service ?
+      service->extensions()->GetByID(menu_item_id.extension_id) : NULL;
+
   if (item->type() == ExtensionMenuItem::RADIO)
     RadioItemSelected(item);
 
   ListValue args;
 
   DictionaryValue* properties = new DictionaryValue();
   SetIdKeyValue(properties, "menuItemId", item->id());
   if (item->parent_id())
     SetIdKeyValue(properties, "parentMenuItemId", item->id());
 
@@ skipping 16 matching lines @@
   AddURLProperty(properties, "frameUrl", params.frame_url);
 
   if (params.selection_text.length() > 0)
     properties->SetString("selectionText", params.selection_text);
 
   properties->SetBoolean("editable", params.is_editable);
 
   args.Append(properties);
 
   // Add the tab info to the argument list.
-  if (web_contents) {
+  // Note: web_contents only NULL in unit tests :(
+  if (web_contents)
     args.Append(ExtensionTabUtil::CreateTabValue(web_contents));
-  } else {
+  else
     args.Append(new DictionaryValue());
-  }
 
   if (item->type() == ExtensionMenuItem::CHECKBOX ||
       item->type() == ExtensionMenuItem::RADIO) {
     bool was_checked = item->checked();
     properties->SetBoolean("wasChecked", was_checked);
 
     // RADIO items always get set to true when you click on them, but CHECKBOX
     // items get their state toggled.
     bool checked =
         (item->type() == ExtensionMenuItem::RADIO) ? true : !was_checked;
 
     item->SetChecked(checked);
     properties->SetBoolean("checked", item->checked());
 
-    const extensions::Extension* extension = ExtensionSystem::Get(profile_)->
-        extension_service()->GetExtensionById(menuItemId.extension_id, false);
-    WriteToStorage(extension);
+    if (extension)
+      WriteToStorage(extension);
+  }
+
+  TabContents* tab_contents = web_contents ?
+      TabContents::FromWebContents(web_contents) : NULL;
+  if (tab_contents && extension) {
+    tab_contents->extension_tab_helper()->active_tab_permission_manager()->
+        GrantIfRequested(extension);
   }
 
   std::string json_args;
   base::JSONWriter::Write(&args, &json_args);
   event_router->DispatchEventToExtension(
       item->extension_id(), extension_event_names::kOnContextMenus,
       json_args, profile, GURL(),
       ExtensionEventRouter::USER_GESTURE_ENABLED);
   event_router->DispatchEventToExtension(
       item->extension_id(), extension_event_names::kOnContextMenuClicked,
@@ skipping 161 matching lines @@
       return true;
     if (extension_id == other.extension_id) {
       if (uid < other.uid)
         return true;
       if (uid == other.uid)
         return string_uid < other.string_uid;
     }
   }
   return false;
 }