Merge "Allow overriding managed profile notification badge"

package android.app;

import static android.annotation.Dimension.DP;

import static android.app.admin.DevicePolicyResources.Drawables.Source.NOTIFICATION;

import static android.app.admin.DevicePolicyResources.Drawables.Style.SOLID_COLORED;

import static android.app.admin.DevicePolicyResources.Drawables.UNDEFINED;

import static android.app.admin.DevicePolicyResources.Drawables.WORK_PROFILE_ICON;

import static android.graphics.drawable.Icon.TYPE_URI;

import static android.graphics.drawable.Icon.TYPE_URI_ADAPTIVE_BITMAP;

import android.annotation.SuppressLint;

import android.annotation.SystemApi;

import android.annotation.TestApi;

import android.app.admin.DevicePolicyManager;

import android.compat.annotation.UnsupportedAppUsage;

import android.content.Context;

import android.content.Intent;

import android.os.SystemClock;

import android.os.SystemProperties;

import android.os.UserHandle;

import android.os.UserManager;

import android.provider.Settings;

import android.text.BidiFormatter;

import android.text.SpannableStringBuilder;

            }

            // Note: This assumes that the current user can read the profile badge of the

            // originating user.

            DevicePolicyManager dpm = mContext.getSystemService(DevicePolicyManager.class);

            return dpm.getDrawable(

                    getUpdatableProfileBadgeId(), SOLID_COLORED, NOTIFICATION,

                    this::getDefaultProfileBadgeDrawable);

        }

        private String getUpdatableProfileBadgeId() {

            return mContext.getSystemService(UserManager.class).isManagedProfile()

                    ? WORK_PROFILE_ICON : UNDEFINED;

        }

        private Drawable getDefaultProfileBadgeDrawable() {

            return mContext.getPackageManager().getUserBadgeForDensityNoBackground(

                    new UserHandle(mContext.getUserId()), 0);

        }