83 lines
1.8 KiB
Go
83 lines
1.8 KiB
Go
package confirmation
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
type InputData struct {
|
|
Notice string
|
|
Question string
|
|
}
|
|
|
|
func New(data InputData) (*bool, error) {
|
|
|
|
if data.Notice != "" {
|
|
fmt.Printf("\033[1m\033[38;5;247m[\033[38;5;214m!\033[38;5;247m]\033[22m \033[3m%s\033[0m\n", data.Notice)
|
|
}
|
|
|
|
if data.Question != "" {
|
|
fmt.Printf("\033[1m\033[38;5;247m[\033[38;5;214m?\033[38;5;247m]\033[0m %s (\033[38;5;34my\033[0m/\033[38;5;167mn\033[0m) \033[38;5;247m>>\033[3m\033[38;5;214m\n", data.Question)
|
|
}
|
|
|
|
var input string
|
|
var chosen bool
|
|
|
|
inputLoop:
|
|
for {
|
|
if _, err := fmt.Scanln(&input); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
width, _, err := term.GetSize(0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
switch strings.ToLower(input) {
|
|
case "y", "yes":
|
|
chosen = true
|
|
break inputLoop
|
|
case "n", "no":
|
|
chosen = false
|
|
break inputLoop
|
|
default:
|
|
var lineNum int
|
|
|
|
if width == 0 {
|
|
if data.Notice != "" {
|
|
lineNum++
|
|
}
|
|
lineNum += 2
|
|
} else {
|
|
if data.Notice == "" {
|
|
lineNum = ((len(data.Question) + 5 + width - 1) / width) + 1
|
|
} else {
|
|
lineNum = ((len(data.Notice) + 5 + width) / width) + ((len(data.Question) + 5 + width - 1) / width) + 1
|
|
}
|
|
|
|
}
|
|
|
|
for i := 0; i < lineNum; i++ {
|
|
fmt.Printf("\033[A\033[K\033[0G")
|
|
}
|
|
fmt.Printf("\033[1m\033[38;5;247m[\033[38;5;167m!\033[38;5;247m]\033[0m Invalid input, please try again!\n")
|
|
|
|
if data.Notice != "" {
|
|
fmt.Printf("\033[1m\033[38;5;247m[\033[38;5;214m!\033[38;5;247m]\033[22m \033[3m%s\033[0m\n", data.Notice)
|
|
}
|
|
|
|
if data.Question != "" {
|
|
fmt.Printf("\033[1m\033[38;5;247m[\033[38;5;214m?\033[38;5;247m]\033[0m %s (\033[38;5;34my\033[0m/\033[38;5;167mn\033[0m) \033[38;5;247m>>\033[3m\033[38;5;214m\n", data.Question)
|
|
}
|
|
continue inputLoop
|
|
}
|
|
}
|
|
|
|
fmt.Println("\033[0m")
|
|
|
|
return &chosen, nil
|
|
|
|
}
|