Improve settings override bubble to indicate policy installed extensions

chrome/browser/ui/extensions/extension_message_bubble_bridge.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/extensions/extension_message_bubble_bridge.h"
 
 #include <utility>
 
+#include "base/memory/ptr_util.h"
 #include "chrome/browser/extensions/extension_message_bubble_controller.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/grit/generated_resources.h"
+#include "components/grit/components_scaled_resources.h"
+#include "extensions/browser/extension_registry.h"
+#include "ui/base/l10n/l10n_util.h"
+#include "ui/base/resource/resource_bundle.h"
 
 ExtensionMessageBubbleBridge::ExtensionMessageBubbleBridge(
     std::unique_ptr<extensions::ExtensionMessageBubbleController> controller)
     : controller_(std::move(controller)) {}
 
 ExtensionMessageBubbleBridge::~ExtensionMessageBubbleBridge() {}
 
 bool ExtensionMessageBubbleBridge::ShouldShow() {
   return controller_->ShouldShow();
 }
@@ skipping 10 matching lines @@
     bool anchored_to_action) {
   return controller_->delegate()->GetMessageBody(
       anchored_to_action, controller_->GetExtensionIdList().size());
 }
 
 base::string16 ExtensionMessageBubbleBridge::GetItemListText() {
   return controller_->GetExtensionListForDisplay();
 }
 
 base::string16 ExtensionMessageBubbleBridge::GetActionButtonText() {
-  return controller_->delegate()->GetActionButtonLabel();
+  const extensions::ExtensionIdList& list = controller_->GetExtensionIdList();

[catmullings, 2016/09/08 15:42:30]

Lines 47 - 59 are obviously duplicated code from the GetExtraViewInfo() method
below. If this is the proper way and place to check whether to remove  the
"Restore Settings" button, then I will tuck this code into a helper function.
WDYT?

[Devlin, 2016/09/12 22:51:43]

Sounds reasonable to me, though note that you have a few stragglers here (e.g.
line 55)

[catmullings, 2016/09/22 00:00:10]

Done.

[catmullings, 2016/09/22 00:00:10]

Mkay. Line 55 should be the only straggler, since I need the list variable to
construct the extension variable, which is used to do the IsPolicyLocation()
check. 

+  const extensions::Extension* extension =
+      extensions::ExtensionRegistry::Get(controller_->profile())
+          ->enabled_extensions()
+          .GetByID(list[0]);
+
+  DCHECK(extension);
+
+  std::unique_ptr<ExtraViewInfo> extra_view_info =
+      base::MakeUnique<ExtraViewInfo>();
+
+  if (controller_->delegate()->SupportsPolicyIndicator() &&
+      extensions::Manifest::IsPolicyLocation(extension->location()))
+    return base::string16();

[catmullings, 2016/09/08 15:42:30]

Initial attempt to remove the "Restore Settings" button if an extension is
policy installed. 

+  else
+    return controller_->delegate()->GetActionButtonLabel();
 }
 
 base::string16 ExtensionMessageBubbleBridge::GetDismissButtonText() {
   return controller_->delegate()->GetDismissButtonLabel();
 }
 
 base::string16 ExtensionMessageBubbleBridge::GetLearnMoreButtonText() {
   return controller_->delegate()->GetLearnMoreLabel();
 }
 
@@ skipping 16 matching lines @@
       break;
     }
     case CLOSE_EXECUTE:
       controller_->OnBubbleAction();
       break;
     case CLOSE_LEARN_MORE:
       controller_->OnLinkClicked();
       break;
   }
 }
+
+std::unique_ptr<ToolbarActionsBarBubbleDelegate::ExtraViewInfo>
+ExtensionMessageBubbleBridge::GetExtraViewInfo() {
+  const extensions::ExtensionIdList& list = controller_->GetExtensionIdList();
+  const extensions::Extension* extension =
+      extensions::ExtensionRegistry::Get(controller_->profile())
+          ->enabled_extensions()
+          .GetByID(list[0]);
+
+  DCHECK(extension);
+
+  std::unique_ptr<ExtraViewInfo> extra_view_info =
+      base::MakeUnique<ExtraViewInfo>();
+
+  if (controller_->delegate()->SupportsPolicyIndicator() &&
+      extensions::Manifest::IsPolicyLocation(extension->location())) {
+    DCHECK_EQ(1u, list.size());
+    extra_view_info->resource_id = IDR_OMNIBOX_HTTPS_POLICY_WARNING;
+    extra_view_info->text =
+        l10n_util::GetStringUTF16(IDS_EXTENSIONS_INSTALLED_BY_ADMIN);
+    extra_view_info->is_text_linked = false;
+  } else {
+    extra_view_info->text = controller_->delegate()->GetLearnMoreLabel();
+    extra_view_info->is_text_linked = true;
+  }
+
+  return extra_view_info;
+}