Support various cursor shapes for more input modes

prompt-toolkit · Jul 29, 2024 · 9e8bf20 · 9e8bf20
1 parent 6695411
commit 9e8bf20
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/src/prompt_toolkit/cursor_shapes.py b/src/prompt_toolkit/cursor_shapes.py
@@ -69,10 +69,23 @@ class ModalCursorShapeConfig(CursorShapeConfig):
 
     def get_cursor_shape(self, application: Application[Any]) -> CursorShape:
         if application.editing_mode == EditingMode.VI:
-            if application.vi_state.input_mode == InputMode.INSERT:
+            if application.vi_state.input_mode in {
+                InputMode.NAVIGATION,
+            }:
+                return CursorShape.BLOCK
+            if application.vi_state.input_mode in {
+                InputMode.INSERT,
+                InputMode.INSERT_MULTIPLE,
+            }:
                 return CursorShape.BEAM
-            if application.vi_state.input_mode == InputMode.REPLACE:
+            if application.vi_state.input_mode == {
+                InputMode.REPLACE,
+                InputMode.REPLACE_SINGLE,
+            }:
                 return CursorShape.UNDERLINE
+        elif application.editing_mode == EditingMode.EMACS:
+            # like vi's INSERT
+            return CursorShape.BEAM
 
         # Default
         return CursorShape.BLOCK