import traceback

from core.elements import MessageSession, Url
from core.elements.others import ErrorMessage
from core.utils.bot import get_url
from modules.github.utils import dirty_check, darkCheck


async def search(msg: MessageSession):
    try:
        result = await get_url('https://api.github.com/search/repositories?q=' + msg.parsed_msg['<query>'], fmt='json')
        items = result['items']
        item_count_expected = int(result['total_count']) if result['total_count'] < 5 else 5
        items_out = []
        for item in items:
            try:
                items_out.append(str(item['full_name'] + ': ' + str(Url(item['html_url']))))
            except TypeError:
                continue
        footnotes = f"另有 {result['total_count'] - 5} 个结果未显示。" if item_count_expected == 5 else ''
        message = f"搜索成功：共 {result['total_count']} 个结果。\n" + '\n'.join(
            items_out[0:item_count_expected]) + f'\n{footnotes}'

        is_dirty = await dirty_check(message) or darkCheck(message)
        if is_dirty:
            message = 'https://wdf.ink/6OUp'

        await msg.sendMessage(message)
    except Exception as error:
        if result['message'] == 'Not Found':
            await msg.sendMessage('发生错误：查无此人，请检查拼写是否正确。')
        else:
            await msg.sendMessage(ErrorMessage(str(error)))
            traceback.print_exc()