refactor

1 files changed, 34 insertions, 0 deletions

diff --git a/internal/prompt/prompt.go b/internal/prompt/prompt.go
new file mode 100644
index 0000000..0afe98e
--- /dev/null
+++ b/internal/prompt/prompt.go
@@ -0,0 +1,34 @@
+package prompt
+
+import (
+	"bufio"
+	"errors"
+	"fmt"
+	"os"
+	"strings"
+)
+
+var ErrAborted = errors.New("aborted")
+
+func Yes(question string) bool {
+	reader := bufio.NewReader(os.Stdin)
+
+	for {
+		fmt.Print(question, " (y/n): ")
+		input, err := reader.ReadString('\n')
+		if err != nil {
+			fmt.Println("Error reading input:", err)
+			continue
+		}
+
+		input = strings.TrimSpace(input)
+		switch strings.ToLower(input) {
+		case "y", "yes":
+			return true
+		case "n", "no":
+			return false
+		default:
+			fmt.Println("Please enter 'y' or 'n'.")
+		}
+	}
+}