634 lines
16 KiB
Go
634 lines
16 KiB
Go
package ui
|
||
|
||
import (
|
||
"fmt"
|
||
"math/rand"
|
||
"sync"
|
||
"time"
|
||
|
||
"moltbot-installer/internal/style"
|
||
"moltbot-installer/internal/sys"
|
||
|
||
"github.com/charmbracelet/bubbles/spinner"
|
||
"github.com/charmbracelet/bubbles/textinput"
|
||
tea "github.com/charmbracelet/bubbletea"
|
||
)
|
||
|
||
type AppState int
|
||
|
||
const (
|
||
StateInit AppState = iota
|
||
StateChecking
|
||
StateConfirmInstall
|
||
StateInstallingNode
|
||
StateConfiguringNpm
|
||
StateInstallingMoltbot
|
||
StateConfiguring
|
||
StateInstalled
|
||
StateMenu
|
||
StateConfigApiSelect
|
||
StateConfigInput
|
||
StateGatewayRunning
|
||
StateUninstallConfirm
|
||
StateUninstalling
|
||
StateError
|
||
)
|
||
|
||
type Model struct {
|
||
state AppState
|
||
spinner spinner.Model
|
||
err error
|
||
logs []string
|
||
nodeVer string
|
||
nodeOk bool
|
||
installMsg string
|
||
quitting bool
|
||
|
||
// Config Wizard
|
||
input textinput.Model
|
||
configOpts sys.ConfigOptions
|
||
configStep int
|
||
menuIndex int
|
||
|
||
nextState AppState
|
||
nextCmd tea.Cmd
|
||
|
||
DidStartGateway bool
|
||
}
|
||
|
||
type checkMsg struct {
|
||
nodeVer string
|
||
nodeOk bool
|
||
needsNode bool
|
||
moltbotVer string
|
||
moltbotInstalled bool
|
||
gatewayRunning bool
|
||
}
|
||
|
||
type installNodeMsg struct{ err error }
|
||
type configNpmMsg struct{ err error }
|
||
type installMoltbotMsg struct {
|
||
version string
|
||
err error
|
||
}
|
||
type configMsg struct {
|
||
restartPath bool
|
||
err error
|
||
}
|
||
type saveConfigMsg struct{ err error }
|
||
type uninstallMsg struct{ err error }
|
||
|
||
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: StateInit,
|
||
spinner: s,
|
||
input: ti,
|
||
logs: []string{},
|
||
}
|
||
}
|
||
|
||
func (m Model) Init() tea.Cmd {
|
||
return tea.Batch(
|
||
m.spinner.Tick,
|
||
func() tea.Msg {
|
||
time.Sleep(500 * time.Millisecond)
|
||
return checkMsg{}
|
||
},
|
||
)
|
||
}
|
||
|
||
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" {
|
||
m.quitting = true
|
||
return m, tea.Quit
|
||
}
|
||
|
||
// Menu Navigation
|
||
if m.state == StateMenu {
|
||
switch msg.String() {
|
||
case "up", "k":
|
||
if m.menuIndex > 0 {
|
||
m.menuIndex--
|
||
}
|
||
case "down", "j":
|
||
if m.menuIndex < 2 {
|
||
m.menuIndex++
|
||
}
|
||
case "enter":
|
||
switch m.menuIndex {
|
||
case 0: // Start
|
||
if sys.IsGatewayRunning() {
|
||
m.state = StateGatewayRunning
|
||
} else {
|
||
sys.StartGateway()
|
||
m.DidStartGateway = true
|
||
return m, tea.Quit
|
||
}
|
||
case 1: // Configure
|
||
m.state = StateConfigApiSelect
|
||
m.configOpts = sys.ConfigOptions{}
|
||
case 2: // Uninstall
|
||
m.state = StateUninstallConfirm
|
||
case 3: // Exit
|
||
return m, tea.Quit
|
||
}
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
// Uninstall Confirm State
|
||
if m.state == StateUninstallConfirm {
|
||
switch msg.String() {
|
||
case "y", "Y":
|
||
m.state = StateUninstalling
|
||
m.logs = append(m.logs, style.RenderStep("➜", "正在卸载 Moltbot 并清理配置...", "running"))
|
||
return m, uninstallCmd
|
||
case "n", "N", "enter":
|
||
m.state = StateMenu
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
// Config API Selection
|
||
if m.state == StateConfigApiSelect {
|
||
switch msg.String() {
|
||
case "1":
|
||
m.configOpts.ApiType = "anthropic"
|
||
m.state = StateConfigInput
|
||
m.configStep = 0
|
||
m.input.Placeholder = "sk-ant-api03-..."
|
||
m.input.EchoMode = textinput.EchoPassword
|
||
m.input.SetValue("")
|
||
case "2":
|
||
m.configOpts.ApiType = "openai"
|
||
m.state = StateConfigInput
|
||
m.configStep = 0
|
||
m.input.Placeholder = "https://api.openai.com/v1"
|
||
m.input.EchoMode = textinput.EchoNormal
|
||
m.input.SetValue("")
|
||
case "q", "esc":
|
||
m.state = StateMenu // Back to menu
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
// Config Input Steps
|
||
if m.state == StateConfigInput {
|
||
switch msg.String() {
|
||
case "enter":
|
||
val := m.input.Value()
|
||
// Step logic:
|
||
// Anthropic: 0(Key) -> 1(TG Token) -> 2(TG ID) -> Finish
|
||
// OpenAI: 0(BaseURL) -> 1(Key) -> 2(Model) -> 3(TG Token) -> 4(TG ID) -> Finish
|
||
|
||
if m.configOpts.ApiType == "anthropic" {
|
||
switch m.configStep {
|
||
case 0: // API Key
|
||
m.configOpts.AnthropicKey = val
|
||
m.configStep++
|
||
m.input.Placeholder = "123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11"
|
||
m.input.EchoMode = textinput.EchoNormal
|
||
m.input.SetValue("")
|
||
case 1: // TG Token
|
||
m.configOpts.BotToken = val
|
||
m.configStep++
|
||
m.input.Placeholder = "123456789"
|
||
m.input.SetValue("")
|
||
case 2: // TG Admin ID
|
||
m.configOpts.AdminID = val
|
||
return m, saveConfigCmd(m.configOpts)
|
||
}
|
||
} else {
|
||
// OpenAI
|
||
switch m.configStep {
|
||
case 0: // BaseURL
|
||
m.configOpts.OpenAIBaseURL = val
|
||
m.configStep++
|
||
m.input.Placeholder = "sk-..."
|
||
m.input.EchoMode = textinput.EchoPassword
|
||
m.input.SetValue("")
|
||
case 1: // Key
|
||
m.configOpts.OpenAIKey = val
|
||
m.configStep++
|
||
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.configStep++
|
||
m.input.Placeholder = "123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11"
|
||
m.input.EchoMode = textinput.EchoNormal
|
||
m.input.SetValue("")
|
||
case 3: // TG Token
|
||
m.configOpts.BotToken = val
|
||
m.configStep++
|
||
m.input.Placeholder = "123456789"
|
||
m.input.SetValue("")
|
||
case 4: // TG Admin ID
|
||
m.configOpts.AdminID = val
|
||
return m, saveConfigCmd(m.configOpts)
|
||
}
|
||
}
|
||
return m, nil
|
||
}
|
||
m.input, cmd = m.input.Update(msg)
|
||
return m, cmd
|
||
}
|
||
|
||
// Gateway Running Conflict State
|
||
if m.state == StateGatewayRunning {
|
||
proceed := false
|
||
switch msg.String() {
|
||
case "y", "Y":
|
||
sys.KillGateway()
|
||
time.Sleep(1 * time.Second)
|
||
sys.StartGateway()
|
||
proceed = true
|
||
case "n", "N", "esc", "enter":
|
||
proceed = true
|
||
}
|
||
|
||
if proceed {
|
||
m.state = m.nextState
|
||
if m.nextState == StateInstallingNode {
|
||
m.logs = append(m.logs, style.RenderStep("➜", "正在安装 Node.js (可能需要管理员权限)...", "running"))
|
||
return m, m.nextCmd
|
||
} else if m.nextState == StateConfiguringNpm {
|
||
m.logs = append(m.logs, style.RenderStep("➜", "正在配置 npm 淘宝镜像...", "running"))
|
||
return m, m.nextCmd
|
||
}
|
||
return m, nil
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
// Confirm Install State
|
||
if m.state == StateConfirmInstall {
|
||
switch msg.String() {
|
||
case "y", "Y":
|
||
m.logs = append(m.logs, style.RenderStep("➜", "开始安装...", "running"))
|
||
if !m.nodeOk {
|
||
m.state = StateInstallingNode
|
||
m.logs = append(m.logs, style.RenderStep("➜", "正在安装 Node.js (可能需要管理员权限)...", "running"))
|
||
return m, installNodeCmd
|
||
}
|
||
m.state = StateConfiguringNpm
|
||
m.logs = append(m.logs, style.RenderStep("➜", "正在配置 npm 淘宝镜像...", "running"))
|
||
return m, configNpmCmd
|
||
case "n", "N", "enter":
|
||
m.logs = append(m.logs, style.RenderStep("!", "跳过安装步骤", "warning"))
|
||
m.state = StateConfiguring
|
||
return m, configCmd
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
// Installed State (Transition to Config)
|
||
if m.state == StateInstalled {
|
||
if msg.String() == "enter" {
|
||
m.state = StateConfigApiSelect
|
||
m.configOpts = sys.ConfigOptions{}
|
||
}
|
||
return m, nil
|
||
}
|
||
|
||
case checkMsg:
|
||
if m.state == StateInit {
|
||
m.state = StateChecking
|
||
return m, checkEnvCmd
|
||
}
|
||
|
||
m.nodeVer = msg.nodeVer
|
||
m.nodeOk = msg.nodeOk
|
||
m.logs = append(m.logs, style.RenderStep("✓", "Windows 系统检测完毕", "done"))
|
||
|
||
if msg.nodeOk {
|
||
m.logs = append(m.logs, style.RenderStep("✓", fmt.Sprintf("发现 Node.js %s", msg.nodeVer), "done"))
|
||
} else {
|
||
if msg.nodeVer != "" {
|
||
m.logs = append(m.logs, style.RenderStep("!", fmt.Sprintf("发现 Node.js %s (需要 v22+)", msg.nodeVer), "warning"))
|
||
} else {
|
||
m.logs = append(m.logs, style.RenderStep("!", "未检测到 Node.js", "warning"))
|
||
}
|
||
}
|
||
|
||
// Determine Next Step
|
||
var nextState AppState
|
||
var nextCmd tea.Cmd
|
||
|
||
if msg.moltbotInstalled {
|
||
m.logs = append(m.logs, style.RenderStep("!", fmt.Sprintf("检测到 Moltbot 已安装 (%s)", msg.moltbotVer), "warning"))
|
||
nextState = StateConfirmInstall
|
||
nextCmd = nil
|
||
} else if !msg.nodeOk {
|
||
nextState = StateInstallingNode
|
||
nextCmd = installNodeCmd
|
||
} else {
|
||
nextState = StateConfiguringNpm
|
||
nextCmd = configNpmCmd
|
||
}
|
||
|
||
m.nextState = nextState
|
||
m.nextCmd = nextCmd
|
||
|
||
// Check Gateway Conflict
|
||
if msg.gatewayRunning {
|
||
m.state = StateGatewayRunning
|
||
return m, nil
|
||
}
|
||
|
||
// Proceed immediately if no conflict
|
||
m.state = nextState
|
||
if nextState == StateInstallingNode {
|
||
m.logs = append(m.logs, style.RenderStep("➜", "正在安装 Node.js (可能需要管理员权限)...", "running"))
|
||
return m, nextCmd
|
||
} else if nextState == StateConfiguringNpm {
|
||
m.logs = append(m.logs, style.RenderStep("➜", "正在配置 npm 淘宝镜像...", "running"))
|
||
return m, nextCmd
|
||
}
|
||
return m, nil
|
||
|
||
case installNodeMsg:
|
||
if msg.err != nil {
|
||
m.err = msg.err
|
||
m.state = StateError
|
||
return m, nil
|
||
}
|
||
m.logs = append(m.logs, style.RenderStep("✓", "Node.js 安装成功", "done"))
|
||
m.state = StateConfiguringNpm
|
||
m.logs = append(m.logs, style.RenderStep("➜", "正在配置 npm 淘宝镜像...", "running"))
|
||
return m, configNpmCmd
|
||
|
||
case configNpmMsg:
|
||
if msg.err != nil {
|
||
m.logs = append(m.logs, style.RenderStep("!", fmt.Sprintf("配置镜像失败 (跳过): %v", msg.err), "warning"))
|
||
} else {
|
||
m.logs = append(m.logs, style.RenderStep("✓", "npm 淘宝镜像配置成功", "done"))
|
||
}
|
||
m.state = StateInstallingMoltbot
|
||
m.logs = append(m.logs, style.RenderStep("➜", "正在安装 Moltbot...", "running"))
|
||
return m, installMoltbotCmd
|
||
|
||
case installMoltbotMsg:
|
||
if msg.err != nil {
|
||
m.err = msg.err
|
||
m.state = StateError
|
||
return m, nil
|
||
}
|
||
if msg.version != "" {
|
||
m.logs = append(m.logs, style.RenderStep("✓", fmt.Sprintf("Moltbot 安装成功 (%s)", msg.version), "done"))
|
||
} else {
|
||
m.logs = append(m.logs, style.RenderStep("✓", "Moltbot 安装成功", "done"))
|
||
}
|
||
m.state = StateConfiguring
|
||
return m, configCmd
|
||
|
||
case configMsg:
|
||
if msg.err != nil {
|
||
m.logs = append(m.logs, style.RenderStep("!", fmt.Sprintf("配置迁移失败: %v", msg.err), "warning"))
|
||
} else {
|
||
m.logs = append(m.logs, style.RenderStep("✓", "配置迁移/初始化完成", "done"))
|
||
}
|
||
if msg.restartPath {
|
||
m.logs = append(m.logs, style.RenderStep("!", "已添加 PATH 环境变量，请重启终端生效", "warning"))
|
||
}
|
||
m.state = StateInstalled
|
||
m.installMsg = getRandomWelcomeMsg()
|
||
return m, nil
|
||
|
||
case saveConfigMsg:
|
||
if msg.err != nil {
|
||
m.logs = append(m.logs, style.RenderStep("!", fmt.Sprintf("保存配置失败: %v", msg.err), "warning"))
|
||
} else {
|
||
m.logs = append(m.logs, style.RenderStep("✓", "配置文件已生成!", "done"))
|
||
m.logs = append(m.logs, style.RenderStep("✓", "配置完成，准备启动", "done"))
|
||
}
|
||
m.state = StateMenu
|
||
m.menuIndex = 0 // Default to Start
|
||
return m, nil
|
||
|
||
case uninstallMsg:
|
||
if msg.err != nil {
|
||
m.logs = append(m.logs, style.RenderStep("!", fmt.Sprintf("卸载失败: %v", msg.err), "warning"))
|
||
} else {
|
||
m.logs = append(m.logs, style.RenderStep("✓", "Moltbot 已卸载并清理配置", "done"))
|
||
}
|
||
m.state = StateMenu
|
||
m.menuIndex = 0
|
||
return m, nil
|
||
|
||
case spinner.TickMsg:
|
||
m.spinner, cmd = m.spinner.Update(msg)
|
||
return m, cmd
|
||
}
|
||
|
||
return m, nil
|
||
}
|
||
|
||
func (m Model) View() string {
|
||
if m.err != nil {
|
||
return fmt.Sprintf("\n%s\n\n%s: %v\n\n按 q 退出\n",
|
||
style.HeaderStyle.Render("Moltbot 安装程序"),
|
||
style.ErrorStyle.Render("发生错误"),
|
||
m.err,
|
||
)
|
||
}
|
||
|
||
s := fmt.Sprintf("\n%s\n\n", style.HeaderStyle.Render("Moltbot 安装程序"))
|
||
|
||
// Show logs for install process
|
||
if m.state != StateMenu && m.state != StateConfigApiSelect && m.state != StateConfigInput {
|
||
for _, log := range m.logs {
|
||
s += log + "\n"
|
||
}
|
||
}
|
||
|
||
// Dynamic Content based on State
|
||
switch m.state {
|
||
case StateInstallingNode, StateConfiguringNpm, StateInstallingMoltbot, StateConfiguring, StateUninstalling:
|
||
s += fmt.Sprintf("\n%s %s\n", m.spinner.View(), style.SubtleStyle.Render("处理中..."))
|
||
|
||
case StateConfirmInstall:
|
||
s += fmt.Sprintf("\n%s\n", style.SubtleStyle.Render("是否强制重新安装/更新？[y/N]"))
|
||
|
||
case StateGatewayRunning:
|
||
s += fmt.Sprintf("\n%s\n", style.SubtleStyle.Render("检测到 Moltbot 网关已在运行 (端口 18789 被占用)"))
|
||
s += fmt.Sprintf("\n%s\n", style.SubtleStyle.Render("是否停止旧进程并重新启动？[y/N]"))
|
||
|
||
case StateUninstallConfirm:
|
||
s += fmt.Sprintf("\n%s\n", style.SubtleStyle.Render("确定要卸载 Moltbot 吗？(这将删除配置文件) [y/N]"))
|
||
|
||
case StateInstalled:
|
||
s += fmt.Sprintf("\n%s\n", style.SuccessStyle.Render("安装完成!"))
|
||
s += style.SubtleStyle.Render(m.installMsg) + "\n\n"
|
||
s += style.StepStyle.Render("按 Enter 进入配置向导") + "\n"
|
||
|
||
case StateMenu:
|
||
s += style.HeaderStyle.Render("主菜单") + "\n\n"
|
||
choices := []string{"启动 Moltbot 网关", "配置 Moltbot", "卸载 Moltbot", "退出"}
|
||
for i, choice := range choices {
|
||
cursor := " "
|
||
if m.menuIndex == i {
|
||
cursor = "➜"
|
||
choice = style.HighlightStyle.Render(choice)
|
||
}
|
||
s += fmt.Sprintf(" %s %s\n", cursor, choice)
|
||
}
|
||
s += "\n" + style.SubtleStyle.Render("使用 ↑/↓ 选择，Enter 确认") + "\n"
|
||
// Show logs below menu if desired, or keep clean
|
||
if len(m.logs) > 0 {
|
||
s += "\n" + style.SubtleStyle.Render("--- 安装日志 ---") + "\n"
|
||
start := len(m.logs) - 3
|
||
if start < 0 {
|
||
start = 0
|
||
}
|
||
for _, log := range m.logs[start:] {
|
||
s += log + "\n"
|
||
}
|
||
}
|
||
|
||
case StateConfigApiSelect:
|
||
s += style.HeaderStyle.Render("配置向导 - 选择 API 类型") + "\n\n"
|
||
s += "1. Anthropic 官方 API (推荐)\n"
|
||
s += "2. OpenAI 兼容 API (中转站/其他模型)\n\n"
|
||
s += style.SubtleStyle.Render("按 1 或 2 选择，Esc 返回") + "\n"
|
||
|
||
case StateConfigInput:
|
||
s += style.HeaderStyle.Render("配置向导") + "\n\n"
|
||
|
||
label := ""
|
||
if m.configOpts.ApiType == "anthropic" {
|
||
switch m.configStep {
|
||
case 0:
|
||
label = "Anthropic API Key (sk-ant-...):"
|
||
case 1:
|
||
label = "Telegram Bot Token (选填, 回车跳过):"
|
||
case 2:
|
||
label = "Telegram User ID (管理员) (选填, 回车跳过):"
|
||
}
|
||
} else {
|
||
// OpenAI
|
||
switch m.configStep {
|
||
case 0:
|
||
label = "API Base URL (例如 https://api.example.com/v1):"
|
||
case 1:
|
||
label = "API Key:"
|
||
case 2:
|
||
label = "模型名称 (例如 gpt-4o):"
|
||
case 3:
|
||
label = "Telegram Bot Token (选填, 回车跳过):"
|
||
case 4:
|
||
label = "Telegram User ID (管理员) (选填, 回车跳过):"
|
||
}
|
||
}
|
||
|
||
s += fmt.Sprintf("%s\n\n%s\n\n", label, m.input.View())
|
||
s += style.SubtleStyle.Render("按 Enter 确认") + "\n"
|
||
if (m.configOpts.ApiType == "anthropic" && m.configStep >= 1) || (m.configOpts.ApiType == "openai" && m.configStep >= 3) {
|
||
s += style.SubtleStyle.Render("跳过后可通过 http://127.0.0.1:18789/ Web UI 交互") + "\n"
|
||
}
|
||
}
|
||
|
||
return style.AppStyle.Render(s)
|
||
}
|
||
|
||
// Commands
|
||
|
||
func checkEnvCmd() tea.Msg {
|
||
var (
|
||
nodeVer string
|
||
nodeOk bool
|
||
moltbotVer string
|
||
moltbotInstalled bool
|
||
gatewayRunning bool
|
||
wg sync.WaitGroup
|
||
)
|
||
|
||
wg.Add(3)
|
||
|
||
go func() {
|
||
defer wg.Done()
|
||
nodeVer, nodeOk = sys.CheckNode()
|
||
}()
|
||
|
||
go func() {
|
||
defer wg.Done()
|
||
moltbotVer, moltbotInstalled = sys.CheckMoltbot()
|
||
}()
|
||
|
||
go func() {
|
||
defer wg.Done()
|
||
gatewayRunning = sys.IsGatewayRunning()
|
||
}()
|
||
|
||
wg.Wait()
|
||
|
||
return checkMsg{
|
||
nodeVer: nodeVer,
|
||
nodeOk: nodeOk,
|
||
needsNode: !nodeOk,
|
||
moltbotVer: moltbotVer,
|
||
moltbotInstalled: moltbotInstalled,
|
||
gatewayRunning: gatewayRunning,
|
||
}
|
||
}
|
||
|
||
func installNodeCmd() tea.Msg {
|
||
err := sys.InstallNode()
|
||
return installNodeMsg{err: err}
|
||
}
|
||
|
||
func configNpmCmd() tea.Msg {
|
||
err := sys.ConfigureNpmMirror()
|
||
return configNpmMsg{err: err}
|
||
}
|
||
|
||
func installMoltbotCmd() tea.Msg {
|
||
err := sys.InstallMoltbotNpm("latest")
|
||
if err != nil {
|
||
return installMoltbotMsg{err: err}
|
||
}
|
||
ver, _ := sys.CheckMoltbot()
|
||
return installMoltbotMsg{version: ver, err: nil}
|
||
}
|
||
|
||
func configCmd() tea.Msg {
|
||
restart, _ := sys.EnsureOnPath()
|
||
sys.RunDoctor()
|
||
return configMsg{restartPath: restart, err: nil}
|
||
}
|
||
|
||
func saveConfigCmd(opts sys.ConfigOptions) tea.Cmd {
|
||
return func() tea.Msg {
|
||
err := sys.GenerateAndWriteConfig(opts)
|
||
return saveConfigMsg{err: err}
|
||
}
|
||
}
|
||
|
||
func uninstallCmd() tea.Msg {
|
||
err := sys.UninstallMoltbot()
|
||
return uninstallMsg{err: err}
|
||
}
|
||
|
||
func getRandomWelcomeMsg() string {
|
||
msgs := []string{
|
||
"所有系统准备就绪",
|
||
"Moltbot 已就绪，随时为您服务",
|
||
"环境配置完成，开始使用吧",
|
||
"安装成功，期待您的使用",
|
||
}
|
||
return msgs[rand.Intn(len(msgs))]
|
||
}
|