Chromium Code Reviews| 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) { |