jj.wang
40 lines
1.6 KiB
Python
40 lines
1.6 KiB
Python
"""Claude Code CLI: `claude -p --output-format json` with the prompt on stdin and cwd = repo;
|
|
resume with `--resume <session_id>`. The JSON result carries `result` (last message) and `session_id`."""
|
|
import json
|
|
|
|
from .base import ExecResult, RunnerExecutor, parse_exit
|
|
|
|
|
|
def parse_output(text):
|
|
"""Return the result object from `--output-format json` output (whole text or its last JSON line)."""
|
|
text = (text or "").strip()
|
|
candidates = [text] + list(reversed(text.splitlines()))
|
|
for c in candidates:
|
|
try:
|
|
obj = json.loads(c)
|
|
except ValueError:
|
|
continue
|
|
if isinstance(obj, dict):
|
|
return obj
|
|
if isinstance(obj, list): # stream-json style array: take the result event
|
|
res = [o for o in obj if isinstance(o, dict) and o.get("type") == "result"]
|
|
if res:
|
|
return res[-1]
|
|
return {}
|
|
|
|
|
|
class ClaudeCodeExecutor(RunnerExecutor):
|
|
stdout_name = "out.json"
|
|
|
|
COLLECT = [("read", "exit.txt"), ("read", "out.json"), ("tail", "stderr.txt", 5)]
|
|
|
|
def collect(self, handle):
|
|
exit_text, out, tail = self.fetch(handle, self.COLLECT) # one remote call
|
|
obj = parse_output(out)
|
|
err = bool(obj.get("is_error")) or obj.get("subtype", "success") != "success"
|
|
last = "" if err else (obj.get("result") or "")
|
|
if err:
|
|
tail = ("%s %s" % (obj.get("subtype", ""), str(obj.get("result", ""))[:400]) + "\n" + tail).strip()
|
|
return ExecResult(last.strip(), parse_exit(exit_text), obj.get("session_id") or handle.get("thread_id"),
|
|
self.logs(handle, "out.json", "stderr.txt"), tail)
|