import argparse
import os
from pathlib import Path

from wpilibc.generate_first_ds_hids import generate_first_ds_hids
from wpilibc.generate_hids import generate_hids
from wpilibc.generate_pwm_motor_controllers import generate_pwm_motor_controllers


def main():
    dirname, _ = os.path.split(os.path.abspath(__file__))

    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--output_directory",
        help="Optional. If set, will output the generated files to this directory, otherwise it will use a path relative to the script",
        default=os.path.join(dirname, "src/generated"),
        type=Path,
    )
    parser.add_argument(
        "--schema_root",
        help="Optional. If set, will use this directory as the root for discovering the pwm controller schema",
        default=os.path.join(dirname, "../wpilibj/src/generate"),
        type=Path,
    )
    parser.add_argument(
        "--template_root",
        help="Optional. If set, will use this directory as the root for the jinja templates",
        default=os.path.join(dirname, "src/generate"),
        type=Path,
    )
    parser.add_argument(
        "--test_output_directory",
        help="Optional. If set, will output generated tests to this directory",
        default=os.path.join(dirname, "src/generated/test"),
        type=Path,
    )
    args = parser.parse_args()

    generate_hids(
        args.output_directory, args.template_root, args.schema_root / "hids.json"
    )
    test_output_directory = (
        None
        if args.test_output_directory.name == "__none__"
        else args.test_output_directory
    )
    generate_first_ds_hids(
        args.output_directory,
        args.template_root,
        args.schema_root / "first_ds_hids.json",
        test_output_directory,
    )
    generate_pwm_motor_controllers(
        args.output_directory, args.template_root, args.schema_root
    )


if __name__ == "__main__":
    main()