import sys
import argparse


def green(*strings):
    return "".join([f"\033[32m{s}\033[0m" for s in strings])


def red(*strings):
    return "".join([f"\033[31m{s}\033[0m" for s in strings])


def parse_args():
    parser = argparse.ArgumentParser( prog="ft_printf test", description="A ~quicker tester for ft_printf")
    parser.add_argument("-v", "--verbose",
                        help="increase verbosity", action="store_true")
    parser.add_argument("-q", "--quiet",
                        help="decrease vebosity", action="store_true")
    parser.add_argument("-l", "--no-log",
                        help="disable result log", action="store_true")
    parser.add_argument("-f", "--output-file", help="output file name")
    return vars(parser.parse_args(sys.argv[1:]))


def parse():
    logs = {
        "ok": 0,
        "ko": 0,
        "ko_info": []
    }
    for line in sys.stdin:
        line = line.strip()
        if line.find("[OK]") != -1:
            logs["ok"] += 1
            print(green("."), end="")
        elif line.find("[KO]") != -1:
            logs["ko"] += 1
            logs["ko_info"].append({
                "msg": line[line.find("[KO]") + 5:],
                "expected": sys.stdin.readline().rstrip(),
                "actual": sys.stdin.readline().rstrip()
            })
            print(red("!"), end="")
        sys.stdout.flush()
    return logs


def write_logs(logs, options):
    filename = "result.log"
    if options["output_file"] is not None:
        filename = options["output_file"]
    with open(filename, "w") as log_file:
        for ko in logs["ko_info"]:
            log_file.write("- " + ko["msg"] + "\n")
            log_file.write("   " + ko["expected"] + "\n")
            log_file.write("   " + ko["actual"] + "\n\n")


def print_logs(logs, options):
    total_str = f"\n\nTotal     {green('OK: ', logs['ok'])}  {red('KO: ', logs['ko'])}"
    print(total_str)
    print("=" * (len(total_str) - len(green("")) * 2 - len(red("")) * 2 - 2))

    if options["quiet"]:
        return
    for ko in logs["ko_info"]:
        print("-", ko["msg"])
        if options["verbose"]:
            print("   ", ko["expected"])
            print("   ", ko["actual"])
            print()
    print("\nSee result.log for more information\n")


if __name__ == "__main__":
    print()
    options = parse_args()
    logs = parse()
    print_logs(logs, options)
    if not options["no_log"]:
        write_logs(logs, options)