chrome/browser/ui/toolbar/media_router_action.cc

   1 // Copyright 2015 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/ui/toolbar/media_router_action.h"
   6
   7 #include "base/strings/utf_string_conversions.h"
   8 #include "chrome/browser/media/router/issue.h"
   9 #include "chrome/browser/media/router/media_route.h"
  10 #include "chrome/browser/media/router/media_router.h"
  11 #include "chrome/browser/media/router/media_router_factory.h"
  12 #include "chrome/browser/media/router/media_router_mojo_impl.h"
  13 #include "chrome/browser/profiles/profile.h"
  14 #include "chrome/browser/ui/browser.h"
  15 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  16 #include "chrome/browser/ui/toolbar/component_toolbar_actions_factory.h"
  17 #include "chrome/browser/ui/toolbar/media_router_action_platform_delegate.h"
  18 #include "chrome/browser/ui/toolbar/toolbar_action_view_delegate.h"
  19 #include "chrome/browser/ui/webui/media_router/media_router_dialog_controller_impl.h"
  20 #include "chrome/grit/generated_resources.h"
  21 #include "grit/theme_resources.h"
  22 #include "ui/base/l10n/l10n_util.h"
  23 #include "ui/base/resource/resource_bundle.h"
  24 #include "ui/gfx/image/image_skia.h"
  25
  26 using media_router::MediaRouterDialogControllerImpl;
  27
  28 MediaRouterAction::MediaRouterAction(Browser* browser)
  29     : media_router::IssuesObserver(GetMediaRouter(browser)),
  30       media_router::MediaRoutesObserver(GetMediaRouter(browser)),
  31       media_router_active_icon_(
  32           ui::ResourceBundle::GetSharedInstance()
  33               .GetImageNamed(IDR_MEDIA_ROUTER_ACTIVE_ICON)),
  34       media_router_error_icon_(ui::ResourceBundle::GetSharedInstance()
  35                                    .GetImageNamed(IDR_MEDIA_ROUTER_ERROR_ICON)),
  36       media_router_idle_icon_(ui::ResourceBundle::GetSharedInstance()
  37                                   .GetImageNamed(IDR_MEDIA_ROUTER_IDLE_ICON)),
  38       media_router_warning_icon_(
  39           ui::ResourceBundle::GetSharedInstance()
  40               .GetImageNamed(IDR_MEDIA_ROUTER_WARNING_ICON)),
  41       current_icon_(&media_router_idle_icon_),
  42       has_local_route_(false),
  43       delegate_(nullptr),
  44       browser_(browser),
  45       platform_delegate_(MediaRouterActionPlatformDelegate::Create(browser)),
  46       contextual_menu_(browser),
  47       tab_strip_model_observer_(this),
  48       weak_ptr_factory_(this) {
  49   DCHECK(browser_);
  50   tab_strip_model_observer_.Add(browser_->tab_strip_model());
  51 }
  52
  53 MediaRouterAction::~MediaRouterAction() {
  54 }
  55
  56 std::string MediaRouterAction::GetId() const {
  57   return ComponentToolbarActionsFactory::kMediaRouterActionId;
  58 }
  59
  60 void MediaRouterAction::SetDelegate(ToolbarActionViewDelegate* delegate) {
  61   delegate_ = delegate;
  62
  63   // Updates the current popup state if |delegate_| is non-null and has
  64   // WebContents ready.
  65   // In cases such as opening a new browser window, SetDelegate() will be
  66   // called before the WebContents is set. In those cases, we update the popup
  67   // state when ActiveTabChanged() is called.
  68   if (delegate_ && delegate_->GetCurrentWebContents())
  69     UpdatePopupState();
  70 }
  71
  72 gfx::Image MediaRouterAction::GetIcon(content::WebContents* web_contents,
  73                                       const gfx::Size& size) {
  74   return *current_icon_;
  75 }
  76
  77 base::string16 MediaRouterAction::GetActionName() const {
  78   return l10n_util::GetStringUTF16(IDS_MEDIA_ROUTER_TITLE);
  79 }
  80
  81 base::string16 MediaRouterAction::GetAccessibleName(
  82     content::WebContents* web_contents) const {
  83   return GetTooltip(web_contents);
  84 }
  85
  86 base::string16 MediaRouterAction::GetTooltip(
  87     content::WebContents* web_contents) const {
  88   return l10n_util::GetStringUTF16(IDS_MEDIA_ROUTER_SHARE_YOUR_SCREEN_TEXT);
  89 }
  90
  91 bool MediaRouterAction::IsEnabled(
  92     content::WebContents* web_contents) const {
  93   return true;
  94 }
  95
  96 bool MediaRouterAction::WantsToRun(
  97     content::WebContents* web_contents) const {
  98   return false;
  99 }
 100
 101 bool MediaRouterAction::HasPopup(
 102     content::WebContents* web_contents) const {
 103   return true;
 104 }
 105
 106 void MediaRouterAction::HidePopup() {
 107   GetMediaRouterDialogController()->HideMediaRouterDialog();
 108   OnPopupHidden();
 109 }
 110
 111 gfx::NativeView MediaRouterAction::GetPopupNativeView() {
 112   return nullptr;
 113 }
 114
 115 ui::MenuModel* MediaRouterAction::GetContextMenu() {
 116   return contextual_menu_.menu_model();
 117 }
 118
 119 bool MediaRouterAction::ExecuteAction(bool by_user) {
 120   GetMediaRouterDialogController()->ShowMediaRouterDialog();
 121   if (GetPlatformDelegate())
 122     GetPlatformDelegate()->CloseOverflowMenuIfOpen();
 123   return true;
 124 }
 125
 126 void MediaRouterAction::UpdateState() {
 127   if (delegate_)
 128     delegate_->UpdateState();
 129 }
 130
 131 bool MediaRouterAction::DisabledClickOpensMenu() const {
 132   return false;
 133 }
 134
 135 void MediaRouterAction::OnIssueUpdated(const media_router::Issue* issue) {
 136   issue_.reset(issue ? new media_router::Issue(*issue) : nullptr);
 137
 138   MaybeUpdateIcon();
 139 }
 140
 141 void MediaRouterAction::OnRoutesUpdated(
 142     const std::vector<media_router::MediaRoute>& routes) {
 143   has_local_route_ =
 144       std::find_if(routes.begin(), routes.end(),
 145                    [](const media_router::MediaRoute& route) {
 146                       return route.is_local(); }) !=
 147       routes.end();
 148   MaybeUpdateIcon();
 149 }
 150
 151 void MediaRouterAction::ActiveTabChanged(content::WebContents* old_contents,
 152                                          content::WebContents* new_contents,
 153                                          int index,
 154                                          int reason) {
 155   UpdatePopupState();
 156 }
 157
 158 void MediaRouterAction::OnPopupHidden() {
 159   if (delegate_)
 160     delegate_->OnPopupClosed();
 161 }
 162
 163 void MediaRouterAction::OnPopupShown() {
 164   // We depress the action regardless of whether ExecuteAction() was user
 165   // initiated.
 166   if (delegate_)
 167     delegate_->OnPopupShown(true);
 168 }
 169
 170 void MediaRouterAction::UpdatePopupState() {
 171   MediaRouterDialogControllerImpl* controller =
 172       GetMediaRouterDialogController();
 173
 174   if (!controller)
 175     return;
 176
 177   // Immediately keep track of MediaRouterAction in the controller. If it was
 178   // already set, this should be a no-op.
 179   controller->SetMediaRouterAction(weak_ptr_factory_.GetWeakPtr());
 180
 181   // Update the button in case the pressed state is out of sync with dialog
 182   // visibility.
 183   if (controller->IsShowingMediaRouterDialog())
 184     OnPopupShown();
 185   else
 186     OnPopupHidden();
 187 }
 188
 189 MediaRouterDialogControllerImpl*
 190 MediaRouterAction::GetMediaRouterDialogController() {
 191   DCHECK(delegate_);
 192   content::WebContents* web_contents = delegate_->GetCurrentWebContents();
 193   DCHECK(web_contents);
 194   return MediaRouterDialogControllerImpl::GetOrCreateForWebContents(
 195       web_contents);
 196 }
 197
 198 media_router::MediaRouter* MediaRouterAction::GetMediaRouter(Browser* browser) {
 199   return media_router::MediaRouterFactory::GetApiForBrowserContext(
 200       static_cast<content::BrowserContext*>(browser->profile()));
 201 }
 202
 203 MediaRouterActionPlatformDelegate* MediaRouterAction::GetPlatformDelegate() {
 204   return platform_delegate_.get();
 205 }
 206
 207 void MediaRouterAction::MaybeUpdateIcon() {
 208   const gfx::Image* new_icon = GetCurrentIcon();
 209
 210   // Update the current state if it has changed.
 211   if (new_icon != current_icon_) {
 212     current_icon_ = new_icon;
 213
 214     // Tell the associated view to update its icon to reflect the change made
 215     // above.
 216     if (delegate_)
 217       delegate_->UpdateState();
 218   }
 219 }
 220
 221 const gfx::Image* MediaRouterAction::GetCurrentIcon() const {
 222   // Highest priority is to indicate whether there's an issue.
 223   if (issue_) {
 224     if (issue_->severity() == media_router::Issue::FATAL)
 225       return &media_router_error_icon_;
 226     if (issue_->severity() == media_router::Issue::WARNING)
 227       return &media_router_warning_icon_;
 228   }
 229
 230   return has_local_route_ ?
 231       &media_router_active_icon_ : &media_router_idle_icon_;
 232 }