fix: key state not updated after composing & has_menu state completed

Co-authored-by: iSilen <lzlv312@gmail.com>
Co-authored-by: WhiredPlanck <whiredplanck@outlook.com>

Author: 3 people

app/src/main/java/com/osfans/trime/ime/keyboard/KeyboardWindow.kt

@@ -12,6 +12,7 @@ import android.widget.FrameLayout
 import androidx.core.content.ContextCompat
 import com.osfans.trime.R
 import com.osfans.trime.core.RimeMessage
+import com.osfans.trime.core.RimeProto
 import com.osfans.trime.core.SchemaItem
 import com.osfans.trime.daemon.RimeSession
 import com.osfans.trime.data.theme.KeyActionManager
@@ -261,6 +262,13 @@ class KeyboardWindow(
         }
     }
 
+    override fun onCompositionUpdate(data: RimeProto.Context.Composition) {
+        val status = rime.run { statusCached }
+        if (!status.isAsciiMode && data.length == 0 && data.preedit.isNullOrEmpty()) {
+            currentKeyboardView?.invalidateAllKeys()
+        }
+    }
+
     override fun onSelectionUpdate(
         start: Int,
         end: Int,