436 lines
10 KiB
Go
436 lines
10 KiB
Go
package chatlog
|
|
|
|
import (
|
|
"fmt"
|
|
"runtime"
|
|
"time"
|
|
|
|
"github.com/sjzar/chatlog/internal/chatlog/ctx"
|
|
"github.com/sjzar/chatlog/internal/ui/footer"
|
|
"github.com/sjzar/chatlog/internal/ui/help"
|
|
"github.com/sjzar/chatlog/internal/ui/infobar"
|
|
"github.com/sjzar/chatlog/internal/ui/menu"
|
|
|
|
"github.com/gdamore/tcell/v2"
|
|
"github.com/rivo/tview"
|
|
)
|
|
|
|
const (
|
|
RefreshInterval = 1000 * time.Millisecond
|
|
)
|
|
|
|
type App struct {
|
|
*tview.Application
|
|
|
|
ctx *ctx.Context
|
|
m *Manager
|
|
stopRefresh chan struct{}
|
|
|
|
// page
|
|
mainPages *tview.Pages
|
|
infoBar *infobar.InfoBar
|
|
tabPages *tview.Pages
|
|
footer *footer.Footer
|
|
|
|
// tab
|
|
menu *menu.Menu
|
|
help *help.Help
|
|
activeTab int
|
|
tabCount int
|
|
}
|
|
|
|
func NewApp(ctx *ctx.Context, m *Manager) *App {
|
|
app := &App{
|
|
ctx: ctx,
|
|
m: m,
|
|
Application: tview.NewApplication(),
|
|
mainPages: tview.NewPages(),
|
|
infoBar: infobar.New(),
|
|
tabPages: tview.NewPages(),
|
|
footer: footer.New(),
|
|
menu: menu.New("主菜单"),
|
|
help: help.New(),
|
|
}
|
|
|
|
app.initMenu()
|
|
|
|
return app
|
|
}
|
|
|
|
func (a *App) Run() error {
|
|
|
|
flex := tview.NewFlex().
|
|
SetDirection(tview.FlexRow).
|
|
AddItem(a.infoBar, infobar.InfoBarViewHeight, 0, false).
|
|
AddItem(a.tabPages, 0, 1, true).
|
|
AddItem(a.footer, 1, 1, false)
|
|
|
|
a.mainPages.AddPage("main", flex, true, true)
|
|
|
|
a.tabPages.
|
|
AddPage("0", a.menu, true, true).
|
|
AddPage("1", a.help, true, false)
|
|
a.tabCount = 2
|
|
|
|
a.SetInputCapture(a.inputCapture)
|
|
|
|
go a.refresh()
|
|
|
|
if err := a.SetRoot(a.mainPages, true).EnableMouse(false).Run(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *App) Stop() {
|
|
// 添加一个通道用于停止刷新 goroutine
|
|
if a.stopRefresh != nil {
|
|
close(a.stopRefresh)
|
|
}
|
|
a.Application.Stop()
|
|
}
|
|
|
|
func (a *App) switchTab(step int) {
|
|
index := (a.activeTab + step) % a.tabCount
|
|
if index < 0 {
|
|
index = a.tabCount - 1
|
|
}
|
|
a.activeTab = index
|
|
a.tabPages.SwitchToPage(fmt.Sprint(a.activeTab))
|
|
}
|
|
|
|
func (a *App) refresh() {
|
|
tick := time.NewTicker(RefreshInterval)
|
|
defer tick.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-a.stopRefresh:
|
|
return
|
|
case <-tick.C:
|
|
a.infoBar.UpdateAccount(a.ctx.Account)
|
|
a.infoBar.UpdateBasicInfo(a.ctx.PID, a.ctx.FullVersion, a.ctx.ExePath)
|
|
a.infoBar.UpdateStatus(a.ctx.Status)
|
|
a.infoBar.UpdateDataKey(a.ctx.DataKey)
|
|
a.infoBar.UpdateDataUsageDir(a.ctx.DataUsage, a.ctx.DataDir)
|
|
a.infoBar.UpdateWorkUsageDir(a.ctx.WorkUsage, a.ctx.WorkDir)
|
|
if a.ctx.HTTPEnabled {
|
|
a.infoBar.UpdateHTTPServer(fmt.Sprintf("[green][已启动][white] [%s]", a.ctx.HTTPAddr))
|
|
} else {
|
|
a.infoBar.UpdateHTTPServer("[未启动]")
|
|
}
|
|
|
|
a.Draw()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (a *App) inputCapture(event *tcell.EventKey) *tcell.EventKey {
|
|
|
|
// 如果当前页面不是主页面，ESC 键返回主页面
|
|
if a.mainPages.HasPage("submenu") && event.Key() == tcell.KeyEscape {
|
|
a.mainPages.RemovePage("submenu")
|
|
a.mainPages.SwitchToPage("main")
|
|
return nil
|
|
}
|
|
|
|
if a.tabPages.HasFocus() {
|
|
switch event.Key() {
|
|
case tcell.KeyLeft:
|
|
a.switchTab(-1)
|
|
return nil
|
|
case tcell.KeyRight:
|
|
a.switchTab(1)
|
|
return nil
|
|
}
|
|
}
|
|
|
|
switch event.Key() {
|
|
case tcell.KeyCtrlC:
|
|
a.Stop()
|
|
}
|
|
|
|
return event
|
|
}
|
|
|
|
func (a *App) initMenu() {
|
|
getDataKey := &menu.Item{
|
|
Index: 2,
|
|
Name: "获取数据密钥",
|
|
Description: "从进程获取数据密钥",
|
|
Selected: func(i *menu.Item) {
|
|
modal := tview.NewModal()
|
|
if runtime.GOOS == "darwin" {
|
|
modal.SetText("获取数据密钥中...\n预计需要 20 秒左右的时间，期间微信会卡住，请耐心等待")
|
|
} else {
|
|
modal.SetText("获取数据密钥中...")
|
|
}
|
|
a.mainPages.AddPage("modal", modal, true, true)
|
|
a.SetFocus(modal)
|
|
|
|
go func() {
|
|
err := a.m.GetDataKey()
|
|
|
|
// 在主线程中更新UI
|
|
a.QueueUpdateDraw(func() {
|
|
if err != nil {
|
|
// 解密失败
|
|
modal.SetText("获取数据密钥失败: " + err.Error())
|
|
} else {
|
|
// 解密成功
|
|
modal.SetText("获取数据密钥成功")
|
|
}
|
|
|
|
// 添加确认按钮
|
|
modal.AddButtons([]string{"OK"})
|
|
modal.SetDoneFunc(func(buttonIndex int, buttonLabel string) {
|
|
a.mainPages.RemovePage("modal")
|
|
})
|
|
a.SetFocus(modal)
|
|
})
|
|
}()
|
|
},
|
|
}
|
|
|
|
decryptData := &menu.Item{
|
|
Index: 3,
|
|
Name: "解密数据",
|
|
Description: "解密数据文件",
|
|
Selected: func(i *menu.Item) {
|
|
// 创建一个没有按钮的模态框，显示"解密中..."
|
|
modal := tview.NewModal().
|
|
SetText("解密中...")
|
|
|
|
a.mainPages.AddPage("modal", modal, true, true)
|
|
a.SetFocus(modal)
|
|
|
|
// 在后台执行解密操作
|
|
go func() {
|
|
// 执行解密
|
|
err := a.m.DecryptDBFiles()
|
|
|
|
// 在主线程中更新UI
|
|
a.QueueUpdateDraw(func() {
|
|
if err != nil {
|
|
// 解密失败
|
|
modal.SetText("解密失败: " + err.Error())
|
|
} else {
|
|
// 解密成功
|
|
modal.SetText("解密数据成功")
|
|
}
|
|
|
|
// 添加确认按钮
|
|
modal.AddButtons([]string{"OK"})
|
|
modal.SetDoneFunc(func(buttonIndex int, buttonLabel string) {
|
|
a.mainPages.RemovePage("modal")
|
|
})
|
|
a.SetFocus(modal)
|
|
})
|
|
}()
|
|
},
|
|
}
|
|
|
|
httpServer := &menu.Item{
|
|
Index: 4,
|
|
Name: "启动 HTTP 服务",
|
|
Description: "启动本地 HTTP & MCP 服务器",
|
|
Selected: func(i *menu.Item) {
|
|
modal := tview.NewModal()
|
|
|
|
// 根据当前服务状态执行不同操作
|
|
if !a.ctx.HTTPEnabled {
|
|
// HTTP 服务未启动，启动服务
|
|
modal.SetText("正在启动 HTTP 服务...")
|
|
a.mainPages.AddPage("modal", modal, true, true)
|
|
a.SetFocus(modal)
|
|
|
|
// 在后台启动服务
|
|
go func() {
|
|
err := a.m.StartService()
|
|
|
|
// 在主线程中更新UI
|
|
a.QueueUpdateDraw(func() {
|
|
if err != nil {
|
|
// 启动失败
|
|
modal.SetText("启动 HTTP 服务失败: " + err.Error())
|
|
} else {
|
|
// 启动成功
|
|
modal.SetText("已启动 HTTP 服务")
|
|
// 更改菜单项名称
|
|
i.Name = "停止 HTTP 服务"
|
|
i.Description = "停止本地 HTTP & MCP 服务器"
|
|
}
|
|
|
|
// 添加确认按钮
|
|
modal.AddButtons([]string{"OK"})
|
|
modal.SetDoneFunc(func(buttonIndex int, buttonLabel string) {
|
|
a.mainPages.RemovePage("modal")
|
|
})
|
|
a.SetFocus(modal)
|
|
})
|
|
}()
|
|
} else {
|
|
// HTTP 服务已启动，停止服务
|
|
modal.SetText("正在停止 HTTP 服务...")
|
|
a.mainPages.AddPage("modal", modal, true, true)
|
|
a.SetFocus(modal)
|
|
|
|
// 在后台停止服务
|
|
go func() {
|
|
err := a.m.StopService()
|
|
|
|
// 在主线程中更新UI
|
|
a.QueueUpdateDraw(func() {
|
|
if err != nil {
|
|
// 停止失败
|
|
modal.SetText("停止 HTTP 服务失败: " + err.Error())
|
|
} else {
|
|
// 停止成功
|
|
modal.SetText("已停止 HTTP 服务")
|
|
// 更改菜单项名称
|
|
i.Name = "启动 HTTP 服务"
|
|
i.Description = "启动本地 HTTP & MCP 服务器"
|
|
}
|
|
|
|
// 添加确认按钮
|
|
modal.AddButtons([]string{"OK"})
|
|
modal.SetDoneFunc(func(buttonIndex int, buttonLabel string) {
|
|
a.mainPages.RemovePage("modal")
|
|
})
|
|
a.SetFocus(modal)
|
|
})
|
|
}()
|
|
}
|
|
},
|
|
}
|
|
|
|
setting := &menu.Item{
|
|
Index: 5,
|
|
Name: "设置",
|
|
Description: "设置应用程序选项",
|
|
Selected: a.settingSelected,
|
|
}
|
|
|
|
a.menu.AddItem(setting)
|
|
a.menu.AddItem(getDataKey)
|
|
a.menu.AddItem(decryptData)
|
|
a.menu.AddItem(httpServer)
|
|
|
|
a.menu.AddItem(&menu.Item{
|
|
Index: 6,
|
|
Name: "退出",
|
|
Description: "退出程序",
|
|
Selected: func(i *menu.Item) {
|
|
a.Stop()
|
|
},
|
|
})
|
|
}
|
|
|
|
// settingItem 表示一个设置项
|
|
type settingItem struct {
|
|
name string
|
|
description string
|
|
action func()
|
|
}
|
|
|
|
func (a *App) settingSelected(i *menu.Item) {
|
|
|
|
settings := []settingItem{
|
|
{
|
|
name: "设置 HTTP 服务端口",
|
|
description: "配置 HTTP 服务监听的端口",
|
|
action: a.settingHTTPPort,
|
|
},
|
|
{
|
|
name: "设置工作目录",
|
|
description: "配置数据解密后的存储目录",
|
|
action: a.settingWorkDir,
|
|
},
|
|
}
|
|
|
|
subMenu := menu.NewSubMenu("设置")
|
|
for idx, setting := range settings {
|
|
item := &menu.Item{
|
|
Index: idx + 1,
|
|
Name: setting.name,
|
|
Description: setting.description,
|
|
Selected: func(action func()) func(*menu.Item) {
|
|
return func(*menu.Item) {
|
|
action()
|
|
}
|
|
}(setting.action),
|
|
}
|
|
subMenu.AddItem(item)
|
|
}
|
|
|
|
a.mainPages.AddPage("submenu", subMenu, true, true)
|
|
a.SetFocus(subMenu)
|
|
}
|
|
|
|
// settingHTTPPort 设置 HTTP 端口
|
|
func (a *App) settingHTTPPort() {
|
|
// 实现端口设置逻辑
|
|
// 这里可以使用 tview.InputField 让用户输入端口
|
|
form := tview.NewForm().
|
|
AddInputField("端口", a.ctx.HTTPAddr, 20, nil, func(text string) {
|
|
a.ctx.SetHTTPAddr(text)
|
|
}).
|
|
AddButton("保存", func() {
|
|
a.mainPages.RemovePage("submenu2")
|
|
a.showInfo("HTTP 端口已设置为 " + a.ctx.HTTPAddr)
|
|
}).
|
|
AddButton("取消", func() {
|
|
a.mainPages.RemovePage("submenu2")
|
|
})
|
|
form.SetBorder(true).SetTitle("设置 HTTP 端口")
|
|
|
|
a.mainPages.AddPage("submenu2", form, true, true)
|
|
a.SetFocus(form)
|
|
}
|
|
|
|
// settingWorkDir 设置工作目录
|
|
func (a *App) settingWorkDir() {
|
|
// 实现工作目录设置逻辑
|
|
form := tview.NewForm().
|
|
AddInputField("工作目录", a.ctx.WorkDir, 40, nil, func(text string) {
|
|
a.ctx.SetWorkDir(text)
|
|
}).
|
|
AddButton("保存", func() {
|
|
a.mainPages.RemovePage("submenu2")
|
|
a.showInfo("工作目录已设置为 " + a.ctx.WorkDir)
|
|
}).
|
|
AddButton("取消", func() {
|
|
a.mainPages.RemovePage("submenu2")
|
|
})
|
|
form.SetBorder(true).SetTitle("设置工作目录")
|
|
|
|
a.mainPages.AddPage("submenu2", form, true, true)
|
|
a.SetFocus(form)
|
|
}
|
|
|
|
// showModal 显示一个模态对话框
|
|
func (a *App) showModal(text string, buttons []string, doneFunc func(buttonIndex int, buttonLabel string)) {
|
|
modal := tview.NewModal().
|
|
SetText(text).
|
|
AddButtons(buttons).
|
|
SetDoneFunc(doneFunc)
|
|
|
|
a.mainPages.AddPage("modal", modal, true, true)
|
|
a.SetFocus(modal)
|
|
}
|
|
|
|
// showError 显示错误对话框
|
|
func (a *App) showError(err error) {
|
|
a.showModal(err.Error(), []string{"OK"}, func(buttonIndex int, buttonLabel string) {
|
|
a.mainPages.RemovePage("modal")
|
|
})
|
|
}
|
|
|
|
// showInfo 显示信息对话框
|
|
func (a *App) showInfo(text string) {
|
|
a.showModal(text, []string{"OK"}, func(buttonIndex int, buttonLabel string) {
|
|
a.mainPages.RemovePage("modal")
|
|
})
|
|
}
|