1 /** 2 * Copyright (C) 2017 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 package com.android.server.notification; 17 18 import android.content.Context; 19 import android.util.Slog; 20 21 /** 22 * Determines whether a badge should be shown for this notification 23 */ 24 public class BadgeExtractor implements NotificationSignalExtractor { 25 private static final String TAG = "BadgeExtractor"; 26 private static final boolean DBG = false; 27 28 private RankingConfig mConfig; 29 30 public void initialize(Context ctx, NotificationUsageStats usageStats) { 31 if (DBG) Slog.d(TAG, "Initializing " + getClass().getSimpleName() + "."); 32 } 33 34 public RankingReconsideration process(NotificationRecord record) { 35 if (record == null || record.getNotification() == null) { 36 if (DBG) Slog.d(TAG, "skipping empty notification"); 37 return null; 38 } 39 40 if (mConfig == null) { 41 if (DBG) Slog.d(TAG, "missing config"); 42 return null; 43 } 44 boolean userWantsBadges = mConfig.badgingEnabled(record.sbn.getUser()); 45 boolean appCanShowBadge = 46 mConfig.canShowBadge(record.sbn.getPackageName(), record.sbn.getUid()); 47 if (!userWantsBadges || !appCanShowBadge) { 48 record.setShowBadge(false); 49 } else { 50 record.setShowBadge(mConfig.getNotificationChannel(record.sbn.getPackageName(), 51 record.sbn.getUid(), record.getChannel().getId(), false).canShowBadge() 52 && appCanShowBadge); 53 } 54 55 return null; 56 } 57 58 @Override 59 public void setConfig(RankingConfig config) { 60 mConfig = config; 61 } 62 } 63