confirm.go RawHistoryFolder

1package confirm
2
3import (
4	"bufio"
5	"context"
6	"errors"
7	"fmt"
8	"io"
9	"strings"
10)
11
12// Confirmer asks a yes/no question on a pair of streams.
13type Confirmer struct {
14	in  io.Reader
15	out io.Writer
16}
17
18// NewConfirmer builds a confirmation prompt. Only y/yes answers confirm; empty,
19// n/no, and unknown answers all decline.
20func NewConfirmer(in io.Reader, out io.Writer) Confirmer {
21	return Confirmer{in: in, out: out}
22}
23
24// ConfirmContext writes question with a [y/N] default and reads one answer
25// line, releasing the caller with the cancellation error if ctx ends first.
26// The reader goroutine may stay blocked, so use this only where the process
27// exits (or detaches from the console) afterward rather than continuing to
28// interact on the same stream.
29func (c Confirmer) ConfirmContext(ctx context.Context, question string) (bool, error) {
30	if ctx == nil {
31		return c.Confirm(question)
32	}
33	type outcome struct {
34		ok  bool
35		err error
36	}
37	result := make(chan outcome, 1)
38	go func() {
39		ok, err := c.Confirm(question)
40		result <- outcome{ok: ok, err: err}
41	}()
42	select {
43	case <-ctx.Done():
44		return false, ctx.Err()
45	case reply := <-result:
46		return reply.ok, reply.err
47	}
48}
49
50// Confirm writes question with a [y/N] default and reads one answer line.
51func (c Confirmer) Confirm(question string) (bool, error) {
52	if c.out != nil {
53		if _, err := fmt.Fprintf(c.out, "%s [y/N] ", question); err != nil {
54			return false, err
55		}
56	}
57	answer, err := bufio.NewReader(c.in).ReadString('\n')
58	if err != nil && !errors.Is(err, io.EOF) {
59		return false, err
60	}
61	switch strings.ToLower(strings.TrimSpace(answer)) {
62	case "y", "yes":
63		return true, nil
64	default:
65		return false, nil
66	}
67}