pr-conflict-detector/json_writer.py at main · github-community-projects/pr-conflict-detector · GitHub

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
"""JSON output generation for PR conflict detection results."""

import json
import os

from conflict_detector import cluster_conflicts


def write_to_json(
    conflicts_by_repo: dict,
    output_file: str = "pr_conflict_report.json",
) -> str:
    """Write conflict results to a JSON file.

    Args:
        conflicts_by_repo: Mapping of repo full name to list of ConflictResult objects.
        output_file: Path for the output JSON file.

    Returns:
        The JSON string that was written to the file.

    """
    repositories = []
    total_conflicts_found = 0

    for repo_name, conflicts in conflicts_by_repo.items():
        clusters = cluster_conflicts(conflicts)

        cluster_data = []
        for cluster in clusters:
            cluster_prs = [
                {
                    "number": pr.number,
                    "title": pr.title,
                    "author": pr.author,
                    "url": pr.url,
                }
                for pr in cluster.prs
            ]
            cluster_pair_list = []
            for conflict in cluster.conflicts:
                conflicting_files = [
                    {
                        "filename": fo.filename,
                        "overlapping_ranges": list(fo.overlapping_ranges),
                    }
                    for fo in conflict.conflicting_files
                ]
                cluster_pair_list.append(
                    {
                        "pr_a": {
                            "number": conflict.pr_a.number,
                            "title": conflict.pr_a.title,
                            "author": conflict.pr_a.author,
                            "url": conflict.pr_a.url,
                        },
                        "pr_b": {
                            "number": conflict.pr_b.number,
                            "title": conflict.pr_b.title,
                            "author": conflict.pr_b.author,
                            "url": conflict.pr_b.url,
                        },
                        "conflicting_files": conflicting_files,
                        "verified": (
                            conflict.verified
                            if hasattr(conflict, "verified")
                            else False
                        ),
                    }
                )

            cluster_data.append(
                {
                    "prs": cluster_prs,
                    "shared_files": cluster.shared_files,
                    "conflicts": cluster_pair_list,
                }
            )

        total_conflicts_found += len(conflicts)
        repositories.append(
            {
                "name": repo_name,
                "clusters": cluster_data,
                "total_conflicts": len(conflicts),
            }
        )

    output = {
        "repositories": repositories,
        "total_repositories_scanned": len(conflicts_by_repo),
        "total_conflicts_found": total_conflicts_found,
    }

    metrics_json = json.dumps(output, indent=4)

    with open(output_file, "w", encoding="utf-8") as f:
        json.dump(output, f, indent=4)

    # Write to GITHUB_OUTPUT if available (multiline requires delimiter)
    if os.environ.get("GITHUB_OUTPUT"):
        with open(os.environ["GITHUB_OUTPUT"], "a", encoding="utf-8") as fh:
            print("conflicts_json<<EOF", file=fh)
            print(metrics_json, file=fh)
            print("EOF", file=fh)

    return metrics_json