src/chrome/browser/extensions/page_action_controller.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/extensions/page_action_controller.h"
   6
   7 #include "chrome/browser/chrome_notification_types.h"
   8 #include "chrome/browser/extensions/api/extension_action/extension_action_api.h"
   9 #include "chrome/browser/extensions/component_loader.h"
  10 #include "chrome/browser/extensions/extension_action.h"
  11 #include "chrome/browser/extensions/extension_action_manager.h"
  12 #include "chrome/browser/extensions/extension_service.h"
  13 #include "chrome/browser/extensions/extension_system.h"
  14 #include "chrome/browser/extensions/extension_tab_util.h"
  15 #include "chrome/browser/extensions/tab_helper.h"
  16 #include "chrome/browser/profiles/profile.h"
  17 #include "chrome/browser/sessions/session_id.h"
  18 #include "chrome/common/extensions/extension_set.h"
  19 #include "content/public/browser/invalidate_type.h"
  20 #include "content/public/browser/navigation_details.h"
  21 #include "content/public/browser/notification_service.h"
  22 #include "content/public/browser/web_contents.h"
  23
  24 namespace extensions {
  25
  26 PageActionController::PageActionController(content::WebContents* web_contents)
  27     : content::WebContentsObserver(web_contents) {}
  28
  29 PageActionController::~PageActionController() {}
  30
  31 std::vector<ExtensionAction*> PageActionController::GetCurrentActions() const {
  32   ExtensionService* service = GetExtensionService();
  33   if (!service)
  34     return std::vector<ExtensionAction*>();
  35
  36   // Accumulate the list of all page actions to display.
  37   std::vector<ExtensionAction*> current_actions;
  38
  39   ExtensionActionManager* extension_action_manager =
  40       ExtensionActionManager::Get(profile());
  41
  42   for (ExtensionSet::const_iterator i = service->extensions()->begin();
  43        i != service->extensions()->end(); ++i) {
  44     ExtensionAction* action =
  45         extension_action_manager->GetPageAction(*i->get());
  46     if (action)
  47       current_actions.push_back(action);
  48   }
  49
  50   return current_actions;
  51 }
  52
  53 LocationBarController::Action PageActionController::OnClicked(
  54     const std::string& extension_id, int mouse_button) {
  55   ExtensionService* service = GetExtensionService();
  56   if (!service)
  57     return ACTION_NONE;
  58
  59   const Extension* extension = service->extensions()->GetByID(extension_id);
  60   CHECK(extension);
  61   ExtensionAction* page_action =
  62       ExtensionActionManager::Get(profile())->GetPageAction(*extension);
  63   CHECK(page_action);
  64   int tab_id = ExtensionTabUtil::GetTabId(web_contents());
  65
  66   extensions::TabHelper::FromWebContents(web_contents())->
  67       active_tab_permission_granter()->GrantIfRequested(extension);
  68
  69   switch (mouse_button) {
  70     case 1:  // left
  71     case 2:  // middle
  72       if (page_action->HasPopup(tab_id))
  73         return ACTION_SHOW_POPUP;
  74
  75       ExtensionActionAPI::PageActionExecuted(
  76           profile(), *page_action, tab_id,
  77           web_contents()->GetURL().spec(), mouse_button);
  78       return ACTION_NONE;
  79
  80     case 3:  // right
  81       return extension->ShowConfigureContextMenus() ?
  82           ACTION_SHOW_CONTEXT_MENU : ACTION_NONE;
  83   }
  84
  85   return ACTION_NONE;
  86 }
  87
  88 void PageActionController::NotifyChange() {
  89   web_contents()->NotifyNavigationStateChanged(
  90       content::INVALIDATE_TYPE_PAGE_ACTIONS);
  91 }
  92
  93 void PageActionController::DidNavigateMainFrame(
  94     const content::LoadCommittedDetails& details,
  95     const content::FrameNavigateParams& params) {
  96   if (details.is_in_page)
  97     return;
  98
  99   const std::vector<ExtensionAction*> current_actions = GetCurrentActions();
 100
 101   if (current_actions.empty())
 102     return;
 103
 104   for (size_t i = 0; i < current_actions.size(); ++i) {
 105     current_actions[i]->ClearAllValuesForTab(
 106         SessionID::IdForTab(web_contents()));
 107   }
 108
 109   NotifyChange();
 110 }
 111
 112 Profile* PageActionController::profile() const {
 113   content::WebContents* web_contents = this->web_contents();
 114   if (web_contents)
 115     return Profile::FromBrowserContext(web_contents->GetBrowserContext());
 116
 117   return NULL;
 118 }
 119
 120 ExtensionService* PageActionController::GetExtensionService() const {
 121   Profile* profile = this->profile();
 122   if (profile)
 123     return ExtensionSystem::Get(profile)->extension_service();
 124
 125   return NULL;
 126 }
 127
 128 }  // namespace extensions