From b1f387d281aad811104bb4e4f4de6f0a6bcd351b Mon Sep 17 00:00:00 2001
From: mecaneer23 <mecaneer23@gmail.com>
Date: Tue, 3 Oct 2023 22:49:40 -0500
Subject: [PATCH] simplify wgetnstr

---
 src/get_todo.py | 37 ++++++++++++++++++++++++-------------
 1 file changed, 24 insertions(+), 13 deletions(-)

diff --git a/src/get_todo.py b/src/get_todo.py
index 8c8e568..fc2281a 100644
--- a/src/get_todo.py
+++ b/src/get_todo.py
@@ -224,6 +224,24 @@ def toggle_note_todo(todo: Todo) -> None:
     todo.box_char = None
 
 
+def get_chars_position(
+    input_char: int,
+    stdscr_win: tuple[Any, Any],
+    chars: list[str],
+    position: int,
+    mode: Mode | None,
+    todo: Todo,
+    backspace_table: dict[int, Callable[..., tuple[list[str], int]]],
+) -> tuple[list[str], int] | None:
+    if input_char == 27:  # any escape sequence `^[`
+        return handle_escape(stdscr_win, chars, position, mode, todo)
+    if input_char == 9:  # tab
+        return handle_indent_dedent(stdscr_win[0], todo, "indent", chars, position)
+    if input_char in backspace_table:
+        return backspace_table[input_char](chars, position)
+    return handle_ascii(chars, position, input_char)
+
+
 def wgetnstr(
     stdscr: Any,
     win: Any,
@@ -299,22 +317,15 @@ def wgetnstr(
             return original
         if input_char in (10, 13):  # enter
             break
-        if input_char == 27:  # any escape sequence `^[`
-            next_step = handle_escape((stdscr, win), chars, position, mode, todo)
-            if next_step is None:
-                return original
-            chars, position = next_step
-            continue
         if input_char in (24, 11):  # ctrl + x/k
             toggle_mode(mode)
             break
-        if input_char == 9:  # tab
-            handle_indent_dedent(stdscr, todo, "indent", chars, position)
-            continue
-        if input_char in backspace_table:
-            chars, position = backspace_table[input_char](chars, position)
-            continue
-        chars, position = handle_ascii(chars, position, input_char)
+        next_step = get_chars_position(
+            input_char, (stdscr, win), chars, position, mode, todo, backspace_table
+        )
+        if next_step is None:
+            return original
+        chars, position = next_step
 
     todo.set_display_text("".join(chars))
     return todo