Fix duplicated item on MIUI

Xposed-Modules-Repo · Mar 25, 2022 · b47be37 · b47be37
1 parent 778ff8f
commit b47be37
Showing 1 changed file with 17 additions and 7 deletions.
diff --git a/app/src/main/java/io/github/tehcneko/undo/UndoHook.java b/app/src/main/java/io/github/tehcneko/undo/UndoHook.java
@@ -32,14 +32,24 @@ protected void beforeHookedMethod(MethodHookParam param) throws Throwable {
                             var editor = editorField.get(param.thisObject);
                             var textView = textViewField.get(editor);
                             var menu = (Menu) param.args[0];
-                            if ((boolean) XposedHelpers.callMethod(textView, "canUndo")) {
-                                menu.add(Menu.NONE, android.R.id.undo, 2, undoId)
-                                        .setAlphabeticShortcut('z')
-                                        .setShowAsAction(MenuItem.SHOW_AS_ACTION_ALWAYS);
+                            var undoItem = menu.findItem(undoId);
+                            if (undoItem == null) {
+                                if ((boolean) XposedHelpers.callMethod(textView, "canUndo")) {
+                                    menu.add(Menu.NONE, android.R.id.undo, 2, undoId)
+                                            .setAlphabeticShortcut('z')
+                                            .setShowAsAction(MenuItem.SHOW_AS_ACTION_ALWAYS);
+                                }
+                            } else {
+                                undoItem.setVisible((boolean) XposedHelpers.callMethod(textView, "canUndo"));
                             }
-                            if ((boolean) XposedHelpers.callMethod(textView, "canRedo")) {
-                                menu.add(Menu.NONE, android.R.id.redo, 3, redoId)
-                                        .setShowAsAction(MenuItem.SHOW_AS_ACTION_ALWAYS);
+                            var redoItem = menu.findItem(redoId);
+                            if (redoItem == null) {
+                                if ((boolean) XposedHelpers.callMethod(textView, "canRedo")) {
+                                    menu.add(Menu.NONE, android.R.id.redo, 3, redoId)
+                                            .setShowAsAction(MenuItem.SHOW_AS_ACTION_ALWAYS);
+                                }
+                            } else {
+                                redoItem.setVisible((boolean) XposedHelpers.callMethod(textView, "canRedo"));
                             }
                         }
                     });