Plumb Notification action icons through to the UI layer.

chrome/android/java/src/org/chromium/chrome/browser/notifications/NotificationBuilderBase.java

 // 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.
 
 package org.chromium.chrome.browser.notifications;
 
+import android.annotation.TargetApi;
 import android.app.Notification;
 import android.app.PendingIntent;
 import android.graphics.Bitmap;
-import android.support.v4.app.NotificationCompat.Action;
+import android.graphics.drawable.Icon;
+import android.os.Build;
 
 import org.chromium.base.VisibleForTesting;
 
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 
 import javax.annotation.Nullable;
 
 /**
  * Abstract base class for building a notification. Stores all given arguments for later use.
  */
 public abstract class NotificationBuilderBase {
+    protected static class Action {
+        public int iconId;
+        public Bitmap iconBitmap;
+        public CharSequence title;
+        public PendingIntent intent;
+
+        Action(int iconId, CharSequence title, PendingIntent intent) {
+            this.iconId = iconId;
+            this.title = title;
+            this.intent = intent;
+        }
+
+        Action(Bitmap iconBitmap, CharSequence title, PendingIntent intent) {
+            this.iconBitmap = iconBitmap;
+            this.title = title;
+            this.intent = intent;
+        }
+    }
+
     /**
      * Maximum length of CharSequence inputs to prevent excessive memory consumption. At current
      * screen sizes we display about 500 characters at most, so this is a pretty generous limit, and
      * it matches what the Notification class does.
      */
     @VisibleForTesting
     static final int MAX_CHARSEQUENCE_LENGTH = 5 * 1024;
 
     /**
-     * The maximum number of action buttons. One is for the settings button, and two more slots are
-     * for developer provided buttons.
+     * The maximum number of author provided action buttons. The settings button is not part of this
+     * count.
      */
-    private static final int MAX_ACTION_BUTTONS = 3;
+    private static final int MAX_AUTHOR_PROVIDED_ACTION_BUTTONS = 2;
 
     protected CharSequence mTitle;
     protected CharSequence mBody;
     protected CharSequence mOrigin;
     protected CharSequence mTickerText;
     protected Bitmap mLargeIcon;
     protected int mSmallIconId;
     protected PendingIntent mContentIntent;
     protected PendingIntent mDeleteIntent;
-    protected List<Action> mActions = new ArrayList<>(MAX_ACTION_BUTTONS);
+    protected List<Action> mActions = new ArrayList<>(MAX_AUTHOR_PROVIDED_ACTION_BUTTONS);
     protected Action mSettingsAction;
     protected int mDefaults = Notification.DEFAULT_ALL;
     protected long[] mVibratePattern;
     protected long mTimestamp;
 
     /**
      * Combines all of the options that have been set and returns a new Notification object.
      */
     public abstract Notification build();
 
@@ skipping 61 matching lines @@
         mDeleteIntent = intent;
         return this;
     }
 
     /**
      * Adds an action to the notification. Actions are typically displayed as a button adjacent to
      * the notification content.
      */
     public NotificationBuilderBase addAction(
             int iconId, @Nullable CharSequence title, @Nullable PendingIntent intent) {
-        if (mActions.size() == MAX_ACTION_BUTTONS) {
+        if (mActions.size() == MAX_AUTHOR_PROVIDED_ACTION_BUTTONS) {
             throw new IllegalStateException(
-                    "Cannot add more than " + MAX_ACTION_BUTTONS + " actions.");
+                    "Cannot add more than " + MAX_AUTHOR_PROVIDED_ACTION_BUTTONS + " actions.");
         }
         mActions.add(new Action(iconId, limitLength(title), intent));
         return this;
     }
 
     /**
+     * Adds an action to the notification. Actions are typically displayed as a button adjacent to
+     * the notification content.
+     */
+    public NotificationBuilderBase addAction(@Nullable Bitmap iconBitmap,
+            @Nullable CharSequence title, @Nullable PendingIntent intent) {
+        if (mActions.size() == MAX_AUTHOR_PROVIDED_ACTION_BUTTONS) {
+            throw new IllegalStateException(
+                    "Cannot add more than " + MAX_AUTHOR_PROVIDED_ACTION_BUTTONS + " actions.");
+        }
+        mActions.add(new Action(iconBitmap, limitLength(title), intent));
+        return this;
+    }
+
+    /**
      * Adds an action to the notification for opening the settings screen.
      */
     public NotificationBuilderBase addSettingsAction(
             int iconId, @Nullable CharSequence title, @Nullable PendingIntent intent) {
         mSettingsAction = new Action(iconId, limitLength(title), intent);
         return this;
     }
 
     /**
      * Sets the default notification options that will be used.
@@ skipping 29 matching lines @@
     @Nullable
     private static CharSequence limitLength(@Nullable CharSequence input) {
         if (input == null) {
             return input;
         }
         if (input.length() > MAX_CHARSEQUENCE_LENGTH) {
             return input.subSequence(0, MAX_CHARSEQUENCE_LENGTH);
         }
         return input;
     }
+
+    /**
+     * Adds an action to {@code builder} using a {@code Bitmap} if a bitmap is provided and the API
+     * level is high enough, otherwise a resource id is used.
+     */
+    @SuppressWarnings("deprecation") // For addAction(int, CharSequence, PendingIntent)
+    @TargetApi(Build.VERSION_CODES.M) // For the Icon class.
+    protected static void addActionToBuilder(Notification.Builder builder, Action action) {
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M && action.iconBitmap != null) {
+            Icon icon = Icon.createWithBitmap(action.iconBitmap);
+            builder.addAction(
+                    new Notification.Action.Builder(icon, action.title, action.intent).build());
+        } else {
+            builder.addAction(action.iconId, action.title, action.intent);
+        }
+    }
 }