From 86222d23e28a09a7a26e7a0d4b4830ded2efb384 Mon Sep 17 00:00:00 2001 From: Tony Wickham Date: Wed, 29 Mar 2017 15:30:43 -0700 Subject: Add support for getting widgets/shortucts for a particular package/user Bug: 34940468 Bug: 33553066 Change-Id: I5d0131df206c6a13d4227ad28c5b094bbf1343df --- .../launcher3/compat/AppWidgetManagerCompat.java | 14 ++++-- .../launcher3/compat/AppWidgetManagerCompatVL.java | 26 ++++++++--- .../launcher3/compat/AppWidgetManagerCompatVO.java | 53 ++++++++++++++++++++++ .../launcher3/compat/LauncherAppsCompat.java | 4 +- .../launcher3/compat/LauncherAppsCompatVL.java | 15 ++++-- .../launcher3/compat/LauncherAppsCompatVO.java | 22 +++++++-- 6 files changed, 116 insertions(+), 18 deletions(-) create mode 100644 src/com/android/launcher3/compat/AppWidgetManagerCompatVO.java (limited to 'src/com/android/launcher3/compat') diff --git a/src/com/android/launcher3/compat/AppWidgetManagerCompat.java b/src/com/android/launcher3/compat/AppWidgetManagerCompat.java index 7911bb213..3efbbfba5 100644 --- a/src/com/android/launcher3/compat/AppWidgetManagerCompat.java +++ b/src/com/android/launcher3/compat/AppWidgetManagerCompat.java @@ -22,12 +22,15 @@ import android.appwidget.AppWidgetManager; import android.appwidget.AppWidgetProviderInfo; import android.content.ComponentName; import android.content.Context; -import android.graphics.Bitmap; import android.os.Bundle; import android.os.UserHandle; +import android.support.annotation.NonNull; +import android.support.annotation.Nullable; import com.android.launcher3.LauncherAppWidgetProviderInfo; +import com.android.launcher3.Utilities; import com.android.launcher3.util.ComponentKey; +import com.android.launcher3.util.PackageUserKey; import java.util.HashMap; import java.util.List; @@ -40,7 +43,11 @@ public abstract class AppWidgetManagerCompat { public static AppWidgetManagerCompat getInstance(Context context) { synchronized (sInstanceLock) { if (sInstance == null) { - sInstance = new AppWidgetManagerCompatVL(context.getApplicationContext()); + if (Utilities.isAtLeastO()) { + sInstance = new AppWidgetManagerCompatVO(context.getApplicationContext()); + } else { + sInstance = new AppWidgetManagerCompatVL(context.getApplicationContext()); + } } return sInstance; } @@ -63,7 +70,8 @@ public abstract class AppWidgetManagerCompat { return info == null ? null : LauncherAppWidgetProviderInfo.fromProviderInfo(mContext, info); } - public abstract List getAllProviders(); + public abstract List getAllProviders( + @Nullable PackageUserKey packageUser); public abstract boolean bindAppWidgetIdIfAllowed( int appWidgetId, AppWidgetProviderInfo info, Bundle options); diff --git a/src/com/android/launcher3/compat/AppWidgetManagerCompatVL.java b/src/com/android/launcher3/compat/AppWidgetManagerCompatVL.java index 61dd1a5b7..f239f5c31 100644 --- a/src/com/android/launcher3/compat/AppWidgetManagerCompatVL.java +++ b/src/com/android/launcher3/compat/AppWidgetManagerCompatVL.java @@ -22,36 +22,48 @@ import android.appwidget.AppWidgetProviderInfo; import android.content.ActivityNotFoundException; import android.content.ComponentName; import android.content.Context; -import android.content.pm.PackageManager; import android.os.Bundle; import android.os.UserHandle; import android.os.UserManager; +import android.support.annotation.Nullable; import android.widget.Toast; import com.android.launcher3.LauncherAppWidgetProviderInfo; import com.android.launcher3.R; import com.android.launcher3.util.ComponentKey; +import com.android.launcher3.util.PackageUserKey; import java.util.ArrayList; import java.util.HashMap; +import java.util.Iterator; import java.util.List; class AppWidgetManagerCompatVL extends AppWidgetManagerCompat { private final UserManager mUserManager; - private final PackageManager mPm; AppWidgetManagerCompatVL(Context context) { super(context); - mPm = context.getPackageManager(); mUserManager = (UserManager) context.getSystemService(Context.USER_SERVICE); } @Override - public List getAllProviders() { - ArrayList providers = new ArrayList(); - for (UserHandle user : mUserManager.getUserProfiles()) { - providers.addAll(mAppWidgetManager.getInstalledProvidersForProfile(user)); + public List getAllProviders(@Nullable PackageUserKey packageUser) { + if (packageUser == null) { + ArrayList providers = new ArrayList(); + for (UserHandle user : mUserManager.getUserProfiles()) { + providers.addAll(mAppWidgetManager.getInstalledProvidersForProfile(user)); + } + return providers; + } + // Only get providers for the given package/user. + List providers = new ArrayList<>(mAppWidgetManager + .getInstalledProvidersForProfile(packageUser.mUser)); + Iterator iterator = providers.iterator(); + while (iterator.hasNext()) { + if (!iterator.next().provider.getPackageName().equals(packageUser.mPackageName)) { + iterator.remove(); + } } return providers; } diff --git a/src/com/android/launcher3/compat/AppWidgetManagerCompatVO.java b/src/com/android/launcher3/compat/AppWidgetManagerCompatVO.java new file mode 100644 index 000000000..bde8b7801 --- /dev/null +++ b/src/com/android/launcher3/compat/AppWidgetManagerCompatVO.java @@ -0,0 +1,53 @@ +/* + * Copyright (C) 2017 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.launcher3.compat; + +import android.appwidget.AppWidgetManager; +import android.appwidget.AppWidgetProviderInfo; +import android.content.Context; +import android.os.UserHandle; +import android.support.annotation.NonNull; +import android.support.annotation.Nullable; +import android.util.Log; + +import com.android.launcher3.util.PackageUserKey; + +import java.lang.reflect.InvocationTargetException; +import java.util.List; + +class AppWidgetManagerCompatVO extends AppWidgetManagerCompatVL { + + AppWidgetManagerCompatVO(Context context) { + super(context); + } + + @Override + public List getAllProviders(@Nullable PackageUserKey packageUser) { + if (packageUser == null) { + return super.getAllProviders(null); + } + // TODO: don't use reflection once API and sdk are ready. + try { + return (List) AppWidgetManager.class.getMethod( + "getInstalledProvidersForPackage", String.class, UserHandle.class) + .invoke(mAppWidgetManager, packageUser.mPackageName, packageUser.mUser); + } catch (IllegalAccessException | InvocationTargetException | NoSuchMethodException e) { + Log.e("AppWidgetManagerCompat", "Failed to call new API", e); + } + return super.getAllProviders(packageUser); + } +} diff --git a/src/com/android/launcher3/compat/LauncherAppsCompat.java b/src/com/android/launcher3/compat/LauncherAppsCompat.java index 4580b8108..f17d8deef 100644 --- a/src/com/android/launcher3/compat/LauncherAppsCompat.java +++ b/src/com/android/launcher3/compat/LauncherAppsCompat.java @@ -33,6 +33,7 @@ import com.android.launcher3.Utilities; import com.android.launcher3.graphics.LauncherIcons; import com.android.launcher3.shortcuts.ShortcutInfoCompat; import com.android.launcher3.util.LooperExecuter; +import com.android.launcher3.util.PackageUserKey; import java.util.List; @@ -83,7 +84,8 @@ public abstract class LauncherAppsCompat { public abstract boolean isPackageEnabledForProfile(String packageName, UserHandle user); public abstract boolean isActivityEnabledForProfile(ComponentName component, UserHandle user); - public abstract List getCustomShortcutActivityList(); + public abstract List getCustomShortcutActivityList( + @Nullable PackageUserKey packageUser); /** * request.accept() will initiate the following flow: diff --git a/src/com/android/launcher3/compat/LauncherAppsCompatVL.java b/src/com/android/launcher3/compat/LauncherAppsCompatVL.java index 459017392..58683db75 100644 --- a/src/com/android/launcher3/compat/LauncherAppsCompatVL.java +++ b/src/com/android/launcher3/compat/LauncherAppsCompatVL.java @@ -29,9 +29,11 @@ import android.graphics.Rect; import android.os.Bundle; import android.os.Process; import android.os.UserHandle; +import android.support.annotation.Nullable; import com.android.launcher3.compat.ShortcutConfigActivityInfo.ShortcutConfigActivityInfoVL; import com.android.launcher3.shortcuts.ShortcutInfoCompat; +import com.android.launcher3.util.PackageUserKey; import java.util.ArrayList; import java.util.HashMap; @@ -175,12 +177,19 @@ public class LauncherAppsCompatVL extends LauncherAppsCompat { } @Override - public List getCustomShortcutActivityList() { - PackageManager pm = mContext.getPackageManager(); + public List getCustomShortcutActivityList( + @Nullable PackageUserKey packageUser) { List result = new ArrayList<>(); + if (packageUser != null && !packageUser.mUser.equals(Process.myUserHandle())) { + return result; + } + PackageManager pm = mContext.getPackageManager(); for (ResolveInfo info : pm.queryIntentActivities(new Intent(Intent.ACTION_CREATE_SHORTCUT), 0)) { - result.add(new ShortcutConfigActivityInfoVL(info.activityInfo, pm)); + if (packageUser == null || packageUser.mPackageName + .equals(info.activityInfo.packageName)) { + result.add(new ShortcutConfigActivityInfoVL(info.activityInfo, pm)); + } } return result; } diff --git a/src/com/android/launcher3/compat/LauncherAppsCompatVO.java b/src/com/android/launcher3/compat/LauncherAppsCompatVO.java index d7e35a2e5..95f2e74f4 100644 --- a/src/com/android/launcher3/compat/LauncherAppsCompatVO.java +++ b/src/com/android/launcher3/compat/LauncherAppsCompatVO.java @@ -23,12 +23,15 @@ import android.content.pm.LauncherApps; import android.os.Build; import android.os.Process; import android.os.UserHandle; +import android.support.annotation.Nullable; import android.util.Log; +import com.android.launcher3.compat.ShortcutConfigActivityInfo.ShortcutConfigActivityInfoVO; +import com.android.launcher3.util.PackageUserKey; + import java.lang.reflect.Method; import java.util.ArrayList; import java.util.List; -import com.android.launcher3.compat.ShortcutConfigActivityInfo.*; public class LauncherAppsCompatVO extends LauncherAppsCompatVL { @@ -44,17 +47,28 @@ public class LauncherAppsCompatVO extends LauncherAppsCompatVL { } @Override - public List getCustomShortcutActivityList() { + public List getCustomShortcutActivityList( + @Nullable PackageUserKey packageUser) { List result = new ArrayList<>(); UserHandle myUser = Process.myUserHandle(); try { Method m = LauncherApps.class.getDeclaredMethod("getShortcutConfigActivityList", String.class, UserHandle.class); - for (UserHandle user : UserManagerCompat.getInstance(mContext).getUserProfiles()) { + final List users; + final String packageName; + if (packageUser == null) { + users = UserManagerCompat.getInstance(mContext).getUserProfiles(); + packageName = null; + } else { + users = new ArrayList<>(1); + users.add(packageUser.mUser); + packageName = packageUser.mPackageName; + } + for (UserHandle user : users) { boolean ignoreTargetSdk = myUser.equals(user); List activities = - (List) m.invoke(mLauncherApps, null, user); + (List) m.invoke(mLauncherApps, packageName, user); for (LauncherActivityInfo activityInfo : activities) { if (ignoreTargetSdk || activityInfo.getApplicationInfo().targetSdkVersion >= Build.VERSION_CODES.O) { -- cgit v1.2.3