Index: chrome/browser/ui/views/toolbar/app_menu_button.cc |
diff --git a/chrome/browser/ui/views/toolbar/app_menu_button.cc b/chrome/browser/ui/views/toolbar/app_menu_button.cc |
index 9db62bc1d38cccf88dee9237922739e7215db784..78820ce35db2569dd58873e3576a76462f85c39b 100644 |
--- a/chrome/browser/ui/views/toolbar/app_menu_button.cc |
+++ b/chrome/browser/ui/views/toolbar/app_menu_button.cc |
@@ -11,6 +11,7 @@ |
#include "base/time/time.h" |
#include "chrome/browser/themes/theme_properties.h" |
#include "chrome/browser/ui/browser.h" |
+#include "chrome/browser/ui/browser_otr_state.h" |
#include "chrome/browser/ui/layout_constants.h" |
#include "chrome/browser/ui/toolbar/app_menu_model.h" |
#include "chrome/browser/ui/views/extensions/browser_action_drag_data.h" |
@@ -50,10 +51,12 @@ AppMenuButton::AppMenuButton(ToolbarView* toolbar_view) |
AppMenuButton::~AppMenuButton() {} |
-void AppMenuButton::SetSeverity(AppMenuIconPainter::Severity severity, |
+void AppMenuButton::SetSeverity(AppMenuBadgeController::BadgeType type, |
+ AppMenuIconPainter::Severity severity, |
bool animate) { |
if (ui::MaterialDesignController::IsModeMaterial()) { |
severity_ = severity; |
+ type_ = type; |
UpdateIcon(); |
return; |
} |
@@ -151,8 +154,25 @@ void AppMenuButton::UpdateIcon() { |
break; |
} |
+ gfx::VectorIconId icon_id = gfx::VectorIconId::VECTOR_ICON_NONE; |
+ switch (type_) { |
+ case AppMenuBadgeController::BadgeType::NONE: |
+ icon_id = gfx::VectorIconId::BROWSER_TOOLS; |
Evan Stade
2016/06/03 22:06:20
nit: indent off
kylix_rd
2016/06/03 22:20:04
Done.
|
+ DCHECK(severity_ == AppMenuIconPainter::SEVERITY_NONE); |
+ break; |
+ case AppMenuBadgeController::BadgeType::UPGRADE_NOTIFICATION: |
+ icon_id = gfx::VectorIconId::UPDATE_MENU_SEVERITY_LOW_MEDIUM_HIGH; |
+ break; |
+ case AppMenuBadgeController::BadgeType::GLOBAL_ERROR: |
+ icon_id = gfx::VectorIconId::UPDATE_MENU_FAILED; |
+ break; |
+ case AppMenuBadgeController::BadgeType::INCOMPATIBILITY_WARNING: |
+ icon_id = gfx::VectorIconId::UPDATE_MENU_SEVERITY_WARNING; |
+ break; |
+ } |
+ |
SetImage(views::Button::STATE_NORMAL, |
- gfx::CreateVectorIcon(gfx::VectorIconId::BROWSER_TOOLS, color)); |
+ gfx::CreateVectorIcon(icon_id, 16, color)); |
} |
void AppMenuButton::SetTrailingMargin(int margin) { |