97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
"""Build the minimal TSL playbook bundle."""
|
|
import argparse
|
|
import shutil
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
DEFAULT_OUTPUT = Path("tmp") / "tsl-playbook"
|
|
API_REFERENCE_BULLET = (
|
|
"- 查 API / 函数的名字、参数、返回值和示例时，使用 "
|
|
"`tsl-api-reference` skill；已知名走 `--name`，未知名走 `--kw`。"
|
|
)
|
|
|
|
|
|
def copy_tree(src: Path, dst: Path) -> None:
|
|
shutil.copytree(src, dst)
|
|
|
|
|
|
def build_agents_text(ruleset_path: Path) -> str:
|
|
text = ruleset_path.read_text(encoding="utf-8")
|
|
text = text.replace("# TSL 智能体规则", "# TSL Agent Instructions", 1)
|
|
if API_REFERENCE_BULLET not in text:
|
|
marker = (
|
|
"- 需要任何 TSL 语法、文件模型、函数、模块或金融数据事实时，"
|
|
"第一跳统一进 `docs/tsl/index.md`，由它路由到具体页；"
|
|
"不在本文件内猜文件路径，也不全目录搜索。"
|
|
)
|
|
text = text.replace(marker, f"{marker}\n{API_REFERENCE_BULLET}", 1)
|
|
return text.rstrip("\n") + "\n"
|
|
|
|
|
|
def ensure_sources(repo_root: Path) -> tuple[Path, Path, Path]:
|
|
docs_tsl = repo_root / "docs" / "tsl"
|
|
skill = repo_root / "skills" / "tsl-api-reference"
|
|
ruleset = repo_root / "rulesets" / "tsl" / "index.md"
|
|
missing = [str(path) for path in (docs_tsl, skill, ruleset) if not path.exists()]
|
|
if missing:
|
|
raise FileNotFoundError("missing source path(s): " + ", ".join(missing))
|
|
return docs_tsl, skill, ruleset
|
|
|
|
|
|
def clean_output(output: Path, repo_root: Path) -> None:
|
|
resolved = output.resolve()
|
|
forbidden = {
|
|
repo_root.resolve(),
|
|
(repo_root / "docs").resolve(),
|
|
(repo_root / "skills").resolve(),
|
|
(repo_root / "rulesets").resolve(),
|
|
}
|
|
if resolved in forbidden:
|
|
raise ValueError(f"refusing to replace source directory: {output}")
|
|
if output.exists():
|
|
shutil.rmtree(output)
|
|
output.mkdir(parents=True)
|
|
|
|
|
|
def build(output: Path, repo_root: Path) -> None:
|
|
docs_tsl, skill, ruleset = ensure_sources(repo_root)
|
|
clean_output(output, repo_root)
|
|
|
|
(output / "AGENTS.md").write_text(
|
|
build_agents_text(ruleset), encoding="utf-8", newline="\n"
|
|
)
|
|
copy_tree(docs_tsl, output / "docs" / "tsl")
|
|
copy_tree(skill, output / "skills" / "tsl-api-reference")
|
|
|
|
|
|
def main(argv=None) -> int:
|
|
if hasattr(sys.stdout, "reconfigure"):
|
|
sys.stdout.reconfigure(encoding="utf-8")
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument(
|
|
"--output",
|
|
default=str(DEFAULT_OUTPUT),
|
|
help=f"output directory (default: {DEFAULT_OUTPUT.as_posix()})",
|
|
)
|
|
args = parser.parse_args(argv)
|
|
|
|
repo_root = Path(__file__).resolve().parents[1]
|
|
output = Path(args.output)
|
|
if not output.is_absolute():
|
|
output = repo_root / output
|
|
|
|
try:
|
|
build(output, repo_root)
|
|
except (FileNotFoundError, ValueError, OSError) as exc:
|
|
print(f"ERROR: {exc}", file=sys.stderr)
|
|
return 1
|
|
|
|
print(f"wrote {output}")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|