common/device-side/bedstead/nene/src/main/java/com/android/bedstead/nene/appops/AppOps.java

/*
 * Copyright (C) 2021 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.bedstead.nene.appops;

import static android.os.Build.VERSION_CODES.Q;

import static com.android.bedstead.nene.permissions.Permissions.MANAGE_APP_OPS_MODES;

import android.annotation.TargetApi;
import android.app.AppOpsManager;

import com.android.bedstead.nene.TestApis;
import com.android.bedstead.nene.annotations.Experimental;
import com.android.bedstead.nene.packages.Package;
import com.android.bedstead.nene.permissions.PermissionContext;
import com.android.bedstead.nene.utils.Versions;

@Experimental
/** Manage AppOps for a package. */
public final class AppOps {

    /** Valid modes for an AppOp. */
    public enum AppOpsMode {
        ALLOWED(AppOpsManager.MODE_ALLOWED),
        IGNORED(AppOpsManager.MODE_IGNORED),
        ERRORED(AppOpsManager.MODE_ERRORED),
        DEFAULT(AppOpsManager.MODE_DEFAULT),
        FOREGROUND(AppOpsManager.MODE_FOREGROUND);

        final int mValue;

        AppOpsMode(int value) {
            this.mValue = value;
        }

        static AppOpsMode forValue(int value) {
            switch (value) {
                case AppOpsManager.MODE_ALLOWED:
                    return ALLOWED;
                case AppOpsManager.MODE_IGNORED:
                    return IGNORED;
                case AppOpsManager.MODE_ERRORED:
                    return ERRORED;
                case AppOpsManager.MODE_DEFAULT:
                    return DEFAULT;
                case AppOpsManager.MODE_FOREGROUND:
                    return FOREGROUND;
                default:
                    throw new IllegalStateException("Unknown AppOpsMode");
            }
        }
    }

    private static final AppOpsManager sAppOpsManager =
            TestApis.context().instrumentedContext().getSystemService(AppOpsManager.class);
    private final Package mPackage;

    public AppOps(Package pkg) {
        this.mPackage = pkg;
    }

    /** Set an AppOp for the given package. */
    public void set(String appOpName, AppOpsMode mode) {
        try (PermissionContext p = TestApis.permissions().withPermission(MANAGE_APP_OPS_MODES)) {
            sAppOpsManager.setMode(appOpName, mPackage.uid(), mPackage.packageName(), mode.mValue);
        }
    }

    /** Get the AppOp mode for the given package. */
    @TargetApi(Q)
    public AppOpsMode get(String appOpName) {
        Versions.requireMinimumVersion(Q);
        return AppOpsMode.forValue(
                sAppOpsManager.unsafeCheckOpNoThrow(
                        appOpName, mPackage.uid(), mPackage.packageName()));
    }
}