1 files changed, 266 insertions, 0 deletions
diff --git a/src/com/android/packageinstaller/permission/model/PermissionApps.java b/src/com/android/packageinstaller/permission/model/PermissionApps.java
new file mode 100644
index 00000000..8f28433b
--- /dev/null
+++ b/src/com/android/packageinstaller/permission/model/PermissionApps.java
@@ -0,0 +1,266 @@
+/*
+ * Copyright (C) 2015 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.permission.model;
+
+import android.content.Context;
+import android.content.pm.ApplicationInfo;
+import android.content.pm.PackageInfo;
+import android.content.pm.PackageItemInfo;
+import android.content.pm.PackageManager;
+import android.content.pm.PackageManager.NameNotFoundException;
+import android.content.pm.PermissionInfo;
+import android.graphics.LightingColorFilter;
+import android.graphics.drawable.Drawable;
+import android.os.AsyncTask;
+import android.os.UserHandle;
+import android.os.UserManager;
+import android.util.ArrayMap;
+import android.util.Log;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+
+public class PermissionApps {
+    private static final String LOG_TAG = "PermissionApps";
+
+    private final Context mContext;
+    private final String mGroupName;
+    private final PackageManager mPm;
+    private final Callback mCallback;
+
+    private CharSequence mLabel;
+    private Drawable mIcon;
+    private List<PermissionApp> mPermApps;
+    // Map (pkg|uid) -> AppPermission
+    private ArrayMap<String, PermissionApp> mAppLookup;
+
+    public PermissionApps(Context context, String groupName, Callback callback) {
+        mContext = context;
+        mPm = mContext.getPackageManager();
+        mGroupName = groupName;
+        mCallback = callback;
+        loadGroupInfo();
+        new PermissionAppsLoader().execute();
+    }
+
+    public void refresh() {
+        new PermissionAppsLoader().execute();
+    }
+
+    public Collection<PermissionApp> getApps() {
+        return mPermApps;
+    }
+
+    public PermissionApp getApp(String key) {
+        return mAppLookup.get(key);
+    }
+
+    public CharSequence getLabel() {
+        return mLabel;
+    }
+
+    public Drawable getIcon() {
+        return mIcon;
+    }
+
+    private void loadGroupInfo() {
+        PackageItemInfo info;
+        try {
+            info = mPm.getPermissionGroupInfo(mGroupName, 0);
+        } catch (PackageManager.NameNotFoundException e) {
+            try {
+                PermissionInfo permInfo = mPm.getPermissionInfo(mGroupName, 0);
+                if (permInfo.protectionLevel != PermissionInfo.PROTECTION_DANGEROUS) {
+                    Log.w(LOG_TAG, mGroupName + " is not a runtime permission");
+                    return;
+                }
+                info = permInfo;
+            } catch (NameNotFoundException reallyNotFound) {
+                Log.w(LOG_TAG, "Can't find permission: " + mGroupName, reallyNotFound);
+                return;
+            }
+        }
+        mLabel = info.loadLabel(mPm);
+        mIcon = info.loadUnbadgedIcon(mPm);
+        LightingColorFilter filter = new LightingColorFilter(0, 0xffffffff);
+        mIcon.setColorFilter(filter);
+    }
+
+    public static class PermissionApp implements Comparable<PermissionApp> {
+        private final PermissionGroup mPermissionGroup;
+        private final String mLabel;
+        private final Drawable mIcon;
+
+        public PermissionApp(PermissionGroup permissionGroup, String label,
+                Drawable icon) {
+            mPermissionGroup = permissionGroup;
+            mLabel = label;
+            mIcon = icon;
+        }
+
+        public String getKey() {
+            return Integer.toString(getUid());
+        }
+
+        public String getLabel() {
+            return mLabel;
+        }
+
+        public Drawable getIcon() {
+            return mIcon;
+        }
+
+        public boolean areRuntimePermissionsGranted() {
+            return mPermissionGroup.areRuntimePermissionsGranted();
+        }
+
+        public void grantRuntimePermissions() {
+            mPermissionGroup.grantRuntimePermissions();
+        }
+
+        public void revokeRuntimePermissions() {
+            mPermissionGroup.revokeRuntimePermissions();
+        }
+
+        @Override
+        public int compareTo(PermissionApp another) {
+            final int result = mLabel.compareTo(another.mLabel);
+            if (result == 0) {
+                // Unbadged before badged.
+                return getUid() - another.getUid();
+            }
+            return result;
+        }
+
+        private int getUid() {
+            return mPermissionGroup.getApp().applicationInfo.uid;
+        }
+    }
+
+    private class PermissionAppsLoader extends AsyncTask<Void, Void, List<PermissionApp>> {
+        @Override
+        protected List<PermissionApp> doInBackground(Void... args) {
+            PackageItemInfo groupInfo = getGroupInfo(mGroupName);
+            if (groupInfo == null) {
+                return Collections.emptyList();
+            }
+
+            List<PermissionInfo> groupPermInfos = getGroupPermissionInfos(mGroupName);
+            if (groupPermInfos == null) {
+                return Collections.emptyList();
+            }
+
+            ArrayList<PermissionApp> permApps = new ArrayList<>();
+
+            for (UserHandle user : UserManager.get(mContext).getUserProfiles()) {
+                List<PackageInfo> apps = mPm.getInstalledPackages(
+                        PackageManager.GET_PERMISSIONS, user.getIdentifier());
+
+                final int N = apps.size();
+                for (int i = 0; i < N; i++) {
+                    PackageInfo app = apps.get(i);
+                    if (app.requestedPermissions == null) {
+                        continue;
+                    }
+
+                    for (int j = 0; j < app.requestedPermissions.length; j++) {
+                        String requestedPerm = app.requestedPermissions[j];
+
+                        boolean requestsPermissionInGroup = false;
+
+                        for (PermissionInfo groupPermInfo : groupPermInfos) {
+                            if (groupPermInfo.name.equals(requestedPerm)) {
+                                requestsPermissionInGroup = true;
+                                break;
+                            }
+                        }
+
+                        if (!requestsPermissionInGroup) {
+                            continue;
+                        }
+
+                        PermissionGroup group = PermissionGroup.create(mContext,
+                                app, groupInfo, groupPermInfos);
+
+                        PermissionApp permApp = new PermissionApp(group,
+                                app.applicationInfo.loadLabel(mPm).toString(),
+                                getBadgedIcon(app.applicationInfo));
+
+                        permApps.add(permApp);
+                    }
+                }
+            }
+
+            Collections.sort(permApps);
+
+            return permApps;
+        }
+
+        private PackageItemInfo getGroupInfo(String groupName) {
+            try {
+                return mContext.getPackageManager().getPermissionGroupInfo(groupName, 0);
+            } catch (NameNotFoundException e) {
+                /* ignore */
+            }
+            try {
+                return mContext.getPackageManager().getPermissionInfo(groupName, 0);
+            } catch (NameNotFoundException e2) {
+                /* ignore */
+            }
+            return null;
+        }
+
+        private List<PermissionInfo> getGroupPermissionInfos(String groupName) {
+            try {
+                return mContext.getPackageManager().queryPermissionsByGroup(groupName, 0);
+            } catch (NameNotFoundException e) {
+                /* ignore */
+            }
+            try {
+                PermissionInfo permissionInfo = mContext.getPackageManager()
+                        .getPermissionInfo(groupName, 0);
+                List<PermissionInfo> permissions = new ArrayList<>();
+                permissions.add(permissionInfo);
+                return permissions;
+            } catch (NameNotFoundException e2) {
+                /* ignore */
+            }
+            return null;
+        }
+
+        private Drawable getBadgedIcon(ApplicationInfo appInfo) {
+            Drawable unbadged = appInfo.loadUnbadgedIcon(mPm);
+            return mPm.getUserBadgedIcon(unbadged,
+                    new UserHandle(UserHandle.getUserId(appInfo.uid)));
+        }
+
+        @Override
+        protected void onPostExecute(List<PermissionApp> result) {
+            mAppLookup = new ArrayMap<>();
+            for (PermissionApp app : result) {
+                mAppLookup.put(app.getKey(), app);
+            }
+            mPermApps = result;
+            mCallback.onPermissionsLoaded();
+        }
+    }
+
+    public interface Callback {
+        void onPermissionsLoaded();
+    }
+}