diff --git a/input/boolean/boolean.go b/input/boolean/boolean.go index 0d43ced..7b31bfe 100644 --- a/input/boolean/boolean.go +++ b/input/boolean/boolean.go @@ -2,20 +2,23 @@ package boolean import "github.com/rivo/tview" -func Choose(title string) bool { +func Choose(text string) bool { choice := false app := tview.NewApplication() - list := tview.NewList(). - AddItem("Yes", "", 'y', func() { - choice = true - app.Stop() - }). - AddItem("No", "", 'n', func() { - choice = false + modal := tview.NewModal() + if text != "" { + modal.SetText(text) + } + + modal.AddButtons([]string{"Yes", "No"}). + SetDoneFunc(func(buttonIndex int, buttonLabel string) { + if buttonLabel == "Yes" { + choice = true + } app.Stop() }) - list.Box.SetBorder(true).SetTitle(title) - if err := app.SetRoot(list, false).EnableMouse(true).Run(); err != nil { + + if err := app.SetRoot(modal, false).EnableMouse(true).Run(); err != nil { panic(err) }