#!/usr/bin/env python3

"""Run local VCS correctness diagnostics; stdout is diagnostic JSONL, not timing rows."""

import argparse

import json

from pathlib import Path

from oakbench.conformance import SCENARIOS, run_scenario

def main():

    parser = argparse.ArgumentParser(description=__doc__)

    parser.add_argument("--workdir", type=Path, required=True)

    parser.add_argument("--subjects", default="git,oak")

    parser.add_argument("--git-bin", default="git")

    parser.add_argument("--oak-bin", default="oak")

    parser.add_argument("--scenarios", default=",".join(SCENARIOS))

    args = parser.parse_args()

    subjects = args.subjects.split(",")

    scenarios = args.scenarios.split(",")

    if not set(subjects) <= {"git", "oak"} or not set(scenarios) <= set(SCENARIOS):

        parser.error("unknown subject or scenario")

    failed = False

    for subject in subjects:

        for scenario in scenarios:

            row = run_scenario(args.workdir.resolve(), subject, getattr(args, f"{subject}_bin"), scenario)

            print(json.dumps(row, sort_keys=True), flush=True)

            failed |= row["returncode"] == 1

    return int(failed)

if __name__ == "__main__":

    raise SystemExit(main())