codebuff-agent 2.1.74__py3-none-any.whl

jupyter_ext/agent/codex_autofix.py

@@ -0,0 +1,160 @@
+"""내장 PostToolUse hook — 고친 파일에 ruff/biome autofix 를 돌린다 (이슈 #145).
+
+`codex_hooks.BUILTIN_HOOKS` 가 `python -m jupyter_ext.agent.codex_autofix` 로 부른다.
+stdin 은 codex `PostToolUse` 페이로드이고, 사용자 hooks.json 의 hook 과 같은 러너·같은
+규격으로 돈다. 표준 라이브러리만 쓴다 — hook 은 매 편집마다 새 프로세스로 뜬다.
+
+| 파일 | 도구 | 조건 |
+|---|---|---|
+| `.py` `.pyi` | `ruff check --fix --unfixable F401` → `ruff format` | 항상 (`ruff_command`) |
+| JS·TS·JSON·CSS | `biome check --write` | 파일부터 워크스페이스까지 `biome.json(c)` 이 있을 때 |
+
+- ruff 는 PATH 의 것을 먼저 쓰고, 없으면 wheel 이 같이 깐 것(`python -m ruff`)을 쓴다.
+  ruff 는 `codebuff-agent` 의 런타임 의존성이다 (이슈 #145) — 서버를 venv 활성 없이
+  띄워 PATH 에 없어도 돈다.
+- `write_file` 의 `.py` 는 건너뛴다. `langchain_tools` 가 이미 ruff 를 돌린다.
+- biome 은 설정 디렉터리의 `node_modules/.bin/biome` 또는 PATH 의 것만 쓴다.
+  `npx` 로 받아 오지 않는다 — 편집 한 번에 네트워크를 타면 안 된다.
+- 도구가 없으면 `systemMessage` 로 알린다 (브리지가 세션당 한 번만 띄운다).
+- 항상 exit 0 이다. 포맷 실패가 편집을 막으면 안 된다.
+"""
+
+from __future__ import annotations
+
+import importlib.util
+import json
+import os
+import re
+import shutil
+import subprocess
+import sys
+from pathlib import Path
+from typing import Dict, List, Optional
+
+PYTHON_SUFFIXES = {".py", ".pyi"}
+BIOME_SUFFIXES = {
+    ".js",
+    ".jsx",
+    ".mjs",
+    ".cjs",
+    ".ts",
+    ".tsx",
+    ".json",
+    ".jsonc",
+    ".css",
+}
+BIOME_CONFIGS = ("biome.json", "biome.jsonc")
+#: `ruff check` 인자. F401(안 쓰인 import) 은 고치지 않는다 — 에이전트는 import 를 먼저
+#: 쓰고 쓰는 곳을 다음 편집에 쓰는데, 그 사이에 지우면 import 가 사라진다. `--unfixable` 은
+#: 프로젝트 설정의 unfixable 에 더해지므로 사용자 설정을 덮지 않는다
+RUFF_FIX = ("check", "--fix", "--unfixable", "F401")
+#: 한 명령의 한도. 내장 hook 전체 한도(`codex_hooks.BUILTIN_TIMEOUT`) 안에 둘이 들어간다
+STEP_TIMEOUT = 25
+
+_PATCH_HEADER = re.compile(r"^\*\*\* (?:Add|Update) File: (.+)$", re.MULTILINE)
+
+
+def _target(payload: Dict, cwd: Path) -> Optional[Path]:
+    """고친 파일의 절대경로. 워크스페이스 밖이면 None 이다."""
+    tool_input = payload.get("tool_input") or {}
+    raw = tool_input.get("path")
+    if not raw:
+        m = _PATCH_HEADER.search(str(tool_input.get("command") or ""))
+        raw = m.group(1).strip() if m else ""
+    if not raw:
+        return None
+    path = (cwd / raw).resolve()
+    try:
+        path.relative_to(cwd)
+    except ValueError:
+        return None
+    return path if path.is_file() else None
+
+
+def ruff_command() -> Optional[List[str]]:
+    """ruff 를 부르는 명령 머리. PATH → wheel 이 깐 것 순이고, 둘 다 없으면 None."""
+    found = shutil.which("ruff")
+    if found:
+        return [found]
+    if importlib.util.find_spec("ruff") is not None:
+        return [sys.executable, "-m", "ruff"]
+    return None
+
+
+def _run(cmd: List[str], cwd: Path) -> None:
+    try:
+        subprocess.run(cmd, cwd=cwd, capture_output=True, timeout=STEP_TIMEOUT)
+    except (OSError, subprocess.TimeoutExpired):
+        pass
+
+
+def _biome(path: Path, cwd: Path) -> Optional[str]:
+    """biome 실행 파일. 설정이 없으면 빈 문자열, 설정은 있는데 실행 파일이 없으면 None."""
+    for d in [path.parent, *path.parent.parents]:
+        if any((d / name).is_file() for name in BIOME_CONFIGS):
+            local = d / "node_modules" / ".bin" / "biome"
+            if local.is_file() and os.access(local, os.X_OK):
+                return str(local)
+            return shutil.which("biome")
+        if d == cwd:
+            break
+    return ""
+
+
+def autofix(payload: Dict) -> Dict:
+    cwd = Path(payload.get("cwd") or os.getcwd()).resolve()
+    path = _target(payload, cwd)
+    if path is None:
+        return {}
+    suffix = path.suffix.lower()
+    tool = (payload.get("tool_input") or {}).get("codebuff_tool")
+    if suffix in PYTHON_SUFFIXES:
+        if tool == "write_file" and suffix == ".py":
+            return {}
+        ruff = ruff_command()
+        if not ruff:
+            return {"systemMessage": "ruff 가 없어 .py 자동 포맷을 건너뜁니다"}
+        cmds = [[*ruff, *RUFF_FIX, str(path)], [*ruff, "format", str(path)]]
+        name = "ruff"
+    elif suffix in BIOME_SUFFIXES:
+        biome = _biome(path, cwd)
+        if biome == "":
+            return {}
+        if biome is None:
+            return {
+                "systemMessage": "biome 설정은 있는데 biome 이 없어 자동 포맷을 건너뜁니다"
+            }
+        cmds = [[biome, "check", "--write", str(path)]]
+        name = "biome"
+    else:
+        return {}
+    before = path.read_bytes()
+    for cmd in cmds:
+        _run(cmd, cwd)
+    if path.read_bytes() == before:
+        return {}
+    rel = path.relative_to(cwd)
+    return {
+        "hookSpecificOutput": {
+            "hookEventName": "PostToolUse",
+            "additionalContext": (
+                f"autofix: {name} 가 {rel} 를 고쳤다. 이 파일을 다시 edit_file 로 "
+                "고치기 전에 read_file 로 새 내용을 읽는다."
+            ),
+        }
+    }
+
+
+def main() -> int:
+    try:
+        payload = json.loads(sys.stdin.read() or "{}")
+        out = autofix(payload) if isinstance(payload, dict) else {}
+    except Exception as e:  # noqa: BLE001 — 포맷 실패가 편집을 막으면 안 된다
+        out = {"systemMessage": f"autofix 실패: {e}"}
+    if out:
+        print(json.dumps(out, ensure_ascii=False))
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())