PermissionController/src/com/android/packageinstaller/role/ui/RoleListLiveData.java

/*
 * 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.packageinstaller.role.ui;

import android.app.role.RoleManager;
import android.content.Context;
import android.content.pm.ApplicationInfo;
import android.util.ArrayMap;
import android.util.Log;

import androidx.annotation.NonNull;
import androidx.annotation.WorkerThread;
import androidx.lifecycle.LiveData;

import com.android.packageinstaller.role.model.Role;
import com.android.packageinstaller.role.model.Roles;
import com.android.packageinstaller.role.utils.PackageUtils;

import java.util.ArrayList;
import java.util.List;

/**
 * {@link LiveData} for a list of roles.
 */
public class RoleListLiveData extends AsyncTaskLiveData<List<RoleItem>> {

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

    private final boolean mExclusive;
    private final Context mContext;

    public RoleListLiveData(boolean exclusive, @NonNull Context context) {
        mExclusive = exclusive;
        mContext = context;

        loadValue();
    }

    @Override
    @WorkerThread
    protected List<RoleItem> loadValueInBackground() {
        ArrayMap<String, Role> roles = Roles.getRoles(mContext);

        List<RoleItem> roleItems = new ArrayList<>();
        RoleManager roleManager = mContext.getSystemService(RoleManager.class);
        int rolesSize = roles.size();
        for (int rolesIndex = 0; rolesIndex < rolesSize; rolesIndex++) {
            Role role = roles.valueAt(rolesIndex);

            if (role.isExclusive() != mExclusive) {
                continue;
            }

            List<ApplicationInfo> holderApplicationInfos = new ArrayList<>();
            List<String> holderPackageNames = roleManager.getRoleHolders(role.getName());
            int holderPackageNamesSize = holderPackageNames.size();
            for (int holderPackageNamesIndex = 0; holderPackageNamesIndex < holderPackageNamesSize;
                    holderPackageNamesIndex++) {
                String holderPackageName = holderPackageNames.get(holderPackageNamesIndex);

                ApplicationInfo holderApplicationInfo = PackageUtils.getApplicationInfo(
                        holderPackageName, mContext);
                if (holderApplicationInfo == null) {
                    Log.w(LOG_TAG, "Cannot get ApplicationInfo for application, skipping: "
                            + holderPackageName);
                    continue;
                }
                holderApplicationInfos.add(holderApplicationInfo);
            }

            roleItems.add(new RoleItem(role, holderApplicationInfos));
        }

        return roleItems;
    }
}