Leo
f10d10747c
事故：daemon 完全沒有 vault 辨識，把萃取卡片寫進使用者的 Logseq vault （system-dev/wiki/cards/ 對一般資料夾沒事，對 vault＝憑空多出頁面， 2026-08-10 leo 實撞：25 個簡體字卡片污染 ~/Documents/KB）。 修法（照票上指示搬現成的，不重新設計）： - collector/vault.go：DetectVaultType/IsVault，判準逐條抄自 system-dev-template/scripts/install.sh:209-221（先查 logseq/、再查 .obsidian/）。 已用同一批 fixture 資料夾跑過 install.sh 與這支 Go 版，五種情境（logseq／ obsidian／一般資料夾／兩者皆有／空資料夾）IS_VAULT 判斷逐一比對一致。 - extract.go：cardsRelDirFor() 依 IsVault 決定卡片相對路徑——非 vault 不變 （system-dev/wiki/cards/），vault 改落 .arcrun-rag/wiki/cards/（點開頭隱藏 目錄，Logseq/Obsidian 預設不掃描，跟 daemon 自己 scan.go 的隱藏目錄跳過規則 一致）。extract_workersai.go／extract_gemma.go／direct.go 的落卡與下架清除 都改用這個函式，三處對同一個 absRoot 保證同一個答案。 - safewrite.go：safeWriteCard() 取代兩處無條件 os.WriteFile——目標已存在且 內容不同就先備份成 <dest>.bak-<unixnano> 才覆寫；內容相同則不動（不產生 垃圾備份）；備份失敗就整個中止，不無聲蓋掉使用者機器上已有的東西。 驗證（見 PR/commit 說明附的實測輸出）： - 用 fake Logseq vault fixture 重現舊行為（卡片確實落在 system-dev/wiki/cards/）， 再用同一份 fixture 驗新行為（卡片改落 .arcrun-rag/wiki/cards/，vault 根目錄 非隱藏 .md 數量不變、journals/ 原稿位元不動）。 - 故意放同名既有卡片，跑完既有內容被備份、新內容確實寫入，未無聲遺失。 - 非 vault 既有測試（extract_gemma_test.go 原有三支）全數不動照過，確認 一般資料夾行為零改變。 範圍外（留給下一輪）：ExtractWithClaude（extract.go 的 claude 路）與其 templatefs/.claude/commands/rag-extract-file.md 技能檔仍硬寫 system-dev/wiki/cards/， 但這條路目前在 direct.go 的 RunDirectOnce 是不支援狀態（cfg.Extractor 只認 workers-ai／gemma），非本次事故的作用路徑，故未動。 不影響：leo 機器上的 daemon（未重啟、未重新打包）；此修復要生效還要 ①總管審過併 main ②重新打包桌面版裝上他機器 ③總管用新二進位實測， 三件缺一不可（見 issue #60 leo 的重啟條件）。
100 lines
3.0 KiB
Go
100 lines
3.0 KiB
Go
// extract_workersai_test.go — arcrun-rag#60：workers-ai 是預設/主線萃取路（t181），
|
||
// 同一套 vault 保護要對這條路也成立，不能只顧 gemma。
|
||
package collector
|
||
|
||
import (
|
||
"encoding/json"
|
||
"net/http"
|
||
"net/http/httptest"
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
"testing"
|
||
)
|
||
|
||
func workersAIStub(t *testing.T, handler http.HandlerFunc) (url string, closeFn func()) {
|
||
t.Helper()
|
||
srv := httptest.NewServer(handler)
|
||
return srv.URL, srv.Close
|
||
}
|
||
|
||
func TestExtractWithWorkersAI_VaultRedirectsAndDoesNotClobber(t *testing.T) {
|
||
root := t.TempDir()
|
||
mustMkdir(t, filepath.Join(root, ".obsidian")) // Obsidian vault
|
||
srcRel := "note.md"
|
||
if err := os.WriteFile(filepath.Join(root, srcRel), []byte("# 原稿"), 0o644); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
// 隱藏卡片目錄裡先放一份既有內容，驗證不會被無聲蓋掉。
|
||
cardDir := filepath.Join(root, ".arcrun-rag", "wiki", "cards")
|
||
mustMkdir(t, cardDir)
|
||
preexisting := "# note\n既有內容"
|
||
if err := os.WriteFile(filepath.Join(cardDir, "note.md"), []byte(preexisting), 0o644); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
url, closeFn := workersAIStub(t, func(w http.ResponseWriter, r *http.Request) {
|
||
if r.Header.Get("X-Arcrun-API-Key") != "key123" {
|
||
t.Errorf("api key 未帶到 header")
|
||
}
|
||
_ = json.NewEncoder(w).Encode(map[string]any{
|
||
"success": true,
|
||
"card": "# note\n## 一句話定義\n新卡\n",
|
||
})
|
||
})
|
||
defer closeFn()
|
||
|
||
cards, err := ExtractWithWorkersAI(url, "key123", root, srcRel)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if len(cards) != 1 || cards[0] != ".arcrun-rag/wiki/cards/note.md" {
|
||
t.Fatalf("vault 目標的卡片路徑不對：%v", cards)
|
||
}
|
||
|
||
// pages/.obsidian 之外沒有新增任何非隱藏 .md（Obsidian 不掃描 .arcrun-rag/）。
|
||
topLevel := countMD(t, root)
|
||
if topLevel != 1 { // 只有原本的 srcRel
|
||
t.Fatalf("vault 根目錄多出非預期的 .md：count=%d", topLevel)
|
||
}
|
||
|
||
// 既有卡片必須被備份，不能無聲覆蓋。
|
||
entries, err := os.ReadDir(cardDir)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
var foundBackup bool
|
||
for _, e := range entries {
|
||
if strings.HasPrefix(e.Name(), "note.md.bak-") {
|
||
foundBackup = true
|
||
}
|
||
}
|
||
if !foundBackup {
|
||
t.Fatalf("既有卡片沒有被備份，目錄內容：%v", entries)
|
||
}
|
||
}
|
||
|
||
// 非 vault：既有行為不變（回歸）——卡片仍落 system-dev/wiki/cards/。
|
||
func TestExtractWithWorkersAI_NonVaultUnchanged(t *testing.T) {
|
||
root := t.TempDir()
|
||
srcRel := "note.md"
|
||
if err := os.WriteFile(filepath.Join(root, srcRel), []byte("# 原稿"), 0o644); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
url, closeFn := workersAIStub(t, func(w http.ResponseWriter, r *http.Request) {
|
||
_ = json.NewEncoder(w).Encode(map[string]any{
|
||
"success": true,
|
||
"card": "# note\n## 一句話定義\n新卡\n",
|
||
})
|
||
})
|
||
defer closeFn()
|
||
|
||
cards, err := ExtractWithWorkersAI(url, "key123", root, srcRel)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if len(cards) != 1 || cards[0] != "system-dev/wiki/cards/note.md" {
|
||
t.Fatalf("非 vault 行為變了：%v，want [system-dev/wiki/cards/note.md]", cards)
|
||
}
|
||
}
|