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 的重啟條件）。
87 lines
2.4 KiB
Go
87 lines
2.4 KiB
Go
// safewrite_test.go — arcrun-rag#60 第二條：落卡前不得無條件覆蓋既有檔案。
|
||
package collector
|
||
|
||
import (
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
"testing"
|
||
)
|
||
|
||
// 目標不存在＝正常首次落卡，行為不變。
|
||
func TestSafeWriteCard_NewFile(t *testing.T) {
|
||
dest := filepath.Join(t.TempDir(), "sub", "foo.md")
|
||
if err := safeWriteCard(dest, []byte("# foo\n內容")); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
got, err := os.ReadFile(dest)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if string(got) != "# foo\n內容" {
|
||
t.Fatalf("內容不對：%q", got)
|
||
}
|
||
}
|
||
|
||
// 核心紅線：目標已存在且內容不同 → 先備份既有內容，不能無聲蓋掉。
|
||
func TestSafeWriteCard_ExistingFile_BacksUpBeforeOverwrite(t *testing.T) {
|
||
dir := t.TempDir()
|
||
dest := filepath.Join(dir, "foo.md")
|
||
original := "# foo\n這是使用者原本就有的東西，不是 daemon 寫的"
|
||
if err := os.WriteFile(dest, []byte(original), 0o644); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
if err := safeWriteCard(dest, []byte("# foo\n機器新產出的卡")); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
// 新內容確實寫進去了（這是萃取的目的，不是完全拒寫）。
|
||
got, err := os.ReadFile(dest)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if string(got) != "# foo\n機器新產出的卡" {
|
||
t.Fatalf("新內容未寫入：%q", got)
|
||
}
|
||
|
||
// 但舊內容必須找得到備份，不能憑空消失。
|
||
entries, err := os.ReadDir(dir)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
var backup string
|
||
for _, e := range entries {
|
||
if strings.HasPrefix(e.Name(), "foo.md.bak-") {
|
||
backup = filepath.Join(dir, e.Name())
|
||
}
|
||
}
|
||
if backup == "" {
|
||
t.Fatalf("沒有找到備份檔，既有內容可能已經被無聲覆蓋。目錄內容：%v", entries)
|
||
}
|
||
backupData, err := os.ReadFile(backup)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if string(backupData) != original {
|
||
t.Fatalf("備份內容不對：%q，want %q", backupData, original)
|
||
}
|
||
}
|
||
|
||
// 內容完全相同（重複萃取同一份卡）＝不必動它，也不必產生垃圾備份。
|
||
func TestSafeWriteCard_SameContent_NoOp(t *testing.T) {
|
||
dir := t.TempDir()
|
||
dest := filepath.Join(dir, "foo.md")
|
||
content := "# foo\n一樣的內容"
|
||
if err := os.WriteFile(dest, []byte(content), 0o644); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := safeWriteCard(dest, []byte(content)); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
entries, _ := os.ReadDir(dir)
|
||
if len(entries) != 1 {
|
||
t.Fatalf("內容沒變不該產生額外檔案，目錄內容：%v", entries)
|
||
}
|
||
}
|