Adding theme engine interface

Change-Id: I64d455835389acb2e591ff0b623797e1e122f06d
Signed-off-by: rmp22 <195054967+rmp22@users.noreply.github.com>
Signed-off-by: Pranav Vashi <neobuddy89@gmail.com>

Author: rmp22

cmds/idmap2/libidmap2/ResourceMapping.cpp

@@ -53,9 +53,9 @@ Result<Unit> CheckOverlayable(const TargetResourceContainer& target,
                               const PolicyBitmask& fulfilled_policies,
                               const ResourceId& target_resource) {
   constexpr const PolicyBitmask kDefaultPolicies =
-      PolicyFlags::ODM_PARTITION | PolicyFlags::OEM_PARTITION | PolicyFlags::SYSTEM_PARTITION |
-      PolicyFlags::VENDOR_PARTITION | PolicyFlags::PRODUCT_PARTITION | PolicyFlags::SIGNATURE |
-      PolicyFlags::CONFIG_SIGNATURE;
+      PolicyFlags::PUBLIC | PolicyFlags::ODM_PARTITION | PolicyFlags::OEM_PARTITION |
+      PolicyFlags::SYSTEM_PARTITION | PolicyFlags::VENDOR_PARTITION |
+      PolicyFlags::PRODUCT_PARTITION | PolicyFlags::SIGNATURE | PolicyFlags::CONFIG_SIGNATURE;
 
   // If the resource does not have an overlayable definition, allow the resource to be overlaid if
   // the overlay is preinstalled, signed with the same signature as the target or signed with the
@@ -131,7 +131,6 @@ Result<ResourceMapping> ResourceMapping::FromContainers(const TargetResourceCont
     }
 
     if (enforce_overlayable) {
-      // Filter out resources the overlay is not allowed to override.
       auto overlayable = CheckOverlayable(target, overlay_info, fulfilled_policies, *target_resid);
       if (!overlayable) {
         log_info.Warning(LogMessage() << "overlay '" << overlay.GetPath()

core/java/android/content/Context.java

@@ -6617,6 +6617,11 @@ public final <T> T getSystemService(@NonNull Class<T> serviceClass) {
      */
     public static final String RESOURCES_SERVICE = "resources";
 
+    /**
+     * @hide
+     */
+    public static final String THEME_ENGINE_SERVICE = "theme_engine";
+
     /**
      * Use with {@link #getSystemService(String)} to retrieve a
      * {android.os.IIdmap2} for managing idmap files (used by overlay

core/java/android/content/pm/LauncherActivityInfo.java

@@ -127,6 +127,21 @@ public CharSequence getLabel() {
      * @return The drawable associated with the activity.
      */
     public Drawable getIcon(int density) {
+        try {
+            ComponentName cn = getComponentName();
+            final Resources resources = mPm.getResourcesForApplication(
+                    getActivityInfo().applicationInfo);
+            if (cn != null && resources != null) {
+                Drawable themedIcon = resources.getIconPackOverride(
+                        cn.getPackageName(), cn.getClassName(), density);
+                if (themedIcon != null) {
+                    return themedIcon;
+                }
+            }
+        } catch (Exception e) {
+            android.util.Log.d("LauncherActivityInfo", "Icon pack override failed", e);
+        }
+
         // TODO: Go through LauncherAppsService
         final int iconRes = getActivityInfo().getIconResource();
         Drawable icon = null;

core/java/android/content/res/IThemeEngineCallback.aidl

@@ -0,0 +1,6 @@
+package android.content.res;
+
+/** @hide */
+oneway interface IThemeEngineCallback {
+    void onThemeChanged(String category);
+}

core/java/android/content/res/IThemeEngineManager.aidl

@@ -0,0 +1,31 @@
+package android.content.res;
+
+import android.content.ComponentName;
+import android.content.res.IThemeEngineCallback;
+import android.graphics.Bitmap;
+
+/** @hide */
+interface IThemeEngineManager {
+    Bitmap getIconPackIcon(in ComponentName component, int density);
+    boolean hasActiveIconPack();
+    String getIconPackPackage();
+
+    Bitmap getPerAppIconPackIcon(String packageName, String className, int density);
+    void setPerAppIconPack(String packageName, String iconPackPackage);
+    void clearPerAppIconPack(String packageName);
+    String getPerAppIconPack(String packageName);
+    List<String> getInstalledIconPacks();
+
+    Bitmap getSystemThemeIconDrawable(String resourceName, int density);
+    boolean isTargetedResource(String resourceName);
+    boolean hasActiveSystemThemeIcons();
+    String getActiveSystemThemeIcons();
+
+    String getCategoryTheme(String category);
+
+    void registerCallback(IThemeEngineCallback callback);
+    void unregisterCallback(IThemeEngineCallback callback);
+    void notifyThemeChanged(String category);
+
+    List<String> getAvailableOverlays(String category);
+}

core/java/android/content/res/Resources.java

@@ -45,6 +45,7 @@
 import android.app.LocaleConfig;
 import android.app.ResourcesManager;
 import android.compat.annotation.UnsupportedAppUsage;
+import android.content.ComponentName;
 import android.content.Context;
 import android.content.pm.ActivityInfo;
 import android.content.pm.ActivityInfo.Config;
@@ -251,6 +252,47 @@ public static Resources getSystem() {
         }
     }
 
+    /** @hide */
+    @Nullable
+    public Drawable getIconPackOverride(
+            @NonNull String packageName, @NonNull String className, int density) {
+        try {
+            ThemeEngine themeEngine = ThemeEngine.getInstance();
+            if (themeEngine == null) return null;
+            ComponentName cn = new ComponentName(packageName, className);
+            return themeEngine.getIconPackDrawable(cn, density);
+        } catch (Exception e) {
+            return null;
+        }
+    }
+
+    /** @hide */
+    @Nullable
+    public Drawable getIconPackOverride(
+            @NonNull String packageName, @NonNull String className) {
+        return getIconPackOverride(packageName, className, 0);
+    }
+
+    /** @hide */
+    public boolean hasActiveIconPack() {
+        try {
+            ThemeEngine themeEngine = ThemeEngine.getInstance();
+            return themeEngine != null && themeEngine.hasActiveIconPack();
+        } catch (Exception e) {
+            return false;
+        }
+    }
+
+    /** @hide */
+    public String getIconPackPackage() {
+        try {
+            ThemeEngine themeEngine = ThemeEngine.getInstance();
+            return themeEngine != null ? themeEngine.getIconPackPackage() : null;
+        } catch (Exception e) {
+            return null;
+        }
+    }
+
     /**
      * This exception is thrown by the resource APIs when a requested resource
      * can not be found.
@@ -1026,9 +1068,31 @@ public Drawable getDrawableForDensity(@DrawableRes int id, int density, @Nullabl
     @UnsupportedAppUsage(maxTargetSdk = Build.VERSION_CODES.R, trackingBug = 170729553)
     Drawable loadDrawable(@NonNull TypedValue value, int id, int density, @Nullable Theme theme)
             throws NotFoundException {
+        return loadDrawableInternal(value, id, density, theme);
+    }
+
+    /** @hide */
+    @NonNull
+    public Drawable loadDrawableInternal(@NonNull TypedValue value, int id, int density,
+            @Nullable Theme theme) throws NotFoundException {
         return mResourcesImpl.loadDrawable(this, value, id, density, theme);
     }
 
+    /** @hide */
+    @Nullable
+    public Drawable getDrawableInternal(@DrawableRes int id) {
+        final TypedValue value = obtainTempTypedValue();
+        try {
+            final ResourcesImpl impl = mResourcesImpl;
+            impl.getValueForDensity(id, 0, value, true);
+            return loadDrawableInternal(value, id, 0, null);
+        } catch (Exception e) {
+            return null;
+        } finally {
+            releaseTempTypedValue(value);
+        }
+    }
+
     /**
      * Return a movie object associated with the particular resource ID.
      * @param id The desired resource identifier, as generated by the aapt