Chromium Code Reviews| 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 <algorithm> | 7 #include <algorithm> |
| 8 | 8 |
| 9 #include "chrome/browser/extensions/active_script_controller.h" | 9 #include "chrome/browser/extensions/active_script_controller.h" |
| 10 #include "chrome/browser/extensions/api/extension_action/extension_action_api.h" | 10 #include "chrome/browser/extensions/api/extension_action/extension_action_api.h" |
| (...skipping 12 matching lines...) Expand all Loading... | |
| 23 namespace extensions { | 23 namespace extensions { |
| 24 | 24 |
| 25 LocationBarController::LocationBarController( | 25 LocationBarController::LocationBarController( |
| 26 content::WebContents* web_contents) | 26 content::WebContents* web_contents) |
| 27 : web_contents_(web_contents), | 27 : web_contents_(web_contents), |
| 28 browser_context_(web_contents->GetBrowserContext()), | 28 browser_context_(web_contents->GetBrowserContext()), |
| 29 action_manager_(ExtensionActionManager::Get(browser_context_)), | 29 action_manager_(ExtensionActionManager::Get(browser_context_)), |
| 30 should_show_page_actions_( | 30 should_show_page_actions_( |
| 31 !FeatureSwitch::extension_action_redesign()->IsEnabled()), | 31 !FeatureSwitch::extension_action_redesign()->IsEnabled()), |
| 32 extension_registry_observer_(this) { | 32 extension_registry_observer_(this) { |
| 33 if (should_show_page_actions_) | 33 extension_registry_observer_.Add(ExtensionRegistry::Get(browser_context_)); |
| 34 extension_registry_observer_.Add(ExtensionRegistry::Get(browser_context_)); | |
| 35 } | 34 } |
| 36 | 35 |
| 37 LocationBarController::~LocationBarController() { | 36 LocationBarController::~LocationBarController() { |
| 38 } | 37 } |
| 39 | 38 |
| 40 std::vector<ExtensionAction*> LocationBarController::GetCurrentActions() { | 39 std::vector<ExtensionAction*> LocationBarController::GetCurrentActions() { |
| 41 const ExtensionSet& extensions = | 40 const ExtensionSet& extensions = |
| 42 ExtensionRegistry::Get(browser_context_)->enabled_extensions(); | 41 ExtensionRegistry::Get(browser_context_)->enabled_extensions(); |
| 43 std::vector<ExtensionAction*> current_actions; | 42 std::vector<ExtensionAction*> current_actions; |
| 44 if (!should_show_page_actions_) | 43 if (!should_show_page_actions_) |
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 86 ->permissions_data() | 85 ->permissions_data() |
| 87 ->HasAPIPermission( | 86 ->HasAPIPermission( |
| 88 extensions::APIPermission::kBookmarkManagerPrivate); | 87 extensions::APIPermission::kBookmarkManagerPrivate); |
| 89 }); | 88 }); |
| 90 return current_actions; | 89 return current_actions; |
| 91 } | 90 } |
| 92 | 91 |
| 93 void LocationBarController::OnExtensionLoaded( | 92 void LocationBarController::OnExtensionLoaded( |
| 94 content::BrowserContext* browser_context, | 93 content::BrowserContext* browser_context, |
| 95 const Extension* extension) { | 94 const Extension* extension) { |
| 96 if (action_manager_->GetPageAction(*extension)) { | 95 if (should_show_page_actions_ && action_manager_->GetPageAction(*extension)) { |
| 97 ExtensionActionAPI::Get(browser_context)-> | 96 ExtensionActionAPI::Get(browser_context)-> |
| 98 NotifyPageActionsChanged(web_contents_); | 97 NotifyPageActionsChanged(web_contents_); |
| 99 } | 98 } |
| 100 | 99 |
| 101 // We might also need to update the location bar if the extension can remove | 100 // We might also need to update the location bar if the extension can remove |
| 102 // the bookmark star. | 101 // the bookmark star. |
| 103 if (UIOverrides::RemovesBookmarkButton(extension)) { | 102 if (UIOverrides::RemovesBookmarkButton(extension)) { |
|
Devlin
2015/09/21 23:28:15
We need to do this part, even if the page actions
| |
| 104 Browser* browser = chrome::FindBrowserWithWebContents(web_contents_); | 103 Browser* browser = chrome::FindBrowserWithWebContents(web_contents_); |
| 105 // In a perfect world, this can never be NULL. Unfortunately, since a | 104 // In a perfect world, this can never be NULL. Unfortunately, since a |
| 106 // LocationBarController is attached to most WebContents, we can't make that | 105 // LocationBarController is attached to most WebContents, we can't make that |
| 107 // guarantee. | 106 // guarantee. |
| 108 if (!browser) | 107 if (!browser) |
| 109 return; | 108 return; |
| 110 // window() can be NULL if this is called before CreateBrowserWindow() | 109 // window() can be NULL if this is called before CreateBrowserWindow() |
| 111 // completes, and there won't be a location bar if the window has no toolbar | 110 // completes, and there won't be a location bar if the window has no toolbar |
| 112 // (e.g., and app window). | 111 // (e.g., and app window). |
| 113 LocationBar* location_bar = | 112 LocationBar* location_bar = |
| 114 browser->window() ? browser->window()->GetLocationBar() : NULL; | 113 browser->window() ? browser->window()->GetLocationBar() : NULL; |
| 115 if (!location_bar) | 114 if (!location_bar) |
| 116 return; | 115 return; |
| 117 location_bar->UpdateBookmarkStarVisibility(); | 116 location_bar->UpdateBookmarkStarVisibility(); |
| 118 } | 117 } |
| 119 } | 118 } |
| 120 | 119 |
| 121 void LocationBarController::OnExtensionUnloaded( | 120 void LocationBarController::OnExtensionUnloaded( |
| 122 content::BrowserContext* browser_context, | 121 content::BrowserContext* browser_context, |
| 123 const Extension* extension, | 122 const Extension* extension, |
| 124 UnloadedExtensionInfo::Reason reason) { | 123 UnloadedExtensionInfo::Reason reason) { |
| 125 if (action_manager_->GetPageAction(*extension)) { | 124 if (should_show_page_actions_ && action_manager_->GetPageAction(*extension)) { |
| 126 ExtensionActionAPI::Get(browser_context)-> | 125 ExtensionActionAPI::Get(browser_context)-> |
| 127 NotifyPageActionsChanged(web_contents_); | 126 NotifyPageActionsChanged(web_contents_); |
| 128 } | 127 } |
| 129 } | 128 } |
| 130 | 129 |
| 131 } // namespace extensions | 130 } // namespace extensions |
| OLD | NEW |