#!/home/bugabuga/anaconda3/bin/python
import sys
import enumerate_markdown


def main():
    if len(sys.argv) not in (3, 4):
        print("Usage: {} <input-markdown-file> <minimal-level> [<output-file>] ")
        sys.exit(1)
    input_filename = sys.argv[1]
    minimal_level = int(sys.argv[2])

    with open(input_filename, "r") as f_in:
        enumerated_content = enumerate_markdown.enumerate_headers(f_in.read(), minimal_level=minimal_level)
    if len(sys.argv) == 4:
        output_filename = sys.argv[3]
        with open(output_filename, "w") as f_out:
            f_out.write(enumerated_content)
    else:
        print(enumerated_content)


if __name__ == '__main__':
    main()
