Master-skill/scripts/cite.py

#!/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()