Fix bug in Custom Wallpaper

Signed-off-by: Dev4Mod <[email protected]>

Author: Dev4Mod

app/src/main/java/com/wmods/wppenhacer/xposed/features/customization/CustomTheme.java

@@ -32,6 +32,7 @@
 import com.wmods.wppenhacer.utils.IColors;
 import com.wmods.wppenhacer.views.WallpaperView;
 import com.wmods.wppenhacer.xposed.core.Feature;
+import com.wmods.wppenhacer.xposed.core.WppCore;
 import com.wmods.wppenhacer.xposed.core.devkit.Unobfuscator;
 import com.wmods.wppenhacer.xposed.utils.DesignUtils;
 import com.wmods.wppenhacer.xposed.utils.ReflectionUtils;
@@ -91,6 +92,7 @@ protected void afterHookedMethod(MethodHookParam param) throws Throwable {
                 new XC_MethodHook() {
                     @Override
                     protected void afterHookedMethod(MethodHookParam param) throws Throwable {
+                        if (!checkHomeActivity()) return;
                         var viewGroup = (ViewGroup) param.getResult();
                         replaceColors(viewGroup, wallAlpha);
                     }
@@ -101,6 +103,7 @@ protected void afterHookedMethod(MethodHookParam param) throws Throwable {
             @Override
             protected void afterHookedMethod(MethodHookParam param) throws Throwable {
                 if (!loadTabFrameClass.isInstance(param.thisObject)) return;
+                if (!checkHomeActivity()) return;
                 var viewGroup = (ViewGroup) param.thisObject;
                 var background = viewGroup.getBackground();
                 try {
@@ -327,4 +330,12 @@ protected void beforeHookedMethod(MethodHookParam param) throws Throwable {
             param.args[0] = IColors.getFromIntColor(color, IColors.colors);
         }
     }
+
+    private boolean checkHomeActivity() {
+        var homeClass = XposedHelpers.findClass("com.whatsapp.HomeActivity", classLoader);
+        var currentActivity = WppCore.getCurrentActivity();
+        return currentActivity != null && homeClass.isInstance(currentActivity);
+    }
+
+
 }