1 // Copyright 2014 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/location_bar_controller.h"
6
7 #include "base/logging.h"
8 #include "chrome/browser/extensions/active_script_controller.h"
9 #include "chrome/browser/extensions/extension_action.h"
10 #include "chrome/browser/extensions/page_action_controller.h"
11 #include "chrome/common/extensions/api/extension_action/action_info.h"
12 #include "content/public/browser/invalidate_type.h"
13 #include "content/public/browser/navigation_details.h"
14 #include "content/public/browser/web_contents.h"
15 #include "extensions/browser/extension_registry.h"
16
17 namespace extensions {
18
LocationBarController(content::WebContents * web_contents)19 LocationBarController::LocationBarController(
20 content::WebContents* web_contents)
21 : WebContentsObserver(web_contents),
22 web_contents_(web_contents),
23 active_script_controller_(new ActiveScriptController(web_contents_)),
24 page_action_controller_(new PageActionController(web_contents_)),
25 extension_registry_observer_(this) {
26 extension_registry_observer_.Add(
27 ExtensionRegistry::Get(web_contents_->GetBrowserContext()));
28 }
29
~LocationBarController()30 LocationBarController::~LocationBarController() {
31 }
32
GetCurrentActions()33 std::vector<ExtensionAction*> LocationBarController::GetCurrentActions() {
34 const ExtensionSet& extensions =
35 ExtensionRegistry::Get(web_contents_->GetBrowserContext())
36 ->enabled_extensions();
37 std::vector<ExtensionAction*> current_actions;
38 for (ExtensionSet::const_iterator iter = extensions.begin();
39 iter != extensions.end();
40 ++iter) {
41 // Right now, we can consolidate these actions because we only want to show
42 // one action per extension. If clicking on an active script action ever
43 // has a response, then we will need to split the actions.
44 ExtensionAction* action =
45 page_action_controller_->GetActionForExtension(*iter);
46 if (!action)
47 action = active_script_controller_->GetActionForExtension(*iter);
48 if (action)
49 current_actions.push_back(action);
50 }
51
52 return current_actions;
53 }
54
OnClicked(const ExtensionAction * action)55 LocationBarController::Action LocationBarController::OnClicked(
56 const ExtensionAction* action) {
57 const Extension* extension =
58 ExtensionRegistry::Get(web_contents_->GetBrowserContext())
59 ->enabled_extensions().GetByID(action->extension_id());
60 CHECK(extension) << action->extension_id();
61
62 Action page_action =
63 page_action_controller_->GetActionForExtension(extension) ?
64 page_action_controller_->OnClicked(extension) :
65 ACTION_NONE;
66 Action active_script_action =
67 active_script_controller_->GetActionForExtension(extension) ?
68 active_script_controller_->OnClicked(extension) :
69 ACTION_NONE;
70
71 // PageAction response takes priority.
72 return page_action != ACTION_NONE ? page_action : active_script_action;
73 }
74
75 // static
NotifyChange(content::WebContents * web_contents)76 void LocationBarController::NotifyChange(content::WebContents* web_contents) {
77 web_contents->NotifyNavigationStateChanged(
78 content::INVALIDATE_TYPE_PAGE_ACTIONS);
79 }
80
DidNavigateMainFrame(const content::LoadCommittedDetails & details,const content::FrameNavigateParams & params)81 void LocationBarController::DidNavigateMainFrame(
82 const content::LoadCommittedDetails& details,
83 const content::FrameNavigateParams& params) {
84 if (details.is_in_page)
85 return;
86
87 page_action_controller_->OnNavigated();
88 active_script_controller_->OnNavigated();
89 }
90
OnExtensionUnloaded(content::BrowserContext * browser_context,const Extension * extension,UnloadedExtensionInfo::Reason reason)91 void LocationBarController::OnExtensionUnloaded(
92 content::BrowserContext* browser_context,
93 const Extension* extension,
94 UnloadedExtensionInfo::Reason reason) {
95 bool should_update = false;
96 if (page_action_controller_->GetActionForExtension(extension)) {
97 page_action_controller_->OnExtensionUnloaded(extension);
98 should_update = true;
99 }
100 if (active_script_controller_->GetActionForExtension(extension)) {
101 active_script_controller_->OnExtensionUnloaded(extension);
102 should_update = true;
103 }
104
105 if (should_update)
106 NotifyChange(web_contents());
107 }
108
109 } // namespace extensions
110