1 /*
2  * Copyright (C) 2020 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.collection
18 
19 import android.app.Notification
20 import android.content.Context
21 import android.content.pm.ApplicationInfo
22 import android.content.pm.PackageManager
23 import android.service.notification.StatusBarNotification
24 import android.util.Log
25 import com.android.systemui.dagger.SysUISingleton
26 import com.android.systemui.statusbar.notification.collection.notifcollection.CommonNotifCollection
27 import com.android.systemui.statusbar.notification.collection.notifcollection.NotifCollectionListener
28 import com.android.systemui.statusbar.phone.CentralSurfaces
29 import javax.inject.Inject
30 
31 @SysUISingleton
32 class TargetSdkResolver @Inject constructor(
33     private val context: Context
34 ) {
initializenull35     fun initialize(collection: CommonNotifCollection) {
36         collection.addCollectionListener(object : NotifCollectionListener {
37             override fun onEntryBind(entry: NotificationEntry, sbn: StatusBarNotification) {
38                 entry.targetSdk = resolveNotificationSdk(sbn)
39             }
40         })
41     }
42 
resolveNotificationSdknull43     private fun resolveNotificationSdk(sbn: StatusBarNotification): Int {
44         val applicationInfo = getApplicationInfoFromExtras(sbn.notification)
45                 ?: getApplicationInfoFromPackageManager(sbn)
46 
47         return applicationInfo?.targetSdkVersion ?: 0
48     }
49 
getApplicationInfoFromExtrasnull50     private fun getApplicationInfoFromExtras(notification: Notification): ApplicationInfo? =
51             notification.extras.getParcelable(
52                     Notification.EXTRA_BUILDER_APPLICATION_INFO,
53                     ApplicationInfo::class.java
54             )
55 
56     private fun getApplicationInfoFromPackageManager(sbn: StatusBarNotification): ApplicationInfo? {
57         val pmUser = CentralSurfaces.getPackageManagerForUser(context, sbn.user.identifier)
58 
59         return try {
60             pmUser.getApplicationInfo(sbn.packageName, 0)
61         } catch (ex: PackageManager.NameNotFoundException) {
62             Log.e(TAG, "Failed looking up ApplicationInfo for " + sbn.packageName, ex)
63             null
64         }
65     }
66 
67     private val TAG = "TargetSdkResolver"
68 }
69