50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bufio"
|
|||
|
|
"os"
|
|||
|
|
"strconv"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const defaultMenuAutoServeTimeout = 10 * time.Second
|
|||
|
|
|
|||
|
|
// menuAutoServeTimeout 交互菜单自动进入常驻的超时；可由 MENU_AUTO_SERVE_TIMEOUT_SEC 覆盖。
|
|||
|
|
func menuAutoServeTimeout() time.Duration {
|
|||
|
|
v := strings.TrimSpace(os.Getenv("MENU_AUTO_SERVE_TIMEOUT_SEC"))
|
|||
|
|
if v == "" {
|
|||
|
|
return defaultMenuAutoServeTimeout
|
|||
|
|
}
|
|||
|
|
sec, err := strconv.Atoi(v)
|
|||
|
|
if err != nil || sec <= 0 {
|
|||
|
|
return defaultMenuAutoServeTimeout
|
|||
|
|
}
|
|||
|
|
return time.Duration(sec) * time.Second
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// readLineTimeout 在 timeout 内等待一行；超时返回 timedOut=true。
|
|||
|
|
// 注意：Windows 控制台无法在超时后取消阻塞读，后台 goroutine 仍会等到用户按 Enter。
|
|||
|
|
func readLineTimeout(r *bufio.Reader, timeout time.Duration) (line string, timedOut bool) {
|
|||
|
|
ch := make(chan string, 1)
|
|||
|
|
go func() {
|
|||
|
|
s, _ := r.ReadString('\n')
|
|||
|
|
ch <- s
|
|||
|
|
}()
|
|||
|
|
select {
|
|||
|
|
case s := <-ch:
|
|||
|
|
return strings.TrimSpace(s), false
|
|||
|
|
case <-time.After(timeout):
|
|||
|
|
return "", true
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// serveConfirmYes 是否启动定时常驻（超时视为确认）。
|
|||
|
|
func serveConfirmYes(confirm string, timedOut bool) bool {
|
|||
|
|
if timedOut {
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
c := strings.TrimSpace(strings.ToLower(confirm))
|
|||
|
|
return c != "n" && c != "no"
|
|||
|
|
}
|