From 3cd9717e114a559dfbbc98f2ef8cd03e37698a1c Mon Sep 17 00:00:00 2001
From: Anthony Liao <anthonyliao@ufl.edu>
Date: Tue, 24 Oct 2023 15:21:46 -0400
Subject: [PATCH] added UI functionality

---
 .../scripts/preflight/preflightUI.py          | 184 +++++++++++++-----
 1 file changed, 131 insertions(+), 53 deletions(-)

diff --git a/mil_common/utils/mil_tools/scripts/preflight/preflightUI.py b/mil_common/utils/mil_tools/scripts/preflight/preflightUI.py
index 389fb6bb0..82c5e5075 100755
--- a/mil_common/utils/mil_tools/scripts/preflight/preflightUI.py
+++ b/mil_common/utils/mil_tools/scripts/preflight/preflightUI.py
@@ -1,67 +1,145 @@
 #!/usr/bin/env python3
 import blessed
 
-inp = ""
-term = blessed.Terminal()
-pos = 0
-
-machine = ["SubjuGator", "NaviGator", "Exit"]
-options = [
-    "Hardware Checklist",
-    "Software Checklist",
-    "Rosnode Command",
-    "Back",
-    "Exit",
-]
-currmenu = machine
-
-
-while inp != "q":
-    maxPos = len(currmenu)
-    # reprint line selected -> term.on_green"message"
-    print(term.home + term.clear + term.blink + term.on_green("pressed ") + repr(inp))
-
-    print("Preflight Checklist\n")
-
-    for x in range(0, maxPos):
-        if x == pos:
-            print(term.blink + term.on_green(currmenu[x]))
-        else:
-            print(currmenu[x])
-
-    with term.cbreak():
-        inp = term.inkey()
-        if inp == "k" or inp.name == "KEY_UP":
-            if pos == 0:
-                pos = maxPos - 1
-            else:
-                pos -= 1
 
-        if inp == "j" or inp.name == "KEY_DOWN":
-            if pos == maxPos - 1:
-                pos = 0
-            else:
-                pos += 1
-        if inp.name == "KEY_ENTER":
-            if pos == maxPos - 1:
-                inp = "q"
-                continue
-            if currmenu == machine:
-                currmenu = options
-            elif currmenu == options:
-                pass
+def main():
+    inp = ""
+    term = blessed.Terminal()
+    pos = 0
+
+    # TESTING VARS BEFORE REAL FUNCTIONS WORK
+    hwtest = [
+        "test 1",
+        "test 2",
+        "test 3",
+        "test 4",
+        "test 5",
+        "test 6",
+        "test 7",
+        "test 8",
+        "test 9",
+        "test 10",
+        "test 11",
+    ]
+
+    # ================
+
+    machine = ["SubjuGator", "NaviGator", "Exit"]
+    options = [
+        "Hardware Checklist",
+        "Software Checklist",
+        "Rosnode Command",
+        "Add Software Test",
+        "Remove Software Test",
+        "Back",
+        "Exit",
+    ]
+    currMenu = machine
+
+    hardwarePass = False
+    softwarePass = False
+
+    while inp != "q":
+        maxPos = len(currMenu)
+        # reprint line selected -> term.on_green"message"
+        print(term.home + term.clear + term.center("Preflight Checklist\n"))
+
+        for x in range(0, maxPos):
+            if x == pos:
+                print(term.blink + term.center(term.on_yellow(currMenu[x])))
             else:
-                pass
+                if currMenu == options:
+                    # if menu is options, set hardware/software to red or green depending on if it's completed
+                    if x == 0:
+                        #   pass
+                        print(
+                            term.center(term.on_green(currMenu[x]))
+                            if hardwarePass
+                            else term.center(term.on_red(currMenu[x])),
+                        )
+                    elif x == 1:
+                        print(
+                            term.center(term.on_green(currMenu[x]))
+                            if softwarePass
+                            else term.center(term.on_red(currMenu[x])),
+                        )
+                    else:
+                        print(term.center(currMenu[x]))
 
+                else:
+                    print(term.center(currMenu[x]))
+
+        with term.cbreak(), term.hidden_cursor():
+            inp = term.inkey()
+            if inp == "k" or inp.name == "KEY_UP":
+                if pos == 0:
+                    pos = maxPos - 1
+                else:
+                    pos -= 1
+
+            if inp == "j" or inp.name == "KEY_DOWN":
+                if pos == maxPos - 1:
+                    pos = 0
+                else:
+                    pos += 1
+            if inp.name == "KEY_ENTER" or inp == " ":
+                # if selection is exit
+                if pos == maxPos - 1:
+                    inp = "q"
+                    continue
+                # if in first screen go to second
+                if currMenu == machine:
+                    currMenu = options
+                    machine[pos]
+                # if in second menu
+                elif currMenu == options:
+                    # Hardware Check
+                    if pos == 0:
+                        hardware_test(term, hwtest)
+                    # Software Check
+                    if pos == 1:
+                        software_test(term)
+                    # ROSnode command
+                    if pos == 2:
+                        run_command(term)
+                    # add test to the checklist
+                    if pos == 3:
+                        write_test(term)
+                    # if back selected
+                    if pos == 5:
+                        currMenu = machine
+                        pos = 0
+                else:
+                    pass
 
-def hardware_test(terminal):
-    pass
 
+def hardware_test(term, hwtest):
+    inp = ""
+    while inp != "q":
+        print(term.clear + term.center("Hardware Tests\n"))
+        for x in hwtest:
+            print(term.on_red(x))
 
-def software_test(terminal):
+        with term.cbreak(), term.hidden_cursor():
+            inp = term.inkey()
+
+
+def software_test(term):
     # this should run ros and do all the init tests
     pass
 
 
-def run_command(terminal):
+def run_command(term):
+    pass
+
+
+def write_test(term):
     pass
+
+
+def delete_test(term):
+    pass
+
+
+if __name__ == "__main__":
+    main()