BadgeExtractor.java revision 924eed1ca6d3fec5dae7eb0f9c11b8f23f628697 
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(record.getChannel().canShowBadge() && appCanShowBadge);
50        }
51
52        return null;
53    }
54
55    @Override
56    public void setConfig(RankingConfig config) {
57        mConfig = config;
58    }
59}
60