feat(scripts): add cite.py and query.py offline tools

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

scripts/cite.py

@@ -0,0 +1,74 @@
+#!/usr/bin/env python3
+"""给定关键词，在指定 master 的 sources/*.md 中查找匹配段落，输出标准 CBETA 引用。"""
+
+import argparse
+import json
+import os
+import re
+import sys
+
+BASE = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "prebuilt")
+
+
+def parse_sections(text):
+    """按 ## 标题分段，返回 [(title, body), ...]"""
+    sections = []
+    parts = re.split(r'^## ', text, flags=re.MULTILINE)
+    for part in parts[1:]:
+        lines = part.split('\n', 1)
+        title = lines[0].strip()
+        body = lines[1] if len(lines) > 1 else ''
+        sections.append((title, body))
+    return sections
+
+
+def find_citations(master, text):
+    sources_dir = os.path.join(BASE, master, "sources")
+    if not os.path.isdir(sources_dir):
+        print(f"错误：找不到目录 {sources_dir}", file=sys.stderr)
+        sys.exit(1)
+
+    results = []
+    for fname in sorted(os.listdir(sources_dir)):
+        if fname == "INDEX.md" or not fname.endswith(".md"):
+            continue
+        fpath = os.path.join(sources_dir, fname)
+        content = open(fpath, encoding="utf-8").read()
+        for title, body in parse_sections(content):
+            full = title + "\n" + body
+            if text not in full:
+                continue
+            # 提取引用格式行
+            for line in body.split('\n'):
+                line_s = line.strip()
+                if "【《" in line_s:
+                    results.append({
+                        "section": title,
+                        "citation": line_s,
+                        "file": fname,
+                    })
+    return results
+
+
+def main():
+    parser = argparse.ArgumentParser(description="在 master 的 sources 中查找关键词并输出 CBETA 引用")
+    parser.add_argument("--master", required=True, help="大师 ID，如 zhiyi")
+    parser.add_argument("--text", required=True, help="搜索关键词")
+    parser.add_argument("--json", action="store_true", dest="as_json", help="JSON 格式输出")
+    args = parser.parse_args()
+
+    results = find_citations(args.master, args.text)
+
+    if not results:
+        print(f"未找到包含「{args.text}」的段落。")
+        return
+
+    if args.as_json:
+        print(json.dumps(results, ensure_ascii=False, indent=2))
+    else:
+        for r in results:
+            print(f"[{r['section']}] {r['citation']}")
+
+
+if __name__ == "__main__":
+    main()

scripts/query.py

@@ -0,0 +1,84 @@
+#!/usr/bin/env python3
+"""离线检索指定 master 的 sources/ 和 references/。"""
+
+import argparse
+import json
+import os
+import re
+import sys
+
+BASE = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "prebuilt")
+
+
+def parse_sections(text):
+    """按 ## 标题分段，返回 [(title, body), ...]"""
+    sections = []
+    parts = re.split(r'^## ', text, flags=re.MULTILINE)
+    for part in parts[1:]:
+        lines = part.split('\n', 1)
+        title = lines[0].strip()
+        body = lines[1] if len(lines) > 1 else ''
+        sections.append((title, body))
+    return sections
+
+
+def search(master, query, brief=False):
+    keywords = query.split()
+    results = []
+
+    for subdir in ("sources", "references"):
+        dirpath = os.path.join(BASE, master, subdir)
+        if not os.path.isdir(dirpath):
+            continue
+        for fname in sorted(os.listdir(dirpath)):
+            if fname == "INDEX.md" or not fname.endswith(".md"):
+                continue
+            fpath = os.path.join(dirpath, fname)
+            content = open(fpath, encoding="utf-8").read()
+            for title, body in parse_sections(content):
+                full = title + "\n" + body
+                # OR 匹配：任一关键词命中即可
+                if not any(kw in full for kw in keywords):
+                    continue
+                # 清理 body 前 200 字
+                clean = re.sub(r'\n{2,}', '\n', body).strip()
+                preview = clean[:200]
+                results.append({
+                    "section": title,
+                    "preview": preview,
+                    "file": os.path.join(subdir, fname),
+                })
+
+    return results
+
+
+def main():
+    parser = argparse.ArgumentParser(description="离线检索 master 的 sources 和 references")
+    parser.add_argument("--master", required=True, help="大师 ID，如 zhiyi")
+    parser.add_argument("--q", required=True, help="搜索关键词（空格分隔，OR 匹配）")
+    parser.add_argument("--json", action="store_true", dest="as_json", help="JSON 格式输出")
+    parser.add_argument("--brief", action="store_true", help="仅输出段标题和文件路径")
+    args = parser.parse_args()
+
+    results = search(args.master, args.q, args.brief)
+
+    if not results:
+        print(f"未找到包含「{args.q}」的段落。")
+        return
+
+    if args.as_json:
+        print(json.dumps(results, ensure_ascii=False, indent=2))
+    elif args.brief:
+        for r in results:
+            print(f"[{r['section']}] → {r['file']}")
+    else:
+        for i, r in enumerate(results):
+            if i > 0:
+                print("---")
+            print(f"## {r['section']}")
+            print(r['preview'])
+            print(f"📂 {r['file']}")
+
+
+if __name__ == "__main__":
+    main()