From c449f96bf01642fc4c251e10eb4f1701adfb241b Mon Sep 17 00:00:00 2001
From: Bryn Phillips <brynphillips1@gmail.com>
Date: Wed, 1 Apr 2020 20:42:07 -0700
Subject: [PATCH] Added up, down, left, right wch codes for win

---
 babi/screen.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/babi/screen.py b/babi/screen.py
index ec9728f..42aac86 100644
--- a/babi/screen.py
+++ b/babi/screen.py
@@ -61,6 +61,28 @@ SEQUENCE_KEYNAME = {
     '\x1b[1;6H': b'kHOM6',  # Shift + ^Home
     '\x1b[1;6F': b'kEND6',  # Shift + ^End
 }
+# windows-curses has different keynames for some keys
+KEYNAME_REWRITE = {
+    # numeric pad arrow keys
+    # - some overlay keyboards pick these as well
+    # - in xterm it seems these are mapped automatically
+    b'KEY_A2': b'KEY_UP',
+    b'KEY_C2': b'KEY_DOWN',
+    b'KEY_B3': b'KEY_RIGHT',
+    b'KEY_B1': b'KEY_LEFT',
+    # map to our M- names
+    b'ALT_U': b'M-u',
+    # arguably these names are better than the xterm names
+    b'CTL_UP': b'kUP5',
+    b'CTL_DOWN': b'kDN5',
+    b'CTL_RIGHT': b'kRIT5',
+    b'CTL_LEFT': b'kLFT5',
+    b'ALT_RIGHT': b'kRIT3',
+    b'ALT_LEFT': b'kLFT3',
+    # idk why these are different
+    b'KEY_SUP': b'KEY_SR',
+    b'KEY_SDOWN': b'KEY_SF',
+}
 
 
 class Key(NamedTuple):
@@ -208,6 +230,7 @@ class Screen:
 
         key = wch if isinstance(wch, int) else ord(wch)
         keyname = curses.keyname(key)
+        keyname = KEYNAME_REWRITE.get(keyname, keyname)
         return Key(wch, keyname)
 
     def get_char(self) -> Key: