stario/io.go

package stario

import (
	"bufio"
	"fmt"
	"golang.org/x/crypto/ssh/terminal"
	"os"
	"runtime"
	"strconv"
	"strings"
)

type InputMsg struct {
	msg             string
	err             error
	skipSliceSigErr bool
}

func Passwd(hint string, defaultVal string) InputMsg {
	return passwd(hint, defaultVal, "", false)
}

func PasswdWithMask(hint string, defaultVal string, mask string) InputMsg {
	return passwd(hint, defaultVal, mask, false)
}

func PasswdResponseSignal(hint string, defaultVal string) InputMsg {
	return passwd(hint, defaultVal, "", true)
}

func PasswdResponseSignalWithMask(hint string, defaultVal string, mask string) InputMsg {
	return passwd(hint, defaultVal, mask, true)
}

func MessageBoxRaw(hint string, defaultVal string) InputMsg {
	return messageBox(hint, defaultVal)
}

func messageBox(hint string, defaultVal string) InputMsg {
	var ioBuf []rune
	if hint != "" {
		fmt.Print(hint)
	}
	if strings.Index(hint, "\n") >= 0 {
		hint = strings.TrimSpace(hint[strings.LastIndex(hint, "\n"):])
	}
	fd := int(os.Stdin.Fd())
	state, err := terminal.MakeRaw(fd)
	if err != nil {
		return InputMsg{msg: "", err: err}
	}
	defer fmt.Println()
	defer terminal.Restore(fd, state)
	inputReader := bufio.NewReader(os.Stdin)
	for {
		b, _, err := inputReader.ReadRune()
		if err != nil {
			return InputMsg{msg: "", err: err}
		}
		if b == 0x0d {
			strValue := strings.TrimSpace(string(ioBuf))
			if len(strValue) == 0 {
				strValue = defaultVal
			}
			return InputMsg{msg: strValue, err: err}
		}
		if b == 0x08 || b == 0x7F {
			if len(ioBuf) > 0 {
				ioBuf = ioBuf[:len(ioBuf)-1]
			}
			fmt.Print("\r")
			for i := 0; i < len(ioBuf)+2+len(hint); i++ {
				fmt.Print(" ")
			}
		} else {
			ioBuf = append(ioBuf, b)
		}
		fmt.Print("\r")
		if hint != "" {
			fmt.Print(hint)
		}
		fmt.Print(string(ioBuf))
	}
}

func isSiganl(s rune) bool {
	switch s {
	case 0x03, 0x1a, 0x1c:
		return true
	default:
		return false
	}
}

func passwd(hint string, defaultVal string, mask string, handleSignal bool) InputMsg {
	var ioBuf []rune
	if hint != "" {
		fmt.Print(hint)
	}
	if strings.Index(hint, "\n") >= 0 {
		hint = strings.TrimSpace(hint[strings.LastIndex(hint, "\n"):])
	}
	fd := int(os.Stdin.Fd())
	state, err := terminal.MakeRaw(fd)
	if err != nil {
		return InputMsg{msg: "", err: err}
	}
	defer fmt.Println()
	defer terminal.Restore(fd, state)
	inputReader := bufio.NewReader(os.Stdin)
	for {
		b, _, err := inputReader.ReadRune()
		if err != nil {
			return InputMsg{msg: "", err: err}
		}
		if handleSignal && isSiganl(b) {
			if runtime.GOOS != "windows" {
				terminal.Restore(fd, state)
			}
			if err := signal(b); err != nil {
				return InputMsg{
					msg: "",
					err: err,
				}
			}
		}
		if b == 0x0d {
			strValue := strings.TrimSpace(string(ioBuf))
			if len(strValue) == 0 {
				strValue = defaultVal
			}
			return InputMsg{msg: strValue, err: err}
		}
		if b == 0x08 || b == 0x7F {
			if len(ioBuf) > 0 {
				ioBuf = ioBuf[:len(ioBuf)-1]
			}
			fmt.Print("\r")
			for i := 0; i < len(ioBuf)+2+len(hint); i++ {
				fmt.Print(" ")
			}
		} else {
			ioBuf = append(ioBuf, b)
		}
		fmt.Print("\r")
		if hint != "" {
			fmt.Print(hint)
		}
		for i := 0; i < len(ioBuf); i++ {
			fmt.Print(mask)
		}
	}
}

func MessageBox(hint string, defaultVal string) InputMsg {
	if hint != "" {
		fmt.Print(hint)
	}
	inputReader := bufio.NewReader(os.Stdin)
	str, err := inputReader.ReadString('\n')
	if err != nil {
		return InputMsg{msg: str, err: err}
	}
	str = strings.TrimSpace(str)
	if len(str) == 0 {
		str = defaultVal
	}
	return InputMsg{msg: str, err: err}
}

func (im *InputMsg) IgnoreSliceParseError(i bool) *InputMsg {
	im.skipSliceSigErr = i
	return im
}

func (im InputMsg) String() (string, error) {
	if im.err != nil {
		return "", im.err
	}
	return im.msg, nil
}

func (im InputMsg) MustString() string {
	res, _ := im.String()
	return res
}

func (im InputMsg) SliceString(sep string) ([]string, error) {
	if im.err != nil {
		return nil, im.err
	}
	if len(strings.TrimSpace(im.msg)) == 0 {
		return []string{}, nil
	}
	return strings.Split(im.msg, sep), nil
}

func (im InputMsg) MustSliceString(sep string) []string {
	res, _ := im.SliceString(sep)
	return res
}

func (im InputMsg) sliceFn(sep string, fn func(string) (interface{}, error)) ([]interface{}, error) {
	var res []interface{}
	data, err := im.SliceString(sep)
	if err != nil {
		return res, err
	}
	for _, v := range data {
		code, err := fn(v)
		if err != nil && !im.skipSliceSigErr {
			return nil, err
		} else if err == nil {
			res = append(res, code)
		}
	}
	return res, nil
}

func (im InputMsg) Int() (int, error) {
	if im.err != nil {
		return 0, im.err
	}
	return strconv.Atoi(im.msg)
}

func (im InputMsg) SliceInt(sep string) ([]int, error) {
	data, err := im.sliceFn(sep, func(v string) (interface{}, error) {
		return strconv.Atoi(v)
	})
	var res []int
	for _, v := range data {
		res = append(res, v.(int))
	}
	return res, err
}

func (im InputMsg) MustSliceInt(sep string) []int {
	res, _ := im.SliceInt(sep)
	return res
}

func (im InputMsg) MustInt() int {
	res, _ := im.Int()
	return res
}

func (im InputMsg) Int64() (int64, error) {
	if im.err != nil {
		return 0, im.err
	}
	return strconv.ParseInt(im.msg, 10, 64)
}

func (im InputMsg) MustInt64() int64 {
	res, _ := im.Int64()
	return res
}

func (im InputMsg) SliceInt64(sep string) ([]int64, error) {
	data, err := im.sliceFn(sep, func(v string) (interface{}, error) {
		return strconv.ParseInt(v, 10, 64)
	})
	var res []int64
	for _, v := range data {
		res = append(res, v.(int64))
	}
	return res, err
}

func (im InputMsg) MustSliceInt64(sep string) []int64 {
	res, _ := im.SliceInt64(sep)
	return res
}

func (im InputMsg) Uint64() (uint64, error) {
	if im.err != nil {
		return 0, im.err
	}
	return strconv.ParseUint(im.msg, 10, 64)
}

func (im InputMsg) MustUint64() uint64 {
	res, _ := im.Uint64()
	return res
}
func (im InputMsg) SliceUint64(sep string) ([]uint64, error) {
	data, err := im.sliceFn(sep, func(v string) (interface{}, error) {
		return strconv.ParseUint(v, 10, 64)
	})
	var res []uint64
	for _, v := range data {
		res = append(res, v.(uint64))
	}
	return res, err
}

func (im InputMsg) MustSliceUint64(sep string) []uint64 {
	res, _ := im.SliceUint64(sep)
	return res
}

func (im InputMsg) Bool() (bool, error) {
	if im.err != nil {
		return false, im.err
	}
	return strconv.ParseBool(im.msg)
}

func (im InputMsg) MustBool() bool {
	res, _ := im.Bool()
	return res
}

func (im InputMsg) SliceBool(sep string) ([]bool, error) {
	data, err := im.sliceFn(sep, func(v string) (interface{}, error) {
		return strconv.ParseBool(v)
	})
	var res []bool
	for _, v := range data {
		res = append(res, v.(bool))
	}
	return res, err
}

func (im InputMsg) MustSliceBool(sep string) []bool {
	res, _ := im.SliceBool(sep)
	return res
}

func (im InputMsg) Float64() (float64, error) {
	if im.err != nil {
		return 0, im.err
	}
	return strconv.ParseFloat(im.msg, 64)
}

func (im InputMsg) MustFloat64() float64 {
	res, _ := im.Float64()
	return res
}

func (im InputMsg) SliceFloat64(sep string) ([]float64, error) {
	data, err := im.sliceFn(sep, func(v string) (interface{}, error) {
		return strconv.ParseFloat(v, 64)
	})
	var res []float64
	for _, v := range data {
		res = append(res, v.(float64))
	}
	return res, err
}

func (im InputMsg) MustSliceFloat64(sep string) []float64 {
	res, _ := im.SliceFloat64(sep)
	return res
}

func (im InputMsg) Float32() (float32, error) {
	if im.err != nil {
		return 0, im.err
	}
	res, err := strconv.ParseFloat(im.msg, 32)
	return float32(res), err
}

func (im InputMsg) MustFloat32() float32 {
	res, _ := im.Float32()
	return res
}

func (im InputMsg) SliceFloat32(sep string) ([]float32, error) {
	data, err := im.sliceFn(sep, func(v string) (interface{}, error) {
		return strconv.ParseFloat(v, 32)
	})
	var res []float32
	for _, v := range data {
		res = append(res, v.(float32))
	}
	return res, err
}

func (im InputMsg) MustSliceFloat32(sep string) []float32 {
	res, _ := im.SliceFloat32(sep)
	return res
}

func YesNo(hint string, defaults bool) bool {
	for {
		res := strings.ToUpper(MessageBox(hint, "").MustString())
		if res == "" {
			return defaults
		}
		res = res[0:1]
		if res == "Y" {
			return true
		} else if res == "N" {
			return false
		}
	}
}

func StopUntil(hint string, trigger string, repeat bool) error {
	pressLen := len([]rune(trigger))
	if trigger == "" {
		pressLen = 1
	}
	fd := int(os.Stdin.Fd())
	if hint != "" {
		fmt.Print(hint)
	}
	state, err := terminal.MakeRaw(fd)
	if err != nil {
		return err
	}
	defer terminal.Restore(fd, state)
	inputReader := bufio.NewReader(os.Stdin)
	//ioBuf := make([]byte, pressLen)
	i := 0
	for {
		b, _, err := inputReader.ReadRune()
		if err != nil {
			return err
		}
		if trigger == "" {
			break
		}
		if b == []rune(trigger)[i] {
			i++
			if i == pressLen {
				break
			}
			continue
		}
		i = 0
		if hint != "" && repeat {
			fmt.Print("\r\n")
			fmt.Print(hint)
		}
	}
	return nil
}