334 lines
8.0 KiB
Go
334 lines
8.0 KiB
Go
package wizard
|
||
|
||
import (
|
||
"github.com/charmbracelet/bubbles/textinput"
|
||
tea "github.com/charmbracelet/bubbletea"
|
||
)
|
||
|
||
// Config 系统配置结构
|
||
type Config struct {
|
||
// 协议
|
||
AgreementAccepted bool `json:"agreement_accepted"`
|
||
|
||
// 数据接收
|
||
Hostname string `json:"hostname"`
|
||
Country string `json:"country"`
|
||
Region string `json:"region"`
|
||
Timezone string `json:"timezone"`
|
||
HomePage string `json:"homepage"`
|
||
DBAddress string `json:"db_address"`
|
||
DBName string `json:"db_name"`
|
||
DataAddress string `json:"data_address"`
|
||
|
||
// 公网设置
|
||
PublicInterface string `json:"public_interface"`
|
||
IPAddress string `json:"ip_address"`
|
||
Netmask string `json:"netmask"`
|
||
Gateway string `json:"gateway"`
|
||
|
||
// 内网配置
|
||
InternalInterface string `json:"internal_interface"`
|
||
InternalIP string `json:"internal_ip"`
|
||
InternalMask string `json:"internal_mask"`
|
||
|
||
// DNS 配置
|
||
DNSPrimary string `json:"dns_primary"`
|
||
DNSSecondary string `json:"dns_secondary"`
|
||
}
|
||
|
||
// PageType 页面类型
|
||
type PageType int
|
||
|
||
const (
|
||
PageAgreement PageType = iota
|
||
PageData
|
||
PagePublicNetwork
|
||
PageInternalNetwork
|
||
PageDNS
|
||
PageSummary
|
||
)
|
||
|
||
const (
|
||
FocusTypeInput int = 0
|
||
FocusTypePrev int = 1
|
||
FocusTypeNext int = 2
|
||
)
|
||
|
||
// model TUI 主模型
|
||
type model struct {
|
||
config Config
|
||
currentPage PageType
|
||
totalPages int
|
||
textInputs []textinput.Model
|
||
inputLabels []string // 存储标签
|
||
focusIndex int
|
||
focusType int // 0=输入框, 1=上一步按钮, 2=下一步按钮
|
||
agreementIdx int // 0=拒绝，1=接受
|
||
width int
|
||
height int
|
||
err error
|
||
quitting bool
|
||
done bool
|
||
force bool
|
||
}
|
||
|
||
// defaultConfig 返回默认配置
|
||
func defaultConfig() Config {
|
||
return Config{
|
||
Hostname: "sunhpc01",
|
||
Country: "China",
|
||
Region: "Beijing",
|
||
Timezone: "Asia/Shanghai",
|
||
HomePage: "https://sunhpc.example.com",
|
||
DBAddress: "127.0.0.1",
|
||
DBName: "sunhpc_db",
|
||
DataAddress: "/data/sunhpc",
|
||
PublicInterface: "eth0",
|
||
InternalInterface: "eth1",
|
||
IPAddress: "192.168.1.100",
|
||
Netmask: "255.255.255.0",
|
||
Gateway: "192.168.1.1",
|
||
InternalIP: "10.0.0.100",
|
||
InternalMask: "255.255.255.0",
|
||
DNSPrimary: "8.8.8.8",
|
||
DNSSecondary: "8.8.4.4",
|
||
}
|
||
}
|
||
|
||
// initialModel 初始化模型
|
||
func initialModel() model {
|
||
cfg := defaultConfig()
|
||
m := model{
|
||
config: cfg,
|
||
totalPages: 6,
|
||
textInputs: make([]textinput.Model, 0),
|
||
inputLabels: make([]string, 0),
|
||
agreementIdx: 1,
|
||
focusIndex: 0,
|
||
focusType: 0, // 0=输入框, 1=上一步按钮, 2=下一步按钮
|
||
width: 80,
|
||
height: 24,
|
||
}
|
||
m.initPageInputs()
|
||
return m
|
||
}
|
||
|
||
// Init 初始化命令
|
||
func (m model) Init() tea.Cmd {
|
||
return textinput.Blink
|
||
}
|
||
|
||
// Update 处理消息更新
|
||
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||
var cmds []tea.Cmd
|
||
|
||
switch msg := msg.(type) {
|
||
case tea.KeyMsg:
|
||
switch msg.String() {
|
||
case "ctrl+c":
|
||
m.quitting = true
|
||
return m, tea.Quit
|
||
|
||
case "esc":
|
||
if m.currentPage > 0 {
|
||
return m.prevPage()
|
||
}
|
||
|
||
case "enter":
|
||
return m.handleEnter()
|
||
|
||
case "tab", "shift+tab", "up", "down", "left", "right":
|
||
return m.handleNavigation(msg)
|
||
}
|
||
|
||
case tea.WindowSizeMsg:
|
||
m.width = msg.Width
|
||
m.height = msg.Height
|
||
|
||
// 动态调整容器宽度
|
||
/*
|
||
if msg.Width > 100 {
|
||
containerStyle = containerStyle.Width(90)
|
||
} else if msg.Width > 80 {
|
||
containerStyle = containerStyle.Width(70)
|
||
} else {
|
||
containerStyle = containerStyle.Width(msg.Width - 10)
|
||
}
|
||
*/
|
||
|
||
// ✅ 动态计算容器宽度（终端宽度的 80%）
|
||
containerWidth := msg.Width * 80 / 100
|
||
|
||
// ✅ 重新设置容器样式宽度
|
||
containerStyle = containerStyle.Width(containerWidth)
|
||
|
||
// 动态设置协议框宽度（容器宽度的 90%）
|
||
agreementWidth := containerWidth * 80 / 100
|
||
agreementBox = agreementBox.Width(agreementWidth)
|
||
|
||
// 动态设置输入框宽度
|
||
inputWidth := containerWidth * 60 / 100
|
||
if inputWidth < 40 {
|
||
inputWidth = 40
|
||
}
|
||
inputBox = inputBox.Width(inputWidth)
|
||
|
||
// 动态设置总结框宽度
|
||
summaryWidth := containerWidth * 90 / 100
|
||
summaryBox = summaryBox.Width(summaryWidth)
|
||
|
||
return m, nil
|
||
}
|
||
|
||
// 更新当前焦点的输入框
|
||
if len(m.textInputs) > 0 && m.focusIndex < len(m.textInputs) {
|
||
var cmd tea.Cmd
|
||
m.textInputs[m.focusIndex], cmd = m.textInputs[m.focusIndex].Update(msg)
|
||
cmds = append(cmds, cmd)
|
||
}
|
||
|
||
return m, tea.Batch(cmds...)
|
||
}
|
||
|
||
// handleEnter 处理回车事件
|
||
func (m *model) handleEnter() (tea.Model, tea.Cmd) {
|
||
switch m.currentPage {
|
||
case PageAgreement:
|
||
if m.agreementIdx == 1 {
|
||
m.config.AgreementAccepted = true
|
||
return m.nextPage()
|
||
} else {
|
||
m.quitting = true
|
||
return m, tea.Quit
|
||
}
|
||
|
||
case PageData, PagePublicNetwork, PageInternalNetwork, PageDNS:
|
||
// 根据焦点类型执行不同操作
|
||
switch m.focusType {
|
||
case FocusTypeInput:
|
||
// 在输入框上，保存并下一页
|
||
m.saveCurrentPage()
|
||
return m.nextPage()
|
||
case FocusTypePrev:
|
||
// 上一步按钮，返回上一页
|
||
return m.prevPage()
|
||
case FocusTypeNext:
|
||
// 下一步按钮，切换到下一页
|
||
m.saveCurrentPage()
|
||
return m.nextPage()
|
||
}
|
||
|
||
case PageSummary:
|
||
switch m.focusIndex {
|
||
case 0: // 执行
|
||
m.done = true
|
||
if err := m.saveConfig(); err != nil {
|
||
m.err = err
|
||
return m, nil
|
||
}
|
||
return m, tea.Quit
|
||
case 1: // 取消
|
||
m.quitting = true
|
||
return m, tea.Quit
|
||
}
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
// handleNavigation 处理导航
|
||
func (m *model) handleNavigation(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
|
||
// debug
|
||
//fmt.Fprintf(os.Stderr, "DEBUG: key=%s page=%d\n", msg.String(), m.currentPage)
|
||
|
||
switch m.currentPage {
|
||
case PageAgreement:
|
||
switch msg.String() {
|
||
case "left", "right", "tab", "shift+tab", "up", "down":
|
||
m.agreementIdx = 1 - m.agreementIdx
|
||
}
|
||
|
||
case PageSummary:
|
||
switch msg.String() {
|
||
case "left", "right", "tab", "shift+tab":
|
||
m.focusIndex = 1 - m.focusIndex
|
||
}
|
||
|
||
default:
|
||
// 输入框页面: 支持输入框和按钮之间切换
|
||
// totalFocusable := len(m.textInputs) + 2
|
||
|
||
switch msg.String() {
|
||
case "down", "tab":
|
||
// 当前在输入框
|
||
switch m.focusType {
|
||
case FocusTypeInput:
|
||
if m.focusIndex < len(m.textInputs)-1 {
|
||
// 切换到下一个输入框
|
||
m.textInputs[m.focusIndex].Blur()
|
||
m.focusIndex++
|
||
m.textInputs[m.focusIndex].Focus()
|
||
} else {
|
||
// 最后一个输入框，切换到“下一步”按钮
|
||
m.textInputs[m.focusIndex].Blur()
|
||
m.focusIndex = 0
|
||
m.focusType = FocusTypeNext // 下一步按钮
|
||
}
|
||
case FocusTypePrev:
|
||
// 当前在“上一步”按钮，切换到第一个输入框
|
||
m.focusType = FocusTypeInput
|
||
m.focusIndex = 0
|
||
m.textInputs[0].Focus()
|
||
case FocusTypeNext:
|
||
// 当前在“下一步”按钮，切换到“上一步”按钮
|
||
m.focusType = FocusTypePrev
|
||
}
|
||
case "up", "shift+tab":
|
||
// 当前在输入框
|
||
switch m.focusType {
|
||
case FocusTypeInput:
|
||
if m.focusIndex > 0 {
|
||
// 切换到上一个输入框
|
||
m.textInputs[m.focusIndex].Blur()
|
||
m.focusIndex--
|
||
m.textInputs[m.focusIndex].Focus()
|
||
} else {
|
||
// 第一个输入框，切换到“上一步”按钮
|
||
m.textInputs[m.focusIndex].Blur()
|
||
m.focusIndex = 0
|
||
m.focusType = FocusTypePrev // 上一步按钮
|
||
}
|
||
case FocusTypeNext:
|
||
// 当前在“下一步”按钮，切换到最后一个输入框
|
||
m.focusType = FocusTypeInput
|
||
m.focusIndex = len(m.textInputs) - 1
|
||
m.textInputs[m.focusIndex].Focus()
|
||
case FocusTypePrev:
|
||
// 当前在“上一步”按钮，切换到“下一步”按钮
|
||
m.focusType = FocusTypeNext
|
||
}
|
||
}
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
// nextPage 下一页
|
||
func (m *model) nextPage() (tea.Model, tea.Cmd) {
|
||
if m.currentPage < PageSummary {
|
||
m.currentPage++
|
||
m.focusIndex = 0
|
||
m.initPageInputs()
|
||
}
|
||
return m, textinput.Blink
|
||
}
|
||
|
||
// prevPage 上一页
|
||
func (m *model) prevPage() (tea.Model, tea.Cmd) {
|
||
if m.currentPage > 0 {
|
||
m.saveCurrentPage()
|
||
m.currentPage--
|
||
m.focusIndex = 0
|
||
m.initPageInputs()
|
||
}
|
||
return m, textinput.Blink
|
||
}
|