668 lines
15 KiB
Go
668 lines
15 KiB
Go
package ui
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
"time"
|
||
|
||
"moltbot-installer/internal/style"
|
||
"moltbot-installer/internal/sys"
|
||
|
||
"github.com/charmbracelet/bubbles/spinner"
|
||
"github.com/charmbracelet/bubbles/textinput"
|
||
tea "github.com/charmbracelet/bubbletea"
|
||
"github.com/charmbracelet/lipgloss"
|
||
)
|
||
|
||
// SessionState defines the high-level mode of the application
|
||
type SessionState int
|
||
|
||
const (
|
||
StateDashboard SessionState = iota
|
||
StateWizard
|
||
StateAction
|
||
)
|
||
|
||
// Sub-states for specific flows
|
||
type WizardStep int
|
||
|
||
const (
|
||
StepApiSelect WizardStep = iota
|
||
StepApiInput
|
||
StepConfirm
|
||
)
|
||
|
||
type ActionType int
|
||
|
||
const (
|
||
ActionCheckEnv ActionType = iota
|
||
ActionInstall
|
||
ActionUninstall
|
||
ActionStartGateway
|
||
ActionKillGateway
|
||
)
|
||
|
||
// Model is the main application state
|
||
type Model struct {
|
||
// Global State
|
||
state SessionState
|
||
width int
|
||
height int
|
||
|
||
// Flag to signal main.go (legacy/compatibility)
|
||
DidStartGateway bool
|
||
|
||
// Dashboard State
|
||
menuIndex int
|
||
|
||
// Wizard State
|
||
wizardStep WizardStep
|
||
configOpts sys.ConfigOptions
|
||
input textinput.Model
|
||
inputStep int // For multi-field steps like API Input
|
||
|
||
// Action/Progress State
|
||
actionType ActionType
|
||
spinner spinner.Model
|
||
progressMsg string
|
||
actionErr error
|
||
actionDone bool
|
||
|
||
// System Status Cache
|
||
nodeVer string
|
||
nodeOk bool
|
||
moltbotVer string
|
||
moltbotOk bool
|
||
gatewayOk bool
|
||
checkDone bool
|
||
}
|
||
|
||
// Messages
|
||
type checkMsg struct {
|
||
nodeVer string
|
||
nodeOk bool
|
||
moltbotVer string
|
||
moltbotInstalled bool
|
||
gatewayRunning bool
|
||
}
|
||
|
||
type actionResultMsg struct {
|
||
err error
|
||
}
|
||
|
||
type progressMsg string
|
||
|
||
type tickMsg time.Time
|
||
|
||
type gatewayStatusMsg bool
|
||
|
||
func InitialModel() Model {
|
||
s := spinner.New()
|
||
s.Spinner = spinner.Dot
|
||
s.Style = style.HeaderStyle
|
||
|
||
ti := textinput.New()
|
||
ti.Cursor.Style = style.HeaderStyle
|
||
ti.Focus()
|
||
|
||
return Model{
|
||
state: StateDashboard,
|
||
spinner: s,
|
||
input: ti,
|
||
menuIndex: 0,
|
||
}
|
||
}
|
||
|
||
func (m Model) Init() tea.Cmd {
|
||
return tea.Batch(
|
||
m.spinner.Tick,
|
||
checkEnvCmd,
|
||
tickCmd(),
|
||
)
|
||
}
|
||
|
||
func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||
var cmd tea.Cmd
|
||
|
||
switch msg := msg.(type) {
|
||
case tea.KeyMsg:
|
||
if msg.String() == "ctrl+c" {
|
||
return m, tea.Quit
|
||
}
|
||
// Global Back handler for Wizard
|
||
if m.state == StateWizard && msg.String() == "esc" {
|
||
m.state = StateDashboard
|
||
return m, nil
|
||
}
|
||
// Global Back handler for Action Result
|
||
if m.state == StateAction && m.actionDone && (msg.String() == "enter" || msg.String() == "esc") {
|
||
m.state = StateDashboard
|
||
// Refresh env after action
|
||
return m, checkEnvCmd
|
||
}
|
||
|
||
case tea.WindowSizeMsg:
|
||
m.width = msg.Width
|
||
m.height = msg.Height
|
||
|
||
case spinner.TickMsg:
|
||
m.spinner, cmd = m.spinner.Update(msg)
|
||
return m, cmd
|
||
|
||
case checkMsg:
|
||
m.nodeVer = msg.nodeVer
|
||
m.nodeOk = msg.nodeOk
|
||
m.moltbotVer = msg.moltbotVer
|
||
m.moltbotOk = msg.moltbotInstalled
|
||
m.gatewayOk = msg.gatewayRunning
|
||
m.checkDone = true
|
||
|
||
// If we are in Action mode checking env, this might be a refresh
|
||
if m.state == StateAction && m.actionType == ActionCheckEnv {
|
||
m.state = StateDashboard
|
||
}
|
||
return m, nil
|
||
|
||
case progressMsg:
|
||
m.progressMsg = string(msg)
|
||
return m, nil
|
||
|
||
case tickMsg:
|
||
return m, tea.Batch(checkGatewayCmd, tickCmd())
|
||
|
||
case gatewayStatusMsg:
|
||
m.gatewayOk = bool(msg)
|
||
return m, nil
|
||
|
||
case actionResultMsg:
|
||
m.actionErr = msg.err
|
||
m.actionDone = true
|
||
if msg.err == nil {
|
||
m.progressMsg = "操作成功完成！"
|
||
if m.actionType == ActionStartGateway {
|
||
m.DidStartGateway = true
|
||
}
|
||
} else {
|
||
m.progressMsg = fmt.Sprintf("操作失败: %v", msg.err)
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
// State-specific updates
|
||
switch m.state {
|
||
case StateDashboard:
|
||
return m.updateDashboard(msg)
|
||
case StateWizard:
|
||
return m.updateWizard(msg)
|
||
case StateAction:
|
||
// In action state, mostly waiting for msgs, but maybe handle quit
|
||
return m, nil
|
||
}
|
||
|
||
return m, nil
|
||
}
|
||
|
||
func (m Model) updateDashboard(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||
switch msg := msg.(type) {
|
||
case tea.KeyMsg:
|
||
switch msg.String() {
|
||
case "up", "k":
|
||
if m.menuIndex > 0 {
|
||
m.menuIndex--
|
||
}
|
||
case "down", "j":
|
||
if m.menuIndex < 4 { // 5 items (0-4)
|
||
m.menuIndex++
|
||
}
|
||
case "enter":
|
||
return m.handleMenuSelect()
|
||
case "q":
|
||
return m, tea.Quit
|
||
}
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
func (m Model) handleMenuSelect() (tea.Model, tea.Cmd) {
|
||
switch m.menuIndex {
|
||
case 0: // Start/Restart Gateway
|
||
m.state = StateAction
|
||
m.actionDone = false
|
||
m.actionErr = nil
|
||
if m.gatewayOk {
|
||
m.actionType = ActionKillGateway
|
||
m.progressMsg = "正在停止网关..."
|
||
return m, runKillGatewayCmd
|
||
} else {
|
||
m.actionType = ActionStartGateway
|
||
m.progressMsg = "正在启动网关..."
|
||
return m, runStartGatewayCmd
|
||
}
|
||
case 1: // Configure
|
||
m.state = StateWizard
|
||
m.wizardStep = StepApiSelect
|
||
m.configOpts = sys.ConfigOptions{}
|
||
return m, nil
|
||
case 2: // Install/Update
|
||
m.state = StateAction
|
||
m.actionType = ActionInstall
|
||
m.actionDone = false
|
||
m.actionErr = nil
|
||
m.progressMsg = "准备安装..."
|
||
return m, runInstallFlowCmd
|
||
case 3: // Uninstall
|
||
m.state = StateAction
|
||
m.actionType = ActionUninstall
|
||
m.actionDone = false
|
||
m.actionErr = nil
|
||
m.progressMsg = "正在卸载..."
|
||
return m, runUninstallCmd
|
||
case 4: // Exit
|
||
return m, tea.Quit
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
func (m Model) updateWizard(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||
var cmd tea.Cmd
|
||
|
||
switch m.wizardStep {
|
||
case StepApiSelect:
|
||
if k, ok := msg.(tea.KeyMsg); ok {
|
||
switch k.String() {
|
||
case "1":
|
||
m.configOpts.ApiType = "anthropic"
|
||
m.wizardStep = StepApiInput
|
||
m.inputStep = 0
|
||
m.input.Placeholder = "sk-ant-api03-..."
|
||
m.input.EchoMode = textinput.EchoPassword
|
||
m.input.SetValue("")
|
||
case "2":
|
||
m.configOpts.ApiType = "openai"
|
||
m.wizardStep = StepApiInput
|
||
m.inputStep = 0
|
||
m.input.Placeholder = "https://api.openai.com/v1"
|
||
m.input.EchoMode = textinput.EchoNormal
|
||
m.input.SetValue("")
|
||
}
|
||
}
|
||
|
||
case StepApiInput:
|
||
switch k := msg.(type) {
|
||
case tea.KeyMsg:
|
||
if k.String() == "enter" {
|
||
val := m.input.Value()
|
||
// Logic handles both Anthropic and OpenAI flows
|
||
isAnthropic := m.configOpts.ApiType == "anthropic"
|
||
|
||
if isAnthropic {
|
||
switch m.inputStep {
|
||
case 0: // Key
|
||
m.configOpts.AnthropicKey = val
|
||
m.inputStep++
|
||
m.input.Placeholder = "123456:ABC-DEF..."
|
||
m.input.EchoMode = textinput.EchoNormal
|
||
m.input.SetValue("")
|
||
case 1: // Bot Token
|
||
m.configOpts.BotToken = val
|
||
m.inputStep++
|
||
m.input.Placeholder = "123456789"
|
||
m.input.SetValue("")
|
||
case 2: // Admin ID
|
||
m.configOpts.AdminID = val
|
||
m.wizardStep = StepConfirm
|
||
}
|
||
} else { // OpenAI
|
||
switch m.inputStep {
|
||
case 0: // Base URL
|
||
m.configOpts.OpenAIBaseURL = val
|
||
m.inputStep++
|
||
m.input.Placeholder = "sk-..."
|
||
m.input.EchoMode = textinput.EchoPassword
|
||
m.input.SetValue("")
|
||
case 1: // Key
|
||
m.configOpts.OpenAIKey = val
|
||
m.inputStep++
|
||
m.input.Placeholder = "gpt-4o / claude-3-5-sonnet"
|
||
m.input.EchoMode = textinput.EchoNormal
|
||
m.input.SetValue("")
|
||
case 2: // Model
|
||
m.configOpts.OpenAIModel = val
|
||
m.inputStep++
|
||
m.input.Placeholder = "123456:ABC-DEF..."
|
||
m.input.SetValue("")
|
||
case 3: // Bot Token
|
||
m.configOpts.BotToken = val
|
||
m.inputStep++
|
||
m.input.Placeholder = "123456789"
|
||
m.input.SetValue("")
|
||
case 4: // Admin ID
|
||
m.configOpts.AdminID = val
|
||
m.wizardStep = StepConfirm
|
||
}
|
||
}
|
||
return m, nil
|
||
}
|
||
}
|
||
m.input, cmd = m.input.Update(msg)
|
||
return m, cmd
|
||
|
||
case StepConfirm:
|
||
if k, ok := msg.(tea.KeyMsg); ok {
|
||
if k.String() == "enter" {
|
||
// Run Save Action
|
||
m.state = StateAction
|
||
m.actionDone = false
|
||
m.actionErr = nil
|
||
m.progressMsg = "正在保存配置..."
|
||
return m, runSaveConfigCmd(m.configOpts)
|
||
}
|
||
}
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
// VIEW RENDERING
|
||
|
||
func (m Model) View() string {
|
||
if m.width == 0 {
|
||
return "Loading..."
|
||
}
|
||
|
||
switch m.state {
|
||
case StateDashboard:
|
||
return m.renderDashboard()
|
||
case StateWizard:
|
||
return m.renderWizard()
|
||
case StateAction:
|
||
return m.renderAction()
|
||
}
|
||
return ""
|
||
}
|
||
|
||
func (m Model) renderDashboard() string {
|
||
// 1. Header
|
||
header := style.HeaderStyle.Render("Moltbot Installer")
|
||
|
||
// 2. Status Bar
|
||
nodeStatus := style.Badge("检测中...", "info")
|
||
if m.checkDone {
|
||
if m.nodeOk {
|
||
nodeStatus = style.Badge(m.nodeVer, "success")
|
||
} else {
|
||
nodeStatus = style.Badge("缺失", "error")
|
||
}
|
||
}
|
||
|
||
moltStatus := style.Badge("检测中...", "info")
|
||
if m.checkDone {
|
||
if m.moltbotOk {
|
||
ver := m.moltbotVer
|
||
if ver == "" {
|
||
ver = "已安装"
|
||
}
|
||
moltStatus = style.Badge(ver, "success")
|
||
} else {
|
||
moltStatus = style.Badge("未安装", "warning")
|
||
}
|
||
}
|
||
|
||
gwStatus := style.Badge("...", "info")
|
||
if m.checkDone {
|
||
if m.gatewayOk {
|
||
gwStatus = style.Badge("运行中", "success")
|
||
} else {
|
||
gwStatus = style.Badge("已停止", "warning")
|
||
}
|
||
}
|
||
|
||
statusPanel := style.PanelStyle.Render(lipgloss.JoinVertical(lipgloss.Left,
|
||
style.SubHeaderStyle.Render("系统状态"),
|
||
fmt.Sprintf("Node.js 环境: %s", nodeStatus),
|
||
fmt.Sprintf("Moltbot 核心: %s", moltStatus),
|
||
fmt.Sprintf("网关进程: %s", gwStatus),
|
||
))
|
||
|
||
// 3. Menu
|
||
menuItems := []struct{ title, desc string }{
|
||
{"启动/重启服务", "管理后台网关进程"},
|
||
{"配置向导", "设置 API 密钥与机器人参数"},
|
||
{"安装/更新环境", "一键部署 Node.js 与核心组件"},
|
||
{"卸载 Moltbot", "清理所有文件与配置"},
|
||
{"退出", "关闭控制台"},
|
||
}
|
||
|
||
// Dynamic text for toggle
|
||
if m.gatewayOk {
|
||
menuItems[0].title = "重启服务"
|
||
menuItems[0].desc = "停止当前进程并重新启动"
|
||
} else {
|
||
menuItems[0].title = "启动服务"
|
||
menuItems[0].desc = "启动后台网关进程"
|
||
}
|
||
|
||
var menuView string
|
||
for i, item := range menuItems {
|
||
if i == m.menuIndex {
|
||
menuView += style.MenuSelectedStyle.Render(fmt.Sprintf("%s\n%s", item.title, style.DescriptionStyle.Render(item.desc)))
|
||
} else {
|
||
menuView += style.MenuNormalStyle.Render(item.title)
|
||
}
|
||
menuView += "\n\n"
|
||
}
|
||
|
||
menuPanel := style.FocusedPanelStyle.Render(lipgloss.JoinVertical(lipgloss.Left,
|
||
style.SubHeaderStyle.Render("主菜单"),
|
||
menuView,
|
||
style.SubtleStyle.Render("使用 ↑/↓ 选择，Enter 确认"),
|
||
))
|
||
|
||
// Layout
|
||
if m.width > 100 {
|
||
// Side by Side
|
||
return style.AppStyle.Render(lipgloss.JoinVertical(lipgloss.Left,
|
||
header,
|
||
"",
|
||
lipgloss.JoinHorizontal(lipgloss.Top, statusPanel, " ", menuPanel),
|
||
))
|
||
}
|
||
|
||
// Vertical Stack
|
||
return style.AppStyle.Render(lipgloss.JoinVertical(lipgloss.Left,
|
||
header,
|
||
"",
|
||
statusPanel,
|
||
"",
|
||
menuPanel,
|
||
))
|
||
}
|
||
|
||
func (m Model) renderWizard() string {
|
||
var content string
|
||
|
||
switch m.wizardStep {
|
||
case StepApiSelect:
|
||
content = lipgloss.JoinVertical(lipgloss.Left,
|
||
style.SubHeaderStyle.Render("Step 1: 选择 API 提供商"),
|
||
"",
|
||
style.MenuNormalStyle.Render("[1] Anthropic 官方 API (推荐)"),
|
||
style.DescriptionStyle.Render(" 直接连接 Claude 服务，最稳定"),
|
||
"",
|
||
style.MenuNormalStyle.Render("[2] OpenAI 兼容 API"),
|
||
style.DescriptionStyle.Render(" 支持 DeepSeek, GPT-4 等第三方模型"),
|
||
"",
|
||
style.SubtleStyle.Render("按 1 或 2 选择，Esc 返回"),
|
||
)
|
||
case StepApiInput:
|
||
label := "配置项"
|
||
if m.configOpts.ApiType == "anthropic" {
|
||
switch m.inputStep {
|
||
case 0:
|
||
label = "Anthropic API Key"
|
||
case 1:
|
||
label = "Telegram Bot Token (可选)"
|
||
case 2:
|
||
label = "Telegram 账户ID (可选)"
|
||
}
|
||
} else {
|
||
switch m.inputStep {
|
||
case 0:
|
||
label = "API 地址"
|
||
case 1:
|
||
label = "API Key"
|
||
case 2:
|
||
label = "模型名称"
|
||
case 3:
|
||
label = "Telegram Bot Token (可选)"
|
||
case 4:
|
||
label = "Telegram 账户ID (可选)"
|
||
}
|
||
}
|
||
|
||
var extraHelp string
|
||
if strings.Contains(label, "(可选)") {
|
||
extraHelp = style.InputHelpStyle.Render("如无需配置，请直接按 Enter 跳过。")
|
||
}
|
||
|
||
content = lipgloss.JoinVertical(lipgloss.Left,
|
||
style.SubHeaderStyle.Render("Step 2: 录入凭证"),
|
||
"",
|
||
style.InputHelpStyle.Render("请在下方输入您的 "+label+"。"),
|
||
style.InputHelpStyle.Render("支持使用 Ctrl+V 或 鼠标右键粘贴内容。"),
|
||
extraHelp,
|
||
"",
|
||
style.TitleStyle.Render(label),
|
||
"",
|
||
style.InputFocusedStyle.Render(m.input.View()),
|
||
"",
|
||
style.DescriptionStyle.Render("示例格式: "+m.input.Placeholder),
|
||
"",
|
||
style.SubtleStyle.Render("Enter 下一步"),
|
||
)
|
||
case StepConfirm:
|
||
content = lipgloss.JoinVertical(lipgloss.Left,
|
||
style.SubHeaderStyle.Render("Step 3: 确认配置"),
|
||
"",
|
||
"配置已就绪，准备写入文件。",
|
||
style.DescriptionStyle.Render("路径: ~/.clawdbot/clawdbot.json"),
|
||
"",
|
||
style.SubtleStyle.Render("Enter 确认写入，Esc 取消"),
|
||
)
|
||
}
|
||
|
||
return style.AppStyle.Render(style.WizardPanelStyle.Render(content))
|
||
}
|
||
|
||
func (m Model) renderAction() string {
|
||
icon := m.spinner.View()
|
||
title := "正在处理..."
|
||
|
||
if m.actionDone {
|
||
icon = "✅"
|
||
if m.actionErr != nil {
|
||
icon = "❌"
|
||
title = "操作失败"
|
||
} else {
|
||
title = "操作完成"
|
||
}
|
||
}
|
||
|
||
content := lipgloss.JoinVertical(lipgloss.Center,
|
||
style.SubHeaderStyle.Render(title),
|
||
"",
|
||
fmt.Sprintf("%s %s", icon, m.progressMsg),
|
||
"",
|
||
)
|
||
|
||
if m.actionDone {
|
||
content = lipgloss.JoinVertical(lipgloss.Center, content, style.SubtleStyle.Render("按 Enter 返回主菜单"))
|
||
}
|
||
|
||
return style.AppStyle.Render(style.PanelStyle.Render(content))
|
||
}
|
||
|
||
// COMMANDS
|
||
|
||
func checkEnvCmd() tea.Msg {
|
||
nodeVer, nodeOk := sys.CheckNode()
|
||
moltVer, moltOk := sys.CheckMoltbot()
|
||
gwRun := sys.IsGatewayRunning()
|
||
return checkMsg{
|
||
nodeVer: nodeVer,
|
||
nodeOk: nodeOk,
|
||
moltbotVer: moltVer,
|
||
moltbotInstalled: moltOk,
|
||
gatewayRunning: gwRun,
|
||
}
|
||
}
|
||
|
||
func runStartGatewayCmd() tea.Msg {
|
||
sys.StartGateway()
|
||
time.Sleep(1 * time.Second) // Wait for startup
|
||
return actionResultMsg{err: nil}
|
||
}
|
||
|
||
func runKillGatewayCmd() tea.Msg {
|
||
sys.KillGateway()
|
||
time.Sleep(1 * time.Second)
|
||
return actionResultMsg{err: nil}
|
||
}
|
||
|
||
func runUninstallCmd() tea.Msg {
|
||
// 1. Try to stop gateway if running
|
||
_ = sys.KillGateway()
|
||
time.Sleep(1 * time.Second)
|
||
|
||
// 2. Uninstall files
|
||
err := sys.UninstallMoltbot()
|
||
return actionResultMsg{err: err}
|
||
}
|
||
|
||
func runSaveConfigCmd(opts sys.ConfigOptions) tea.Cmd {
|
||
return func() tea.Msg {
|
||
err := sys.GenerateAndWriteConfig(opts)
|
||
return actionResultMsg{err: err}
|
||
}
|
||
}
|
||
|
||
func runInstallFlowCmd() tea.Msg {
|
||
// Linear flow: Check Node -> Install Node -> Config NPM -> Install Moltbot -> Config System
|
||
// Simplified to a blocking chain for "Action" state simplicity,
|
||
// or we can use tea.Sequence if we want granular updates.
|
||
// For now, we do a blocking sequence in a goroutine wrapper.
|
||
|
||
err := sys.InstallNode()
|
||
if err != nil {
|
||
return actionResultMsg{err: fmt.Errorf("node.js 安装失败: %v", err)}
|
||
}
|
||
|
||
err = sys.ConfigureNpmMirror()
|
||
if err != nil {
|
||
return actionResultMsg{err: fmt.Errorf("npm 配置失败: %v", err)}
|
||
}
|
||
|
||
err = sys.InstallMoltbotNpm("latest")
|
||
if err != nil {
|
||
return actionResultMsg{err: fmt.Errorf("moltbot 安装失败: %v", err)}
|
||
}
|
||
|
||
_, err = sys.EnsureOnPath()
|
||
if err != nil {
|
||
// Non-fatal
|
||
}
|
||
|
||
sys.RunDoctor()
|
||
|
||
return actionResultMsg{err: nil}
|
||
}
|
||
|
||
func tickCmd() tea.Cmd {
|
||
return tea.Tick(2*time.Second, func(t time.Time) tea.Msg {
|
||
return tickMsg(t)
|
||
})
|
||
}
|
||
|
||
func checkGatewayCmd() tea.Msg {
|
||
return gatewayStatusMsg(sys.IsGatewayRunning())
|
||
}
|