#!/usr/bin/env python3
import sys
from typing import List

from lib.pretty_print import pretty_print_html


def clean_html(filenames: List[str]) -> None:
    for fn in filenames:
        print(f"Prettifying: {fn}")
        with open(fn) as f:
            html = f.read()
        phtml = pretty_print_html(html)
        with open(fn, "w") as f:
            f.write(phtml)


if __name__ == "__main__":
    # If command arguments are provided, we only check those filenames.
    # Otherwise, we display a message
    filenames = sys.argv[1:]
    if filenames:
        clean_html(filenames)
        sys.exit(0)
    print("Please specify at least one file to prettify")
    sys.exit(1)