Wallpaper: Allow multiple partners to be loaded

Current implementation only allowed for one partner, and any
additional partner wallpapers would not be loaded.
Partrner.get() still returns the first partner and a new method,
getAllPartners is introduced which returns a list of all partners.

Change-Id: I06b6cd4817d3f812e2110967f075d68ee31cb318

2 files changed, 45 insertions, 6 deletions

diff --git a/src/com/android/launcher3/Partner.java b/src/com/android/launcher3/Partner.java
index e1913193b..de4eecdda 100644
--- a/src/com/android/launcher3/Partner.java
+++ b/src/com/android/launcher3/Partner.java
@@ -23,6 +23,8 @@ import android.util.Log;
 import android.util.Pair;
 
 import java.io.File;
+import java.util.ArrayList;
+import java.util.List;
 
 /**
  * Utilities to discover and interact with partner customizations. There can
@@ -54,20 +56,33 @@ public class Partner {
     public static final String RES_GRID_ICON_SIZE_DP = "grid_icon_size_dp";
 
     private static boolean sSearched = false;
-    private static Partner sPartner;
+    private static List<Partner> sPartners;
+
+    static {
+        sPartners = new ArrayList<Partner>();
+    }
 
     /**
-     * Find and return partner details, or {@code null} if none exists.
+     * Find and return first partner details, or {@code null} if none exists.
      */
     public static synchronized Partner get(PackageManager pm) {
+        getAllPartners(pm);
+        return sPartners.size() > 0 ? sPartners.get(0) : null;
+    }
+
+    /**
+     * Find and return all partner details, or {@code null} if none exists.
+     */
+    public static synchronized List<Partner> getAllPartners(PackageManager pm) {
         if (!sSearched) {
-            Pair<String, Resources> apkInfo = Utilities.findSystemApk(ACTION_PARTNER_CUSTOMIZATION, pm);
-            if (apkInfo != null) {
-                sPartner = new Partner(apkInfo.first, apkInfo.second);
+            List<Pair<String, Resources>> apkInfos =
+                    Utilities.findSystemApks(ACTION_PARTNER_CUSTOMIZATION, pm);
+            for (Pair<String, Resources> apkInfo : apkInfos) {
+                sPartners.add(new Partner(apkInfo.first, apkInfo.second));
             }
             sSearched = true;
         }
-        return sPartner;
+        return sPartners;
     }
 
     private final String mPackageName;
diff --git a/src/com/android/launcher3/Utilities.java b/src/com/android/launcher3/Utilities.java
index a594fd9f9..1e8bc2f73 100644
--- a/src/com/android/launcher3/Utilities.java
+++ b/src/com/android/launcher3/Utilities.java
@@ -50,6 +50,7 @@ import android.view.View;
 import android.widget.Toast;
 
 import java.util.ArrayList;
+import java.util.List;
 
 /**
  * Various utilities shared amongst the Launcher's classes.
@@ -561,6 +562,29 @@ public final class Utilities {
         return null;
     }
 
+    /*
+     * Finds all system apks which had a broadcast receiver listening to a particular action.
+     * @param action intent action used to find the apk
+     * @return a list of pairs of apk package name and the resources.
+     */
+    static List<Pair<String, Resources>> findSystemApks(String action, PackageManager pm) {
+        final Intent intent = new Intent(action);
+        List<Pair<String, Resources>> systemApks = new ArrayList<Pair<String, Resources>>();
+        for (ResolveInfo info : pm.queryBroadcastReceivers(intent, 0)) {
+            if (info.activityInfo != null &&
+                    (info.activityInfo.applicationInfo.flags & ApplicationInfo.FLAG_SYSTEM) != 0) {
+                final String packageName = info.activityInfo.packageName;
+                try {
+                    final Resources res = pm.getResourcesForApplication(packageName);
+                    systemApks.add(Pair.create(packageName, res));
+                } catch (NameNotFoundException e) {
+                    Log.w(TAG, "Failed to find resources for " + packageName);
+                }
+            }
+        }
+        return systemApks;
+    }
+
     public static float convertDpToPixel(float dp, Context context){
         Resources resources = context.getResources();
         DisplayMetrics metrics = resources.getDisplayMetrics();