141 lines
5.6 KiB
Python
141 lines
5.6 KiB
Python
#!/usr/bin/env python
|
|
"""Stop hook 안전망: 이번 세션에서 코드/문서 고쳐놓고 worklog 안 남겼으면 딱 한 번 붙잡는다.
|
|
|
|
훅이 문장을 대신 못 써준다(그냥 셸이라). 대신 "안 적었으면 멈추지 못하게" 막아서
|
|
내가 직접 한 줄 남기게 만든다. gen_worklog.py add 흔적이 마지막 편집보다 뒤에 있으면
|
|
통과, 아니면 block(한 턴 더 돌려서 남기게 함).
|
|
|
|
stdin : Claude Code Stop hook JSON (transcript_path, stop_hook_active)
|
|
stdout: 통과면 아무것도. 걸리면 {"decision":"block","reason":...}
|
|
|
|
python worklog_guard.py --selftest 로 자체검증.
|
|
"""
|
|
|
|
import json
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
TAIL = 256 * 1024 # ponytail: 트랜스크립트 꼬리만 스캔. 이보다 오래된 미기재 편집은 못 잡음 —
|
|
# 편집은 보통 그 턴에 바로 붙잡히니 실무상 충분. 느려지면 전체읽기로.
|
|
REPO = Path(__file__).resolve().parents[2].as_posix().lower() # 레포 루트 — 이 안의 편집만 프로젝트 스텝
|
|
|
|
|
|
def tool_uses(obj):
|
|
"""파싱된 한 줄 안의 모든 tool_use dict 를 훑어서 뽑는다(중첩 상관없이)."""
|
|
out = []
|
|
|
|
def walk(x):
|
|
if isinstance(x, dict):
|
|
if x.get("type") == "tool_use":
|
|
out.append(x)
|
|
for v in x.values():
|
|
walk(v)
|
|
elif isinstance(x, list):
|
|
for v in x:
|
|
walk(v)
|
|
|
|
walk(obj)
|
|
return out
|
|
|
|
|
|
def scan(transcript_path):
|
|
"""미기재면 마지막 편집 파일경로, 아니면 None."""
|
|
p = Path(transcript_path)
|
|
if not p.exists():
|
|
return None
|
|
data = p.read_bytes()
|
|
if len(data) > TAIL:
|
|
data = data[-TAIL:]
|
|
data = data[data.find(b"\n") + 1 :] # 잘린 첫 줄 버림
|
|
last_edit_i, last_edit_file, last_log_i = -1, None, -1
|
|
for i, raw in enumerate(data.decode("utf-8", "ignore").splitlines()):
|
|
if not raw.strip():
|
|
continue
|
|
try:
|
|
obj = json.loads(raw)
|
|
except ValueError:
|
|
continue
|
|
for tu in tool_uses(obj):
|
|
name = tu.get("name", "")
|
|
inp = tu.get("input", {}) or {}
|
|
if name in ("Edit", "Write", "NotebookEdit"):
|
|
fp = str(inp.get("file_path", "")).replace("\\", "/")
|
|
low = fp.lower()
|
|
if not low.startswith(REPO):
|
|
continue # 레포 밖(메모리·스크래치패드 등)은 프로젝트 스텝 아님
|
|
if "z-my-docs" in low:
|
|
continue # worklog 자체 편집은 self-trigger 방지로 제외
|
|
last_edit_i, last_edit_file = i, fp
|
|
elif name in ("Bash", "PowerShell") and "gen_worklog.py add" in str(
|
|
inp.get("command", "")
|
|
):
|
|
# 윈도우 세션은 Bash 대신 PowerShell 툴로 돎 — 둘 다 인정해야 헛붙잡기 안 함
|
|
last_log_i = i
|
|
if last_edit_i >= 0 and last_edit_i > last_log_i:
|
|
return last_edit_file or "코드"
|
|
return None
|
|
|
|
|
|
def main():
|
|
try:
|
|
payload = json.load(sys.stdin)
|
|
except (ValueError, OSError):
|
|
return
|
|
if payload.get("stop_hook_active"):
|
|
return # 이미 한 번 붙잡았음 — 두 번은 안 함(덫·무한루프 방지). 그래도 안 적으면 그냥 통과.
|
|
edited = scan(payload.get("transcript_path", ""))
|
|
if not edited:
|
|
return
|
|
hint = Path(edited).name if edited != "코드" else "코드"
|
|
print(
|
|
json.dumps(
|
|
{
|
|
"decision": "block",
|
|
"reason": (
|
|
f"이번에 {hint} 고쳤는데 worklog 를 안 남겼어. "
|
|
f'`python z-my-docs/work-log/gen_worklog.py add "<한 일 한 줄>"` 로 '
|
|
f"한 줄 남기고 끝내. 진짜 남길 것 없으면 그냥 다시 멈추면 통과됨."
|
|
),
|
|
},
|
|
ensure_ascii=False,
|
|
)
|
|
)
|
|
|
|
|
|
def _selftest():
|
|
import tempfile
|
|
|
|
def w(objs):
|
|
f = tempfile.NamedTemporaryFile("w", suffix=".jsonl", delete=False, encoding="utf-8")
|
|
for o in objs:
|
|
f.write(json.dumps(o) + "\n")
|
|
f.close()
|
|
return f.name
|
|
|
|
def tu(name, **inp):
|
|
return {"message": {"content": [{"type": "tool_use", "name": name, "input": inp}]}}
|
|
|
|
root = Path(__file__).resolve().parents[2]
|
|
edit = tu("Edit", file_path=str(root / "1_backend/src/x.py"))
|
|
log = tu("Bash", command='python z-my-docs/work-log/gen_worklog.py add "x"')
|
|
pslog = tu("PowerShell", command='python z-my-docs/work-log/gen_worklog.py add "x"')
|
|
logedit = tu("Edit", file_path=str(root / "z-my-docs/work-log/2026-07/x.tsv"))
|
|
memedit = tu("Edit", file_path="C:/Users/user/.claude/projects/foo/memory/bar.md")
|
|
|
|
r = scan(w([edit]))
|
|
assert r and r.endswith("x.py"), r # 레포 안 편집만 → 미기재
|
|
assert scan(w([edit, log])) is None # 편집 뒤 로그 → 통과
|
|
assert scan(w([edit, pslog])) is None # PowerShell 툴로 남긴 로그도 인정(윈도우 세션)
|
|
assert scan(w([log, edit])) is not None # 로그 뒤 편집 → 미기재
|
|
assert scan(w([logedit])) is None # worklog 편집만 → 무시
|
|
assert scan(w([memedit])) is None # 레포 밖(메모리) 편집 → 무시
|
|
assert scan(w([])) is None # 빈 트랜스크립트 → 통과
|
|
print("selftest ok")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if sys.argv[1:2] == ["--selftest"]:
|
|
_selftest()
|
|
else:
|
|
main()
|