Migrate web 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.os.Build;
+import android.widget.RemoteViews;
 
 /**
  * Wraps a Notification.Builder object.
  *
  * Created by awdf on 2/17/17.
  */
 public class NotificationBuilder implements ChromeNotificationBuilder {
     protected final Notification.Builder mBuilder;
 
     public NotificationBuilder(Context context) {
         mBuilder = new Notification.Builder(context);
     }
 
     @Override
     public ChromeNotificationBuilder setAutoCancel(boolean autoCancel) {
         mBuilder.setAutoCancel(autoCancel);
         return this;
     }
 
     @Override
     public ChromeNotificationBuilder setContentIntent(PendingIntent contentIntent) {
         mBuilder.setContentIntent(contentIntent);
         return this;
     }
 
     @Override
-    public ChromeNotificationBuilder setContentTitle(String title) {
+    public ChromeNotificationBuilder setContentTitle(CharSequence title) {
         mBuilder.setContentTitle(title);
         return this;
     }
 
     @Override
-    public ChromeNotificationBuilder setContentText(String text) {
+    public ChromeNotificationBuilder setContentText(CharSequence text) {
         mBuilder.setContentText(text);
         return this;
     }
 
     @Override
     public ChromeNotificationBuilder setSmallIcon(int icon) {
         mBuilder.setSmallIcon(icon);
         return this;
     }
 
     @Override
-    public ChromeNotificationBuilder setTicker(String text) {
+    public ChromeNotificationBuilder setSmallIcon(Icon icon) {
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
+            mBuilder.setSmallIcon(icon);
+        }
+        return this;
+    }
+
+    @Override
+    public ChromeNotificationBuilder setTicker(CharSequence text) {
         mBuilder.setTicker(text);
         return this;
     }
 
     @Override
     public ChromeNotificationBuilder setLocalOnly(boolean localOnly) {
         if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.KITKAT_WATCH) {
             mBuilder.setLocalOnly(localOnly);
         }
         return this;
@@ skipping 23 matching lines @@
 
     @Override
     public ChromeNotificationBuilder setShowWhen(boolean showWhen) {
         if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.JELLY_BEAN_MR1) {
             mBuilder.setShowWhen(showWhen);
         }
         return this;
     }
 
     @Override
-    public ChromeNotificationBuilder addAction(int icon, String title, PendingIntent intent) {
+    public ChromeNotificationBuilder addAction(int icon, CharSequence title, PendingIntent intent) {
         mBuilder.addAction(icon, title, intent);
         return this;
     }
 
     @Override
+    public ChromeNotificationBuilder addAction(Notification.Action action) {
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.KITKAT_WATCH) {
+            mBuilder.addAction(action);
+        }
+        return this;
+    }
+
+    @Override
     public ChromeNotificationBuilder setDeleteIntent(PendingIntent intent) {
         mBuilder.setDeleteIntent(intent);
         return this;
     }
 
     @Override
     public ChromeNotificationBuilder setPriority(int pri) {
         mBuilder.setPriority(pri);
         return this;
     }
 
     @Override
     public ChromeNotificationBuilder setProgress(int max, int percentage, boolean indeterminate) {
         mBuilder.setProgress(max, percentage, indeterminate);
         return this;
     }
 
     @Override
-    public ChromeNotificationBuilder setSubText(String text) {
+    public ChromeNotificationBuilder setSubText(CharSequence text) {
         mBuilder.setSubText(text);
         return this;
     }
 
     @Override
     public ChromeNotificationBuilder setContentInfo(String info) {
         if (Build.VERSION.SDK_INT < Build.VERSION_CODES.N) {
             mBuilder.setContentInfo(info);
         } else {
             mBuilder.setSubText(info);
@@ skipping 13 matching lines @@
         return this;
     }
 
     @Override
     public ChromeNotificationBuilder setVibrate(long[] vibratePattern) {
         mBuilder.setVibrate(vibratePattern);
         return this;
     }
 
     @Override
+    public ChromeNotificationBuilder setDefaults(int defaults) {
+        mBuilder.setDefaults(defaults);
+        return this;
+    }
+
+    @Override
+    public ChromeNotificationBuilder setOnlyAlertOnce(boolean onlyAlertOnce) {
+        mBuilder.setOnlyAlertOnce(onlyAlertOnce);
+        return this;
+    }
+
+    @Override
+    public ChromeNotificationBuilder setPublicVersion(Notification publicNotification) {
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.LOLLIPOP) {
+            mBuilder.setPublicVersion(publicNotification);
+        }
+        return this;
+    }
+
+    @Override
+    public ChromeNotificationBuilder setContent(RemoteViews views) {
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N) {
+            mBuilder.setCustomContentView(views);
+        } else {
+            mBuilder.setContent(views);
+        }
+        return this;
+    }
+
+    @Override
+    public ChromeNotificationBuilder setStyle(Notification.BigPictureStyle style) {
+        mBuilder.setStyle(style);
+        return this;
+    }
+
+    @Override
+    public ChromeNotificationBuilder setStyle(Notification.BigTextStyle style) {
+        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();
     }
 }