1 /*
2  * Copyright (C) 2019 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 
17 package com.android.systemui.statusbar.notification
18 
19 import android.content.Context
20 import android.provider.DeviceConfig
21 
22 import com.android.internal.annotations.VisibleForTesting
23 import com.android.internal.config.sysui.SystemUiDeviceConfigFlags.NOTIFICATIONS_USE_PEOPLE_FILTERING
24 import com.android.systemui.statusbar.notification.stack.BUCKET_ALERTING
25 import com.android.systemui.statusbar.notification.stack.BUCKET_FOREGROUND_SERVICE
26 import com.android.systemui.statusbar.notification.stack.BUCKET_HEADS_UP
27 import com.android.systemui.statusbar.notification.stack.BUCKET_MEDIA_CONTROLS
28 import com.android.systemui.statusbar.notification.stack.BUCKET_PEOPLE
29 import com.android.systemui.statusbar.notification.stack.BUCKET_SILENT
30 import com.android.systemui.util.DeviceConfigProxy
31 import com.android.systemui.util.Utils
32 
33 import javax.inject.Inject
34 
35 private var sUsePeopleFiltering: Boolean? = null
36 
37 /**
38  * Feature controller for the NOTIFICATIONS_USE_PEOPLE_FILTERING config.
39  */
40 class NotificationSectionsFeatureManager @Inject constructor(
41     val proxy: DeviceConfigProxy,
42     val context: Context
43 ) {
44 
45     fun isFilteringEnabled(): Boolean {
46         return usePeopleFiltering(proxy)
47     }
48 
49     fun isMediaControlsEnabled(): Boolean {
50         return Utils.useQsMediaPlayer(context)
51     }
52 
53     fun getNotificationBuckets(): IntArray {
54         return when {
55             isFilteringEnabled() && isMediaControlsEnabled() ->
56                 intArrayOf(BUCKET_HEADS_UP, BUCKET_FOREGROUND_SERVICE, BUCKET_MEDIA_CONTROLS,
57                         BUCKET_PEOPLE, BUCKET_ALERTING, BUCKET_SILENT)
58             !isFilteringEnabled() && isMediaControlsEnabled() ->
59                 intArrayOf(BUCKET_HEADS_UP, BUCKET_FOREGROUND_SERVICE, BUCKET_MEDIA_CONTROLS,
60                         BUCKET_ALERTING, BUCKET_SILENT)
61             isFilteringEnabled() && !isMediaControlsEnabled() ->
62                 intArrayOf(BUCKET_HEADS_UP, BUCKET_FOREGROUND_SERVICE, BUCKET_PEOPLE,
63                         BUCKET_ALERTING, BUCKET_SILENT)
64             else ->
65                 intArrayOf(BUCKET_ALERTING, BUCKET_SILENT)
66         }
67     }
68 
69     fun getNumberOfBuckets(): Int {
70         return getNotificationBuckets().size
71     }
72 
73     @VisibleForTesting
74     fun clearCache() {
75         sUsePeopleFiltering = null
76     }
77 }
78 
79 private fun usePeopleFiltering(proxy: DeviceConfigProxy): Boolean {
80     if (sUsePeopleFiltering == null) {
81         sUsePeopleFiltering = proxy.getBoolean(
82                 DeviceConfig.NAMESPACE_SYSTEMUI, NOTIFICATIONS_USE_PEOPLE_FILTERING, true)
83     }
84 
85     return sUsePeopleFiltering!!
86 }
87