gitee.com/KyleChenSource/lib-robot@v1.0.2/robottest/common/interaction/stdinput.go

package interaction

import (
	"bufio"
	"fmt"
	"os"
	"sync"
)

var (
	input *StdInput
)

const (
	ctxkey_input_flag = "input_flag"
)

type StdInput struct {
	reader_ *bufio.Reader
	cmds_   chan string // 输入指令， 用于允许输入开始，结束输入，清理之类的指令
	lines_  chan string // 终端输入

	input_signal_ chan any
	ctx_          sync.Map
}

func Input() *StdInput {
	if input == nil {
		// info, _ := os.Stdin.Stat()
		// fmt.Printf("std:%T %v", info, info)
		input = &StdInput{
			reader_:       bufio.NewReader(os.Stdin),
			cmds_:         make(chan string, 1024),
			lines_:        make(chan string),
			input_signal_: make(chan any),
		}

		go co_input(input)
	}

	return input
}

func (this *StdInput) Want() {
	this.clear()

	this.input_signal_ <- struct{}{}
}

func (this *StdInput) Line() <-chan string {
	return this.lines_
}

func (this *StdInput) clear() {
	for {
		select {
		case <-this.lines_:
			// 丢弃之前的所有输入

		default:
			return
		}
	}
}

func co_input(in *StdInput) {
	line := make([]byte, 0)
	for {
		<-in.input_signal_

		fmt.Print("$请输入指令:")
		line = line[:0]
		bLine := true
		for bLine {
			r, _, e := in.reader_.ReadRune()
			if e != nil {
				fmt.Printf("输入终端错误:%s，进行恢复尝试", e.Error())
				in.reader_ = bufio.NewReader(os.Stdin)
				in.lines_ <- ""
				break
			}

			switch r {
			case '\b':
				fmt.Print('\b')
				if len(line) > 0 {
					line = line[0 : len(line)-1]
				}
			case '\n':
				bLine = false
				break
			default:
				line = append(line, []byte(string(r))...)
			}
		}

		in.lines_ <- string(line)
	}
}