BadgeExtractor.java revision eb3dca71b5df7fdf6299a3e65eb5d6fe8cb7bcbc 
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*/
16package com.android.server.notification;
17
18import android.content.Context;
19import android.util.Slog;
20
21/**
22 * Determines whether a badge should be shown for this notification
23 */
24public 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            if (record.getChannel() != null) {
51                record.setShowBadge(record.getChannel().canShowBadge() && appCanShowBadge);
52            } else {
53                record.setShowBadge(appCanShowBadge);
54            }
55        }
56
57        return null;
58    }
59
60    @Override
61    public void setConfig(RankingConfig config) {
62        mConfig = config;
63    }
64}
65