140 lines
No EOL
5.8 KiB
Python
140 lines
No EOL
5.8 KiB
Python
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')) |