PermissionController/src/com/android/permissioncontroller/role/utils/RoleUiBehaviorUtils.java

/*
 * Copyright (C) 2022 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.role.utils;

import android.content.Context;
import android.content.Intent;
import android.content.pm.ApplicationInfo;
import android.os.UserHandle;
import android.os.UserManager;
import android.util.Log;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;

import com.android.modules.utils.build.SdkLevel;
import com.android.permissioncontroller.role.ui.RoleApplicationPreference;
import com.android.permissioncontroller.role.ui.RolePreference;
import com.android.permissioncontroller.role.ui.UserRestrictionAwarePreference;
import com.android.permissioncontroller.role.ui.behavior.RoleUiBehavior;
import com.android.role.controller.model.Role;

/**
 * Utility methods for Role UI behavior
 */
public final class RoleUiBehaviorUtils {

    private static final String LOG_TAG = RoleUiBehaviorUtils.class.getSimpleName();

    /**
     * Get the role ui behavior if available
     */
    @Nullable
    private static RoleUiBehavior getUiBehavior(@NonNull Role role) {
        String uiBehaviorName = role.getUiBehaviorName();
        if (uiBehaviorName == null) {
            return null;
        }
        RoleUiBehavior uiBehavior;
        String uiBehaviorClassName = RoleUiBehavior.class.getPackage().getName() + '.'
                + uiBehaviorName;
        try {
            uiBehavior = (RoleUiBehavior) Class.forName(uiBehaviorClassName).newInstance();
        } catch (ClassNotFoundException | IllegalAccessException | InstantiationException e) {
            Log.e(LOG_TAG, "Unable to instantiate UI behavior: " + uiBehaviorClassName, e);
            return null;
        }
        return uiBehavior;
    }

    /**
     * @see RoleUiBehavior#getManageIntentAsUser
     */
    @Nullable
    public static Intent getManageIntentAsUser(@NonNull Role role, @NonNull UserHandle user,
            @NonNull Context context) {
        RoleUiBehavior uiBehavior = getUiBehavior(role);
        if (uiBehavior == null) {
            return null;
        }
        return uiBehavior.getManageIntentAsUser(role, user, context);
    }

    /**
     * @see RoleUiBehavior#preparePreferenceAsUser
     */
    public static void preparePreferenceAsUser(@NonNull Role role,
            @NonNull RolePreference preference, @NonNull UserHandle user,
            @NonNull Context context) {
        prepareUserRestrictionAwarePreferenceAsUser(role, preference, user, context);

        RoleUiBehavior uiBehavior = getUiBehavior(role);
        if (uiBehavior == null) {
            return;
        }
        uiBehavior.preparePreferenceAsUser(role, preference, user, context);
    }

    /**
     * @see RoleUiBehavior#prepareApplicationPreferenceAsUser
     */
    public static void prepareApplicationPreferenceAsUser(@NonNull Role role,
            @NonNull RoleApplicationPreference preference,
            @NonNull ApplicationInfo applicationInfo, @NonNull UserHandle user,
            @NonNull Context context) {
        prepareUserRestrictionAwarePreferenceAsUser(role, preference, user, context);

        RoleUiBehavior uiBehavior = getUiBehavior(role);
        if (uiBehavior == null) {
            return;
        }
        uiBehavior.prepareApplicationPreferenceAsUser(
                role, preference.asTwoStatePreference(), applicationInfo, user,
                context);
    }

    private static void prepareUserRestrictionAwarePreferenceAsUser(@NonNull Role role,
            @NonNull UserRestrictionAwarePreference preference, @NonNull UserHandle user,
            @NonNull Context context) {
        if (SdkLevel.isAtLeastU() && role.isExclusive()) {
            UserManager userManager = context.getSystemService(UserManager.class);
            boolean hasDisallowConfigDefaultApps = userManager.hasUserRestrictionForUser(
                    UserManager.DISALLOW_CONFIG_DEFAULT_APPS, user);
            preference.setUserRestriction(hasDisallowConfigDefaultApps
                    ? UserManager.DISALLOW_CONFIG_DEFAULT_APPS : null);
        }
    }

    /**
     * @see RoleUiBehavior#getConfirmationMessage
     */
    @Nullable
    public static CharSequence getConfirmationMessage(@NonNull Role role,
            @NonNull String packageName, @NonNull Context context) {
        RoleUiBehavior uiBehavior = getUiBehavior(role);
        if (uiBehavior == null) {
            return null;
        }
        return uiBehavior.getConfirmationMessage(role, packageName, context);
    }
}