313 lines
8.9 KiB
Go
313 lines
8.9 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/charmbracelet/bubbles/key"
|
|
inf "github.com/fzdwx/infinite"
|
|
"github.com/fzdwx/infinite/color"
|
|
"github.com/fzdwx/infinite/components"
|
|
"github.com/fzdwx/infinite/components/input/text"
|
|
"github.com/fzdwx/infinite/components/selection/confirm"
|
|
"github.com/fzdwx/infinite/components/selection/singleselect"
|
|
"github.com/fzdwx/infinite/style"
|
|
"github.com/fzdwx/infinite/theme"
|
|
"github.com/spf13/pflag"
|
|
"go.bug.st/serial"
|
|
"log"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type ptrVal struct {
|
|
*string
|
|
sl *[]string
|
|
*int
|
|
il *[]int
|
|
*bool
|
|
*float64
|
|
*float32
|
|
ext *string
|
|
}
|
|
type Val struct {
|
|
string
|
|
int
|
|
bool
|
|
float64
|
|
float32
|
|
extdef string
|
|
}
|
|
type Flag struct {
|
|
v ptrVal
|
|
sStr string
|
|
lStr string
|
|
dv Val
|
|
help string
|
|
}
|
|
|
|
var (
|
|
portName = Flag{ptrVal{string: &config.portName}, "p", "port", Val{string: ""}, "要连接的串口\t(/dev/ttyUSB0、COMx)"}
|
|
baudRate = Flag{ptrVal{int: &config.baudRate}, "b", "baud", Val{int: 115200}, "波特率"}
|
|
dataBits = Flag{ptrVal{int: &config.dataBits}, "d", "data", Val{int: 8}, "数据位"}
|
|
stopBits = Flag{ptrVal{int: &config.stopBits}, "s", "stop", Val{int: 0}, "停止位停止位(0: 1停止 1:1.5停止 2:2停止)"}
|
|
outputCode = Flag{ptrVal{string: &config.outputCode}, "o", "out", Val{string: "UTF-8"}, "输出编码"}
|
|
inputCode = Flag{ptrVal{string: &config.inputCode}, "i", "in", Val{string: "UTF-8"}, "输入编码"}
|
|
endStr = Flag{ptrVal{string: &config.endStr}, "e", "end", Val{string: "\n"}, "终端换行符"}
|
|
logExt = Flag{v: ptrVal{ext: &config.logFilePath}, sStr: "l", lStr: "log", dv: Val{extdef: "./%s-$s.txt", string: ""}, help: "日志保存路径"}
|
|
timeExt = Flag{v: ptrVal{ext: &config.timesFmt}, sStr: "t", lStr: "time", dv: Val{extdef: "[06-01-02 15:04:05.000]", string: ""}, help: "时间戳格式化字段"}
|
|
forWard = Flag{ptrVal{il: &config.forWard}, "f", "forward", Val{int: 0}, "转发模式(0: 无 1:TCP-C 2:UDP-C 支持多次传入)"}
|
|
address = Flag{ptrVal{sl: &config.address}, "a", "address", Val{string: "127.0.0.1:12345"}, "转发服务地址(支持多次传入)"}
|
|
frameSize = Flag{ptrVal{int: &config.frameSize}, "F", "Frame", Val{int: 16}, "帧大小"}
|
|
parityBit = Flag{ptrVal{int: &config.parityBit}, "v", "verify", Val{int: 0}, "奇偶校验(0:无校验、1:奇校验、2:偶校验、3:1校验、4:0校验)"}
|
|
flags = []Flag{portName, baudRate, dataBits, stopBits, outputCode, inputCode, endStr, forWard, address, frameSize, parityBit, logExt, timeExt}
|
|
)
|
|
|
|
var (
|
|
bauds = []string{"自定义", "300", "600", "1200", "2400", "4800", "9600",
|
|
"14400", "19200", "38400", "56000", "57600", "115200", "128000",
|
|
"256000", "460800", "512000", "750000", "921600", "1500000"}
|
|
datas = []string{"5", "6", "7", "8"}
|
|
stops = []string{"1", "1.5", "2"}
|
|
paritys = []string{"无校验", "奇校验", "偶校验", "1校验", "0校验"}
|
|
forwards = []string{"No", "TCP-C", "UDP-C"}
|
|
)
|
|
|
|
type ValType int
|
|
|
|
const (
|
|
notVal ValType = iota
|
|
stringVal
|
|
intVal
|
|
boolVal
|
|
extVal
|
|
)
|
|
|
|
func printUsage(ports []string) {
|
|
fmt.Printf("\n参数帮助:\n")
|
|
for _, f := range flags {
|
|
flagprint(f)
|
|
}
|
|
fmt.Printf("\n在线串口: %v\n", strings.Join(ports, ","))
|
|
}
|
|
func flagFindValue(v ptrVal) ValType {
|
|
if v.string != nil {
|
|
return stringVal
|
|
}
|
|
if v.bool != nil {
|
|
return boolVal
|
|
}
|
|
if v.int != nil {
|
|
return intVal
|
|
}
|
|
if v.ext != nil {
|
|
return extVal
|
|
}
|
|
return notVal
|
|
}
|
|
func flagprint(f Flag) {
|
|
switch flagFindValue(f.v) {
|
|
case stringVal:
|
|
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%q\n", f.sStr, f.lStr, f.dv.string, f.help, f.dv.string)
|
|
case intVal:
|
|
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", f.sStr, f.lStr, f.dv.int, f.help, f.dv.int)
|
|
case boolVal:
|
|
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", f.sStr, f.lStr, f.dv.bool, f.help, f.dv.bool)
|
|
case extVal:
|
|
fmt.Printf("\t-%v -%v %T \n\t %v\t默认值:%v\n", f.sStr, f.lStr, f.dv.extdef, f.help, f.dv.extdef)
|
|
default:
|
|
panic("unhandled default case")
|
|
}
|
|
}
|
|
func flagInit(f *Flag) {
|
|
if f.v.string != nil {
|
|
pflag.StringVarP(f.v.string, f.lStr, f.sStr, f.dv.string, f.help)
|
|
}
|
|
if f.v.bool != nil {
|
|
pflag.BoolVarP(f.v.bool, f.lStr, f.sStr, f.dv.bool, f.help)
|
|
}
|
|
if f.v.int != nil {
|
|
pflag.IntVarP(f.v.int, f.lStr, f.sStr, f.dv.int, f.help)
|
|
}
|
|
if f.v.ext != nil {
|
|
pflag.StringVarP(f.v.ext, f.lStr, f.sStr, f.dv.string, f.help)
|
|
pflag.Lookup(f.lStr).NoOptDefVal = f.dv.extdef
|
|
}
|
|
if f.v.sl != nil {
|
|
pflag.StringArrayVarP(f.v.sl, f.lStr, f.sStr, []string{f.dv.string}, f.help)
|
|
}
|
|
if f.v.il != nil {
|
|
pflag.IntSliceVarP(f.v.il, f.lStr, f.sStr, []int{f.dv.int}, f.help)
|
|
}
|
|
}
|
|
func flagExt() {
|
|
if config.logFilePath != "" {
|
|
config.enableLog = true
|
|
}
|
|
if config.timesFmt != "" {
|
|
config.timesTamp = true
|
|
}
|
|
}
|
|
func getCliFlag() {
|
|
ports, err := serial.GetPortsList()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
inputs := components.NewInput()
|
|
inputs.Prompt = "Filtering: "
|
|
inputs.PromptStyle = style.New().Bold().Italic().Fg(color.LightBlue)
|
|
|
|
selectKeymap := singleselect.DefaultSingleKeyMap()
|
|
selectKeymap.Confirm = key.NewBinding(
|
|
key.WithKeys("enter"),
|
|
key.WithHelp("enter", "finish select"),
|
|
)
|
|
selectKeymap.Choice = key.NewBinding(
|
|
key.WithKeys("enter"),
|
|
key.WithHelp("enter", "finish select"),
|
|
)
|
|
selectKeymap.NextPage = key.NewBinding(
|
|
key.WithKeys("right"),
|
|
key.WithHelp("->", "next page"),
|
|
)
|
|
selectKeymap.PrevPage = key.NewBinding(
|
|
key.WithKeys("left"),
|
|
key.WithHelp("<-", "prev page"),
|
|
)
|
|
|
|
s, _ := inf.NewSingleSelect(
|
|
ports,
|
|
singleselect.WithKeyBinding(selectKeymap),
|
|
singleselect.WithPageSize(4),
|
|
singleselect.WithFilterInput(inputs),
|
|
).Display("选择串口")
|
|
config.portName = ports[s]
|
|
|
|
s, _ = inf.NewSingleSelect(
|
|
bauds,
|
|
singleselect.WithKeyBinding(selectKeymap),
|
|
singleselect.WithPageSize(4),
|
|
).Display("选择波特率")
|
|
if s != 0 {
|
|
config.baudRate, _ = strconv.Atoi(bauds[s])
|
|
} else {
|
|
b, _ := inf.NewText(
|
|
text.WithPrompt("BaudRate:"),
|
|
text.WithPromptStyle(theme.DefaultTheme.PromptStyle),
|
|
text.WithDefaultValue("115200"),
|
|
).Display()
|
|
config.baudRate, _ = strconv.Atoi(b)
|
|
}
|
|
v, _ := inf.NewConfirmWithSelection(
|
|
confirm.WithPrompt("启用Hex"),
|
|
).Display()
|
|
if v {
|
|
config.inputCode = "hex"
|
|
b, _ := inf.NewText(
|
|
text.WithPrompt("Frames:"),
|
|
text.WithPromptStyle(theme.DefaultTheme.PromptStyle),
|
|
text.WithDefaultValue("16"),
|
|
).Display()
|
|
config.frameSize, _ = strconv.Atoi(b)
|
|
}
|
|
v, _ = inf.NewConfirmWithSelection(
|
|
confirm.WithPrompt("启用时间戳"),
|
|
).Display()
|
|
config.timesTamp = v
|
|
if v {
|
|
b, _ := inf.NewText(
|
|
text.WithPrompt("格式化字段:"),
|
|
text.WithPromptStyle(theme.DefaultTheme.PromptStyle),
|
|
text.WithDefaultValue(timeExt.dv.extdef),
|
|
).Display()
|
|
config.timesFmt = b
|
|
}
|
|
v, _ = inf.NewConfirmWithSelection(
|
|
confirm.WithPrompt("启用高级配置"),
|
|
).Display()
|
|
if v {
|
|
s, _ = inf.NewSingleSelect(
|
|
datas,
|
|
singleselect.WithKeyBinding(selectKeymap),
|
|
singleselect.WithPageSize(4),
|
|
singleselect.WithFilterInput(inputs),
|
|
).Display("选择数据位")
|
|
config.dataBits, _ = strconv.Atoi(datas[s])
|
|
|
|
s, _ = inf.NewSingleSelect(
|
|
stops,
|
|
singleselect.WithKeyBinding(selectKeymap),
|
|
singleselect.WithPageSize(4),
|
|
singleselect.WithFilterInput(inputs),
|
|
).Display("选择停止位")
|
|
config.stopBits = s
|
|
|
|
s, _ = inf.NewSingleSelect(
|
|
paritys,
|
|
singleselect.WithKeyBinding(selectKeymap),
|
|
singleselect.WithPageSize(4),
|
|
singleselect.WithFilterInput(inputs),
|
|
).Display("选择校验位")
|
|
config.parityBit = s
|
|
|
|
t, _ := inf.NewText(
|
|
text.WithPrompt("换行符:"),
|
|
text.WithPromptStyle(theme.DefaultTheme.PromptStyle),
|
|
text.WithDefaultValue(endStr.dv.string),
|
|
).Display()
|
|
config.endStr = t
|
|
|
|
v, _ = inf.NewConfirmWithSelection(
|
|
confirm.WithDefaultYes(),
|
|
confirm.WithPrompt("启用编码转换"),
|
|
).Display()
|
|
|
|
if v {
|
|
t, _ = inf.NewText(
|
|
text.WithPrompt("输入编码:"),
|
|
text.WithPromptStyle(theme.DefaultTheme.PromptStyle),
|
|
text.WithDefaultValue(inputCode.dv.string),
|
|
).Display()
|
|
config.inputCode = t
|
|
|
|
t, _ = inf.NewText(
|
|
text.WithPrompt("输出编码:"),
|
|
text.WithPromptStyle(theme.DefaultTheme.PromptStyle),
|
|
text.WithDefaultValue(outputCode.dv.string),
|
|
).Display()
|
|
config.outputCode = t
|
|
}
|
|
G_F_mode:
|
|
s, _ = inf.NewSingleSelect(
|
|
forwards,
|
|
singleselect.WithKeyBinding(selectKeymap),
|
|
singleselect.WithPageSize(3),
|
|
singleselect.WithFilterInput(inputs),
|
|
).Display("选择转发模式")
|
|
if s != 0 {
|
|
config.forWard = append(config.forWard, s)
|
|
t, _ = inf.NewText(
|
|
text.WithPrompt("地址:"),
|
|
text.WithPromptStyle(theme.DefaultTheme.PromptStyle),
|
|
text.WithDefaultValue(address.dv.string),
|
|
).Display()
|
|
config.address = append(config.address, t)
|
|
goto G_F_mode
|
|
}
|
|
|
|
e, _ := inf.NewConfirmWithSelection(
|
|
confirm.WithDefaultYes(),
|
|
confirm.WithPrompt("启用日志"),
|
|
).Display()
|
|
config.enableLog = e
|
|
if e {
|
|
t, _ = inf.NewText(
|
|
text.WithPrompt("Path:"),
|
|
text.WithPromptStyle(theme.DefaultTheme.PromptStyle),
|
|
text.WithDefaultValue("./%s-$s.txt"),
|
|
).Display()
|
|
config.logFilePath = t
|
|
}
|
|
}
|
|
|
|
}
|