# -*- coding: utf-8 -*-
import hashlib

from petrel_client.common.exception import ConfigKeyValueError

_SUPPORTED_TYPES = ('blake2b', 'blake2s', 'md5', 'pbkdf2_hmac', 'sha1', 'sha224', 'sha256',
                    'sha384', 'sha3_224', 'sha3_256', 'sha3_384', 'sha3_512', 'sha512', 'shake_128', 'shake_256')


def get_hash_fn(hash_type):
    if hash_type in _SUPPORTED_TYPES:
        return getattr(hashlib, hash_type)
    else:
        raise ConfigKeyValueError(f"'{hash_type}' is not a valid hash type.")


def to_bytes(key):
    if isinstance(key, str):
        key = key.encode('utf-8')
    else:
        assert isinstance(key, bytes)
    return key


def hexdigest(key, hash_fn):
    key = to_bytes(key)
    m = hash_fn()
    m.update(key)
    return m.hexdigest()