Merge pull request #81 from jiehuadai/sonyBadge

Support badge to latest Sony launcher

ShortcutBadger/src/main/AndroidManifest.xml

@@ -20,6 +20,7 @@
 
     <!--for sony-->
     <uses-permission android:name="com.sonyericsson.home.permission.BROADCAST_BADGE"/>
+    <uses-permission android:name="com.sonymobile.home.permission.PROVIDER_INSERT_BADGE"/>
 
     <!--for apex-->
     <uses-permission android:name="com.anddoes.launcher.permission.UPDATE_COUNT"/>

ShortcutBadger/src/main/java/me/leolin/shortcutbadger/impl/SonyHomeBadger.java

@@ -1,8 +1,12 @@
 package me.leolin.shortcutbadger.impl;
 
+import android.content.AsyncQueryHandler;
 import android.content.ComponentName;
+import android.content.ContentValues;
 import android.content.Context;
 import android.content.Intent;
+import android.content.pm.ProviderInfo;
+import android.net.Uri;
 
 import me.leolin.shortcutbadger.Badger;
 import me.leolin.shortcutbadger.ShortcutBadgeException;
@@ -11,6 +15,7 @@ import me.leolin.shortcutbadger.ShortcutBadger;
 import java.util.Arrays;
 import java.util.List;
 
+
 /**
  * @author Leo Lin
  */
@@ -22,8 +27,33 @@ public class SonyHomeBadger implements Badger {
     private static final String INTENT_EXTRA_MESSAGE = "com.sonyericsson.home.intent.extra.badge.MESSAGE";
     private static final String INTENT_EXTRA_SHOW_MESSAGE = "com.sonyericsson.home.intent.extra.badge.SHOW_MESSAGE";
 
+    private static final String PROVIDER_CONTENT_URI = "content://com.sonymobile.home.resourceprovider/badge";
+    private static final String PROVIDER_COLUMNS_BADGE_COUNT = "badge_count";
+    private static final String PROVIDER_COLUMNS_PACKAGE_NAME = "package_name";
+    private static final String PROVIDER_COLUMNS_ACTIVITY_NAME = "activity_name";
+    private static final String SONY_HOME_PROVIDER_NAME = "com.sonymobile.home.resourceprovider";
+    private final Uri BADGE_CONTENT_URI = Uri.parse(PROVIDER_CONTENT_URI);
+
+    private AsyncQueryHandler mQueryHandler;
+
     @Override
-    public void executeBadge(Context context, ComponentName componentName, int badgeCount) throws ShortcutBadgeException {
+    public void executeBadge(Context context, ComponentName componentName,
+            int badgeCount) throws ShortcutBadgeException {
+        if (sonyBadgeContentProviderExists(context)) {
+            executeBadgeByContentProvider(context, componentName, badgeCount);
+        } else {
+            executeBadgeByBroadcast(context, componentName, badgeCount);
+        }
+
+    }
+
+    @Override
+    public List<String> getSupportLaunchers() {
+        return Arrays.asList("com.sonyericsson.home");
+    }
+
+    private static void executeBadgeByBroadcast(Context context, ComponentName componentName,
+            int badgeCount) {
         Intent intent = new Intent(INTENT_ACTION);
         intent.putExtra(INTENT_EXTRA_PACKAGE_NAME, componentName.getPackageName());
         intent.putExtra(INTENT_EXTRA_ACTIVITY_NAME, componentName.getClassName());
@@ -32,8 +62,67 @@ public class SonyHomeBadger implements Badger {
         context.sendBroadcast(intent);
     }
 
-    @Override
-    public List<String> getSupportLaunchers() {
-        return Arrays.asList("com.sonyericsson.home");
+    /**
+     * Send request to Sony badge content provider to set badge in Sony home launcher.
+     *
+     * @param context       the context to use
+     * @param componentName the componentName to use
+     * @param badgeCount    the badge count
+     */
+    private void executeBadgeByContentProvider(Context context, ComponentName componentName,
+            int badgeCount) {
+        if (badgeCount < 0) {
+            return;
+        }
+
+        if (mQueryHandler == null) {
+            mQueryHandler = new AsyncQueryHandler(
+                    context.getApplicationContext().getContentResolver()) {
+            };
+        }
+        insertBadgeAsync(badgeCount, componentName.getPackageName(), componentName.getClassName());
+    }
+
+    /**
+     * Insert a badge associated with the specified package and activity names
+     * asynchronously. The package and activity names must correspond to an
+     * activity that holds an intent filter with action
+     * "android.intent.action.MAIN" and category
+     * "android.intent.category.LAUNCHER" in the manifest. Also, it is not
+     * allowed to publish badges on behalf of another client, so the package and
+     * activity names must belong to the process from which the insert is made.
+     * To be able to insert badges, the app must have the PROVIDER_INSERT_BADGE
+     * permission in the manifest file. In case these conditions are not
+     * fulfilled, or any content values are missing, there will be an unhandled
+     * exception on the background thread.
+     *
+     * @param badgeCount   the badge count
+     * @param packageName  the package name
+     * @param activityName the activity name
+     */
+    private void insertBadgeAsync(int badgeCount, String packageName, String activityName) {
+        final ContentValues contentValues = new ContentValues();
+        contentValues.put(PROVIDER_COLUMNS_BADGE_COUNT, badgeCount);
+        contentValues.put(PROVIDER_COLUMNS_PACKAGE_NAME, packageName);
+        contentValues.put(PROVIDER_COLUMNS_ACTIVITY_NAME, activityName);
+
+        // The badge must be inserted on a background thread
+        mQueryHandler.startInsert(0, null, BADGE_CONTENT_URI, contentValues);
+    }
+
+    /**
+     * Check if the latest Sony badge content provider exists .
+     *
+     * @param context the context to use
+     * @return true if Sony badge content provider exists, otherwise false.
+     */
+    private static boolean sonyBadgeContentProviderExists(Context context) {
+        boolean exists = false;
+        ProviderInfo info = context.getPackageManager()
+                .resolveContentProvider(SONY_HOME_PROVIDER_NAME, 0);
+        if (info != null) {
+            exists = true;
+        }
+        return exists;
     }
 }