Bugfix in TabsProcessor: handle situations when the cursor is at the end of the line.

This fixes a bug in the selection, when TabsProcessor is used. (For
pyvim, for instance.)

1 files changed, 4 insertions, 4 deletions

diff --git a/prompt_toolkit/layout/processors.py b/prompt_toolkit/layout/processors.py
index ccc7b1ec..f5e5ab26 100644
--- a/prompt_toolkit/layout/processors.py
+++ b/prompt_toolkit/layout/processors.py
@@ -589,13 +589,13 @@ class TabsProcessor(Processor):
                 pos += 1
 
         position_mappings[len(fragments)] = pos
+        # Add `pos+1` to mapping, because the cursor can be right after the
+        # line as well.
+        position_mappings[len(fragments) + 1] = pos + 1
 
         def source_to_display(from_position):
             " Maps original cursor position to the new one. "
-            try:
-                return position_mappings[from_position]
-            except KeyError:
-                return 0
+            return position_mappings[from_position]
 
         def display_to_source(display_pos):
             " Maps display cursor position to the original one. "