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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160

import importlib.util
import json
import os
import subprocess
import tempfile
import unittest
from pathlib import Path


REPO_ROOT = Path(__file__).resolve().parents[2]
SCRIPT_PATH = REPO_ROOT / "scripts" / "capture-cursor.py"


def load_module():
    spec = importlib.util.spec_from_file_location("capture_cursor", SCRIPT_PATH)
    module = importlib.util.module_from_spec(spec)
    assert spec and spec.loader
    spec.loader.exec_module(module)
    return module


class CaptureCursorTests(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.mod = load_module()

    def test_normalize_windows_style_path(self):
        normalized = self.mod.normalize_path(r"\home\prakh\repo\src\main.rs", "/tmp")
        self.assertEqual(normalized, "/home/prakh/repo/src/main.rs")

    def test_extract_lines_from_cursor_change_ranges(self):
        payload = [
            {"startLine": 3, "endLine": 5},
            {"range": {"start": {"line": 8}, "end": {"line": 9}}},
            {"line_number": "12"},
        ]
        self.assertEqual(self.mod.extract_lines_from_changes(payload), [3, 4, 5, 8, 9, 12])

    def test_extract_lines_prefers_top_level_range_over_nested_selection(self):
        payload = [
            {
                "startLine": 10,
                "endLine": 10,
                "selection": {"start": {"line": 5}, "end": {"line": 20}},
            }
        ]
        self.assertEqual(self.mod.extract_lines_from_changes(payload), [10])

    def test_extract_lines_accepts_mixed_line_lists(self):
        payload = [2, "4", {"start_line": 7, "end_line": 8}, 0, "nope"]
        self.assertEqual(self.mod.extract_lines(payload), [2, 4, 7, 8])

    def test_session_log_uses_repo_hint(self):
        with tempfile.TemporaryDirectory() as tmp:
            repo = Path(tmp) / "repo"
            repo.mkdir()
            subprocess.run(["git", "init"], cwd=repo, check=True, capture_output=True)

            # Simulate agentdiff init — capture only fires when .git/agentdiff/ exists.
            agentdiff_dir = repo / ".git" / "agentdiff"
            agentdiff_dir.mkdir(parents=True, exist_ok=True)

            session_log = self.mod.get_session_log(str(repo), str(repo))
            expected = agentdiff_dir / "session.jsonl"
            self.assertEqual(Path(session_log), expected)

    def test_returns_none_when_agentdiff_init_not_run(self):
        with tempfile.TemporaryDirectory() as tmp:
            repo = Path(tmp) / "repo"
            repo.mkdir()
            subprocess.run(["git", "init"], cwd=repo, check=True, capture_output=True)
            # No .git/agentdiff/ — agentdiff init not run.
            result = self.mod.get_session_log(str(repo), str(repo))
            self.assertIsNone(result, "get_session_log must return None when init not run")

    def test_hook_event_writes_repo_session_when_cwd_not_repo(self):
        """Capture writes to repo-local session.jsonl when cwd is not the repo.

        Cursor often runs hooks with cwd=~/.cursor rather than the repo root.
        The capture script resolves the repo from the edited file's path.
        Capture only fires when agentdiff init has been run (.git/agentdiff/ exists).
        """
        with tempfile.TemporaryDirectory() as tmp:
            repo = Path(tmp) / "repo"
            repo.mkdir()
            subprocess.run(["git", "init"], cwd=repo, check=True, capture_output=True)
            src = repo / "src"
            src.mkdir()
            edited = src / "main.rs"
            edited.write_text("fn main() {}\n", encoding="utf-8")

            # Simulate agentdiff init: create .git/agentdiff/ so capture fires.
            agentdiff_dir = repo / ".git" / "agentdiff"
            agentdiff_dir.mkdir(parents=True, exist_ok=True)

            payload = {
                "hook_event_name": "afterTabFileEdit",
                "cwd": str(Path.home() / ".cursor"),
                "file_path": str(edited).replace("/", "\\"),
                "lineNumber": 7,
                "model": "cursor-test-model",
                "conversationId": "conv-test-1",
            }

            proc = subprocess.run(
                ["python3", str(SCRIPT_PATH)],
                input=json.dumps(payload),
                text=True,
                capture_output=True,
                env=os.environ.copy(),
            )
            self.assertEqual(proc.returncode, 0, msg=proc.stderr)

            session_log = agentdiff_dir / "session.jsonl"
            self.assertTrue(session_log.exists(), "repo-local session log should exist")

            lines = [ln for ln in session_log.read_text(encoding="utf-8").splitlines() if ln.strip()]
            self.assertEqual(len(lines), 1)
            entry = json.loads(lines[0])
            self.assertEqual(entry.get("agent"), "cursor")
            self.assertEqual(entry.get("file"), "src/main.rs")

    def test_no_capture_when_agentdiff_init_not_run(self):
        """Capture must be silent when .git/agentdiff/ does not exist."""
        with tempfile.TemporaryDirectory() as tmp:
            repo = Path(tmp) / "repo"
            repo.mkdir()
            subprocess.run(["git", "init"], cwd=repo, check=True, capture_output=True)
            edited = repo / "main.rs"
            edited.write_text("fn main() {}\n", encoding="utf-8")

            # Do NOT create .git/agentdiff/ — agentdiff init not run.

            payload = {
                "hook_event_name": "afterTabFileEdit",
                "cwd": str(repo),
                "file_path": str(edited),
                "lineNumber": 1,
                "model": "cursor-test-model",
                "conversationId": "conv-test-2",
            }

            proc = subprocess.run(
                ["python3", str(SCRIPT_PATH)],
                input=json.dumps(payload),
                text=True,
                capture_output=True,
                env=os.environ.copy(),
            )
            self.assertEqual(proc.returncode, 0, msg=proc.stderr)

            session_log = repo / ".git" / "agentdiff" / "session.jsonl"
            self.assertFalse(
                session_log.exists(),
                "session.jsonl must not be created when agentdiff init has not been run",
            )


if __name__ == "__main__":
    unittest.main()