// extract.go — 可插拔本地萃取器(daemon-beta task 3/4,四步定稿第 3 步)。 // // leo 定稿:「由現在的 gemma4 幫你本地萃,或是用你自己的訂閱帳號,例如叫起 claude 幫你萃。」 // - claude 路(task 3):執行 `claude -p "/rag-extract-file <檔>"`,cwd=監看根—— // template 既有萃取流**原樣重用**(prompt 都不搬),卡片由 CC 寫進 system-dev/wiki/cards/。 // 用戶自己的訂閱=我們零 API 成本。 // - gemma 路(task 4):Go 內直呼 Gemini API,卡片由 daemon 自己寫進 cards/。 // // 兩路的產出契約相同:回傳「本次新增/變動的卡片相對路徑」,交 task 6 推 ingest。 // 原文永不出本函式(claude 路整段在用戶機器;gemma 路內容過境 API 但不落任何雲儲存)。 package main import ( "context" "fmt" "os" "os/exec" "path/filepath" "time" ) // cardsRelDir 是 template 規約的卡片產物區(相對監看根)。 const cardsRelDir = "system-dev/wiki/cards" // snapshotCards 記 cards/ 目前每檔 mtime+size(偵測萃取後的新卡/變卡)。 func snapshotCards(absRoot string) map[string]fileState { out := map[string]fileState{} base := filepath.Join(absRoot, filepath.FromSlash(cardsRelDir)) _ = filepath.Walk(base, func(p string, info os.FileInfo, err error) error { if err != nil || info.IsDir() || filepath.Ext(p) != ".md" { return nil } rel, rerr := filepath.Rel(absRoot, p) if rerr != nil { return nil } out[filepath.ToSlash(rel)] = fileState{size: info.Size(), mtime: info.ModTime().UnixNano()} return nil }) return out } // diffCards 比對兩次快照,回傳新增或內容變動的卡片相對路徑(排序穩定交由呼叫端)。 func diffCards(before, after map[string]fileState) []string { var changed []string for p, st := range after { if b, ok := before[p]; !ok || b.size != st.size || b.mtime != st.mtime { changed = append(changed, p) } } return changed } // claudeExtractTimeout:單檔萃取上限。CC 冷啟+讀檔+寫卡實測分鐘級,給足裕度。 const claudeExtractTimeout = 10 * time.Minute // ExtractWithClaude 叫起用戶自己的 claude 跑 template 的 /rag-extract-file(task 3)。 // binPath 空="claude"(PATH 尋找)。回傳本次產出的卡片相對路徑。 func ExtractWithClaude(binPath, absRoot, relPath string) ([]string, error) { if binPath == "" { binPath = "claude" } if _, err := exec.LookPath(binPath); err != nil { return nil, fmt.Errorf("找不到 claude 執行檔(%s):%w——請確認已安裝 Claude Code,或改用 gemma 萃取路", binPath, err) } before := snapshotCards(absRoot) ctx, cancel := context.WithTimeout(context.Background(), claudeExtractTimeout) defer cancel() // /rag-extract-file=template 既有萃取 skill;cwd=監看根(template 已代裝,skill 就在 .claude/commands/)。 cmd := exec.CommandContext(ctx, binPath, "-p", fmt.Sprintf("/rag-extract-file %s", relPath)) cmd.Dir = absRoot out, err := cmd.CombinedOutput() if ctx.Err() == context.DeadlineExceeded { return nil, fmt.Errorf("claude 萃取逾時(%s)", claudeExtractTimeout) } if err != nil { snippet := string(out) if len(snippet) > 400 { snippet = snippet[len(snippet)-400:] } return nil, fmt.Errorf("claude 萃取失敗:%w;輸出尾段:%s", err, snippet) } cards := diffCards(before, snapshotCards(absRoot)) if len(cards) == 0 { return nil, fmt.Errorf("claude 跑完但 %s 沒有新卡片(輸出尾段:%.200s)", cardsRelDir, string(out)) } return cards, nil }