Fix bug in Bubble Colors

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

Author: Dev4Mod

app/src/main/java/com/wmods/wppenhacer/xposed/core/devkit/Unobfuscator.java

@@ -1656,6 +1656,14 @@ public synchronized static Method loadBubbleDrawableMethod(ClassLoader classLoad
         });
     }
 
+    public synchronized static Method loadDateWrapper(ClassLoader classLoader) throws Exception {
+        return UnobfuscatorCache.getInstance().getMethod(classLoader, () -> {
+            var methodData = dexkit.findMethod(FindMethod.create().matcher(MethodMatcher.create().name("getDateWrapperBackground")));
+            if (methodData.isEmpty()) throw new Exception("LoadDateWrapper method not found");
+            return methodData.get(0).getMethodInstance(classLoader);
+        });
+    }
+
     public static synchronized Method[] loadRootDetector(ClassLoader classLoader) {
         var methods = findAllMethodUsingStrings(classLoader, StringMatchType.Contains, "/system/bin/su");
         if (methods.length == 0) throw new RuntimeException("RootDetector method not found");

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

@@ -9,6 +9,7 @@
 import androidx.annotation.NonNull;
 
 import com.wmods.wppenhacer.xposed.core.Feature;
+import com.wmods.wppenhacer.xposed.core.components.FMessageWpp;
 import com.wmods.wppenhacer.xposed.core.devkit.Unobfuscator;
 import com.wmods.wppenhacer.xposed.utils.DesignUtils;
 import com.wmods.wppenhacer.xposed.utils.Utils;
@@ -20,6 +21,7 @@
 import de.robv.android.xposed.XC_MethodHook;
 import de.robv.android.xposed.XSharedPreferences;
 import de.robv.android.xposed.XposedBridge;
+import de.robv.android.xposed.XposedHelpers;
 
 public class BubbleColors extends Feature {
     public BubbleColors(ClassLoader loader, XSharedPreferences preferences) {
@@ -88,6 +90,24 @@ public void doHook() throws Exception {
             }
         }
 
+        var dateWrapper = Unobfuscator.loadDateWrapper(classLoader);
+        XposedBridge.hookMethod(dateWrapper, new XC_MethodHook() {
+            @Override
+            protected void afterHookedMethod(MethodHookParam param) throws Throwable {
+                var fmessageObject = XposedHelpers.callMethod(param.thisObject, "getFMessage");
+                if (fmessageObject == null) return;
+                var view = (Drawable) param.getResult();
+                var fmessage = new FMessageWpp(fmessageObject);
+                if (fmessage.getKey().isFromMe) {
+                    if (bubbleRightColor == 0) return;
+                    view.setColorFilter(new PorterDuffColorFilter(bubbleRightColor, PorterDuff.Mode.SRC_IN));
+                } else {
+                    if (bubbleLeftColor == 0) return;
+                    view.setColorFilter(new PorterDuffColorFilter(bubbleLeftColor, PorterDuff.Mode.SRC_IN));
+                }
+
+            }
+        });
 
         var bubbleDrawableMethod = Unobfuscator.loadBubbleDrawableMethod(classLoader);