From 84e6da4de7d8009014487b49887cea9414bd9740 Mon Sep 17 00:00:00 2001
From: Ilya Zhuravlev <whatever@xyz.is>
Date: Thu, 24 Dec 2020 09:41:31 -0500
Subject: [PATCH] macro_optimizer: more optimizations

---
 src/main/python/macro_optimizer.py | 73 +++++++++++++++++++++++++++++-
 src/main/python/test/test_macro.py | 13 ++++--
 2 files changed, 82 insertions(+), 4 deletions(-)

diff --git a/src/main/python/macro_optimizer.py b/src/main/python/macro_optimizer.py
index 6e822dd..4d45aef 100644
--- a/src/main/python/macro_optimizer.py
+++ b/src/main/python/macro_optimizer.py
@@ -1,5 +1,5 @@
 # SPDX-License-Identifier: GPL-2.0-or-later
-from macro_key import KeyUp, KeyDown, KeyTap
+from macro_key import KeyUp, KeyDown, KeyTap, KeyString
 
 
 def remove_repeats(sequence):
@@ -29,7 +29,78 @@ def replace_with_tap(sequence):
     return out
 
 
+PRINTABLE = {
+    "KC_1": "1", "KC_2": "2", "KC_3": "3", "KC_4": "4", "KC_5": "5", "KC_6": "6", "KC_7": "7",
+    "KC_8": "8", "KC_9": "9", "KC_0": "0",
+    "KC_MINUS": "-",
+    "KC_EQUAL": "=",
+    "KC_Q": "q",
+    "KC_W": "w",
+    "KC_E": "e",
+    "KC_R": "r",
+    "KC_T": "t",
+    "KC_Y": "y",
+    "KC_U": "u",
+    "KC_I": "i",
+    "KC_O": "o",
+    "KC_P": "p",
+    "KC_LBRACKET": "[",
+    "KC_RBRACKET": "]",
+    "KC_A": "a",
+    "KC_S": "s",
+    "KC_D": "d",
+    "KC_F": "f",
+    "KC_G": "g",
+    "KC_H": "h",
+    "KC_J": "j",
+    "KC_K": "k",
+    "KC_L": "l",
+    "KC_SCOLON": ";",
+    "KC_QUOTE": "'",
+    "KC_GRAVE": "`",
+    "KC_BSLASH": "\\",
+    "KC_Z": "z",
+    "KC_X": "x",
+    "KC_C": "c",
+    "KC_V": "v",
+    "KC_B": "b",
+    "KC_N": "n",
+    "KC_M": "m",
+    "KC_COMMA": ",",
+    "KC_DOT": ".",
+    "KC_SLASH": "/",
+}
+
+
+def is_printable_tap(k):
+    return isinstance(k, KeyTap) and k.keycode.qmk_id in PRINTABLE
+
+
+def get_printable_char(k):
+    return PRINTABLE[k.keycode.qmk_id]
+
+
+def replace_with_string(sequence):
+    """ Replaces a sequence of printable taps with a sendstring """
+    out = []
+    sequence = sequence[:]
+    while len(sequence) > 0:
+        if len(sequence) >= 2 and is_printable_tap(sequence[0]) and is_printable_tap(sequence[1]):
+            cur = get_printable_char(sequence[0]) + get_printable_char(sequence[1])
+            sequence.pop(0)
+            sequence.pop(0)
+            while len(sequence) and is_printable_tap(sequence[0]):
+                cur += get_printable_char(sequence[0])
+                sequence.pop(0)
+            out.append(KeyString(cur))
+        else:
+            out.append(sequence[0])
+            sequence.pop(0)
+    return out
+
+
 def macro_optimize(sequence):
     sequence = remove_repeats(sequence)
     sequence = replace_with_tap(sequence)
+    sequence = replace_with_string(sequence)
     return sequence
diff --git a/src/main/python/test/test_macro.py b/src/main/python/test/test_macro.py
index c042e08..3805067 100644
--- a/src/main/python/test/test_macro.py
+++ b/src/main/python/test/test_macro.py
@@ -2,9 +2,8 @@
 import unittest
 
 from keycodes import find_keycode
-from macro_key import KeyDown, KeyTap
-from macro_optimizer import remove_repeats
-
+from macro_key import KeyDown, KeyTap, KeyUp, KeyString
+from macro_optimizer import remove_repeats, replace_with_tap, replace_with_string
 
 KC_A = find_keycode(0x04)
 KC_B = find_keycode(0x05)
@@ -20,3 +19,11 @@ class TestMacro(unittest.TestCase):
 
         # don't remove repeated taps
         self.assertEqual(remove_repeats([KeyTap(KC_A), KeyTap(KC_A)]), [KeyTap(KC_A), KeyTap(KC_A)])
+
+    def test_replace_tap(self):
+        self.assertEqual(replace_with_tap([KeyDown(KC_A)]), [KeyDown(KC_A)])
+        self.assertEqual(replace_with_tap([KeyDown(KC_A), KeyUp(KC_A)]), [KeyTap(KC_A)])
+        self.assertEqual(replace_with_tap([KeyUp(KC_A), KeyDown(KC_A)]), [KeyUp(KC_A), KeyDown(KC_A)])
+
+    def test_replace_string(self):
+        self.assertEqual(replace_with_string([KeyTap(KC_A), KeyTap(KC_B)]), [KeyString("ab")])