reafactor(menu.py): Use simple_menu

Signed-off-by: Rahix <rahix@rahix.de>

preload/menu.py

@@ -5,10 +5,10 @@ You can customize this script however you want :)  If you want to go back to
 the default version, just delete this file; the firmware will recreate it on
 next run.
 """
-import buttons
 import color
 import display
 import os
+import simple_menu
 
 
 def list_apps():
@@ -24,96 +24,33 @@ def list_apps():
     return apps
 
 
-def button_events():
-    """Iterate over button presses (event-loop)."""
-    yield 0
-    button_pressed = False
-    while True:
-        v = buttons.read(buttons.BOTTOM_LEFT | buttons.BOTTOM_RIGHT | buttons.TOP_RIGHT)
-
-        if v == 0:
-            button_pressed = False
-
-        if not button_pressed and v & buttons.BOTTOM_LEFT != 0:
-            button_pressed = True
-            yield buttons.BOTTOM_LEFT
-
-        if not button_pressed and v & buttons.BOTTOM_RIGHT != 0:
-            button_pressed = True
-            yield buttons.BOTTOM_RIGHT
-
-        if not button_pressed and v & buttons.TOP_RIGHT != 0:
-            button_pressed = True
-            yield buttons.TOP_RIGHT
-
-
-COLOR1, COLOR2 = (color.CHAOSBLUE_DARK, color.CHAOSBLUE)
-
-
-def draw_menu(disp, applist, idx, offset):
-    disp.clear()
-
-    # Wrap around the app-list and draw entries from idx - 3 to idx + 4
-    for y, i in enumerate(range(len(applist) + idx - 3, len(applist) + idx + 4)):
-        disp.print(
-            " " + applist[i % len(applist)] + "      ",
-            posy=offset + y * 20 - 40,
-            bg=COLOR1 if i % 2 == 0 else COLOR2,
-        )
-
-    disp.print(
-        ">",
-        posy=20,
-        fg=color.COMMYELLOW,
-        bg=COLOR1 if (idx + len(applist)) % 2 == 0 else COLOR2,
-    )
-    disp.update()
+class MainMenu(simple_menu.Menu):
+    def on_select(self, name, index):
+        display.open().clear().update()
+        display.close()
+        try:
+            os.exec(name)
+        except OSError as e:
+            print("Loading failed: ", e)
+            os.exit(1)
 
 
 def main():
-    disp = display.open()
     applist = list_apps()
-    numapps = len(applist)
-    current = 0
-    for ev in button_events():
-        if numapps == 0:
+
+    if applist == []:
+        disp = display.open()
         disp.clear(color.COMMYELLOW)
         disp.print(
-                " No apps ",
-                posx=17,
-                posy=20,
-                fg=color.COMMYELLOW_DARK,
-                bg=color.COMMYELLOW,
+            " No apps ", posx=17, posy=20, fg=color.COMMYELLOW_DARK, bg=color.COMMYELLOW
         )
         disp.print(
-                "available",
-                posx=17,
-                posy=40,
-                fg=color.COMMYELLOW_DARK,
-                bg=color.COMMYELLOW,
+            "available", posx=17, posy=40, fg=color.COMMYELLOW_DARK, bg=color.COMMYELLOW
         )
         disp.update()
-            continue
-
-        if ev == buttons.BOTTOM_RIGHT:
-            # Scroll down
-            draw_menu(disp, applist, current, -8)
-            current = (current + 1) % numapps
-        elif ev == buttons.BOTTOM_LEFT:
-            # Scroll up
-            draw_menu(disp, applist, current, 8)
-            current = (current + numapps - 1) % numapps
-        elif ev == buttons.TOP_RIGHT:
-            # Select & start
-            disp.clear().update()
-            disp.close()
-            try:
-                os.exec(applist[current])
-            except OSError as e:
-                print("Loading failed: ", e)
-                os.exit(1)
+        return
 
-        draw_menu(disp, applist, current, 0)
+    MainMenu(applist).run()
 
 
 if __name__ == "__main__":