From 4e282232841146ad4eb9cc15ed6fdcf1c9a8ac82 Mon Sep 17 00:00:00 2001
From: ItsDrike <itsdrikeofficial@gmail.com>
Date: Thu, 22 Oct 2020 19:31:59 +0200
Subject: [PATCH] Add input questions

---
 src/util/user.py | 43 ++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 42 insertions(+), 1 deletion(-)

diff --git a/src/util/user.py b/src/util/user.py
index 5782b56..4188ac4 100644
--- a/src/util/user.py
+++ b/src/util/user.py
@@ -9,7 +9,7 @@ class Print:
         text = [f"{Color.GREEN} // {question}{Color.RESET}"]
         if options:
             for option in options:
-                text.append(f"{Color.GREEN}    # {options}{Color.RESET}")
+                text.append(f"{Color.GREEN}    # {option}{Color.RESET}")
         print("\n".join(text))
 
     def action(action: str) -> None:
@@ -31,3 +31,44 @@ class Print:
     def warning(text: str) -> None:
         """Print syntax for warnings"""
         print(f"{Color.YELLOW} ** {text}{Color.RESET}")
+
+
+class Input:
+    def yes_no(question: str) -> bool:
+        """Ask a yes/no `question`, return True(y)/False(n)"""
+        while True:
+            Print.question(question)
+            ans = input("   Y/N: ").lower()
+            if ans == "y" or ans == "":
+                return True
+            elif ans == "n":
+                return False
+            else:
+                Print.err("Invalid option (Y/N)")
+
+    def multiple(question: str, options: t.List[str], abort: bool = False) -> int:
+        """
+        Ask `question` with multiple `options`
+        Return index from options list as chosen option
+
+        You can also specify `abort` which will add option `-1` for No/Cancel
+        """
+        num_options = [f"{index + 1}: {option}" for index, option in enumerate(options)]
+        if abort:
+            num_options.append("-1: No/Cancel")
+        while True:
+            Print.question(question, num_options)
+            try:
+                ans = int(input(f"   Choice (1-{len(options)}{'/-1' if abort else None}): "))
+            except TypeError:
+                Print.err(f"Invalid option, must be a number between 1-{len(options)}")
+                continue
+
+            if ans in range(len(options) + 1):
+                return ans - 1
+            else:
+                Print.err(f"Invalid option, outside of the number range 1-{len(options)}")
+
+    def text(text: str) -> str:
+        Print.question(text)
+        return input("  >>")