✨ Ask for user confirmation before restoring

cmd/restore/restore.go

@@ -3,6 +3,7 @@ package restore
 import (
 	"bufio"
 	"compress/gzip"
+	"github.com/clevyr/kubedb/internal/cli"
 	"github.com/clevyr/kubedb/internal/database/sqlformat"
 	"github.com/clevyr/kubedb/internal/kubernetes"
 	"github.com/clevyr/kubedb/internal/postgres"
@@ -91,6 +92,10 @@ func run(cmd *cobra.Command, args []string) (err error) {
 
 	log.Println("Restoring \"" + args[0] + "\" to \"" + postgresPod.Name + "\"")
 
+	if err = cli.Confirm(os.Stdin, false); err != nil {
+		return err
+	}
+
 	ch := make(chan error)
 	go func() {
 		if clean {

internal/cli/input.go

@@ -0,0 +1,40 @@
+package cli
+
+import (
+	"bufio"
+	"errors"
+	"fmt"
+	"io"
+	"strings"
+)
+
+var UserDeclinedErr = errors.New("user declined")
+
+func Confirm(r io.Reader, defaultVal bool) (err error) {
+	fmt.Print("Continue? ")
+	if defaultVal {
+		fmt.Print("[Y/n]: ")
+	} else {
+		fmt.Print("[y/N]: ")
+	}
+
+	buf := bufio.NewReader(r)
+	var response string
+	response, err = buf.ReadString('\n')
+	if err != nil {
+		return err
+	}
+
+	response = strings.ToLower(strings.TrimSpace(response))
+	switch response {
+	case "yes", "y":
+		return nil
+	case "no", "n":
+		return UserDeclinedErr
+	}
+	if defaultVal {
+		return nil
+	} else {
+		return UserDeclinedErr
+	}
+}

internal/cli/input_test.go

@@ -0,0 +1,39 @@
+package cli
+
+import (
+	"os"
+	"strings"
+	"testing"
+)
+
+func testConfirm(response string, defaultVal bool) error {
+	temp := os.Stdout
+	os.Stdout = nil
+	result := Confirm(strings.NewReader(response + "\n"), defaultVal)
+	os.Stdout = temp
+	return result
+}
+
+func TestConfirm(t *testing.T) {
+	type confirmTestCase struct {
+		response string
+		defaultVal bool
+		error error
+	}
+
+	testCases := []confirmTestCase{
+		{response: "yes", defaultVal: true, error: nil},
+		{response: "yes", defaultVal: false, error: nil},
+		{response: "no", defaultVal: true, error: UserDeclinedErr},
+		{response: "no", defaultVal: false, error: UserDeclinedErr},
+		{response: "", defaultVal: true, error: nil},
+		{response: "", defaultVal: false, error: UserDeclinedErr},
+	}
+
+	for key, testCase := range testCases {
+		err := testConfirm(testCase.response, testCase.defaultVal)
+		if err != testCase.error {
+			t.Errorf("case %d: got %v; expected %v", key, err, testCase.error)
+		}
+	}
+}