mirror of
https://github.com/la5nta/pat
synced 2026-08-13 17:46:54 -04:00
129 lines
2.9 KiB
Go
129 lines
2.9 KiB
Go
package cli
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"unicode"
|
|
|
|
"github.com/la5nta/pat/app"
|
|
"github.com/la5nta/pat/internal/debug"
|
|
"github.com/la5nta/wl2k-go/fbb"
|
|
"github.com/la5nta/wl2k-go/mailbox"
|
|
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
func shiftArgs(s []string) (string, []string) {
|
|
if len(s) == 0 {
|
|
return "", nil
|
|
}
|
|
return strings.TrimSpace(s[0]), s[1:]
|
|
}
|
|
|
|
var stdin *bufio.Reader
|
|
|
|
func readLine() string {
|
|
if stdin == nil {
|
|
stdin = bufio.NewReader(os.Stdin)
|
|
}
|
|
|
|
str, _ := stdin.ReadString('\n')
|
|
return strings.TrimSpace(str)
|
|
}
|
|
|
|
func isTerminal(f interface{ Fd() uintptr }) bool { return term.IsTerminal(int(f.Fd())) }
|
|
|
|
func prompt2(w io.Writer, question, defaultValue string, options ...string) string {
|
|
var suffix string
|
|
if len(options) > 0 {
|
|
// Ensure default is included in options if not already present
|
|
allOptions := options
|
|
defaultFound := false
|
|
for _, opt := range options {
|
|
if strings.EqualFold(opt, defaultValue) {
|
|
defaultFound = true
|
|
break
|
|
}
|
|
}
|
|
if !defaultFound && defaultValue != "" {
|
|
allOptions = append([]string{defaultValue}, options...)
|
|
}
|
|
|
|
// Use standard (Y/n) format where uppercase indicates default
|
|
formatted := make([]string, len(allOptions))
|
|
for i, opt := range allOptions {
|
|
if strings.EqualFold(opt, defaultValue) {
|
|
formatted[i] = strings.ToUpper(opt)
|
|
} else {
|
|
formatted[i] = strings.ToLower(opt)
|
|
}
|
|
}
|
|
suffix = fmt.Sprintf(" (%s)", strings.Join(formatted, "/"))
|
|
} else if defaultValue != "" {
|
|
// Free-text field with default value
|
|
suffix = fmt.Sprintf(" [%s]", defaultValue)
|
|
}
|
|
|
|
fmt.Fprintf(w, "%s%s: ", question, suffix)
|
|
response := readLine()
|
|
if response == "" {
|
|
return defaultValue
|
|
}
|
|
return response
|
|
}
|
|
|
|
func prompt(question, defaultValue string, options ...string) string {
|
|
return prompt2(os.Stdout, question, defaultValue, options...)
|
|
}
|
|
|
|
func promptPassword(prompt string) (string, error) {
|
|
defer fmt.Fprintln(os.Stdout)
|
|
fmt.Fprint(os.Stdout, prompt)
|
|
passwd, err := term.ReadPassword(int(os.Stdin.Fd()))
|
|
return string(passwd), err
|
|
}
|
|
|
|
func SplitFunc(c rune) bool {
|
|
return unicode.IsSpace(c) || c == ',' || c == ';'
|
|
}
|
|
|
|
func exitOnContextCancellation(ctx context.Context) (cancel func()) {
|
|
done := make(chan struct{}, 1)
|
|
go func() {
|
|
select {
|
|
case <-ctx.Done():
|
|
fmt.Println()
|
|
os.Exit(1)
|
|
case <-done:
|
|
}
|
|
}()
|
|
return func() {
|
|
select {
|
|
case done <- struct{}{}:
|
|
default:
|
|
}
|
|
}
|
|
}
|
|
|
|
func openMessage(a *app.App, path string) (*fbb.Message, error) {
|
|
// Search if only MID is specified.
|
|
if filepath.Dir(path) == "." && filepath.Ext(path) == "" {
|
|
debug.Printf("openMessage(%q): Searching...", path)
|
|
path += mailbox.Ext
|
|
fs.WalkDir(os.DirFS(a.Mailbox().MBoxPath), ".", func(p string, d fs.DirEntry, err error) error {
|
|
if d.Name() != path {
|
|
return nil
|
|
}
|
|
debug.Printf("openMessage(%q): Found %q", d.Name(), p)
|
|
path = filepath.Join(a.Mailbox().MBoxPath, p)
|
|
return io.EOF
|
|
})
|
|
}
|
|
return mailbox.OpenMessage(path)
|
|
}
|