agentdiff 0.1.26

Audit and trace autonomous AI code contributions in git repositories
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

#!/usr/bin/env python3
"""
Record MCP context into .git/agentdiff/pending.json

Usage:
  record-context.py [--cwd <path>] [--agent <name>] [--model-id <id>] [--session-id <id>]
                    [--prompt <text>] [--files-read <json-array>] [--intent <text>]
                    [--trust <0-100>] [--flags <json-array>]

If stdin contains JSON object, it is used as payload and CLI flags override fields.
"""
import argparse
import json
import os
import select
import subprocess
import sys
from datetime import datetime, timezone


def find_repo_root(cwd: str) -> str:
    try:
        result = subprocess.run(
            ["git", "rev-parse", "--show-toplevel"],
            capture_output=True,
            text=True,
            cwd=cwd,
        )
        return result.stdout.strip() if result.returncode == 0 else ""
    except Exception:
        return ""


def parse_json_array(raw: str):
    if not raw:
        return []
    try:
        data = json.loads(raw)
        if isinstance(data, list):
            return data
    except Exception:
        pass
    return []


def read_available_stdin() -> str:
    if sys.stdin.isatty():
        return ""
    try:
        ready, _, _ = select.select([sys.stdin], [], [], 0)
    except Exception:
        return ""
    if not ready:
        return ""
    return sys.stdin.read()


def main() -> int:
    parser = argparse.ArgumentParser(add_help=True)
    parser.add_argument("--cwd", default=os.getcwd())
    parser.add_argument("--agent", default="")
    parser.add_argument("--model-id", default="")
    parser.add_argument("--session-id", default="")
    parser.add_argument("--prompt", default="")
    parser.add_argument("--files-read", default="")
    parser.add_argument("--intent", default="")
    parser.add_argument("--trust", type=int, default=None)
    parser.add_argument("--flags", default="")
    args = parser.parse_args()

    payload = {}
    stdin = read_available_stdin()
    if stdin.strip():
        try:
            obj = json.loads(stdin)
            if isinstance(obj, dict):
                payload = obj
        except Exception:
            payload = {}

    repo_root = find_repo_root(args.cwd or os.getcwd())
    if not repo_root:
        return 0

    pending = {
        "recorded_at": datetime.now(timezone.utc).isoformat(),
        "agent": args.agent or str(payload.get("agent") or ""),
        "model_id": args.model_id or str(payload.get("model_id") or payload.get("model") or ""),
        "session_id": args.session_id or str(payload.get("session_id") or ""),
        "prompt": args.prompt or str(payload.get("prompt") or ""),
        "files_read": parse_json_array(args.files_read) or payload.get("files_read") or [],
        "intent": args.intent or str(payload.get("intent") or ""),
        "flags": parse_json_array(args.flags) or payload.get("flags") or [],
    }

    trust = args.trust if args.trust is not None else payload.get("trust")
    if isinstance(trust, int):
        pending["trust"] = max(0, min(100, trust))

    session_dir = os.path.join(repo_root, ".git", "agentdiff")
    os.makedirs(session_dir, exist_ok=True)
    out_path = os.path.join(session_dir, "pending.json")
    with open(out_path, "w", encoding="utf-8") as f:
        json.dump(pending, f, separators=(",", ":"))
    return 0


if __name__ == "__main__":
    raise SystemExit(main())