from core.builtins import Bot
from core.component import module
from core.utils.http import get_url

hitokoto_types = ["a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l"]

hitokoto = module(
    'hitokoto',
    developers=['bugungu','DoroWolf'],
    desc='{hitokoto.help.desc}',
    alias='htkt',
    support_languages=['zh_cn'])


@hitokoto.handle()
@hitokoto.handle('[<type>] {{hitokoto.help.type}}')
async def _(msg: Bot.MessageSession):
    try:
        set_type = msg.parsed_msg.get('<type>')
    except AttributeError:
        set_type = None
    url = 'https://v1.hitokoto.cn/?'
    if set_type is not None:
        if set_type not in hitokoto_types:
            await msg.finish(msg.locale.t('hitokoto.message.error.type'))
        else:
            url += "c=" + set_type
    data = await get_url(url, 200, fmt='json')
    from_who = data["from_who"] or ""
    _type = msg.locale.t('hitokoto.message.type') + msg.locale.t('hitokoto.message.type.' + data['type'])
    link = 'https://hitokoto.cn?id='
    await msg.finish(f'''{data["hitokoto"]}\n——{from_who}「{data["from"]}」\n{_type}\n{link}{data["id"]}''')