From e8bac3c74145acea8d55b671d4f1f30163e3abe5 Mon Sep 17 00:00:00 2001
From: HikariKnight <2557889+HikariKnight@users.noreply.github.com>
Date: Fri, 27 Oct 2023 08:18:52 +0200
Subject: [PATCH] add a skip option

---
 pkg/menu/genmenu.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/pkg/menu/genmenu.go b/pkg/menu/genmenu.go
index 52f03ce..f1e917a 100644
--- a/pkg/menu/genmenu.go
+++ b/pkg/menu/genmenu.go
@@ -7,7 +7,7 @@ import (
 	"github.com/nexidian/gocliselect"
 )
 
-func GenIOMMUMenu(msg string, choices []string) string {
+func GenIOMMUMenu(msg string, choices []string, none_option ...int) string {
 	// Make a regex to get the iommu group
 	iommu_group_regex := regexp.MustCompile(`(\d{1,3})`)
 
@@ -23,6 +23,12 @@ func GenIOMMUMenu(msg string, choices []string) string {
 		menu.AddItem(choice, iommuGroup)
 	}
 
+	// If none_option is higher than 0
+	if len(none_option) > 0 {
+		// Add a skip option
+		menu.AddItem(color.Bold.Sprint("Skip/None"), "skip")
+	}
+
 	// Add a go back option
 	menu.AddItem(color.Bold.Sprint("Go Back"), "back")