[MediaSession,Android] Use a badged notification icon for the Work profile

chrome/android/java/src/org/chromium/chrome/browser/media/ui/NotificationMediaPlaybackControls.java

 // Copyright 2015 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.media.ui;
 
 import android.app.PendingIntent;
 import android.app.Service;
 import android.content.Context;
 import android.content.Intent;
+import android.graphics.Bitmap;
+import android.graphics.drawable.BitmapDrawable;
+import android.graphics.drawable.Drawable;
 import android.os.IBinder;
 import android.provider.Browser;
 import android.support.v4.app.NotificationCompat;
 import android.widget.RemoteViews;
 
+import org.chromium.base.ApiCompatibilityUtils;
 import org.chromium.chrome.R;
 import org.chromium.chrome.browser.IntentHandler.TabOpenType;
 
 /**
  * A class for notifications that provide information and optional media controls for a given media.
  * Internally implements a Service for transforming notification Intents into
  * {@link MediaPlaybackListener} calls for all registered listeners.
  */
 public class NotificationMediaPlaybackControls {
     private static final Object LOCK = new Object();
@@ skipping 118 matching lines @@
 
     // ListenerService running for the notification. Only non-null when showing.
     private ListenerService mService;
 
     private final String mPlayDescription;
 
     private final String mPauseDescription;
 
     private NotificationCompat.Builder mNotificationBuilder;
 
+    private Bitmap mNotificationIconBitmap;
+
     private MediaNotificationInfo mMediaNotificationInfo;
 
     private NotificationMediaPlaybackControls(Context context) {
         mContext = context;
         mPlayDescription = context.getResources().getString(R.string.accessibility_play);
         mPauseDescription = context.getResources().getString(R.string.accessibility_pause);
     }
 
     private void showNotification(MediaNotificationInfo mediaNotificationInfo) {
         mContext.startService(new Intent(mContext, ListenerService.class));
@@ skipping 63 matching lines @@
 
     private void updateNotification() {
         if (mService == null) return;
 
         if (mMediaNotificationInfo == null) {
             // Notification was hidden before we could update it.
             assert mNotificationBuilder == null;
             return;
         }
 
+        // Android doesn't badge the icons for RemoteViews automatically when
+        // running the app under the Work profile.
+        if (mNotificationIconBitmap == null) {
+            Drawable notificationIconDrawable = ApiCompatibilityUtils.getUserBadgedIcon(
+                    mContext, R.drawable.audio_playing);
+            mNotificationIconBitmap = drawableToBitmap(notificationIconDrawable);
+        }
+
         if (mNotificationBuilder == null) {
             mNotificationBuilder = new NotificationCompat.Builder(mContext)
                 .setSmallIcon(R.drawable.audio_playing)
                 .setAutoCancel(false)
                 .setOngoing(true);
         }
         mNotificationBuilder.setContentIntent(createContentIntent());
 
         RemoteViews contentView = createContentView();
 
         contentView.setTextViewText(R.id.title, getTitle());
         contentView.setTextViewText(R.id.status, getStatus());
-        contentView.setImageViewResource(R.id.icon, R.drawable.audio_playing);
+        if (mNotificationIconBitmap != null) {
+            contentView.setImageViewBitmap(R.id.icon, mNotificationIconBitmap);
+        } else {
+            contentView.setImageViewResource(R.id.icon, R.drawable.audio_playing);
+        }
 
         if (mMediaNotificationInfo.isPaused) {
             contentView.setImageViewResource(R.id.playpause, R.drawable.ic_vidcontrol_play);
             contentView.setContentDescription(R.id.playpause, mPlayDescription);
             contentView.setOnClickPendingIntent(R.id.playpause,
                     mService.getPendingIntent(ListenerService.ACTION_PLAY));
         } else {
             contentView.setImageViewResource(R.id.playpause, R.drawable.ic_vidcontrol_pause);
             contentView.setContentDescription(R.id.playpause, mPauseDescription);
             contentView.setOnClickPendingIntent(R.id.playpause,
                     mService.getPendingIntent(ListenerService.ACTION_PAUSE));
         }
 
         mNotificationBuilder.setContent(contentView);
         mNotificationBuilder.setVisibility(
                 mMediaNotificationInfo.isPrivate ? NotificationCompat.VISIBILITY_PRIVATE
                                                  : NotificationCompat.VISIBILITY_PUBLIC);
 
         mService.startForeground(R.id.media_playback_notification, mNotificationBuilder.build());
     }
+
+    private Bitmap drawableToBitmap(Drawable drawable) {
+        if (!(drawable instanceof BitmapDrawable)) return null;
+
+        BitmapDrawable bitmapDrawable = (BitmapDrawable) drawable;
+        return bitmapDrawable.getBitmap();
+    }
 }