307 lines
9.4 KiB
Go
307 lines
9.4 KiB
Go
// arcrun-tray — Arcrun RAG 桌面托盤殼（CP-1 第 6 關 daemon 產品化，leo 2026-07-21 定 fyne）。
|
||
//
|
||
// 目的：讓「不開 terminal 的人」也能裝、開關、選資料夾——一個選單列/系統匣 icon 就是全部。
|
||
// 它看守 `collector direct`（子行程，見 supervisor 套件的誠實界定）並顯示狀態。
|
||
//
|
||
// ⚠️ 本檔是 fyne GUI（CGo＋系統 webview/GL），**無法在無螢幕 sandbox build/驗**——
|
||
//
|
||
// build/簽章/公證/TCC 授權＝leo/地端在真機做（紅線②③）。純邏輯（看守/config）已抽到
|
||
// supervisor 套件並在 sandbox `go test` 驗過。
|
||
//
|
||
// 真機 build：
|
||
//
|
||
// cd collector/cmd/arcrun-tray && go mod tidy
|
||
// go run fyne.io/fyne/v2/cmd/fyne@latest package -os darwin -icon icon.png -name "Arcrun RAG"
|
||
// （Windows：-os windows；簽章/公證另見同目錄 README）
|
||
package main
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
neturl "net/url"
|
||
"os"
|
||
"path/filepath"
|
||
|
||
"arcrun-rag/collector/supervisor"
|
||
|
||
"fyne.io/fyne/v2"
|
||
"fyne.io/fyne/v2/app"
|
||
"fyne.io/fyne/v2/container"
|
||
"fyne.io/fyne/v2/dialog"
|
||
"fyne.io/fyne/v2/driver/desktop"
|
||
"fyne.io/fyne/v2/theme"
|
||
"fyne.io/fyne/v2/widget"
|
||
)
|
||
|
||
// directConfig 是 collector direct 的設定（與 collector/direct.go 的 DirectConfig 同結構；
|
||
// 這裡只需讀寫 watch_folder，其餘由安裝器帶入）。
|
||
type directConfig struct {
|
||
WatchFolder string `json:"watch_folder,omitempty"` // 單數舊制（第一個資料夾的鏡像，維持相容）
|
||
WatchFolders []string `json:"watch_folders,omitempty"` // 多資料夾（daemon-beta task 1；完整勾選 UI＝task 7）
|
||
Manifest string `json:"manifest"`
|
||
CypherURL string `json:"cypher_url"`
|
||
Namespace string `json:"namespace"`
|
||
APIKey string `json:"api_key,omitempty"`
|
||
Library string `json:"library,omitempty"`
|
||
IngestWF string `json:"ingest_workflow,omitempty"`
|
||
RemovedWF string `json:"removed_workflow,omitempty"`
|
||
PollSec int `json:"poll_interval_sec,omitempty"`
|
||
MaxRemoved float64 `json:"max_removed_ratio,omitempty"`
|
||
}
|
||
|
||
// appDir 是設定與 manifest 落地處：~/.arcrun-rag/
|
||
func appDir() string {
|
||
home, _ := os.UserHomeDir()
|
||
return filepath.Join(home, ".arcrun-rag")
|
||
}
|
||
|
||
func configPath() string { return filepath.Join(appDir(), "config.json") }
|
||
|
||
// defaultWatchFolder 預設 ~/ArcrunRAG——**刻意避開 ~/Documents**：macOS launchd 背景程序碰
|
||
// ~/Documents 會被 TCC 擋（07-19 實撞）。預設在非 TCC 禁區＝零授權即可用；用戶要改到 Documents
|
||
// 才需走 TCC 同意流（見 README）。
|
||
func defaultWatchFolder() string {
|
||
home, _ := os.UserHomeDir()
|
||
return filepath.Join(home, "ArcrunRAG")
|
||
}
|
||
|
||
func loadConfig() *directConfig {
|
||
c := &directConfig{}
|
||
if data, err := os.ReadFile(configPath()); err == nil {
|
||
_ = json.Unmarshal(data, c)
|
||
}
|
||
if c.WatchFolder == "" && len(c.WatchFolders) == 0 {
|
||
c.WatchFolder = defaultWatchFolder()
|
||
}
|
||
if c.WatchFolder == "" && len(c.WatchFolders) > 0 {
|
||
c.WatchFolder = c.WatchFolders[0] // 單數欄位＝第一根鏡像（舊 collector 相容）
|
||
}
|
||
if c.Manifest == "" {
|
||
c.Manifest = filepath.Join(appDir(), "manifest.json")
|
||
}
|
||
return c
|
||
}
|
||
|
||
// addWatchFolder 把資料夾加進監看清單（去重、保序），並維持單數欄位＝第一根的鏡像。
|
||
// 完整的「勾選/移除」UI 是 task 7；本函式先保證資料層正確。
|
||
func addWatchFolder(c *directConfig, p string) {
|
||
if p == "" {
|
||
return
|
||
}
|
||
if len(c.WatchFolders) == 0 && c.WatchFolder != "" && c.WatchFolder != defaultWatchFolder() {
|
||
c.WatchFolders = []string{c.WatchFolder}
|
||
}
|
||
for _, f := range c.WatchFolders {
|
||
if f == p {
|
||
return
|
||
}
|
||
}
|
||
c.WatchFolders = append(c.WatchFolders, p)
|
||
c.WatchFolder = c.WatchFolders[0]
|
||
}
|
||
|
||
// Folders 回傳監看根清單（正規化：單數舊制併入、去重、保序；與 collector DirectConfig 同語意）。
|
||
func (c *directConfig) Folders() []string {
|
||
seen := map[string]bool{}
|
||
var out []string
|
||
add := func(p string) {
|
||
if p == "" || seen[p] {
|
||
return
|
||
}
|
||
seen[p] = true
|
||
out = append(out, p)
|
||
}
|
||
add(c.WatchFolder)
|
||
for _, f := range c.WatchFolders {
|
||
add(f)
|
||
}
|
||
return out
|
||
}
|
||
|
||
// removeWatchFolder 把資料夾移出監看清單，並維持單數欄位＝第一根鏡像（清單空＝兩欄皆空）。
|
||
func removeWatchFolder(c *directConfig, p string) {
|
||
if len(c.WatchFolders) == 0 && c.WatchFolder == p {
|
||
c.WatchFolder = ""
|
||
return
|
||
}
|
||
var out []string
|
||
for _, f := range c.WatchFolders {
|
||
if f != p {
|
||
out = append(out, f)
|
||
}
|
||
}
|
||
c.WatchFolders = out
|
||
if len(out) > 0 {
|
||
c.WatchFolder = out[0]
|
||
} else {
|
||
c.WatchFolder = ""
|
||
}
|
||
}
|
||
|
||
func saveConfig(c *directConfig) error {
|
||
if err := os.MkdirAll(appDir(), 0o755); err != nil {
|
||
return err
|
||
}
|
||
data, _ := json.MarshalIndent(c, "", " ")
|
||
return os.WriteFile(configPath(), data, 0o600)
|
||
}
|
||
|
||
// collectorBinPath 找同綑的 collector 執行檔（app bundle 內與托盤同層）。
|
||
func collectorBinPath() string {
|
||
exe, err := os.Executable()
|
||
if err != nil {
|
||
return "arcrun-collector"
|
||
}
|
||
name := "arcrun-collector"
|
||
if isWindows() {
|
||
name += ".exe"
|
||
}
|
||
return filepath.Join(filepath.Dir(exe), name)
|
||
}
|
||
|
||
func isWindows() bool { return os.PathSeparator == '\\' }
|
||
|
||
func main() {
|
||
a := app.NewWithID("dev.arcrun.rag.tray")
|
||
cfg := loadConfig()
|
||
|
||
// 設定視窗（平時隱藏；選資料夾/看說明時開）
|
||
win := a.NewWindow("Arcrun RAG")
|
||
win.SetCloseIntercept(func() { win.Hide() }) // 關窗只隱藏，不結束 app（托盤續跑）
|
||
win.Resize(fyne.NewSize(420, 220))
|
||
|
||
sup := supervisor.New(collectorBinPath(), configPath())
|
||
|
||
// 狀態列（tray 選單第一項，只讀）
|
||
statusItem := fyne.NewMenuItem("狀態：尚未開始", nil)
|
||
statusItem.Disabled = true
|
||
|
||
desk, hasTray := a.(desktop.App)
|
||
|
||
// rebuildTray 依現況重建整個 tray 選單（daemon-beta t7 多資料夾）：
|
||
// 每個看守中的資料夾一列（點＝在 Finder 打開）＋各自的「停止看守」項；
|
||
// 重設選單＝刷新（比 (*Menu).Refresh() 跨版本更穩）。
|
||
var rebuildTray func()
|
||
|
||
restartWatch := func() {
|
||
sup.Stop()
|
||
sup.Start()
|
||
}
|
||
|
||
addAction := func() {
|
||
win.Show()
|
||
win.RequestFocus()
|
||
dialog.ShowFolderOpen(func(uri fyne.ListableURI, err error) {
|
||
if err != nil || uri == nil {
|
||
return
|
||
}
|
||
addWatchFolder(cfg, uri.Path())
|
||
if err := saveConfig(cfg); err != nil {
|
||
dialog.ShowError(err, win)
|
||
return
|
||
}
|
||
restartWatch()
|
||
rebuildTray()
|
||
win.Hide()
|
||
}, win)
|
||
}
|
||
|
||
var pauseItem *fyne.MenuItem
|
||
pauseItem = fyne.NewMenuItem("暫停看守", func() {
|
||
st := sup.Status()
|
||
if st.State == supervisor.StateStopped {
|
||
sup.Start()
|
||
pauseItem.Label = "暫停看守"
|
||
} else {
|
||
sup.Stop()
|
||
pauseItem.Label = "繼續看守"
|
||
}
|
||
rebuildTray()
|
||
})
|
||
|
||
rebuildTray = func() {
|
||
if !hasTray {
|
||
return
|
||
}
|
||
items := []*fyne.MenuItem{statusItem, fyne.NewMenuItemSeparator()}
|
||
for _, f := range cfg.Folders() {
|
||
folder := f // capture
|
||
it := fyne.NewMenuItem("📁 "+filepath.Base(folder), func() {
|
||
if u, err := neturl.Parse("file://" + folder); err == nil {
|
||
_ = a.OpenURL(u)
|
||
}
|
||
})
|
||
it.ChildMenu = fyne.NewMenu("",
|
||
fyne.NewMenuItem("停止看守這個資料夾", func() {
|
||
removeWatchFolder(cfg, folder)
|
||
if err := saveConfig(cfg); err != nil {
|
||
dialog.ShowError(err, win)
|
||
return
|
||
}
|
||
restartWatch()
|
||
rebuildTray()
|
||
}),
|
||
)
|
||
items = append(items, it)
|
||
}
|
||
items = append(items,
|
||
fyne.NewMenuItem("＋ 新增知識資料夾…", addAction),
|
||
fyne.NewMenuItemSeparator(),
|
||
pauseItem,
|
||
)
|
||
desk.SetSystemTrayMenu(fyne.NewMenu("Arcrun RAG", items...))
|
||
}
|
||
|
||
if hasTray {
|
||
desk.SetSystemTrayIcon(theme.StorageIcon()) // TODO 換品牌 icon（leo logo 進行中）；Mac 建議 template 單色
|
||
}
|
||
rebuildTray() // 開機即建初始選單（否則第一次狀態回呼前選單是空的）
|
||
refreshTray := rebuildTray
|
||
|
||
// 狀態變更 → 刷新 tray 文案（回呼在 supervisor goroutine，切回 UI thread）
|
||
sup.SetOnChange(func(s supervisor.Status) {
|
||
// 此版 fyne 無 fyne.Do；tray menu label 更新直接做即可
|
||
// （2026-07-21 真機 build 實測修正）
|
||
statusItem.Label = "狀態：" + humanStatus(s)
|
||
refreshTray()
|
||
})
|
||
|
||
// 設定視窗內容：說明現況 + 選資料夾按鈕（給會開窗的人；不會開的人靠 tray 選單）
|
||
win.SetContent(container.NewVBox(
|
||
widget.NewLabelWithStyle("Arcrun RAG 知識同步", fyne.TextAlignCenter, fyne.TextStyle{Bold: true}),
|
||
widget.NewLabel("把檔案丟進你選的資料夾，就會自動進你的知識庫。\n這個程式會待在選單列/系統匣，關掉視窗它仍在背景看守。"),
|
||
widget.NewButton("新增知識資料夾…", func() { addAction() }),
|
||
))
|
||
|
||
// 開機即看守（若設定完整）
|
||
if cfg.CypherURL != "" && cfg.Namespace != "" {
|
||
sup.Start()
|
||
} else {
|
||
// 缺安裝器帶入的 cypher_url/namespace：提示用戶先完成安裝
|
||
statusItem.Label = "狀態：尚未連結（請先完成安裝）"
|
||
win.Show()
|
||
}
|
||
|
||
a.Lifecycle().SetOnStopped(func() { sup.Stop() })
|
||
a.Run()
|
||
}
|
||
|
||
// humanStatus 把狀態機轉成白話（給不懂內部的人）。
|
||
func humanStatus(s supervisor.Status) string {
|
||
switch s.State {
|
||
case supervisor.StateWatching:
|
||
if !s.LastRoundAt.IsZero() {
|
||
return fmt.Sprintf("看守中 · 上次同步 %s", s.LastRoundAt.Local().Format("15:04"))
|
||
}
|
||
return "看守中"
|
||
case supervisor.StateStarting:
|
||
return "啟動中…"
|
||
case supervisor.StateError:
|
||
return "暫時出錯，正在自動重試"
|
||
case supervisor.StateStopped:
|
||
return "已暫停"
|
||
default:
|
||
return "尚未開始"
|
||
}
|
||
}
|