Function main

scripts/test_impact_plan.py:540–585

(argv: list[str] | None = None)

def main(argv: list[str] | None = None) -> int:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--repo", default=".", help="repository path")
    parser.add_argument("--base", default="origin/main", help="base ref")
    parser.add_argument("--head", default="HEAD", help="head ref")
    parser.add_argument(
        "--github-pr-files-json",
        help="analyze GitHub pull request files API JSON instead of running git diff",
    )
    parser.add_argument(
        "--expected-file-count",
        type=int,
        help="warn when GitHub reports more changed files than were provided",
    )
    parser.add_argument("--markdown-output", help="write Markdown report here")
    parser.add_argument("--json-output", help="write JSON report here")
    parser.add_argument(
        "--json",
        action="store_true",
        help="print JSON instead of Markdown to stdout",
    )
    args = parser.parse_args(argv)

    if args.github_pr_files_json:
        with Path(args.github_pr_files_json).open() as handle:
            plan = analyze_github_pr_files(
                json.load(handle),
                expected_file_count=args.expected_file_count,
            )
    else:
        repo = Path(args.repo)
        diff_range = f"{args.base}...{args.head}"
        name_status = run_git(repo, ["diff", "--name-status", diff_range])
        patch = run_git(repo, ["diff", "--unified=0", diff_range])
        plan = analyze_diff(name_status, patch)

    markdown = render_markdown(plan)
    payload = json.dumps(asdict(plan), indent=2, sort_keys=True)
    write_text(args.markdown_output, markdown)
    write_text(args.json_output, payload + "\n")

    if args.json:
        sys.stdout.write(payload + "\n")
    else:
        sys.stdout.write(markdown)
    return 0


if __name__ == "__main__":