from core.builtins import Bot, Plain, Image, Url
from core.component import module
from core.utils.image_table import image_table_render, ImageTable
from core.utils.http import get_url
from core.logger import logger

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


@ncmusic.handle(['search <keyword> {{ncmusic.help.search}}',
                 'search legacy <keyword> {{ncmusic.help.search.legacy}}'])
async def search(msg: Bot.MessageSession, keyword: str):
    url = f"https://ncmusic.akari-bot.top/search?keywords={keyword}"
    result = await get_url(url, 200, fmt='json')
    song_count = result['result']['songCount']
    legacy = True

    if song_count == 0:
        await msg.finish(msg.locale.t('ncmusic.message.search.not_found'))

    songs = result['result']['songs'][:10]

    if not msg.parsed_msg.get('legacy', False) and msg.Feature.image:

        send_msg = [Plain(msg.locale.t('ncmusic.message.search.result') + '\n')]
        data = [[
                str(i),
                song['name'] + (f" ({' / '.join(song['transNames'])})" if 'transNames' in song else ''),
                f"{' / '.join(artist['name'] for artist in song['artists'])}",
                f"{song['album']['name']}" + (f" ({' / '.join(song['album']['transNames'])})" if 'transNames' in song['album'] else ''),
                f"{song['id']}"
            ] for i, song in enumerate(songs, start=1)
        ]

        tables = ImageTable(data, [
            msg.locale.t('ncmusic.message.search.table.header.id'),
            msg.locale.t('ncmusic.message.search.table.header.name'),
            msg.locale.t('ncmusic.message.search.table.header.artists'),
            msg.locale.t('ncmusic.message.search.table.header.album'),
            'ID'
            ])

        img = await image_table_render(tables)
        if img:
            legacy = False

            send_msg.append(Image(img))
            if song_count > 10:
                song_count = 10
                send_msg.append(Plain(msg.locale.t("message.collapse", amount="10")))
        
        if song_count == 1:
            send_msg.append(Plain(msg.locale.t('ncmusic.message.search.confirm')))
            query = await msg.wait_confirm(send_msg, delete=False)
            if query:
                sid = result['result']['songs'][0]['id']
            else:
                return

        else:
            send_msg.append(Plain(msg.locale.t('ncmusic.message.search.prompt')))
            query = await msg.wait_reply(send_msg)
            query = query.as_display(text_only=True)

            if query.isdigit():
                query = int(query)
                if query > song_count:
                    await msg.finish(msg.locale.t("mod_dl.message.invalid.out_of_range"))
                else:
                    sid = result['result']['songs'][query - 1]['id']
            else:
                await msg.finish(msg.locale.t('ncmusic.message.search.invalid.non_digital'))

        await info(msg, sid)

    if legacy:
        send_msg = msg.locale.t('ncmusic.message.search.result') + '\n'

        for i, song in enumerate(songs, start=1):
            send_msg += f"{i}\u200B. {song['name']}"
            if 'transNames' in song:
                send_msg += f"（{' / '.join(song['transNames'])}）"
            send_msg += f"——{' / '.join(artist['name'] for artist in song['artists'])}"
            if song['album']['name']:
                send_msg += f"《{song['album']['name']}》"
            if 'transNames' in song['album']:
                send_msg += f"（{' / '.join(song['album']['transNames'])}）"
            send_msg += f"（{song['id']}）\n"

        if song_count > 10:
            song_count = 10
            send_msg += msg.locale.t("message.collapse", amount="10")

        if song_count == 1:
            send_msg += '\n' + msg.locale.t('ncmusic.message.search.confirm')
            query = await msg.wait_confirm(send_msg, delete=False)
            if query:
                sid = result['result']['songs'][0]['id']
            else:
                return
        else:
            send_msg += '\n' + msg.locale.t('ncmusic.message.search.prompt')
            query = await msg.wait_reply(send_msg)
            query = query.as_display(text_only=True)
        
            if query.isdigit():
                query = int(query)
                if query > song_count:
                    await msg.finish(msg.locale.t("mod_dl.message.invalid.out_of_range"))
                else:
                    sid = result['result']['songs'][query - 1]['id']
            else:
                await msg.finish(msg.locale.t('ncmusic.message.search.invalid.non_digital'))

        await info(msg, sid)
        

@ncmusic.handle('info <sid> {{ncmusic.help.info}}')
async def info(msg: Bot.MessageSession, sid: str):
    url = f"https://ncmusic.akari-bot.top/song/detail?ids={sid}"
    result = await get_url(url, 200, fmt='json')

    if result['songs']:
        info = result['songs'][0]
        artist = ' / '.join([ar['name'] for ar in info['ar']])
        song_url = f"https://music.163.com/#/song?id={info['id']}"

        send_msg = msg.locale.t('ncmusic.message.info',
                                name=info['name'],
                                id=info['id'],
                                artists=artist,
                                album=info['al']['name'],
                                album_id=info['al']['id'])
                                
        await msg.finish([Image(info['al']['picUrl']), Url(song_url), Plain(send_msg)])
    else:
        await msg.finish(msg.locale.t('ncmusic.message.info.not_found'))