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..5e0aa5469f3c4f345119071084b644d992b0c138 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; |
| } |
| @@ -133,7 +136,9 @@ void AppMenuButton::ScheduleAppMenuIconPaint() { |
| } |
| void AppMenuButton::UpdateIcon() { |
| + using BadgeType = AppMenuBadgeController::BadgeType; |
| DCHECK(ui::MaterialDesignController::IsModeMaterial()); |
| + gfx::VectorIconId icon_id = gfx::VectorIconId::BROWSER_TOOLS; |
| SkColor color = gfx::kPlaceholderColor; |
| switch (severity_) { |
| case AppMenuIconPainter::SEVERITY_NONE: |
| @@ -151,8 +156,17 @@ void AppMenuButton::UpdateIcon() { |
| break; |
| } |
| + if (severity_ != AppMenuIconPainter::SEVERITY_NONE) { |
|
Evan Stade
2016/06/03 19:31:26
why this check? Wouldn't type_ be NONE in this cas
kylix_rd
2016/06/03 19:50:43
The code was assuming that if the severity_ is NON
|
| + if (type_ == BadgeType::GLOBAL_ERROR) |
| + icon_id = gfx::VectorIconId::UPDATE_MENU_FAILED; |
| + else if (type_ == BadgeType::INCOMPATIBILITY_WARNING) |
| + icon_id = gfx::VectorIconId::UPDATE_MENU_SEVERITY_WARNING; |
| + else |
| + icon_id = gfx::VectorIconId::UPDATE_MENU_SEVERITY_LOW_MEDIUM_HIGH; |
| + } |
| + |
| SetImage(views::Button::STATE_NORMAL, |
| - gfx::CreateVectorIcon(gfx::VectorIconId::BROWSER_TOOLS, color)); |
| + gfx::CreateVectorIcon(icon_id, 16, color)); |
| } |
| void AppMenuButton::SetTrailingMargin(int margin) { |