import traceback

import ujson as json

from config import Config
from core.elements import Url
from core.utils import get_url


async def urban(term: str):
    '''查询urban dictionary。

    :param term: 需要查询的term。
    :returns: 查询结果。'''
    try:
        url = 'http://api.urbandictionary.com/v0/define?term=' + term
        webrender = Config('web_render')
        if not webrender:
            return
        url = webrender + 'source?url=' + url
        text = await get_url(url, 200, headers={'accept': '*/*',
                                                'accept-encoding': 'gzip, deflate',
                                                'accept-language': 'zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7,en-GB;q=0.6',
                                                'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36 Edg/96.0.1054.62'})
        print(text)
        data = json.loads(text)['list']
        if data == []:
            return '[Urban Dictionary] 没有找到相关结果。'
        else:
            count = data.__len__()
            word = data[0]['word']
            definition = limit_length(data[0]['definition'])
            example = limit_length(data[0]['example'])
            link = data[0]['permalink']
            return f'[Urban Dictionary]（{count}个结果）：\n{word}\n{definition}\nExample: {example}\n{str(Url(link))}'
    except Exception:
        traceback.print_exc()
        return '[Urban Dictionary] 查询出错。'


def limit_length(text, limit=50):
    new = text
    length = new.split(' ').__len__()
    if length > limit:
        new = ' '.join(new.split(' ')[0:limit]) + '…'
    return new