OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/extensions/location_bar_controller.h" | 5 #include "chrome/browser/extensions/location_bar_controller.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "chrome/browser/extensions/active_script_controller.h" | 8 #include "chrome/browser/extensions/active_script_controller.h" |
| 9 #include "chrome/browser/extensions/api/extension_action/extension_action_api.h" |
9 #include "chrome/browser/extensions/page_action_controller.h" | 10 #include "chrome/browser/extensions/page_action_controller.h" |
10 #include "chrome/browser/ui/browser_finder.h" | |
11 #include "chrome/browser/ui/browser_window.h" | |
12 #include "chrome/browser/ui/location_bar/location_bar.h" | |
13 #include "chrome/common/extensions/api/extension_action/action_info.h" | 11 #include "chrome/common/extensions/api/extension_action/action_info.h" |
14 #include "content/public/browser/invalidate_type.h" | |
15 #include "content/public/browser/navigation_details.h" | 12 #include "content/public/browser/navigation_details.h" |
16 #include "content/public/browser/notification_service.h" | |
17 #include "content/public/browser/notification_source.h" | |
18 #include "content/public/browser/web_contents.h" | 13 #include "content/public/browser/web_contents.h" |
19 #include "extensions/browser/extension_registry.h" | 14 #include "extensions/browser/extension_registry.h" |
20 #include "extensions/browser/notification_types.h" | |
21 | 15 |
22 namespace extensions { | 16 namespace extensions { |
23 | 17 |
24 LocationBarController::LocationBarController( | 18 LocationBarController::LocationBarController( |
25 content::WebContents* web_contents) | 19 content::WebContents* web_contents) |
26 : WebContentsObserver(web_contents), | 20 : WebContentsObserver(web_contents), |
27 web_contents_(web_contents), | 21 web_contents_(web_contents), |
28 active_script_controller_(new ActiveScriptController(web_contents_)), | 22 active_script_controller_(new ActiveScriptController(web_contents_)), |
29 page_action_controller_(new PageActionController(web_contents_)), | 23 page_action_controller_(new PageActionController(web_contents_)), |
30 extension_registry_observer_(this) { | 24 extension_registry_observer_(this) { |
(...skipping 19 matching lines...) Expand all Loading... |
50 page_action_controller_->GetActionForExtension(iter->get()); | 44 page_action_controller_->GetActionForExtension(iter->get()); |
51 if (!action) | 45 if (!action) |
52 action = active_script_controller_->GetActionForExtension(iter->get()); | 46 action = active_script_controller_->GetActionForExtension(iter->get()); |
53 if (action) | 47 if (action) |
54 current_actions.push_back(action); | 48 current_actions.push_back(action); |
55 } | 49 } |
56 | 50 |
57 return current_actions; | 51 return current_actions; |
58 } | 52 } |
59 | 53 |
60 // static | |
61 void LocationBarController::NotifyChange(content::WebContents* web_contents) { | |
62 Browser* browser = chrome::FindBrowserWithWebContents(web_contents); | |
63 if (!browser) | |
64 return; | |
65 LocationBar* location_bar = | |
66 browser->window() ? browser->window()->GetLocationBar() : NULL; | |
67 if (!location_bar) | |
68 return; | |
69 location_bar->UpdatePageActions(); | |
70 | |
71 content::NotificationService::current()->Notify( | |
72 NOTIFICATION_EXTENSION_PAGE_ACTIONS_UPDATED, | |
73 content::Source<content::WebContents>(web_contents), | |
74 content::NotificationService::NoDetails()); | |
75 } | |
76 | |
77 void LocationBarController::DidNavigateMainFrame( | 54 void LocationBarController::DidNavigateMainFrame( |
78 const content::LoadCommittedDetails& details, | 55 const content::LoadCommittedDetails& details, |
79 const content::FrameNavigateParams& params) { | 56 const content::FrameNavigateParams& params) { |
80 if (details.is_in_page) | 57 if (details.is_in_page) |
81 return; | 58 return; |
82 | 59 |
83 page_action_controller_->OnNavigated(); | 60 page_action_controller_->OnNavigated(); |
84 active_script_controller_->OnNavigated(); | 61 active_script_controller_->OnNavigated(); |
85 } | 62 } |
86 | 63 |
87 void LocationBarController::OnExtensionUnloaded( | 64 void LocationBarController::OnExtensionUnloaded( |
88 content::BrowserContext* browser_context, | 65 content::BrowserContext* browser_context, |
89 const Extension* extension, | 66 const Extension* extension, |
90 UnloadedExtensionInfo::Reason reason) { | 67 UnloadedExtensionInfo::Reason reason) { |
91 bool should_update = false; | 68 bool should_update = false; |
92 if (page_action_controller_->GetActionForExtension(extension)) { | 69 if (page_action_controller_->GetActionForExtension(extension)) { |
93 page_action_controller_->OnExtensionUnloaded(extension); | 70 page_action_controller_->OnExtensionUnloaded(extension); |
94 should_update = true; | 71 should_update = true; |
95 } | 72 } |
96 if (active_script_controller_->GetActionForExtension(extension)) { | 73 if (active_script_controller_->GetActionForExtension(extension)) { |
97 active_script_controller_->OnExtensionUnloaded(extension); | 74 active_script_controller_->OnExtensionUnloaded(extension); |
98 should_update = true; | 75 should_update = true; |
99 } | 76 } |
100 | 77 |
101 if (should_update) | 78 if (should_update) { |
102 NotifyChange(web_contents()); | 79 ExtensionActionAPI::Get(browser_context)-> |
| 80 NotifyPageActionsChanged(web_contents()); |
| 81 } |
103 } | 82 } |
104 | 83 |
105 } // namespace extensions | 84 } // namespace extensions |
OLD | NEW |