BadgeExtractor.java revision a5d2f235add2b8c1e4199918b5fc18040e8af434 
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 appCanShowBadge =
45                mConfig.canShowBadge(record.sbn.getPackageName(), record.sbn.getUid());
46        if (!appCanShowBadge) {
47            record.setShowBadge(false);
48        } else {
49            record.setShowBadge(mConfig.getNotificationChannel(record.sbn.getPackageName(),
50                    record.sbn.getUid(), record.getChannel().getId(), false).canShowBadge()
51                    && appCanShowBadge);
52        }
53
54        return null;
55    }
56
57    @Override
58    public void setConfig(RankingConfig config) {
59        mConfig = config;
60    }
61}
62