Add a policy to always show the Cast toolbar icon

chrome/browser/ui/toolbar/media_router_contextual_menu.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+#include "chrome/browser/ui/toolbar/media_router_contextual_menu.h"
+
 #include <string>
 
 #include "base/logging.h"
 #include "base/metrics/user_metrics.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "chrome/browser/extensions/component_migration_helper.h"
 #include "chrome/browser/media/router/media_router_factory.h"
 #include "chrome/browser/media/router/mojo/media_router_mojo_impl.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/signin/signin_manager_factory.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/singleton_tabs.h"
 #include "chrome/browser/ui/toolbar/component_toolbar_actions_factory.h"
-#include "chrome/browser/ui/toolbar/media_router_contextual_menu.h"
+#include "chrome/browser/ui/toolbar/media_router_action_controller.h"
 #include "chrome/browser/ui/toolbar/toolbar_actions_model.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/grit/chromium_strings.h"
 #include "chrome/grit/generated_resources.h"
 #include "components/signin/core/browser/signin_manager.h"
 #include "extensions/common/constants.h"
 #include "ui/base/models/menu_model_delegate.h"
-
+#include "ui/gfx/color_palette.h"
+#include "ui/gfx/paint_vector_icon.h"
 
 MediaRouterContextualMenu::MediaRouterContextualMenu(Browser* browser)
+    : MediaRouterContextualMenu(
+          browser,
+          MediaRouterActionController::IsActionShownByPolicy(
+              browser->profile())) {}
+
+MediaRouterContextualMenu::MediaRouterContextualMenu(Browser* browser,
+                                                     bool shown_by_policy)
     : browser_(browser),
       menu_model_(this),
       component_migration_helper_(ToolbarActionsModel::Get(browser->profile())
-                                      ->component_migration_helper()) {
+                                      ->component_migration_helper()),
+      shown_by_policy_(shown_by_policy) {

[msw, 2017/01/05 01:14:57]

This member isn't needed if it's only accessed in the ctor.

[takumif, 2017/01/06 22:06:57]

Right, removed.

   DCHECK(component_migration_helper_);
 
   menu_model_.AddItemWithStringId(IDC_MEDIA_ROUTER_ABOUT,
                                   IDS_MEDIA_ROUTER_ABOUT);
   menu_model_.AddSeparator(ui::NORMAL_SEPARATOR);
   menu_model_.AddItemWithStringId(IDC_MEDIA_ROUTER_LEARN_MORE,
                                   IDS_MEDIA_ROUTER_LEARN_MORE);
   menu_model_.AddItemWithStringId(IDC_MEDIA_ROUTER_HELP,
                                   IDS_MEDIA_ROUTER_HELP);
-  menu_model_.AddCheckItemWithStringId(
-      IDC_MEDIA_ROUTER_ALWAYS_SHOW_TOOLBAR_ACTION,
-      IDS_MEDIA_ROUTER_ALWAYS_SHOW_TOOLBAR_ACTION);
+  if (shown_by_policy_) {
+    menu_model_.AddItemWithStringId(IDC_MEDIA_ROUTER_SHOWN_BY_POLICY,
+                                    IDS_MEDIA_ROUTER_SHOWN_BY_POLICY);
+    menu_model_.SetIcon(
+        menu_model_.GetIndexOfCommandId(IDC_MEDIA_ROUTER_SHOWN_BY_POLICY),
+        gfx::Image(gfx::CreateVectorIcon(gfx::VectorIconId::BUSINESS, 16,
+                                         gfx::kChromeIconGrey)));
+  } else {
+    menu_model_.AddCheckItemWithStringId(
+        IDC_MEDIA_ROUTER_ALWAYS_SHOW_TOOLBAR_ACTION,
+        IDS_MEDIA_ROUTER_ALWAYS_SHOW_TOOLBAR_ACTION);
+  }
   menu_model_.AddSeparator(ui::NORMAL_SEPARATOR);
 #if defined(OS_WIN) || defined(OS_MACOSX) || defined(OS_CHROMEOS)
   menu_model_.AddItemWithStringId(IDC_MEDIA_ROUTER_MANAGE_DEVICES,
                                   IDS_MEDIA_ROUTER_MANAGE_DEVICES);
 #endif
   menu_model_.AddCheckItemWithStringId(IDC_MEDIA_ROUTER_CLOUD_SERVICES_TOGGLE,
                                        IDS_MEDIA_ROUTER_CLOUD_SERVICES_TOGGLE);
   menu_model_.AddItemWithStringId(IDC_MEDIA_ROUTER_REPORT_ISSUE,
                                   IDS_MEDIA_ROUTER_REPORT_ISSUE);
 }
 
 MediaRouterContextualMenu::~MediaRouterContextualMenu() {}
 
 bool MediaRouterContextualMenu::GetAlwaysShowActionPref() const {
   return component_migration_helper_->GetComponentActionPref(
       ComponentToolbarActionsFactory::kMediaRouterActionId);
 }
 
 void MediaRouterContextualMenu::SetAlwaysShowActionPref(bool always_show) {
   component_migration_helper_->SetComponentActionPref(
       ComponentToolbarActionsFactory::kMediaRouterActionId, always_show);
 }
 
 bool MediaRouterContextualMenu::IsCommandIdChecked(int command_id) const {
   if (command_id == IDC_MEDIA_ROUTER_CLOUD_SERVICES_TOGGLE) {
     return browser_->profile()->GetPrefs()->GetBoolean(
         prefs::kMediaRouterEnableCloudServices);
   }
-  if (command_id == IDC_MEDIA_ROUTER_ALWAYS_SHOW_TOOLBAR_ACTION) {
+  if (command_id == IDC_MEDIA_ROUTER_ALWAYS_SHOW_TOOLBAR_ACTION)
     return GetAlwaysShowActionPref();
-  }
   return false;
 }
 
 bool MediaRouterContextualMenu::IsCommandIdEnabled(int command_id) const {
-  return true;
+  return command_id != IDC_MEDIA_ROUTER_SHOWN_BY_POLICY;
 }
 
 bool MediaRouterContextualMenu::IsCommandIdVisible(int command_id) const {
   if (command_id == IDC_MEDIA_ROUTER_CLOUD_SERVICES_TOGGLE) {
     // Cloud services preference is not set or used if the user is not signed
     // in.
     SigninManagerBase* signin_manager =
         SigninManagerFactory::GetForProfile(browser_->profile());
     return signin_manager && signin_manager->IsAuthenticated();
   }
@@ skipping 55 matching lines @@
           media_router::MediaRouterFactory::GetApiForBrowserContext(
               static_cast<content::BrowserContext*>(browser_->profile())));
   if (media_router->media_route_provider_extension_id().empty())
     return;
   std::string feedback_url(extensions::kExtensionScheme +
                            std::string(url::kStandardSchemeSeparator) +
                            media_router->media_route_provider_extension_id() +
                            "/feedback.html");
   chrome::ShowSingletonTab(browser_, GURL(feedback_url));
 }