feat: add experimental force restore backup feature with user confirmation dialog

Author: Dev4Mod

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

@@ -906,61 +906,6 @@ public synchronized static Method loadAntiRevokeMessageMethod(ClassLoader loader
         });
     }
 
-    public synchronized static Class<?> loadSettingsGoogleDriveActivity(ClassLoader loader) throws Exception {
-        return UnobfuscatorCache.getInstance().getClass(loader, () -> {
-            var classes = findAllClassUsingStrings(loader, StringMatchType.Contains, "SettingsGoogleDrive");
-            if (classes == null)
-                throw new Exception("SettingsGoogleDriveActivity not found (No classes with string)");
-
-            StringBuilder candidates = new StringBuilder();
-            for (var cls : classes) {
-                String name = cls.getName();
-                if (name.contains("com.whatsapp.deeplink"))
-                    continue;
-
-                candidates.append(name).append(", ");
-
-                // Check for onCreate method (Standard for Activity/Fragment)
-                try {
-                    // Check declared methods for "onCreate"
-                    for (Method m : cls.getDeclaredMethods()) {
-                        if (m.getName().equals("onCreate")) {
-                            return cls;
-                        }
-                    }
-                } catch (Throwable t) {
-                    // Ignore reflection errors
-                }
-            }
-            throw new Exception("SettingsGoogleDriveActivity not found. Candidates checked: " + candidates.toString());
-
-        });
-    }
-
-    public synchronized static Class<?> loadRestoreBackupActivity(ClassLoader loader) throws Exception {
-        return UnobfuscatorCache.getInstance().getClass(loader, () -> {
-            var strings = new String[] { "RestoreFromBackupActivity", "gdrive/restore/activity",
-                    "gdrive_restore_title" };
-            for (String s : strings) {
-                var classes = findAllClassUsingStrings(loader, StringMatchType.Contains, s);
-                if (classes != null) {
-                    for (var cls : classes) {
-                        try {
-                            for (Method m : cls.getDeclaredMethods()) {
-                                if (m.getName().equals("onCreate")) {
-                                    return cls;
-                                }
-                            }
-                        } catch (Throwable t) {
-                        }
-                    }
-                }
-
-            }
-            throw new Exception("RestoreBackupActivity not found");
-        });
-    }
-
     public synchronized static Field loadMessageKeyField(ClassLoader loader) throws Exception {
         return UnobfuscatorCache.getInstance().getField(loader, () -> {
             var classList = dexkit.findClass(new FindClass().matcher(new ClassMatcher().fieldCount(3)

app/src/main/java/com/wmods/wppenhacer/xposed/features/others/BackupRestore.java

@@ -3,17 +3,19 @@
 import android.app.Activity;
 import android.content.Intent;
 import android.view.Menu;
-import android.view.MenuItem;
 import android.widget.Toast;
 
 import com.wmods.wppenhacer.xposed.core.Feature;
+import com.wmods.wppenhacer.xposed.core.components.AlertDialogWpp;
 import com.wmods.wppenhacer.xposed.core.devkit.Unobfuscator;
+import com.wmods.wppenhacer.xposed.utils.ResId;
 import com.wmods.wppenhacer.xposed.utils.Utils;
 
+import org.luckypray.dexkit.query.enums.StringMatchType;
+
 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 BackupRestore extends Feature {
 
@@ -30,39 +32,39 @@ public String getPluginName() {
     public void doHook() throws Exception {
         if (!prefs.getBoolean("force_restore_backup_feature", false)) return;
 
-        Class<?> settingsDriveClass = Unobfuscator.loadSettingsGoogleDriveActivity(classLoader);
-        
-        XposedHelpers.findAndHookMethod(settingsDriveClass, "onPrepareOptionsMenu", Menu.class, new XC_MethodHook() {
+        var restoreFromBackupClass = Unobfuscator.findFirstClassUsingName(classLoader, StringMatchType.EndsWith, "RestoreFromBackupActivity");
+
+        XposedBridge.hookAllMethods(Activity.class, "onPrepareOptionsMenu", new XC_MethodHook() {
             @Override
             protected void afterHookedMethod(MethodHookParam param) throws Throwable {
+                var name = param.thisObject.getClass().getSimpleName().toLowerCase();
+                if (!(name.contains("drive") && name.contains("google"))) return;
                 Menu menu = (Menu) param.args[0];
-                // Hardcoding string to ensure it appears without resource injection issues
-                String title = "Force Restore Backup (Experimental)";
-                
-                // Use a high ID to avoid conflicts
-                if (menu.findItem(10001) == null) {
-                    menu.add(0, 10001, 0, title);
-                }
-            }
-        });
+                if (menu.findItem(10001) != null) return;
+                var menuItem = menu.add(0, 10001, 0, ResId.string.force_restore_backup_experimental);
+                Activity activity = (Activity) param.thisObject;
+                menuItem.setOnMenuItemClickListener((item) -> {
+                    new AlertDialogWpp(activity)
+                            .setTitle(ResId.string.force_restore_backup)
+                            .setMessage(activity.getString(ResId.string.warning_restore))
+                            .setPositiveButton(activity.getString(ResId.string.yes), (dialog, which) -> {
+                                try {
+                                    Intent intent = new Intent(activity, restoreFromBackupClass);
+                                    intent.setAction("action_show_restore_one_time_setup");
+                                    activity.startActivityForResult(intent, 10001);
+                                } catch (Exception e) {
+                                    XposedBridge.log(e);
+                                    Utils.showToast("Error launching restore activity: " + e.getMessage(), Toast.LENGTH_LONG);
+                                }
+                            })
+                            .setNegativeButton(activity.getString(ResId.string.no), null)
+                            .show();
+
+                    return true;
+                });
 
-        XposedHelpers.findAndHookMethod(settingsDriveClass, "onOptionsItemSelected", MenuItem.class, new XC_MethodHook() {
-            @Override
-            protected void beforeHookedMethod(MethodHookParam param) throws Throwable {
-                MenuItem item = (MenuItem) param.args[0];
-                if (item.getItemId() == 10001) {
-                    Activity activity = (Activity) param.thisObject;
-                    try {
-                        Class<?> restoreClass = Unobfuscator.loadRestoreBackupActivity(classLoader);
-                        Intent intent = new Intent(activity, restoreClass);
-                        activity.startActivity(intent);
-                        param.setResult(true);
-                    } catch (Exception e) {
-                        Utils.showToast("Error launching restore activity: " + e.getMessage(), Toast.LENGTH_LONG);
-                        XposedBridge.log(e);
-                    }
-                }
             }
         });
+
     }
 }

app/src/main/java/com/wmods/wppenhacer/xposed/utils/ResId.java

@@ -109,6 +109,9 @@ public static class string {
         public static int possible_block_detected;
         public static int checking_if_the_contact_is_blocked;
         public static int block_unverified;
+        public static int warning_restore;
+        public static int force_restore_backup_experimental;
+        public static int force_restore_backup;
     }
 
     public static class array {

app/src/main/res/values/strings.xml

@@ -476,4 +476,6 @@
     <string name="action_follow_developer">Coming soon!</string>
     <string name="enable_spy">Enable Spy Tool</string>
     <string name="enable_spy_sum">Logs internal events to Xposed log for debugging.</string>
+    <string name="warning_restore">When restoring the backup all messages after the backup will be lost, are you sure you want to continue?</string>
+    <string name="force_restore_backup_experimental">Force Restore Backup (Experimental)</string>
 </resources>