refactor and start adding actual functionality

pkg/menu/manual.go

@@ -0,0 +1,27 @@
+package menu
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/HikariKnight/ls-iommu/pkg/errorcheck"
+	"github.com/gookit/color"
+)
+
+func ManualInput(msg string, format string) []string {
+	// Print the title
+	color.Bold.Println(msg)
+
+	// Tell user the format to use
+	color.Bold.Printf("The format is %s\n", format)
+
+	// Get the user input
+	var input string
+	_, err := fmt.Scan(&input)
+	errorcheck.ErrorCheck(err)
+
+	input_list := strings.Split(input, ",")
+
+	// Return the input
+	return input_list
+}

pkg/menu/yesno.go

@@ -16,6 +16,20 @@ func YesNo(msg string) string {
 	return choice
 }
 
+func YesNoBack(msg string) string {
+	// Make the menu
+	menu := gocliselect.NewMenu(msg)
+	menu.AddItem("Yes", "y")
+	menu.AddItem("No", "n")
+	menu.AddItem("Go Back", "back")
+
+	// Display the menu
+	choice := menu.Display()
+
+	// Return the value selected
+	return choice
+}
+
 func YesNoEXT(msg string) string {
 	// Make the menu
 	menu := gocliselect.NewMenu(msg)
@@ -29,3 +43,18 @@ func YesNoEXT(msg string) string {
 	// Return the value selected
 	return choice
 }
+
+// Make a YesNo menu
+func YesNoManual(msg string) string {
+	// Make the menu
+	menu := gocliselect.NewMenu(msg)
+	menu.AddItem("Yes", "y")
+	menu.AddItem("No", "n")
+	menu.AddItem("Manual Entry", "manual")
+
+	// Display the menu
+	choice := menu.Display()
+
+	// Return the value selected
+	return choice
+}