Merge pull request leolin310148#65 from auchri/patch-2

Fix for xiaomi devices with custom launcher
DwyaneQ · Mar 20, 2016 · 51e4aa1 · 51e4aa1
2 parents 82a6c75 + 7025e73
commit 51e4aa1
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/ShortcutBadger/src/main/java/me/leolin/shortcutbadger/ShortcutBadger.java b/ShortcutBadger/src/main/java/me/leolin/shortcutbadger/ShortcutBadger.java
@@ -100,18 +100,18 @@ private static void initBadger(Context context) {
             ResolveInfo resolveInfo = context.getPackageManager().resolveActivity(intent, PackageManager.MATCH_DEFAULT_ONLY);
             String currentHomePackage = resolveInfo.activityInfo.packageName;
 
-            if (Build.MANUFACTURER.equalsIgnoreCase("Xiaomi")) {
-                sShortcutBadger = new XiaomiHomeBadger();
-                return;
-            }
-
             for (Class<? extends Badger> badger : BADGERS) {
                 Badger shortcutBadger = badger.newInstance();
                 if (shortcutBadger.getSupportLaunchers().contains(currentHomePackage)) {
                     sShortcutBadger = shortcutBadger;
                     break;
                 }
             }
+
+            if (sShortcutBadger == null && Build.MANUFACTURER.equalsIgnoreCase("Xiaomi")) {
+                sShortcutBadger = new XiaomiHomeBadger();
+                return;
+            }
         } catch (Exception e) {
             Log.e(LOG_TAG, e.getMessage(), e);
         }