Themes: Let ThemeService handle all theme processing [3/3]

Change-Id: Ied877eb5f1e96774d2e415970034d0b892765134

1 files changed, 48 insertions, 12 deletions

diff --git a/src/org/cyanogenmod/themes/provider/AppReceiver.java b/src/org/cyanogenmod/themes/provider/AppReceiver.java
index cf0d1c3..c856abd 100644
--- a/src/org/cyanogenmod/themes/provider/AppReceiver.java
+++ b/src/org/cyanogenmod/themes/provider/AppReceiver.java
@@ -18,32 +18,63 @@ package org.cyanogenmod.themes.provider;
 import android.content.BroadcastReceiver;
 import android.content.Context;
 import android.content.Intent;
+import android.content.pm.PackageManager;
 import android.content.pm.PackageManager.NameNotFoundException;
+import android.content.res.ThemeManager;
 import android.database.Cursor;
 import android.net.Uri;
 import android.provider.ThemesContract;
 import android.util.Log;
 
+import java.util.Set;
+
 public class AppReceiver extends BroadcastReceiver {
     public final static String TAG = AppReceiver.class.getName();
 
     @Override
     public void onReceive(Context context, Intent intent) {
-        Uri uri = intent.getData();
-        String pkgName = uri != null ? uri.getSchemeSpecificPart() : null;
-        boolean isReplacing = intent.getExtras().getBoolean(Intent.EXTRA_REPLACING, false);
-
+        final Uri uri = intent.getData();
+        final String pkgName = uri != null ? uri.getSchemeSpecificPart() : null;
+        final boolean isReplacing = intent.getExtras().getBoolean(Intent.EXTRA_REPLACING, false);
+        final String action = intent.getAction();
         try {
-            if (intent.getAction().equals(Intent.ACTION_PACKAGE_ADDED) && !isReplacing) {
-                ThemePackageHelper.insertPackage(context, pkgName);
-            } else if (intent.getAction().equals(Intent.ACTION_PACKAGE_FULLY_REMOVED)) {
+            if (Intent.ACTION_PACKAGE_ADDED.equals(action) && !isReplacing) {
+                if (!isThemeBeingProcessed(context, pkgName)) {
+                    ThemePackageHelper.insertPackage(context, pkgName);
+                } else {
+                    // store this package name so we know it's being processed and it can be
+                    // added to the DB when ACTION_THEME_RESOURCES_CACHED is received
+                    PreferenceUtils.addThemeBeingProcessed(context, pkgName);
+                }
+            } else if (Intent.ACTION_PACKAGE_FULLY_REMOVED.equals(action)) {
                 ThemePackageHelper.removePackage(context, pkgName);
-            } else if (intent.getAction().equals(Intent.ACTION_PACKAGE_REPLACED)) {
-                if (themeExistsInProvider(context, pkgName)) {
-                    ThemePackageHelper.updatePackage(context, pkgName);
+            } else if (Intent.ACTION_PACKAGE_REPLACED.equals(action)) {
+                if (!isThemeBeingProcessed(context, pkgName)) {
+                    if (themeExistsInProvider(context, pkgName)) {
+                        ThemePackageHelper.updatePackage(context, pkgName);
+                    } else {
+                        // Edge case where app was not a theme in previous install
+                        ThemePackageHelper.insertPackage(context, pkgName);
+                    }
                 } else {
-                    // Edge case where app was not a theme in previous install
-                    ThemePackageHelper.insertPackage(context, pkgName);
+                    // store this package name so we know it's being processed and it can be
+                    // added to the DB when ACTION_THEME_RESOURCES_CACHED is received
+                    PreferenceUtils.addThemeBeingProcessed(context, pkgName);
+                }
+            } else if (Intent.ACTION_THEME_RESOURCES_CACHED.equals(action)) {
+                final String themePkgName = intent.getStringExtra(Intent.EXTRA_THEME_PACKAGE_NAME);
+                final int result = intent.getIntExtra(Intent.EXTRA_THEME_RESULT,
+                        PackageManager.INSTALL_FAILED_THEME_UNKNOWN_ERROR);
+                Set<String> processingThemes =
+                        PreferenceUtils.getInstalledThemesBeingProcessed(context);
+                if (processingThemes != null &&
+                        processingThemes.contains(themePkgName) && result >= 0) {
+                    if (themeExistsInProvider(context, themePkgName)) {
+                        ThemePackageHelper.updatePackage(context, themePkgName);
+                    } else {
+                        // Edge case where app was not a theme in previous install
+                        ThemePackageHelper.insertPackage(context, themePkgName);
+                    }
                 }
             }
         } catch(NameNotFoundException e) {
@@ -65,4 +96,9 @@ public class AppReceiver extends BroadcastReceiver {
         }
         return exists;
     }
+
+    private boolean isThemeBeingProcessed(Context context, String pkgName) {
+        ThemeManager tm = (ThemeManager) context.getSystemService(Context.THEME_SERVICE);
+        return tm.isThemeBeingProcessed(pkgName);
+    }
 }