/*
 * Copyright (C) 2018 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.permissioncontroller.permission.service;

import static android.Manifest.permission.ACCESS_FINE_LOCATION;
import static android.Manifest.permission_group.LOCATION;
import static android.app.AppOpsManager.OPSTR_FINE_LOCATION;
import static android.app.NotificationManager.IMPORTANCE_LOW;
import static android.app.PendingIntent.FLAG_IMMUTABLE;
import static android.app.PendingIntent.FLAG_ONE_SHOT;
import static android.app.PendingIntent.FLAG_UPDATE_CURRENT;
import static android.app.job.JobScheduler.RESULT_SUCCESS;
import static android.content.Context.MODE_PRIVATE;
import static android.content.Intent.ACTION_MANAGE_APP_PERMISSION;
import static android.content.Intent.ACTION_SAFETY_CENTER;
import static android.content.Intent.EXTRA_PACKAGE_NAME;
import static android.content.Intent.EXTRA_PERMISSION_GROUP_NAME;
import static android.content.Intent.EXTRA_UID;
import static android.content.Intent.EXTRA_USER;
import static android.content.Intent.FLAG_ACTIVITY_MULTIPLE_TASK;
import static android.content.Intent.FLAG_ACTIVITY_NEW_TASK;
import static android.content.Intent.FLAG_RECEIVER_FOREGROUND;
import static android.content.pm.PackageManager.GET_PERMISSIONS;
import static android.graphics.Bitmap.Config.ARGB_8888;
import static android.graphics.Bitmap.createBitmap;
import static android.os.UserHandle.getUserHandleForUid;
import static android.os.UserHandle.myUserId;
import static android.provider.Settings.Secure.LOCATION_ACCESS_CHECK_DELAY_MILLIS;
import static android.provider.Settings.Secure.LOCATION_ACCESS_CHECK_INTERVAL_MILLIS;
import static android.safetycenter.SafetyCenterManager.EXTRA_SAFETY_SOURCE_ID;
import static android.safetycenter.SafetyCenterManager.EXTRA_SAFETY_SOURCE_ISSUE_ID;
import static android.safetycenter.SafetyCenterManager.EXTRA_SAFETY_SOURCE_USER_HANDLE;

import static com.android.permissioncontroller.Constants.EXTRA_SESSION_ID;
import static com.android.permissioncontroller.Constants.INVALID_SESSION_ID;
import static com.android.permissioncontroller.Constants.KEY_LAST_LOCATION_ACCESS_NOTIFICATION_SHOWN;
import static com.android.permissioncontroller.Constants.KEY_LOCATION_ACCESS_CHECK_ENABLED_TIME;
import static com.android.permissioncontroller.Constants.LOCATION_ACCESS_CHECK_ALREADY_NOTIFIED_FILE;
import static com.android.permissioncontroller.Constants.LOCATION_ACCESS_CHECK_JOB_ID;
import static com.android.permissioncontroller.Constants.LOCATION_ACCESS_CHECK_NOTIFICATION_ID;
import static com.android.permissioncontroller.Constants.PERIODIC_LOCATION_ACCESS_CHECK_JOB_ID;
import static com.android.permissioncontroller.Constants.PERMISSION_REMINDER_CHANNEL_ID;
import static com.android.permissioncontroller.Constants.PREFERENCES_FILE;
import static com.android.permissioncontroller.PermissionControllerStatsLog.LOCATION_ACCESS_CHECK_NOTIFICATION_ACTION;
import static com.android.permissioncontroller.PermissionControllerStatsLog.LOCATION_ACCESS_CHECK_NOTIFICATION_ACTION__RESULT__NOTIFICATION_DECLINED;
import static com.android.permissioncontroller.PermissionControllerStatsLog.LOCATION_ACCESS_CHECK_NOTIFICATION_ACTION__RESULT__NOTIFICATION_PRESENTED;
import static com.android.permissioncontroller.PermissionControllerStatsLog.PRIVACY_SIGNAL_ISSUE_CARD_INTERACTION;
import static com.android.permissioncontroller.PermissionControllerStatsLog.PRIVACY_SIGNAL_ISSUE_CARD_INTERACTION__ACTION__CARD_DISMISSED;
import static com.android.permissioncontroller.PermissionControllerStatsLog.PRIVACY_SIGNAL_ISSUE_CARD_INTERACTION__ACTION__CLICKED_CTA1;
import static com.android.permissioncontroller.PermissionControllerStatsLog.PRIVACY_SIGNAL_ISSUE_CARD_INTERACTION__PRIVACY_SOURCE__BG_LOCATION;
import static com.android.permissioncontroller.PermissionControllerStatsLog.PRIVACY_SIGNAL_NOTIFICATION_INTERACTION;
import static com.android.permissioncontroller.PermissionControllerStatsLog.PRIVACY_SIGNAL_NOTIFICATION_INTERACTION__ACTION__DISMISSED;
import static com.android.permissioncontroller.PermissionControllerStatsLog.PRIVACY_SIGNAL_NOTIFICATION_INTERACTION__ACTION__NOTIFICATION_SHOWN;
import static com.android.permissioncontroller.PermissionControllerStatsLog.PRIVACY_SIGNAL_NOTIFICATION_INTERACTION__PRIVACY_SOURCE__BG_LOCATION;
import static com.android.permissioncontroller.permission.utils.Utils.OS_PKG;
import static com.android.permissioncontroller.permission.utils.Utils.getParcelableExtraSafe;
import static com.android.permissioncontroller.permission.utils.Utils.getParentUserContext;
import static com.android.permissioncontroller.permission.utils.Utils.getStringExtraSafe;
import static com.android.permissioncontroller.permission.utils.Utils.getSystemServiceSafe;

import static java.lang.System.currentTimeMillis;
import static java.util.concurrent.TimeUnit.DAYS;

import android.app.AppOpsManager;
import android.app.AppOpsManager.OpEntry;
import android.app.AppOpsManager.PackageOps;
import android.app.Application;
import android.app.Notification;
import android.app.NotificationChannel;
import android.app.NotificationManager;
import android.app.PendingIntent;
import android.app.job.JobInfo;
import android.app.job.JobParameters;
import android.app.job.JobScheduler;
import android.app.job.JobService;
import android.content.BroadcastReceiver;
import android.content.ComponentName;
import android.content.ContentResolver;
import android.content.Context;
import android.content.Intent;
import android.content.SharedPreferences;
import android.content.pm.PackageInfo;
import android.content.pm.PackageManager;
import android.graphics.Bitmap;
import android.graphics.Canvas;
import android.graphics.drawable.Drawable;
import android.graphics.drawable.Icon;
import android.location.LocationManager;
import android.net.Uri;
import android.os.AsyncTask;
import android.os.Build;
import android.os.Bundle;
import android.os.UserHandle;
import android.os.UserManager;
import android.provider.DeviceConfig;
import android.provider.Settings;
import android.safetycenter.SafetyCenterManager;
import android.safetycenter.SafetyEvent;
import android.safetycenter.SafetySourceData;
import android.safetycenter.SafetySourceIssue;
import android.safetycenter.SafetySourceIssue.Action;
import android.service.notification.StatusBarNotification;
import android.text.TextUtils;
import android.util.ArrayMap;
import android.util.ArraySet;
import android.util.Log;

import androidx.annotation.ChecksSdkIntAtLeast;
import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.annotation.RequiresApi;
import androidx.annotation.WorkerThread;
import androidx.core.util.Preconditions;

import com.android.modules.utils.build.SdkLevel;
import com.android.permissioncontroller.Constants;
import com.android.permissioncontroller.DeviceUtils;
import com.android.permissioncontroller.PermissionControllerStatsLog;
import com.android.permissioncontroller.R;
import com.android.permissioncontroller.permission.model.AppPermissionGroup;
import com.android.permissioncontroller.permission.utils.KotlinUtils;
import com.android.permissioncontroller.permission.utils.Utils;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Random;
import java.util.Set;
import java.util.function.BooleanSupplier;
import java.util.stream.Collectors;

/**
 * Show notification that double-guesses the user if she/he really wants to grant fine background
 * location access to an app.
 *
 * <p>A notification is scheduled after the background permission access is granted via
 * {@link #checkLocationAccessSoon()} or periodically.
 *
 * <p>We rate limit the number of notification we show and only ever show one notification at a
 * time. Further we only shown notifications if the app has actually accessed the fine location
 * in the background.
 *
 * <p>As there are many cases why a notification should not been shown, we always schedule a
 * {@link #addLocationNotificationIfNeeded check} which then might add a notification.
 */
public class LocationAccessCheck {
    private static final String LOG_TAG = LocationAccessCheck.class.getSimpleName();
    private static final boolean DEBUG = false;
    private static final long DEFAULT_RENOTIFY_DURATION_MILLIS = DAYS.toMillis(90);
    private static final String ISSUE_ID_PREFIX = "bg_location_";
    private static final String ISSUE_TYPE_ID = "bg_location_privacy_issue";
    private static final String REVOKE_LOCATION_ACCESS_ID_PREFIX = "revoke_location_access_";
    private static final String VIEW_LOCATION_ACCESS_ID = "view_location_access";
    public static final String BG_LOCATION_SOURCE_ID = "AndroidBackgroundLocation";

    /**
     * Device config property for delay in milliseconds
     * between granting a permission and the follow up check
     **/
    public static final String PROPERTY_LOCATION_ACCESS_CHECK_DELAY_MILLIS =
            "location_access_check_delay_millis";

    /**
     * Device config property for delay in milliseconds
     * between periodic checks for background location access
     **/
    public static final String PROPERTY_LOCATION_ACCESS_PERIODIC_INTERVAL_MILLIS =
            "location_access_check_periodic_interval_millis";

    /**
     * Device config property for flag that determines whether location check for safety center
     * is enabled.
     */
    public static final String PROPERTY_BG_LOCATION_CHECK_ENABLED = "bg_location_check_is_enabled";

    /**
     * Lock required for all methods called {@code ...Locked}
     */
    private static final Object sLock = new Object();

    private final Random mRandom = new Random();

    private final @NonNull Context mContext;
    private final @NonNull JobScheduler mJobScheduler;
    private final @NonNull ContentResolver mContentResolver;
    private final @NonNull AppOpsManager mAppOpsManager;
    private final @NonNull PackageManager mPackageManager;
    private final @NonNull UserManager mUserManager;
    private final @NonNull SharedPreferences mSharedPrefs;

    /**
     * If the current long running operation should be canceled
     */
    private final @Nullable BooleanSupplier mShouldCancel;

    /**
     * Get time in between two periodic checks.
     *
     * <p>Default: 1 day
     *
     * @return The time in between check in milliseconds
     */
    private long getPeriodicCheckIntervalMillis() {
        return SdkLevel.isAtLeastT() ? DeviceConfig.getLong(DeviceConfig.NAMESPACE_PRIVACY,
                PROPERTY_LOCATION_ACCESS_PERIODIC_INTERVAL_MILLIS, DAYS.toMillis(1))
                : Settings.Secure.getLong(mContentResolver,
                        LOCATION_ACCESS_CHECK_INTERVAL_MILLIS, DAYS.toMillis(1));
    }

    /**
     * Flexibility of the periodic check.
     *
     * <p>10% of {@link #getPeriodicCheckIntervalMillis()}
     *
     * @return The flexibility of the periodic check in milliseconds
     */
    private long getFlexForPeriodicCheckMillis() {
        return getPeriodicCheckIntervalMillis() / 10;
    }

    /**
     * Get the delay in between granting a permission and the follow up check.
     *
     * <p>Default: 1 day
     *
     * @return The delay in milliseconds
     */
    private long getDelayMillis() {
        return SdkLevel.isAtLeastT() ? DeviceConfig.getLong(DeviceConfig.NAMESPACE_PRIVACY,
                PROPERTY_LOCATION_ACCESS_CHECK_DELAY_MILLIS, DAYS.toMillis(1))
                : Settings.Secure.getLong(mContentResolver, LOCATION_ACCESS_CHECK_DELAY_MILLIS,
                        DAYS.toMillis(1));
    }

    /**
     * Minimum time in between showing two notifications.
     *
     * <p>This is just small enough so that the periodic check can always show a notification.
     *
     * @return The minimum time in milliseconds
     */
    private long getInBetweenNotificationsMillis() {
        return getPeriodicCheckIntervalMillis() - (long) (getFlexForPeriodicCheckMillis() * 2.1);
    }

    /**
     * Load the list of {@link UserPackage packages} we already shown a notification for.
     *
     * @return The list of packages we already shown a notification for.
     */
    private @NonNull ArraySet<UserPackage> loadAlreadyNotifiedPackagesLocked() {
        try (BufferedReader reader = new BufferedReader(new InputStreamReader(
            mContext.openFileInput(LOCATION_ACCESS_CHECK_ALREADY_NOTIFIED_FILE)))) {
            ArraySet<UserPackage> packages = new ArraySet<>();

            /*
             * The format of the file is <package> <serial of user> <dismissed in safety center>,
             * Since notification timestamp was added later it is possible that it might be
             * missing during the first check. We need to handle that.
             *
             * e.g.
             * com.one.package 5630633845 true
             * com.two.package 5630633853 false
             * com.three.package 5630633853 false
             */
            while (true) {
                String line = reader.readLine();
                if (line == null) {
                    break;
                }
                String[] lineComponents = line.split(" ");
                String pkg = lineComponents[0];
                UserHandle user = mUserManager.getUserForSerialNumber(
                        Long.valueOf(lineComponents[1]));
                boolean dismissedInSafetyCenter = lineComponents.length == 3
                        ? Boolean.valueOf(lineComponents[2]) : false;
                if (user != null) {
                    UserPackage userPkg = new UserPackage(mContext, pkg, user,
                            dismissedInSafetyCenter);
                    packages.add(userPkg);
                } else {
                    Log.i(LOG_TAG, "Not restoring state \"" + line + "\" as user is unknown");
                }
            }
            return packages;
        } catch (FileNotFoundException ignored) {
            return new ArraySet<>();
        } catch (Exception e) {
            Log.w(LOG_TAG, "Could not read " + LOCATION_ACCESS_CHECK_ALREADY_NOTIFIED_FILE, e);
            return new ArraySet<>();
        }
    }

    /**
     * Persist the list of {@link UserPackage packages} we have already shown a notification for.
     *
     * @param packages The list of packages we already shown a notification for.
     */
    private void persistAlreadyNotifiedPackagesLocked(@NonNull ArraySet<UserPackage> packages) {
        try (BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(
                mContext.openFileOutput(LOCATION_ACCESS_CHECK_ALREADY_NOTIFIED_FILE,
                        MODE_PRIVATE)))) {
            /*
             * The format of the file is <package> <serial of user> <dismissed in safety center>,
             * e.g.
             * com.one.package 5630633845 true
             * com.two.package 5630633853 false
             * com.three.package 5630633853 false
             */
            int numPkgs = packages.size();
            for (int i = 0; i < numPkgs; i++) {
                UserPackage userPkg = packages.valueAt(i);
                writer.append(userPkg.pkg);
                writer.append(' ');
                writer.append(
                        Long.valueOf(mUserManager.getSerialNumberForUser(userPkg.user)).toString());
                writer.append(' ');
                writer.append(Boolean.toString(userPkg.dismissedInSafetyCenter));
                writer.newLine();
            }
        } catch (IOException e) {
            Log.e(LOG_TAG, "Could not write " + LOCATION_ACCESS_CHECK_ALREADY_NOTIFIED_FILE, e);
        }
    }

    /**
     * Remember that we showed a notification for a {@link UserPackage}
     *
     * @param pkg                     The package we notified for
     * @param user                    The user we notified for
     * @param dismissedInSafetyCenter Whether this warning was dismissed by the user in safety
     *                                center
     */
    private void markAsNotified(@NonNull String pkg, @NonNull UserHandle user,
            boolean dismissedInSafetyCenter) {
        synchronized (sLock) {
            ArraySet<UserPackage> alreadyNotifiedPackages = loadAlreadyNotifiedPackagesLocked();
            UserPackage userPackage = new UserPackage(mContext, pkg, user, dismissedInSafetyCenter);
            // Remove stale persisted info
            alreadyNotifiedPackages.remove(userPackage);
            // Persist new info about the package
            alreadyNotifiedPackages.add(userPackage);
            persistAlreadyNotifiedPackagesLocked(alreadyNotifiedPackages);
        }
    }

    /**
     * Create the channel the location access notifications should be posted to.
     *
     * @param user The user to create the channel for
     */
    private void createPermissionReminderChannel(@NonNull UserHandle user) {
        NotificationManager notificationManager = getSystemServiceSafe(mContext,
                NotificationManager.class, user);

        NotificationChannel permissionReminderChannel = new NotificationChannel(
                PERMISSION_REMINDER_CHANNEL_ID, mContext.getString(R.string.permission_reminders),
                IMPORTANCE_LOW);
        notificationManager.createNotificationChannel(permissionReminderChannel);
    }

    /**
     * If {@link #mShouldCancel} throw an {@link InterruptedException}.
     */
    private void throwInterruptedExceptionIfTaskIsCanceled() throws InterruptedException {
        if (mShouldCancel != null && mShouldCancel.getAsBoolean()) {
            throw new InterruptedException();
        }
    }

    /**
     * Create a new {@link LocationAccessCheck} object.
     *
     * @param context      Used to resolve managers
     * @param shouldCancel If supplied, can be used to interrupt long running operations
     */
    public LocationAccessCheck(@NonNull Context context, @Nullable BooleanSupplier shouldCancel) {
        mContext = getParentUserContext(context);
        mJobScheduler = getSystemServiceSafe(mContext, JobScheduler.class);
        mAppOpsManager = getSystemServiceSafe(mContext, AppOpsManager.class);
        mPackageManager = mContext.getPackageManager();
        mUserManager = getSystemServiceSafe(mContext, UserManager.class);
        mSharedPrefs = mContext.getSharedPreferences(PREFERENCES_FILE, MODE_PRIVATE);
        mContentResolver = mContext.getContentResolver();
        mShouldCancel = shouldCancel;
    }

    /**
     * Check if a location access notification should be shown and then add it.
     *
     * <p>Always run async inside a
     * {@link LocationAccessCheckJobService.AddLocationNotificationIfNeededTask}.
     */
    @WorkerThread
    private void addLocationNotificationIfNeeded(@NonNull JobParameters params,
            @NonNull LocationAccessCheckJobService service) {
        synchronized (sLock) {
            try {
                if (currentTimeMillis() - mSharedPrefs.getLong(
                        KEY_LAST_LOCATION_ACCESS_NOTIFICATION_SHOWN, 0)
                        < getInBetweenNotificationsMillis()) {
                    Log.i(LOG_TAG, "location notification interval is not enough.");
                    service.jobFinished(params, false);
                    return;
                }

                if (getCurrentlyShownNotificationLocked() != null) {
                    Log.i(LOG_TAG, "already location notification exist.");
                    service.jobFinished(params, false);
                    return;
                }

                addLocationNotificationIfNeeded(mAppOpsManager.getPackagesForOps(
                        new String[]{OPSTR_FINE_LOCATION}), service.getApplication());
                service.jobFinished(params, false);
            } catch (Exception e) {
                Log.e(LOG_TAG, "Could not check for location access", e);
                service.jobFinished(params, true);
            } finally {
                synchronized (sLock) {
                    service.mAddLocationNotificationIfNeededTask = null;
                }
            }
        }
    }

    private void addLocationNotificationIfNeeded(@NonNull List<PackageOps> ops, Application app)
            throws InterruptedException {
        synchronized (sLock) {
            List<UserPackage> packages = getLocationUsersLocked(ops);
            ArraySet<UserPackage> alreadyNotifiedPackages = loadAlreadyNotifiedPackagesLocked();
            if (DEBUG) {
                Log.d(LOG_TAG, "location packages: " + packages);
                Log.d(LOG_TAG, "already notified packages: " + alreadyNotifiedPackages);
            }
            throwInterruptedExceptionIfTaskIsCanceled();
            // Send these issues to safety center
            if (isSafetyCenterBgLocationReminderEnabled()) {
                SafetyEvent safetyEvent = new SafetyEvent.Builder(
                        SafetyEvent.SAFETY_EVENT_TYPE_SOURCE_STATE_CHANGED).build();
                sendToSafetyCenter(packages, safetyEvent, alreadyNotifiedPackages, null);
            }
            filterAlreadyNotifiedPackagesLocked(packages, alreadyNotifiedPackages);

            // Get a random package and resolve package info
            PackageInfo pkgInfo = null;
            while (pkgInfo == null) {
                throwInterruptedExceptionIfTaskIsCanceled();

                if (packages.isEmpty()) {
                    if (DEBUG) {
                        Log.d(LOG_TAG, "No package found to send a notification");
                    }
                    return;
                }

                UserPackage packageToNotifyFor = null;

                // Prefer to show notification for location controller extra package
                int numPkgs = packages.size();
                for (int i = 0; i < numPkgs; i++) {
                    UserPackage pkg = packages.get(i);

                    LocationManager locationManager = getSystemServiceSafe(mContext,
                            LocationManager.class, pkg.user);
                    if (locationManager.isExtraLocationControllerPackageEnabled() && pkg.pkg.equals(
                            locationManager.getExtraLocationControllerPackage())) {
                        packageToNotifyFor = pkg;
                        break;
                    }
                }

                if (packageToNotifyFor == null) {
                    packageToNotifyFor = packages.get(mRandom.nextInt(packages.size()));
                }

                try {
                    pkgInfo = packageToNotifyFor.getPackageInfo();
                } catch (PackageManager.NameNotFoundException e) {
                    packages.remove(packageToNotifyFor);
                }
            }
            createPermissionReminderChannel(getUserHandleForUid(pkgInfo.applicationInfo.uid));
            createNotificationForLocationUser(pkgInfo, app);
        }
    }

    /**
     * Get the {@link UserPackage packages} which accessed the location
     *
     * <p>This also ignores all packages that are excepted from the notification.
     *
     * @return The packages we might need to show a notification for
     * @throws InterruptedException If {@link #mShouldCancel}
     */
    private @NonNull List<UserPackage> getLocationUsersLocked(
            @NonNull List<PackageOps> allOps) throws InterruptedException {
        List<UserPackage> pkgsWithLocationAccess = new ArrayList<>();
        List<UserHandle> profiles = mUserManager.getUserProfiles();

        LocationManager lm = mContext.getSystemService(LocationManager.class);

        int numPkgs = allOps.size();
        for (int pkgNum = 0; pkgNum < numPkgs; pkgNum++) {
            PackageOps packageOps = allOps.get(pkgNum);

            String pkg = packageOps.getPackageName();
            if (pkg.equals(OS_PKG) || lm.isProviderPackage(pkg)) {
                continue;
            }

            UserHandle user = getUserHandleForUid(packageOps.getUid());
            // Do not handle apps that belong to a different profile user group
            if (!profiles.contains(user)) {
                continue;
            }

            UserPackage userPkg = new UserPackage(mContext, pkg, user, false);
            AppPermissionGroup bgLocationGroup = userPkg.getBackgroundLocationGroup();
            // Do not show notification that do not request the background permission anymore
            if (bgLocationGroup == null) {
                continue;
            }

            // Do not show notification that do not currently have the background permission
            // granted
            if (!bgLocationGroup.areRuntimePermissionsGranted()) {
                continue;
            }

            // Do not show notification for permissions that are not user sensitive
            if (!bgLocationGroup.isUserSensitive()) {
                continue;
            }

            // Never show notification for pregranted permissions as warning the user via the
            // notification and then warning the user again when revoking the permission is
            // confusing
            if (userPkg.getLocationGroup().hasGrantedByDefaultPermission()
                    && bgLocationGroup.hasGrantedByDefaultPermission()) {
                continue;
            }

            int numOps = packageOps.getOps().size();
            for (int opNum = 0; opNum < numOps; opNum++) {
                OpEntry entry = packageOps.getOps().get(opNum);

                // To protect against OEM apps that accidentally blame app ops on other packages
                // since they can hold the privileged UPDATE_APP_OPS_STATS permission for location
                // access in the background we trust only the OS and the location providers. Note
                // that this mitigation only handles usage of AppOpsManager#noteProxyOp and not
                // direct usage of AppOpsManager#noteOp, i.e. handles bad blaming and not bad
                // attribution.
                String proxyPackageName = entry.getProxyPackageName();
                if (proxyPackageName != null && !proxyPackageName.equals(OS_PKG)
                        && !lm.isProviderPackage(proxyPackageName)) {
                    continue;
                }

                // We show only bg accesses since the location access check feature was enabled
                // to handle cases where the feature is remotely toggled since we don't want to
                // notify for accesses before the feature was turned on.
                long featureEnabledTime = getLocationAccessCheckEnabledTime();
                if (entry.getLastAccessBackgroundTime(AppOpsManager.OP_FLAGS_ALL_TRUSTED)
                        >= featureEnabledTime) {
                    pkgsWithLocationAccess.add(userPkg);
                    break;
                }
            }
        }
        return pkgsWithLocationAccess;
    }

    private void filterAlreadyNotifiedPackagesLocked(
            @NonNull List<UserPackage> pkgsWithLocationAccess,
            @NonNull ArraySet<UserPackage> alreadyNotifiedPkgs) throws InterruptedException {
        resetAlreadyNotifiedPackagesWithoutPermissionLocked(alreadyNotifiedPkgs);
        pkgsWithLocationAccess.removeAll(alreadyNotifiedPkgs);
    }

    /**
     * Sets the LocationAccessCheckEnabledTime if not set.
     */
    private void setLocationAccessCheckEnabledTime() {
        if (isLocationAccessCheckEnabledTimeNotSet()) {
            mSharedPrefs.edit().putLong(KEY_LOCATION_ACCESS_CHECK_ENABLED_TIME,
                    currentTimeMillis()).apply();
        }
    }

    /**
     * @return true if the LocationAccessCheckEnabledTime has not been set, else false.
     */
    private boolean isLocationAccessCheckEnabledTimeNotSet() {
        return mSharedPrefs.getLong(KEY_LOCATION_ACCESS_CHECK_ENABLED_TIME, 0) == 0;
    }

    /**
     * @return The time the location access check was enabled, or currentTimeMillis if not set.
     */
    private long getLocationAccessCheckEnabledTime() {
        return mSharedPrefs.getLong(KEY_LOCATION_ACCESS_CHECK_ENABLED_TIME, currentTimeMillis());
    }

    /**
     * Create a notification reminding the user that a package used the location. From this
     * notification the user can directly go to the screen that allows to change the permission.
     *
     * @param pkg The {@link PackageInfo} for the package to to be changed
     */
    private void createNotificationForLocationUser(@NonNull PackageInfo pkg, Application app) {
        CharSequence pkgLabel = mPackageManager.getApplicationLabel(pkg.applicationInfo);

        boolean safetyCenterBgLocationReminderEnabled = isSafetyCenterBgLocationReminderEnabled();

        String pkgName = pkg.packageName;
        int uid = pkg.applicationInfo.uid;
        UserHandle user = getUserHandleForUid(uid);

        NotificationManager notificationManager = getSystemServiceSafe(mContext,
                NotificationManager.class, user);

        long sessionId = INVALID_SESSION_ID;
        while (sessionId == INVALID_SESSION_ID) {
            sessionId = new Random().nextLong();
        }

        CharSequence appName = Utils.getSettingsLabelForNotifications(mPackageManager);

        CharSequence notificationTitle =
                safetyCenterBgLocationReminderEnabled ? mContext.getString(
                        R.string.safety_center_background_location_access_notification_title
                ) : mContext.getString(
                        R.string.background_location_access_reminder_notification_title,
                        pkgLabel);

        CharSequence notificationContent = safetyCenterBgLocationReminderEnabled
                ? mContext.getString(
                R.string.safety_center_background_location_access_reminder_notification_content,
                pkgLabel) : mContext.getString(
                R.string.background_location_access_reminder_notification_content);

        CharSequence appLabel = appName;
        Icon smallIcon;
        int color = mContext.getColor(android.R.color.system_notification_accent_color);
        if (safetyCenterBgLocationReminderEnabled) {
            KotlinUtils.NotificationResources notifRes =
                    KotlinUtils.INSTANCE.getSafetyCenterNotificationResources(mContext);
            appLabel = notifRes.getAppLabel();
            smallIcon = notifRes.getSmallIcon();
            color = notifRes.getColor();
        } else {
            smallIcon = Icon.createWithResource(mContext, R.drawable.ic_pin_drop);
        }

        Notification.Builder b = (new Notification.Builder(mContext,
                PERMISSION_REMINDER_CHANNEL_ID))
                .setLocalOnly(true)
                .setContentTitle(notificationTitle)
                .setContentText(notificationContent)
                .setStyle(new Notification.BigTextStyle().bigText(notificationContent))
                .setSmallIcon(smallIcon)
                .setColor(color)
                .setDeleteIntent(createNotificationDismissIntent(pkgName, sessionId, uid))
                .setContentIntent(createNotificationClickIntent(pkgName, user, sessionId, uid))
                .setAutoCancel(true);

        if (!safetyCenterBgLocationReminderEnabled) {
            Drawable pkgIcon = mPackageManager.getApplicationIcon(pkg.applicationInfo);
            Bitmap pkgIconBmp = createBitmap(pkgIcon.getIntrinsicWidth(),
                    pkgIcon.getIntrinsicHeight(),
                    ARGB_8888);
            Canvas canvas = new Canvas(pkgIconBmp);
            pkgIcon.setBounds(0, 0, pkgIcon.getIntrinsicWidth(), pkgIcon.getIntrinsicHeight());
            pkgIcon.draw(canvas);
            b.setLargeIcon(pkgIconBmp);
        }

        Bundle extras = new Bundle();
        if (DeviceUtils.isAuto(mContext)) {
            Bitmap settingsIcon = KotlinUtils.INSTANCE.getSettingsIcon(app, user, mPackageManager);
            b.setLargeIcon(settingsIcon);
            extras.putBoolean(Constants.NOTIFICATION_EXTRA_USE_LAUNCHER_ICON, false);
        }

        if (!TextUtils.isEmpty(appLabel)) {
            String appNameSubstitute = appLabel.toString();
            extras.putString(Notification.EXTRA_SUBSTITUTE_APP_NAME, appNameSubstitute);
        }
        b.addExtras(extras);

        notificationManager.notify(pkgName, LOCATION_ACCESS_CHECK_NOTIFICATION_ID, b.build());
        markAsNotified(pkgName, user, false);

        if (DEBUG) {
            Log.d(LOG_TAG,
                    "Location access check notification shown with sessionId=" + sessionId + ""
                            + " uid=" + pkg.applicationInfo.uid + " pkgName=" + pkgName);
        }
        if (safetyCenterBgLocationReminderEnabled) {
            PermissionControllerStatsLog.write(
                    PRIVACY_SIGNAL_NOTIFICATION_INTERACTION,
                    PRIVACY_SIGNAL_NOTIFICATION_INTERACTION__PRIVACY_SOURCE__BG_LOCATION,
                    uid,
                    PRIVACY_SIGNAL_NOTIFICATION_INTERACTION__ACTION__NOTIFICATION_SHOWN,
                    sessionId);
        } else {
            PermissionControllerStatsLog.write(LOCATION_ACCESS_CHECK_NOTIFICATION_ACTION, sessionId,
                    pkg.applicationInfo.uid, pkgName,
                    LOCATION_ACCESS_CHECK_NOTIFICATION_ACTION__RESULT__NOTIFICATION_PRESENTED);
        }

        mSharedPrefs.edit().putLong(KEY_LAST_LOCATION_ACCESS_NOTIFICATION_SHOWN,
                currentTimeMillis()).apply();
    }

    /**
     * Get currently shown notification. We only ever show one notification per profile group.
     *
     * @return The notification or {@code null} if no notification is currently shown
     */
    private @Nullable StatusBarNotification getCurrentlyShownNotificationLocked() {
        List<UserHandle> profiles = mUserManager.getUserProfiles();

        int numProfiles = profiles.size();
        for (int profileNum = 0; profileNum < numProfiles; profileNum++) {
            NotificationManager notificationManager;
            try {
                notificationManager = getSystemServiceSafe(mContext, NotificationManager.class,
                        profiles.get(profileNum));
            } catch (IllegalStateException e) {
                continue;
            }

            StatusBarNotification[] notifications = notificationManager.getActiveNotifications();

            int numNotifications = notifications.length;
            for (int notificationNum = 0; notificationNum < numNotifications; notificationNum++) {
                StatusBarNotification notification = notifications[notificationNum];
                if (notification.getId() == LOCATION_ACCESS_CHECK_NOTIFICATION_ID
                        && notification.getUser() != null && notification.getTag() != null) {
                    return notification;
                }
            }
        }
        return null;
    }

    /**
     * Go through the list of packages we already shown a notification for and remove those that do
     * not request fine background location access.
     *
     * @param alreadyNotifiedPkgs The packages we already shown a notification for. This parameter
     *                            is modified inside of this method.
     * @throws InterruptedException If {@link #mShouldCancel}
     */
    private void resetAlreadyNotifiedPackagesWithoutPermissionLocked(
            @NonNull ArraySet<UserPackage> alreadyNotifiedPkgs) throws InterruptedException {
        ArrayList<UserPackage> packagesToRemove = new ArrayList<>();

        for (UserPackage userPkg : alreadyNotifiedPkgs) {
            throwInterruptedExceptionIfTaskIsCanceled();
            AppPermissionGroup bgLocationGroup = userPkg.getBackgroundLocationGroup();
            if (bgLocationGroup == null || !bgLocationGroup.areRuntimePermissionsGranted()) {
                packagesToRemove.add(userPkg);
            }
        }

        if (!packagesToRemove.isEmpty()) {
            alreadyNotifiedPkgs.removeAll(packagesToRemove);
            persistAlreadyNotifiedPackagesLocked(alreadyNotifiedPkgs);
            throwInterruptedExceptionIfTaskIsCanceled();
        }
    }

    /**
     * Remove all persisted state for a package.
     *
     * @param pkg  name of package
     * @param user user the package belongs to
     */
    private void forgetAboutPackage(@NonNull String pkg, @NonNull UserHandle user) {
        synchronized (sLock) {
            StatusBarNotification notification = getCurrentlyShownNotificationLocked();
            if (notification != null && notification.getUser().equals(user)
                    && notification.getTag().equals(pkg)) {
                getSystemServiceSafe(mContext, NotificationManager.class, user).cancel(
                        pkg, LOCATION_ACCESS_CHECK_NOTIFICATION_ID);
            }

            ArraySet<UserPackage> packages = loadAlreadyNotifiedPackagesLocked();
            packages.remove(new UserPackage(mContext, pkg, user, false));
            persistAlreadyNotifiedPackagesLocked(packages);
        }
    }

    /**
     * After a small delay schedule a check if we should show a notification.
     *
     * <p>This is called when location access is granted to an app. In this case it is likely that
     * the app will access the location soon. If this happens the notification will appear only a
     * little after the user granted the location.
     */
    public void checkLocationAccessSoon() {
        JobInfo.Builder b = (new JobInfo.Builder(LOCATION_ACCESS_CHECK_JOB_ID,
                new ComponentName(mContext, LocationAccessCheckJobService.class)))
                .setMinimumLatency(getDelayMillis());

        int scheduleResult = mJobScheduler.schedule(b.build());
        if (scheduleResult != RESULT_SUCCESS) {
            Log.e(LOG_TAG, "Could not schedule location access check " + scheduleResult);
        }
    }

    /**
     * Cancel the background access warning notification for an app if the permission has been
     * revoked for the app and forget persisted information about the app
     */
    public void cancelBackgroundAccessWarningNotification(String packageName, UserHandle user,
            Boolean forgetAboutPackage) {
        // Cancel the current notification if background
        // location access for the package is revoked
        StatusBarNotification notification = getCurrentlyShownNotificationLocked();
        if (notification != null && notification.getUser().equals(user)
                && notification.getTag().equals(packageName)) {
            getSystemServiceSafe(mContext, NotificationManager.class, user).cancel(
                    packageName, LOCATION_ACCESS_CHECK_NOTIFICATION_ID);
        }

        if (isSafetyCenterBgLocationReminderEnabled()) {
            rescanAndPushSafetyCenterData(new SafetyEvent.Builder(
                    SafetyEvent.SAFETY_EVENT_TYPE_SOURCE_STATE_CHANGED)
                    .build(), user);
        }

        if (forgetAboutPackage) {
            forgetAboutPackage(packageName, user);
        }
    }

    /**
     * Cancel the background access warning notification if currently being shown
     */
    public void cancelBackgroundAccessWarningNotification() {
        StatusBarNotification notification = getCurrentlyShownNotificationLocked();
        if (notification != null) {
            getSystemServiceSafe(mContext, NotificationManager.class,
                    notification.getUser()).cancel(
                    notification.getTag(), LOCATION_ACCESS_CHECK_NOTIFICATION_ID);
        }
    }

    @ChecksSdkIntAtLeast(api = Build.VERSION_CODES.TIRAMISU)
    private boolean isSafetyCenterBgLocationReminderEnabled() {
        if (!SdkLevel.isAtLeastT()) {
            return false;
        }

        return DeviceConfig.getBoolean(
                DeviceConfig.NAMESPACE_PRIVACY,
                PROPERTY_BG_LOCATION_CHECK_ENABLED, true)
                && getSystemServiceSafe(mContext,
                SafetyCenterManager.class).isSafetyCenterEnabled();
    }

    @RequiresApi(Build.VERSION_CODES.TIRAMISU)
    private void sendToSafetyCenter(List<UserPackage> userPackages, SafetyEvent safetyEvent,
            @Nullable ArraySet<UserPackage> alreadyNotifiedPackages, @Nullable UserHandle user) {
        try {
            Set<UserPackage> alreadyDismissedPackages =
                    getAlreadyDismissedPackages(alreadyNotifiedPackages);

            // Filter out packages already dismissed by the user in safety center
            List<UserPackage> filteredPackages = userPackages.stream().filter(
                    pkg -> !alreadyDismissedPackages.contains(pkg)).collect(
                    Collectors.toList());

            Map<UserHandle, List<UserPackage>> userHandleToUserPackagesMap =
                    splitUserPackageByUserHandle(filteredPackages);

            if (user == null) {
                // Get all the user profiles
                List<UserHandle> userProfiles = mUserManager.getUserProfiles();
                for (UserHandle userProfile : userProfiles) {
                    sendUserDataToSafetyCenter(userHandleToUserPackagesMap.getOrDefault(userProfile,
                            new ArrayList<>()), safetyEvent, userProfile);
                }
            } else {
                sendUserDataToSafetyCenter(userHandleToUserPackagesMap.getOrDefault(user,
                        new ArrayList<>()), safetyEvent, user);
            }

        } catch (Exception e) {
            Log.e(LOG_TAG, "Could not send to safety center", e);
        }
    }

    private Set<UserPackage> getAlreadyDismissedPackages(
            @Nullable ArraySet<UserPackage> alreadyNotifiedPackages) {
        if (alreadyNotifiedPackages == null) {
            alreadyNotifiedPackages = loadAlreadyNotifiedPackagesLocked();
        }
        return alreadyNotifiedPackages.stream().filter(
                pkg -> pkg.dismissedInSafetyCenter).collect(
                Collectors.toSet());
    }

    @RequiresApi(Build.VERSION_CODES.TIRAMISU)
    private Map<UserHandle, List<UserPackage>> splitUserPackageByUserHandle(
            List<UserPackage> userPackages) {
        Map<UserHandle, List<UserPackage>> userHandleToUserPackagesMap = new ArrayMap<>();
        for (UserPackage userPackage : userPackages) {
            if (userHandleToUserPackagesMap.get(userPackage.user) == null) {
                userHandleToUserPackagesMap.put(userPackage.user, new ArrayList<>());
            }
            userHandleToUserPackagesMap.get(userPackage.user).add(userPackage);
        }
        return userHandleToUserPackagesMap;
    }

    @RequiresApi(Build.VERSION_CODES.TIRAMISU)
    private void sendUserDataToSafetyCenter(List<UserPackage> userPackages,
            SafetyEvent safetyEvent, @Nullable UserHandle user) {
        SafetySourceData.Builder safetySourceDataBuilder = new SafetySourceData.Builder();
        Context userContext = null;
        for (UserPackage userPkg : userPackages) {
            if (userContext == null) {
                userContext = userPkg.mContext;
            }
            SafetySourceIssue sourceIssue = createSafetySourceIssue(userPkg);
            if (sourceIssue != null) {
                safetySourceDataBuilder.addIssue(sourceIssue);
            }
        }
        if (userContext == null && user != null) {
            userContext = mContext.createContextAsUser(user, 0);
        }
        if (userContext != null) {
            getSystemServiceSafe(userContext, SafetyCenterManager.class).setSafetySourceData(
                    BG_LOCATION_SOURCE_ID,
                    safetySourceDataBuilder.build(),
                    safetyEvent
            );
        }
    }

    @RequiresApi(Build.VERSION_CODES.TIRAMISU)
    private SafetySourceIssue createSafetySourceIssue(UserPackage userPackage) {
        PackageInfo pkgInfo = null;
        try {
            pkgInfo = userPackage.getPackageInfo();
        } catch (PackageManager.NameNotFoundException e) {
            Log.e(LOG_TAG, "Could not get package info for " + userPackage, e);
            return null;
        }

        long sessionId = INVALID_SESSION_ID;
        while (sessionId == INVALID_SESSION_ID) {
            sessionId = new Random().nextLong();
        }

        int uid = pkgInfo.applicationInfo.uid;

        Intent primaryActionIntent = new Intent(mContext, SafetyCenterPrimaryActionHandler.class);
        primaryActionIntent.putExtra(EXTRA_PACKAGE_NAME, userPackage.pkg);
        primaryActionIntent.putExtra(EXTRA_USER, userPackage.user);
        primaryActionIntent.putExtra(EXTRA_UID, uid);
        primaryActionIntent.putExtra(EXTRA_SESSION_ID, sessionId);
        primaryActionIntent.setFlags(FLAG_RECEIVER_FOREGROUND);
        primaryActionIntent.setIdentifier(userPackage.pkg + userPackage.user);

        PendingIntent revokeIntent = PendingIntent.getBroadcast(mContext, 0,
                primaryActionIntent,
                FLAG_ONE_SHOT | FLAG_UPDATE_CURRENT | FLAG_IMMUTABLE);

        Action revokeAction = new Action.Builder(createLocationRevokeActionId(userPackage.pkg,
                userPackage.user),
                mContext.getString(R.string.permission_access_only_foreground),
                revokeIntent).setWillResolve(true).setSuccessMessage(mContext.getString(
                R.string.safety_center_background_location_access_revoked)).build();

        Intent secondaryActionIntent = new Intent(Intent.ACTION_REVIEW_PERMISSION_HISTORY);
        secondaryActionIntent.putExtra(Intent.EXTRA_PERMISSION_GROUP_NAME, LOCATION);

        PendingIntent locationUsageIntent = PendingIntent.getActivity(mContext, 0,
                secondaryActionIntent,
                FLAG_UPDATE_CURRENT | FLAG_IMMUTABLE);

        Action viewLocationUsageAction = new Action.Builder(VIEW_LOCATION_ACCESS_ID,
                mContext.getString(R.string.safety_center_view_recent_location_access),
                locationUsageIntent).build();

        String pkgName = userPackage.pkg;
        String id = createSafetySourceIssueId(pkgName, userPackage.user);

        CharSequence pkgLabel = mPackageManager.getApplicationLabel(pkgInfo.applicationInfo);

        return new SafetySourceIssue.Builder(
                id,
                mContext.getString(
                        R.string.safety_center_background_location_access_reminder_title),
                mContext.getString(
                        R.string.safety_center_background_location_access_reminder_summary),
                SafetySourceData.SEVERITY_LEVEL_INFORMATION,
                ISSUE_TYPE_ID)
                .setSubtitle(pkgLabel)
                .addAction(revokeAction)
                .addAction(viewLocationUsageAction)
                .setOnDismissPendingIntent(
                        createWarningCardDismissalIntent(pkgName, sessionId, uid))
                .setIssueCategory(SafetySourceIssue.ISSUE_CATEGORY_DEVICE)
                .build();
    }

    private PendingIntent createNotificationDismissIntent(String pkgName, long sessionId, int uid) {
        Intent dismissIntent = new Intent(mContext, NotificationDeleteHandler.class);
        dismissIntent.putExtra(EXTRA_PACKAGE_NAME, pkgName);
        dismissIntent.putExtra(EXTRA_SESSION_ID, sessionId);
        dismissIntent.putExtra(EXTRA_UID, uid);
        UserHandle user = getUserHandleForUid(uid);
        dismissIntent.putExtra(EXTRA_USER, user);
        dismissIntent.setIdentifier(pkgName + user);
        dismissIntent.setFlags(FLAG_RECEIVER_FOREGROUND);
        return PendingIntent.getBroadcast(mContext, 0, dismissIntent,
                FLAG_ONE_SHOT | FLAG_UPDATE_CURRENT | FLAG_IMMUTABLE);
    }

    private PendingIntent createNotificationClickIntent(String pkg, UserHandle user,
            long sessionId, int uid) {
        Intent clickIntent = null;
        if (isSafetyCenterBgLocationReminderEnabled()) {
            clickIntent = new Intent(ACTION_SAFETY_CENTER);
            clickIntent.putExtra(EXTRA_SAFETY_SOURCE_ID, BG_LOCATION_SOURCE_ID);
            clickIntent.putExtra(
                    EXTRA_SAFETY_SOURCE_ISSUE_ID, createSafetySourceIssueId(pkg, user));
            clickIntent.putExtra(EXTRA_SAFETY_SOURCE_USER_HANDLE, user);
        } else {
            clickIntent = new Intent(ACTION_MANAGE_APP_PERMISSION);
            clickIntent.putExtra(EXTRA_PERMISSION_GROUP_NAME, LOCATION);
        }
        clickIntent.putExtra(EXTRA_PACKAGE_NAME, pkg);
        clickIntent.putExtra(EXTRA_USER, user);
        clickIntent.putExtra(EXTRA_SESSION_ID, sessionId);
        clickIntent.putExtra(EXTRA_UID, uid);
        clickIntent.addFlags(FLAG_ACTIVITY_NEW_TASK | FLAG_ACTIVITY_MULTIPLE_TASK);
        return PendingIntent.getActivity(mContext, 0, clickIntent,
                FLAG_ONE_SHOT | FLAG_UPDATE_CURRENT | FLAG_IMMUTABLE);
    }

    private PendingIntent createWarningCardDismissalIntent(String pkgName, long sessionId,
            int uid) {
        Intent dismissIntent = new Intent(mContext, WarningCardDismissalHandler.class);
        dismissIntent.putExtra(EXTRA_PACKAGE_NAME, pkgName);
        dismissIntent.putExtra(EXTRA_SESSION_ID, sessionId);
        dismissIntent.putExtra(EXTRA_UID, uid);
        UserHandle user = getUserHandleForUid(uid);
        dismissIntent.putExtra(EXTRA_USER, user);
        dismissIntent.setIdentifier(pkgName + user);
        dismissIntent.setFlags(FLAG_RECEIVER_FOREGROUND);
        return PendingIntent.getBroadcast(mContext, 0, dismissIntent,
                FLAG_ONE_SHOT | FLAG_UPDATE_CURRENT | FLAG_IMMUTABLE);
    }

    /**
     * Check if the current user is the profile parent.
     *
     * @return {@code true} if the current user is the profile parent.
     */
    private boolean isRunningInParentProfile() {
        UserHandle user = UserHandle.of(myUserId());
        UserHandle parent = mUserManager.getProfileParent(user);

        return parent == null || user.equals(parent);
    }

    /**
     * Query for packages having background location access and push to safety center
     *
     * @param safetyEvent Safety event for which data is being pushed
     * @param user Optional, if supplied only send safety center data for that user
     */
    @RequiresApi(Build.VERSION_CODES.TIRAMISU)
    public void rescanAndPushSafetyCenterData(SafetyEvent safetyEvent, @Nullable UserHandle user) {
        if (!isSafetyCenterBgLocationReminderEnabled()) {
            return;
        }
        try {
            List<UserPackage> packages = getLocationUsersLocked(mAppOpsManager.getPackagesForOps(
                    new String[]{OPSTR_FINE_LOCATION}));
            sendToSafetyCenter(packages, safetyEvent, null, user);
        } catch (InterruptedException e) {
            Log.e(LOG_TAG, "Couldn't get ops for location");
        }
    }

    /**
     * On boot set up a periodic job that starts checks.
     */
    public static class SetupPeriodicBackgroundLocationAccessCheck extends BroadcastReceiver {
        @Override
        public void onReceive(Context context, Intent intent) {
            LocationAccessCheck locationAccessCheck = new LocationAccessCheck(context, null);
            JobScheduler jobScheduler = getSystemServiceSafe(context, JobScheduler.class);

            if (!locationAccessCheck.isRunningInParentProfile()) {
                // Profile parent handles child profiles too.
                return;
            }

            // Init LocationAccessCheckEnabledTime if needed
            locationAccessCheck.setLocationAccessCheckEnabledTime();

            if (jobScheduler.getPendingJob(PERIODIC_LOCATION_ACCESS_CHECK_JOB_ID) == null) {
                JobInfo.Builder b = (new JobInfo.Builder(PERIODIC_LOCATION_ACCESS_CHECK_JOB_ID,
                        new ComponentName(context, LocationAccessCheckJobService.class)))
                        .setPeriodic(locationAccessCheck.getPeriodicCheckIntervalMillis(),
                                locationAccessCheck.getFlexForPeriodicCheckMillis());

                int scheduleResult = jobScheduler.schedule(b.build());
                if (scheduleResult != RESULT_SUCCESS) {
                    Log.e(LOG_TAG, "Could not schedule periodic location access check "
                            + scheduleResult);
                }
            }
        }
    }

    /**
     * Checks if a new notification should be shown.
     */
    public static class LocationAccessCheckJobService extends JobService {
        private LocationAccessCheck mLocationAccessCheck;

        /**
         * If we currently check if we should show a notification, the task executing the check
         */
        // @GuardedBy("sLock")
        private @Nullable AddLocationNotificationIfNeededTask mAddLocationNotificationIfNeededTask;

        @Override
        public void onCreate() {
            super.onCreate();
            mLocationAccessCheck = new LocationAccessCheck(this, () -> {
                synchronized (sLock) {
                    AddLocationNotificationIfNeededTask task = mAddLocationNotificationIfNeededTask;

                    return task != null && task.isCancelled();
                }
            });
        }

        /**
         * Starts an asynchronous check if a location access notification should be shown.
         *
         * @param params Not used other than for interacting with job scheduling
         * @return {@code false} iff another check if already running
         */
        @Override
        public boolean onStartJob(JobParameters params) {
            synchronized (LocationAccessCheck.sLock) {
                if (mAddLocationNotificationIfNeededTask != null) {
                    Log.i(LOG_TAG, "LocationAccessCheck old job not completed yet.");
                    return false;
                }

                mAddLocationNotificationIfNeededTask =
                        new AddLocationNotificationIfNeededTask();

                mAddLocationNotificationIfNeededTask.execute(params, this);
            }

            return true;
        }

        /**
         * Abort the check if still running.
         *
         * @param params ignored
         * @return false
         */
        @Override
        public boolean onStopJob(JobParameters params) {
            AddLocationNotificationIfNeededTask task;
            synchronized (sLock) {
                if (mAddLocationNotificationIfNeededTask == null) {
                    return false;
                } else {
                    task = mAddLocationNotificationIfNeededTask;
                }
            }

            task.cancel(false);

            try {
                // Wait for task to finish
                task.get();
            } catch (Exception e) {
                Log.e(LOG_TAG, "While waiting for " + task + " to finish", e);
            }

            return false;
        }

        /**
         * A {@link AsyncTask task} that runs the check in the background.
         */
        private class AddLocationNotificationIfNeededTask extends
                AsyncTask<Object, Void, Void> {
            @Override
            protected final Void doInBackground(Object... in) {
                JobParameters params = (JobParameters) in[0];
                LocationAccessCheckJobService service = (LocationAccessCheckJobService) in[1];
                mLocationAccessCheck.addLocationNotificationIfNeeded(params, service);
                return null;
            }
        }
    }

    /**
     * Handle the case where the notification is swiped away without further interaction.
     */
    public static class NotificationDeleteHandler extends BroadcastReceiver {
        @Override
        public void onReceive(Context context, Intent intent) {
            String pkg = getStringExtraSafe(intent, EXTRA_PACKAGE_NAME);
            UserHandle user = getParcelableExtraSafe(intent, EXTRA_USER);
            long sessionId = intent.getLongExtra(EXTRA_SESSION_ID, INVALID_SESSION_ID);
            int uid = intent.getIntExtra(EXTRA_UID, -1);

            Log.i(LOG_TAG,
                    "Location access check notification declined with sessionId=" + sessionId + ""
                            + " uid=" + uid + " pkgName=" + pkg);
            LocationAccessCheck locationAccessCheck = new LocationAccessCheck(context, null);

            if (locationAccessCheck.isSafetyCenterBgLocationReminderEnabled()) {
                PermissionControllerStatsLog.write(
                        PRIVACY_SIGNAL_NOTIFICATION_INTERACTION,
                        PRIVACY_SIGNAL_NOTIFICATION_INTERACTION__PRIVACY_SOURCE__BG_LOCATION,
                        uid,
                        PRIVACY_SIGNAL_NOTIFICATION_INTERACTION__ACTION__DISMISSED,
                        sessionId
                );
            } else {
                PermissionControllerStatsLog.write(LOCATION_ACCESS_CHECK_NOTIFICATION_ACTION,
                        sessionId,
                        uid, pkg,
                        LOCATION_ACCESS_CHECK_NOTIFICATION_ACTION__RESULT__NOTIFICATION_DECLINED);
            }
            locationAccessCheck.markAsNotified(pkg, user, false);
        }
    }

    /**
     * Broadcast receiver to handle the primary action from a safety center warning card
     */
    @RequiresApi(Build.VERSION_CODES.TIRAMISU)
    public static class SafetyCenterPrimaryActionHandler extends BroadcastReceiver {
        @Override
        public void onReceive(Context context, Intent intent) {
            String packageName = getStringExtraSafe(intent, EXTRA_PACKAGE_NAME);
            UserHandle user = getParcelableExtraSafe(intent, EXTRA_USER);
            int uid = intent.getIntExtra(EXTRA_UID, -1);
            long sessionId = intent.getLongExtra(EXTRA_SESSION_ID, INVALID_SESSION_ID);
            // Revoke bg location permission and notify safety center
            KotlinUtils.INSTANCE.revokeBackgroundRuntimePermissions(context, packageName, LOCATION,
                    user, () -> {
                        new LocationAccessCheck(context, null).rescanAndPushSafetyCenterData(
                                new SafetyEvent.Builder(
                                        SafetyEvent.SAFETY_EVENT_TYPE_RESOLVING_ACTION_SUCCEEDED)
                                        .setSafetySourceIssueId(
                                                createSafetySourceIssueId(packageName, user))
                                        .setSafetySourceIssueActionId(
                                                createLocationRevokeActionId(packageName, user))
                                        .build(), user);
                    });
            PermissionControllerStatsLog.write(
                    PRIVACY_SIGNAL_ISSUE_CARD_INTERACTION,
                    PRIVACY_SIGNAL_ISSUE_CARD_INTERACTION__PRIVACY_SOURCE__BG_LOCATION,
                    uid,
                    PRIVACY_SIGNAL_ISSUE_CARD_INTERACTION__ACTION__CLICKED_CTA1,
                    sessionId
            );

        }
    }

    private static String createSafetySourceIssueId(String packageName, UserHandle user) {
        return ISSUE_ID_PREFIX + packageName + user;
    }

    private static String createLocationRevokeActionId(String packageName, UserHandle user) {
        return REVOKE_LOCATION_ACCESS_ID_PREFIX + packageName + user;
    }

    /**
     * Handle the case where the warning card is dismissed by the user in Safety center
     */
    public static class WarningCardDismissalHandler extends BroadcastReceiver {
        @Override
        public void onReceive(Context context, Intent intent) {
            String pkg = getStringExtraSafe(intent, EXTRA_PACKAGE_NAME);
            UserHandle user = getParcelableExtraSafe(intent, EXTRA_USER);
            long sessionId = intent.getLongExtra(EXTRA_SESSION_ID, INVALID_SESSION_ID);
            int uid = intent.getIntExtra(EXTRA_UID, -1);
            Log.i(LOG_TAG,
                    "Location access check warning card dismissed with sessionId=" + sessionId + ""
                            + " uid=" + uid + " pkgName=" + pkg);
            PermissionControllerStatsLog.write(
                    PRIVACY_SIGNAL_ISSUE_CARD_INTERACTION,
                    PRIVACY_SIGNAL_ISSUE_CARD_INTERACTION__PRIVACY_SOURCE__BG_LOCATION,
                    uid,
                    PRIVACY_SIGNAL_ISSUE_CARD_INTERACTION__ACTION__CARD_DISMISSED,
                    sessionId
            );

            LocationAccessCheck locationAccessCheck = new LocationAccessCheck(context, null);
            locationAccessCheck.markAsNotified(pkg, user, true);
            locationAccessCheck.cancelBackgroundAccessWarningNotification(pkg, user, false);
        }
    }

    /**
     * If a package gets removed or the data of the package gets cleared, forget that we showed a
     * notification for it.
     */
    public static class PackageResetHandler extends BroadcastReceiver {
        @Override
        public void onReceive(Context context, Intent intent) {
            String action = intent.getAction();
            if (!(Objects.equals(action, Intent.ACTION_PACKAGE_DATA_CLEARED)
                    || Objects.equals(action, Intent.ACTION_PACKAGE_FULLY_REMOVED))) {
                return;
            }

            Uri data = Preconditions.checkNotNull(intent.getData());
            UserHandle user = getUserHandleForUid(intent.getIntExtra(EXTRA_UID, 0));
            if (DEBUG) Log.i(LOG_TAG, "Reset " + data.getSchemeSpecificPart());
            LocationAccessCheck locationAccessCheck = new LocationAccessCheck(context, null);
            String packageName =  data.getSchemeSpecificPart();
            locationAccessCheck.forgetAboutPackage(packageName, user);
            if (locationAccessCheck.isSafetyCenterBgLocationReminderEnabled()) {
                locationAccessCheck.rescanAndPushSafetyCenterData(
                        new SafetyEvent.Builder(SafetyEvent.SAFETY_EVENT_TYPE_SOURCE_STATE_CHANGED)
                                .build(), user);
            }
        }
    }

    /**
     * A immutable class containing a package name and a {@link UserHandle}.
     */
    private static final class UserPackage {
        private final @NonNull Context mContext;

        public final @NonNull String pkg;
        public final @NonNull UserHandle user;
        public final boolean dismissedInSafetyCenter;

        /**
         * Create a new {@link UserPackage}
         *
         * @param context               A context to be used by methods of this object
         * @param pkg                   The name of the package
         * @param user                  The user the package belongs to
         * @param dismissedInSafetyCenter Optional boolean recording if the safety center
         *                                       warning was dismissed by the user
         */
        UserPackage(@NonNull Context context, @NonNull String pkg, @NonNull UserHandle user,
                boolean dismissedInSafetyCenter) {
            try {
                mContext = context.createPackageContextAsUser(context.getPackageName(), 0, user);
            } catch (PackageManager.NameNotFoundException e) {
                throw new IllegalStateException(e);
            }

            this.pkg = pkg;
            this.user = user;
            this.dismissedInSafetyCenter = dismissedInSafetyCenter;
        }

        /**
         * Get {@link PackageInfo} for this user package.
         *
         * @return The package info
         * @throws PackageManager.NameNotFoundException if package/user does not exist
         */
        @NonNull
        PackageInfo getPackageInfo() throws PackageManager.NameNotFoundException {
            return mContext.getPackageManager().getPackageInfo(pkg, GET_PERMISSIONS);
        }

        /**
         * Get the {@link AppPermissionGroup} for
         * {@link android.Manifest.permission#ACCESS_FINE_LOCATION} and this user package.
         *
         * @return The app permission group or {@code null} if the app does not request location
         */
        @Nullable
        AppPermissionGroup getLocationGroup() {
            try {
                return AppPermissionGroup.create(mContext, getPackageInfo(), ACCESS_FINE_LOCATION,
                        false);
            } catch (PackageManager.NameNotFoundException e) {
                return null;
            }
        }

        /**
         * Get the {@link AppPermissionGroup} for the background location of
         * {@link android.Manifest.permission#ACCESS_FINE_LOCATION} and this user package.
         *
         * @return The app permission group or {@code null} if the app does not request background
         * location
         */
        @Nullable
        AppPermissionGroup getBackgroundLocationGroup() {
            AppPermissionGroup locationGroup = getLocationGroup();
            if (locationGroup == null) {
                return null;
            }

            return locationGroup.getBackgroundPermissions();
        }

        @Override
        public boolean equals(Object o) {
            if (!(o instanceof UserPackage)) {
                return false;
            }

            UserPackage userPackage = (UserPackage) o;
            return pkg.equals(userPackage.pkg) && user.equals(userPackage.user);
        }

        @Override
        public int hashCode() {
            return Objects.hash(pkg, user);
        }

        @Override
        public String toString() {
            return "UserPackage { "
                    + "pkg = " + pkg + ", "
                    + "UserHandle = " + user.toString() + ", "
                    + "dismissedInSafetyCenter = " + dismissedInSafetyCenter + " }";
        }
    }
}