import utils
import sys

_opt_type = sys.argv[1]
_file_path = sys.argv[2]


if _opt_type == "c":
    _file = utils.get_file(_file_path)
    _pbm_content = utils.BitMapFile(_file)
    _result = utils.huffman_encode_image(_pbm_content)
    with open(sys.argv[2]+".bin", "wb") as file:
        file.write(_result)
    print(_result)
elif _opt_type == "d":
    _result = utils.huffman_decode_file(_file_path)
    with open(sys.argv[2].strip('.bin'), "w") as file:
        file.write(_result)