Migrating media notifications to ChromeNotificationBuilder

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

 // Copyright 2017 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.app.Notification;
 import android.app.PendingIntent;
 import android.content.Context;
 import android.graphics.Bitmap;
 import android.graphics.drawable.Icon;
+import android.media.session.MediaSession;
 import android.os.Build;
+import android.support.v4.media.session.MediaSessionCompat;
 import android.widget.RemoteViews;
 
 /**
  * Wraps a Notification.Builder object.
  */
 public class NotificationBuilder implements ChromeNotificationBuilder {
     protected final Notification.Builder mBuilder;
 
     public NotificationBuilder(Context context) {
         mBuilder = new Notification.Builder(context);
@@ skipping 191 matching lines @@
         return this;
     }
 
     @Override
     public ChromeNotificationBuilder setStyle(Notification.BigTextStyle style) {
         mBuilder.setStyle(style);
         return this;
     }
 
     @Override
+    public ChromeNotificationBuilder setMediaStyle(MediaSessionCompat session, int[] actions,
+            PendingIntent intent, boolean showCancelButton) {
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.LOLLIPOP) {
+            Notification.MediaStyle style = new Notification.MediaStyle();
+            style.setMediaSession(((MediaSession) session.getMediaSession()).getSessionToken());
+            style.setShowActionsInCompactView(actions);
+            mBuilder.setStyle(style);
+        }
+        return this;
+    }
+
+    @Override
     public Notification buildWithBigContentView(RemoteViews view) {
         if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N) {
             return mBuilder.setCustomBigContentView(view).build();
         } else {
             Notification notification = mBuilder.build();
             notification.bigContentView = view;
             return notification;
         }
     }
 
     @Override
     public Notification buildWithBigTextStyle(String bigText) {
         Notification.BigTextStyle bigTextStyle = new Notification.BigTextStyle(mBuilder);
         bigTextStyle.bigText(bigText);
         return bigTextStyle.build();
     }
 
     @Override
     public Notification build() {
         return mBuilder.build();
     }
 }