choice: Menu add showQuit option

Author: sunshineplan

choice/choice.go

@@ -25,7 +25,7 @@ func choiceStr(choice any) string {
 }
 
 // Menu function is used to generate a string representation of the menu, associating choices with numbers.
-func Menu[E any](choices []E) string {
+func Menu[E any](choices []E, showQuit bool) string {
 	if len(choices) == 0 {
 		return ""
 	}
@@ -38,7 +38,9 @@ func Menu[E any](choices []E) string {
 	for i, choice := range choices {
 		fmt.Fprintf(&b, "%s. %s\n", fmt.Sprintf(option, i+1), choiceStr(choice))
 	}
-	fmt.Fprintf(&b, "%s. Quit\n", fmt.Sprintf(fmt.Sprintf("%%%ds", digit), "q"))
+	if showQuit {
+		fmt.Fprintf(&b, "%s. Quit\n", fmt.Sprintf(fmt.Sprintf("%%%ds", digit), "q"))
+	}
 	return b.String()
 }
 

choice/choice_test.go

@@ -32,7 +32,7 @@ func TestMenu(t *testing.T) {
 10. jj
  q. Quit
 `
-	if s := Menu(choices); s != expect {
+	if s := Menu(choices, true); s != expect {
 		t.Errorf("expected %q; got %q", expect, s)
 	}
 }