增加交互式配置支持

flag.go

@@ -3,6 +3,18 @@ package main
 import (
 	"flag"
 	"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"
+	"go.bug.st/serial"
+	"log"
+	"strconv"
 	"strings"
 )
 
@@ -45,6 +57,16 @@ var (
 	flags       = []Flag{portName, baudRate, dataBits, stopBits, outputCode, inputCode, endStr, enableLog, logFilePath, forWard, address, frameSize, parityBit}
 )
 
+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 (
@@ -99,3 +121,150 @@ func flagInit(f *Flag) {
 		flag.IntVar(f.v.int, f.lStr, f.dv.int, f.help)
 	}
 }
+
+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(3),
+		singleselect.WithFilterInput(inputs),
+	).Display("选择串口")
+	config.portName = ports[s]
+
+	s, _ = inf.NewSingleSelect(
+		bauds,
+		singleselect.WithKeyBinding(selectKeymap),
+		singleselect.WithPageSize(3),
+	).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"
+	}
+	v, _ = inf.NewConfirmWithSelection(
+		confirm.WithDefaultYes(),
+		confirm.WithPrompt("启用高级配置"),
+	).Display()
+	if v {
+		s, _ = inf.NewSingleSelect(
+			datas,
+			singleselect.WithKeyBinding(selectKeymap),
+			singleselect.WithPageSize(3),
+			singleselect.WithFilterInput(inputs),
+		).Display("选择数据位")
+		config.dataBits, _ = strconv.Atoi(datas[s])
+
+		s, _ = inf.NewSingleSelect(
+			stops,
+			singleselect.WithKeyBinding(selectKeymap),
+			singleselect.WithPageSize(3),
+			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
+		}
+
+		s, _ = inf.NewSingleSelect(
+			forwards,
+			singleselect.WithKeyBinding(selectKeymap),
+			singleselect.WithPageSize(3),
+			singleselect.WithFilterInput(inputs),
+		).Display("选择转发模式")
+		if s != 0 {
+			config.forWard = s
+			t, _ = inf.NewText(
+				text.WithPrompt("地址:"),
+				text.WithPromptStyle(theme.DefaultTheme.PromptStyle),
+				text.WithDefaultValue(address.dv.string),
+			).Display()
+			config.address = t
+		}
+
+		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(logFilePath.dv.string),
+			).Display()
+			config.logFilePath = t
+		}
+	}
+
+}