From a6d0ab5c69964d4d4016817dcde29210d262217d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lilian=20J=C3=B3nsd=C3=B3ttir?=
 <lilian.jonsdottir@gmail.com>
Date: Fri, 28 Jun 2024 11:26:30 -0700
Subject: [PATCH] move promts to new package

---
 internal/prompt/prompt.go | 40 +++++++++++++++++++++++++++++++++++++++
 main.go                   | 35 +++++-----------------------------
 2 files changed, 45 insertions(+), 30 deletions(-)
 create mode 100644 internal/prompt/prompt.go

diff --git a/internal/prompt/prompt.go b/internal/prompt/prompt.go
new file mode 100644
index 0000000..05423b0
--- /dev/null
+++ b/internal/prompt/prompt.go
@@ -0,0 +1,40 @@
+package prompt
+
+import (
+	"bytes"
+	"fmt"
+	"os"
+
+	"golang.org/x/term"
+)
+
+// TODO: use charm stuff for this
+
+func YesNo(prompt string) bool {
+	return AskRune(prompt, "y/n") == 'y'
+}
+
+func AskRune(prompt, options string) byte {
+	// TODO: handle errors better
+	// switch stdin into 'raw' mode
+	oldState, err := term.MakeRaw(int(os.Stdin.Fd()))
+	if err != nil {
+		panic(err)
+	}
+	defer func() {
+		if err := term.Restore(int(os.Stdin.Fd()), oldState); err != nil {
+			panic(err)
+		}
+	}()
+
+	fmt.Printf("%s [%s]: ", prompt, options)
+
+	// read one byte from stdin
+	b := make([]byte, 1)
+	_, err = os.Stdin.Read(b)
+	if err != nil {
+		return 0
+	}
+
+	return bytes.ToLower(b)[0]
+}
diff --git a/main.go b/main.go
index c8a6ed4..ecfd185 100644
--- a/main.go
+++ b/main.go
@@ -1,7 +1,6 @@
 package main
 
 import (
-	"bytes"
 	"fmt"
 	"io/fs"
 	"os"
@@ -12,6 +11,7 @@ import (
 	"git.burning.moe/celediel/gt/internal/files"
 	"git.burning.moe/celediel/gt/internal/filter"
 	"git.burning.moe/celediel/gt/internal/modes"
+	"git.burning.moe/celediel/gt/internal/prompt"
 	"git.burning.moe/celediel/gt/internal/tables"
 	"git.burning.moe/celediel/gt/internal/trash"
 
@@ -459,7 +459,7 @@ func interactiveMode() error {
 }
 
 func confirmRestore(is trash.Infos) error {
-	if !askconfirm || confirm(fmt.Sprintf("restore %d selected files?", len(is))) {
+	if !askconfirm || prompt.YesNo(fmt.Sprintf("restore %d selected files?", len(is))) {
 		log.Info("doing the thing")
 		restored, err := trash.Restore(is)
 		if err != nil {
@@ -473,8 +473,8 @@ func confirmRestore(is trash.Infos) error {
 }
 
 func confirmClean(is trash.Infos) error {
-	if confirm(fmt.Sprintf("remove %d selected files permanently from the trash?", len(is))) &&
-		(!askconfirm || confirm(fmt.Sprintf("really remove all these %d selected files permanently from the trash forever??", len(is)))) {
+	if prompt.YesNo(fmt.Sprintf("remove %d selected files permanently from the trash?", len(is))) &&
+		(!askconfirm || prompt.YesNo(fmt.Sprintf("really remove all these %d selected files permanently from the trash forever??", len(is)))) {
 		log.Info("gonna remove some files forever")
 		removed, err := trash.Remove(is)
 		if err != nil {
@@ -488,7 +488,7 @@ func confirmClean(is trash.Infos) error {
 }
 
 func confirmTrash(fs files.Files) error {
-	if !askconfirm || confirm(fmt.Sprintf("trash %d selected files?", len(fs))) {
+	if !askconfirm || prompt.YesNo(fmt.Sprintf("trash %d selected files?", len(fs))) {
 		tfs := make([]string, 0, len(fs))
 		for _, file := range fs {
 			log.Debugf("gonna trash %s", file.Filename())
@@ -506,28 +506,3 @@ func confirmTrash(fs files.Files) error {
 	}
 	return nil
 }
-
-func confirm(prompt string) bool {
-	// TODO: handle errors better
-	// switch stdin into 'raw' mode
-	oldState, err := term.MakeRaw(int(os.Stdin.Fd()))
-	if err != nil {
-		panic(err)
-	}
-	defer func() {
-		if err := term.Restore(int(os.Stdin.Fd()), oldState); err != nil {
-			panic(err)
-		}
-	}()
-
-	fmt.Printf("%s [%s/%s]: ", prompt, string(yes), string(no))
-
-	// read one byte from stdin
-	b := make([]byte, 1)
-	_, err = os.Stdin.Read(b)
-	if err != nil {
-		return false
-	}
-
-	return bytes.ToLower(b)[0] == 'y'
-}