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  */
16 package android.util;
17 
18 import android.annotation.UserIdInt;
19 import android.content.Context;
20 import android.content.pm.ApplicationInfo;
21 import android.content.pm.PackageItemInfo;
22 import android.content.pm.PackageManager;
23 import android.content.res.Resources;
24 import android.graphics.Color;
25 import android.graphics.drawable.Drawable;
26 import android.os.UserHandle;
27 import android.os.UserManager;
28 
29 import com.android.internal.annotations.VisibleForTesting;
30 
31 /**
32  * Utility class to load app drawables with appropriate badging.
33  *
34  * @hide
35  */
36 public class IconDrawableFactory {
37 
38     protected final Context mContext;
39     protected final PackageManager mPm;
40     protected final UserManager mUm;
41     protected final LauncherIcons mLauncherIcons;
42     protected final boolean mEmbedShadow;
43 
IconDrawableFactory(Context context, boolean embedShadow)44     private IconDrawableFactory(Context context, boolean embedShadow) {
45         mContext = context;
46         mPm = context.getPackageManager();
47         mUm = context.getSystemService(UserManager.class);
48         mLauncherIcons = new LauncherIcons(context);
49         mEmbedShadow = embedShadow;
50     }
51 
needsBadging(ApplicationInfo appInfo, @UserIdInt int userId)52     protected boolean needsBadging(ApplicationInfo appInfo, @UserIdInt int userId) {
53         return appInfo.isInstantApp() || mUm.isManagedProfile(userId);
54     }
55 
getBadgedIcon(ApplicationInfo appInfo)56     public Drawable getBadgedIcon(ApplicationInfo appInfo) {
57         return getBadgedIcon(appInfo, UserHandle.getUserId(appInfo.uid));
58     }
59 
getBadgedIcon(ApplicationInfo appInfo, @UserIdInt int userId)60     public Drawable getBadgedIcon(ApplicationInfo appInfo, @UserIdInt int userId) {
61         return getBadgedIcon(appInfo, appInfo, userId);
62     }
63 
getBadgedIcon(PackageItemInfo itemInfo, ApplicationInfo appInfo, @UserIdInt int userId)64     public Drawable getBadgedIcon(PackageItemInfo itemInfo, ApplicationInfo appInfo,
65             @UserIdInt int userId) {
66         Drawable icon = mPm.loadUnbadgedItemIcon(itemInfo, appInfo);
67         if (!mEmbedShadow && !needsBadging(appInfo, userId)) {
68             return icon;
69         }
70 
71         // Before badging, add shadow to adaptive icon if needed.
72         icon = mLauncherIcons.wrapIconDrawableWithShadow(icon);
73         if (appInfo.isInstantApp()) {
74             int badgeColor = Resources.getSystem().getColor(
75                     com.android.internal.R.color.instant_app_badge, null);
76             icon = mLauncherIcons.getBadgedDrawable(icon,
77                     com.android.internal.R.drawable.ic_instant_icon_badge_bolt,
78                     badgeColor);
79         }
80         if (mUm.isManagedProfile(userId)) {
81             icon = mLauncherIcons.getBadgedDrawable(icon,
82                     com.android.internal.R.drawable.ic_corp_icon_badge_case,
83                     getUserBadgeColor(mUm, userId));
84         }
85         return icon;
86     }
87 
88     // Should have enough colors to cope with UserManagerService.getMaxManagedProfiles()
89     @VisibleForTesting
90     public static final int[] CORP_BADGE_COLORS = new int[] {
91             com.android.internal.R.color.profile_badge_1,
92             com.android.internal.R.color.profile_badge_2,
93             com.android.internal.R.color.profile_badge_3
94     };
95 
getUserBadgeColor(UserManager um, @UserIdInt int userId)96     public static int getUserBadgeColor(UserManager um, @UserIdInt int userId) {
97         int badge = um.getManagedProfileBadge(userId);
98         if (badge < 0) {
99             badge = 0;
100         }
101         int resourceId = CORP_BADGE_COLORS[badge % CORP_BADGE_COLORS.length];
102         return Resources.getSystem().getColor(resourceId, null);
103     }
104 
newInstance(Context context)105     public static IconDrawableFactory newInstance(Context context) {
106         return new IconDrawableFactory(context, true);
107     }
108 
newInstance(Context context, boolean embedShadow)109     public static IconDrawableFactory newInstance(Context context, boolean embedShadow) {
110         return new IconDrawableFactory(context, embedShadow);
111     }
112 }
113