Fix Seen tick button for View Once for audios

Author: Dev4Mod

app/src/main/java/com/wmods/wppenhacer/xposed/core/WppCore.java

@@ -59,13 +59,12 @@ public class WppCore {
     private static Object mActionUser;
     private static SQLiteDatabase mWaDatabase;
     public static BaseClient client;
-    private static boolean isBridgeInitialized;
+    private static Object mCachedMessageStore;
 
 
     public static void Initialize(ClassLoader loader) throws Exception {
         privPrefs = Utils.getApplication().getSharedPreferences("WaGlobal", Context.MODE_PRIVATE);
 
-
         // init UserJID
         var mSendReadClass = XposedHelpers.findClass("com.whatsapp.jobqueue.job.SendReadReceiptJob", loader);
         var subClass = ReflectionUtils.findConstructorUsingFilter(mSendReadClass, (constructor) -> constructor.getParameterCount() == 8).getParameterTypes()[0];
@@ -86,13 +85,24 @@ protected void beforeHookedMethod(MethodHookParam param) throws Throwable {
             }
         });
 
+        // ActionUser
         var actionUser = Unobfuscator.loadActionUser(loader);
         XposedBridge.hookAllConstructors(actionUser, new XC_MethodHook() {
             @Override
             protected void afterHookedMethod(MethodHookParam param) throws Throwable {
                 mActionUser = param.thisObject;
             }
         });
+
+        // CachedMessageStore
+        var cachedMessageStore = Unobfuscator.loadCachedMessageStore(loader);
+        XposedBridge.hookAllConstructors(cachedMessageStore, new XC_MethodHook() {
+            @Override
+            protected void afterHookedMethod(MethodHookParam param) throws Throwable {
+                mCachedMessageStore = param.thisObject;
+            }
+        });
+
         // Load wa database
         loadWADatabase();
         initBridge(Utils.getApplication());
@@ -135,7 +145,6 @@ private static boolean tryConnectBridge(BaseClient baseClient) throws Exception
             CompletableFuture<Boolean> canLoadFuture = baseClient.connect();
             Boolean canLoad = canLoadFuture.get();
             if (!canLoad) throw new Exception();
-            isBridgeInitialized = true;
         } catch (Exception e) {
             return false;
         }
@@ -258,6 +267,17 @@ public static String getWppContactName(Object userJid) {
         return name == null ? "" : name;
     }
 
+    public static Object getFMessageFromKey(Object messageKey) {
+        if (messageKey == null) return null;
+        try {
+            var methodResult = ReflectionUtils.findMethodUsingFilter(mCachedMessageStore.getClass(), (method) -> method.getParameterCount() == 1 && FMessageWpp.Key.TYPE.isAssignableFrom(method.getParameterTypes()[0]));
+            return ReflectionUtils.callMethod(methodResult, mCachedMessageStore, messageKey);
+        } catch (Exception e) {
+            XposedBridge.log(e);
+            return null;
+        }
+    }
+
 
     public static Object createUserJid(String rawjid) {
         var genInstance = XposedHelpers.newInstance(mGenJidClass);
@@ -356,7 +376,8 @@ public static Activity getCurrentConversation() {
         // for tablet UI, they're using HomeActivity instead of Conversation
         // TODO: Add more checks for ConversationFragment
         Class<?> home = XposedHelpers.findClass("com.whatsapp.HomeActivity", mCurrentActivity.getClassLoader());
-        if (mCurrentActivity.getResources().getConfiguration().smallestScreenWidthDp >= 600 && home.isInstance(mCurrentActivity)) return mCurrentActivity;
+        if (mCurrentActivity.getResources().getConfiguration().smallestScreenWidthDp >= 600 && home.isInstance(mCurrentActivity))
+            return mCurrentActivity;
         return null;
     }
 

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

@@ -1770,4 +1770,12 @@ public static synchronized Method loadStateChangeMethod(ClassLoader classLoader)
         return UnobfuscatorCache.getInstance().getMethod(classLoader, () -> findFirstMethodUsingStrings(classLoader, StringMatchType.Contains, "presencestatemanager/startTransitionToUnavailable/new-state"));
     }
 
+    public static synchronized Class loadCachedMessageStore(ClassLoader loader) throws Exception {
+        return UnobfuscatorCache.getInstance().getClass(loader, () -> {
+            var cacheMsClass = findFirstClassUsingStrings(loader, StringMatchType.Contains, "CachedMessageStore/getMessage/key");
+            if (cacheMsClass == null)
+                throw new RuntimeException("CachedMessageStore class not found");
+            return cacheMsClass;
+        });
+    }
 }

app/src/main/java/com/wmods/wppenhacer/xposed/features/general/SeenTick.java

@@ -297,13 +297,38 @@ protected void afterHookedMethod(MethodHookParam param) throws Throwable {
                         var messageField = ReflectionUtils.getFieldByExtendType(menuMethod.getDeclaringClass(), classThreadMessage);
                         var messageObject = XposedHelpers.getObjectField(param.thisObject, messageField.getName());
                         sendBlueTickMedia(messageObject, true);
-                        Toast.makeText(Utils.getApplication(), ResId.string.sending_read_blue_tick, Toast.LENGTH_SHORT).show();
+                        Utils.showToast(Utils.getApplication().getString(ResId.string.sending_read_blue_tick), Toast.LENGTH_SHORT);
                         return true;
                     });
                 }
 
             }
         });
+
+        XposedHelpers.findAndHookMethod("com.whatsapp.messaging.ViewOnceViewerActivity", classLoader, "onCreateOptionsMenu", classLoader.loadClass("android.view.Menu"),
+                new XC_MethodHook() {
+                    @Override
+                    protected void afterHookedMethod(MethodHookParam param) throws Throwable {
+                        Menu menu = (Menu) param.args[0];
+                        MenuItem item = menu.add(0, 0, 0, ResId.string.send_blue_tick).setIcon(Utils.getID("ic_notif_mark_read", "drawable"));
+                        if (ticktype == 1) item.setShowAsAction(MenuItem.SHOW_AS_ACTION_ALWAYS);
+                        item.setOnMenuItemClickListener(item1 -> {
+                            CompletableFuture.runAsync(() -> {
+                                var keyClass = FMessageWpp.Key.TYPE;
+                                var fieldType = ReflectionUtils.getFieldByType(param.thisObject.getClass(), keyClass);
+                                var keyMessage = ReflectionUtils.getField(fieldType, param.thisObject);
+                                var fMessage = WppCore.getFMessageFromKey(keyMessage);
+                                if (fMessage == null) return;
+                                sendBlueTickMedia(fMessage, true);
+                                Utils.showToast(Utils.getApplication().getString(ResId.string.sending_read_blue_tick), Toast.LENGTH_SHORT);
+                            });
+                            return true;
+                        });
+
+                    }
+                });
+
+
     }
 
     private void hookOnSendMessages() throws Exception {