import hashlib

from core.builtins import Bot
from core.component import module

h = module('hash', developers=['Dianliang233'], desc='{hash.help}', )


@h.handle('<algorithm> <str> [<encoding>] {{hash.help}}')
async def _(msg: Bot.MessageSession):
    algorithm = msg.parsed_msg['<algorithm>']
    string = msg.parsed_msg['<str>']
    encoding = msg.parsed_msg.get('<encoding>', 'utf-8')
    try:
        hash_ = hashlib.new(algorithm, string.encode(encoding))
    except ValueError:
        await msg.finish(f"{msg.locale.t('hash.message.unsupported_algorithm', algorithm=algorithm)}\n"
                         f"{msg.locale.t('hash.message.algorithms_list', algorithms=', '.join(hashlib.algorithms_available))}")
    await msg.finish(msg.locale.t('hash.message.output', algorithm=hash_.name, digest=hash_.hexdigest()))


@h.handle('list {{hash.help.list}}')
async def _(msg: Bot.MessageSession):
    await msg.finish(msg.locale.t('hash.message.algorithms_list', algorithms=', '.join(hashlib.algorithms_available)))