Update app menu button MD colors for the notification case.

chrome/browser/ui/views/toolbar/wrench_toolbar_button.cc

 // Copyright 2013 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/views/toolbar/wrench_toolbar_button.h"
 
 #include "base/location.h"
 #include "base/single_thread_task_runner.h"
 #include "base/thread_task_runner_handle.h"
 #include "base/time/time.h"
+#include "chrome/browser/themes/theme_properties.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/toolbar/wrench_menu_model.h"
 #include "chrome/browser/ui/views/extensions/browser_action_drag_data.h"
 #include "chrome/browser/ui/views/layout_constants.h"
 #include "chrome/browser/ui/views/toolbar/toolbar_view.h"
 #include "chrome/browser/ui/views/toolbar/wrench_menu.h"
 #include "extensions/common/feature_switch.h"
 #include "grit/theme_resources.h"
 #include "ui/base/resource/material_design/material_design_controller.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "ui/base/theme_provider.h"
+#include "ui/gfx/color_palette.h"
+#include "ui/gfx/paint_vector_icon.h"
+#include "ui/gfx/vector_icons_public.h"
 #include "ui/keyboard/keyboard_controller.h"
 #include "ui/views/controls/button/label_button_border.h"
 #include "ui/views/controls/menu/menu_listener.h"
 #include "ui/views/metrics.h"
 #include "ui/views/painter.h"
 
 // static
 bool WrenchToolbarButton::g_open_app_immediately_for_testing = false;
 
 WrenchToolbarButton::WrenchToolbarButton(ToolbarView* toolbar_view)
     : views::MenuButton(NULL, base::string16(), toolbar_view, false),
-      wrench_icon_painter_(nullptr),
+      severity_(WrenchIconPainter::SEVERITY_NONE),
       toolbar_view_(toolbar_view),
       allow_extension_dragging_(
           extensions::FeatureSwitch::extension_action_redesign()
               ->IsEnabled()),
       weak_factory_(this) {
   if (!ui::MaterialDesignController::IsModeMaterial())
     wrench_icon_painter_.reset(new WrenchIconPainter(this));
 }
 
 WrenchToolbarButton::~WrenchToolbarButton() {
 }
 
 void WrenchToolbarButton::SetSeverity(WrenchIconPainter::Severity severity,
                                       bool animate) {
-  if (ui::MaterialDesignController::IsModeMaterial())
+  if (ui::MaterialDesignController::IsModeMaterial()) {
+    severity_ = severity;
+    UpdateIcon();
     return;
+  }
 
   wrench_icon_painter_->SetSeverity(severity, animate);
   SchedulePaint();
 }
 
 void WrenchToolbarButton::ShowMenu(bool for_drop) {
   if (menu_ && menu_->IsShowing())
     return;
 
 #if defined(USE_AURA)
@@ skipping 46 matching lines @@
   }
 
   return ResourceBundle::GetSharedInstance().
       GetImageSkiaNamed(IDR_TOOLBAR_BEZEL_HOVER)->size();
 }
 
 void WrenchToolbarButton::ScheduleWrenchIconPaint() {
   SchedulePaint();
 }
 
+void WrenchToolbarButton::UpdateIcon() {
+  DCHECK(ui::MaterialDesignController::IsModeMaterial());
+  SkColor color = SK_ColorRED;
+  switch (severity_) {
+    case WrenchIconPainter::SEVERITY_NONE:
+      color = GetThemeProvider()->GetColor(
+          ThemeProperties::COLOR_TOOLBAR_BUTTON_ICON);
+      break;
+    case WrenchIconPainter::SEVERITY_LOW:
+      color = gfx::kGoogleGreen700;
+      break;
+    case WrenchIconPainter::SEVERITY_MEDIUM:
+      color = gfx::kGoogleYellow700;
+      break;
+    case WrenchIconPainter::SEVERITY_HIGH:
+      color = gfx::kGoogleRed700;
+      break;
+  }
+
+  // TODO(estade): find a home for this constant.
+  const int kButtonSize = 16;
+  SetImage(views::Button::STATE_NORMAL,
+           gfx::CreateVectorIcon(gfx::VectorIconId::BROWSER_TOOLS, kButtonSize,
+                                 color));
+}
+
 const char* WrenchToolbarButton::GetClassName() const {
   return "WrenchToolbarButton";
 }
 
 bool WrenchToolbarButton::GetDropFormats(
     int* formats,
     std::set<ui::Clipboard::FormatType>* format_types) {
   return allow_extension_dragging_ ?
       BrowserActionDragData::GetDropFormats(format_types) :
       views::View::GetDropFormats(formats, format_types);
@@ skipping 40 matching lines @@
 
 void WrenchToolbarButton::OnPaint(gfx::Canvas* canvas) {
   views::MenuButton::OnPaint(canvas);
   if (ui::MaterialDesignController::IsModeMaterial())
     return;
   wrench_icon_painter_->Paint(canvas,
                               GetThemeProvider(),
                               gfx::Rect(size()),
                               WrenchIconPainter::BEZEL_NONE);
 }