[Mac] App Menu Animated Icon

chrome/browser/ui/cocoa/toolbar/app_toolbar_button.mm

 // Copyright 2016 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.
 
 #import "chrome/browser/ui/cocoa/toolbar/app_toolbar_button.h"
 
+#include "base/command_line.h"
 #include "base/macros.h"
 #include "chrome/app/vector_icons/vector_icons.h"
 #import "chrome/browser/ui/cocoa/themed_window.h"
+#import "chrome/browser/ui/cocoa/toolbar/app_toolbar_button_icon.h"
 #import "chrome/browser/ui/cocoa/view_id_util.h"
+#include "chrome/common/chrome_switches.h"
 #include "chrome/grit/chromium_strings.h"
 #include "ui/base/l10n/l10n_util_mac.h"
 #include "ui/base/material_design/material_design_controller.h"
 #include "ui/base/theme_provider.h"
 #include "ui/gfx/color_palette.h"
 
 @interface AppToolbarButton ()
 - (void)commonInit;
 @end
 
 @implementation AppToolbarButton
 
 - (instancetype)initWithFrame:(NSRect)frame {
   if ((self = [super initWithFrame:frame])) {
     [self commonInit];
+
+    base::CommandLine* commandLine = base::CommandLine::ForCurrentProcess();
+    if (commandLine->HasSwitch(switches::kEnableNewAppMenuIcon)) {
+      app_icon_.reset(new AppToolbarButtonIcon(
+          self, [self vectorIconColor:[[self window] hasDarkTheme]]));
+    }
   }
   return self;
 }
 
 - (void)awakeFromNib {
   [self commonInit];
 }
 
+- (void)drawRect:(NSRect)frame {
+  [super drawRect:frame];
+
+  if (app_icon_)
+    app_icon_->DrawIcon(frame);
+}
+
 - (void)commonInit {
   view_id_util::SetID(self, VIEW_ID_APP_MENU);
   severity_ = AppMenuIconController::Severity::NONE;
   type_ = AppMenuIconController::IconType::NONE;
   [self setToolTip:l10n_util::GetNSString(IDS_APPMENU_TOOLTIP)];
 }
 
 - (const gfx::VectorIcon*)vectorIcon {
+  // The new app menu icon doesn't use vectors.
+  if (app_icon_)
+    return nullptr;
+
   switch (type_) {
     case AppMenuIconController::IconType::NONE:
       DCHECK_EQ(severity_, AppMenuIconController::Severity::NONE);
       return &kBrowserToolsIcon;
     case AppMenuIconController::IconType::UPGRADE_NOTIFICATION:
       return &kBrowserToolsUpdateIcon;
     case AppMenuIconController::IconType::GLOBAL_ERROR:
     case AppMenuIconController::IconType::INCOMPATIBILITY_WARNING:
       return &kBrowserToolsErrorIcon;
   }
@@ skipping 27 matching lines @@
       break;
   }
 }
 
 - (void)setSeverity:(AppMenuIconController::Severity)severity
            iconType:(AppMenuIconController::IconType)type
       shouldAnimate:(BOOL)shouldAnimate {
   if (severity != severity_ || type != type_) {
     severity_ = severity;
     type_ = type;
-    // Update the button state images with the new severity color or icon type.
-    [self resetButtonStateImages];
+
+    if (app_icon_) {
+      app_icon_->set_target_color(
+          [self vectorIconColor:[[self window] hasDarkTheme]]);
+      app_icon_->StartAnimation();
+    } else {
+      // Update the button state images with the new severity color or icon
+      // type.
+      [self resetButtonStateImages];
+    }
   }
 }
 
+- (void)onTabInsertedInForeground {

[Robert Sesek, 2017/05/03 22:44:29]

What do you think of this?

Rename this -maybeStartAnimation and call it directly from
BrowserWindowController, and have -willShowMenu in this class call it too. It
seems a little weird to have the button care about tabs being inserted, but I
can also see why you did it this way.

+  if (app_icon_ && severity_ != AppMenuIconController::Severity::NONE)
+    app_icon_->StartAnimation();
+}
+
+- (void)willShowMenu {
+  if (app_icon_ && severity_ != AppMenuIconController::Severity::NONE)
+    app_icon_->StartAnimation();
+}
+
 @end