RichInputConnection: Force batch edit restart when querying text near cursor

Issue: calyxos#2916
Change-Id: I7daad9ab62f87d4dcff02f0993b33a85a815896f
Signed-off-by: Aayush Gupta <aayushgupta219@gmail.com>

Author: theimpulson

java/src/com/android/inputmethod/latin/RichInputConnection.java

@@ -705,6 +705,15 @@ public TextRange getWordRangeAtCursor(final SpacingAndPunctuations spacingAndPun
         if (!isConnected()) {
             return null;
         }
+
+        // Restart batch edit to force state update for broken web apps (for e.g. ProtonMail)
+        // as batch edit mode avoids sending state updates until batch edit end.
+        // See beginBatchEdit mode documentation for further information.
+        if (mNestLevel > 0) {
+            endBatchEdit();
+            beginBatchEdit();
+        }
+
         final CharSequence before = getTextBeforeCursorAndDetectLaggyConnection(
                 OPERATION_GET_WORD_RANGE_AT_CURSOR,
                 SLOW_INPUT_CONNECTION_ON_PARTIAL_RELOAD_MS,