#!/usr/bin/env python3
"""Generate a file with all md5 hashes of the assets."""

from collections import OrderedDict
import glob
import hashlib
import json
import argparse
from os import path
import re

parser = argparse.ArgumentParser(description='Generate fingerprints of frontend files.')
parser.add_argument('--base_dir', type=str, help='Base dir to look for files.', default='hass_frontend')
args = parser.parse_args()
base_dir = args.base_dir + '/'
fingerprint_file = path.join(base_dir, '__init__.py')
panel_match = re.compile(r'ha-panel-((\w|-)+)\.html')


def fingerprint():
    """Fingerprint the panels."""
    files = glob.glob(base_dir + 'panels/*.html')
    md5s = OrderedDict()

    for fil in sorted(files):
        panel = panel_match.search(fil).groups(0)[0]
        with open(fil) as fp:
            md5 = hashlib.md5(fp.read().encode('utf-8')).hexdigest()
        md5s[panel] = md5

    template = "FINGERPRINTS = {}\n"
    result = template.format(json.dumps(md5s, indent=4))

    with open(fingerprint_file, 'at') as fp:
        fp.write(result)


if __name__ == '__main__':
    fingerprint()