akari-bot/modules/maimai/__init__.py

import re

from core.builtins import Bot, Plain, Image as BImage
from core.component import module
from core.logger import Logger
from core.utils.image import msgchain2image
from modules.maimai.libraries.image import *
from modules.maimai.libraries.maimai_best_40 import generate
from modules.maimai.libraries.maimai_best_50 import generate50
from modules.maimai.libraries.maimaidx_music import *
from modules.maimai.libraries.tool import hash_

total_list = TotalList()

diff_label = ['Basic', 'Advanced', 'Expert', 'Master', 'Re:MASTER']
diff_label_abbr = ['bas', 'adv', 'exp', 'mas', 'rem']
diff_label_zhs = ['绿', '黄', '红', '紫', '白']
diff_label_zht = ['綠', '黃', '紅']


def song_txt(music: Music):
    return [Plain(f"{music.id} {music.title}{' (DX)' if music['type'] == 'DX' else ''}\n"),
            BImage(f"https://www.diving-fish.com/covers/{get_cover_len4_id(music.id)}.png", ),
            Plain(f"\n{'/'.join(music.level)}")]


def get_label(diff):
    diff = diff.lower()
    diff_label_lower = [label.lower() for label in diff_label]
    
    if diff in diff_label_zhs:
        level = diff_label_zhs.index(diff)
    elif diff in diff_label_zht:
        level = diff_label_zht.index(diff)
    elif diff in diff_label_abbr:
        level = diff_label_abbr.index(diff)
    elif diff in diff_label_lower:
        level = diff_label_lower.index(diff)
    else:
        level = None
    return level



mai = module('maimai', developers=['mai-bot', 'OasisAkari', 'DoroWolf'], alias='mai',
             desc='{maimai.help.desc}')



@mai.handle('level <level> {{maimai.help.level}}')
async def _(msg: Bot.MessageSession):
    level = msg.parsed_msg['<level>']
    result_set = await diff_level_q(level)
    s = msg.locale.t("maimai.message.level", level=level) + "\n"
    for elem in result_set:
        s += f"{elem[0]} {elem[1]}{' (DX)' if elem[5] == 'DX' else ''} {elem[3]} {elem[4]} ({elem[2]})\n"
    if len(result_set) == 0:
        return await msg.finish(msg.locale.t("maimai.message.music_not_found"))
    if len(result_set) <= 10:
        await msg.finish(s.strip())
    else:
        img = await msgchain2image([Plain(s)])
        await msg.finish([BImage(img)])

async def diff_level_q(level):
    result_set = []
    music_data = (await total_list.get()).filter(level=level)
    for music in sorted(music_data, key=lambda i: int(i['id'])):
        for i in music.diff:
            result_set.append((music['id'], music['title'], music['ds'][i], diff_label[i], music['level'][i], music['type']))
    return result_set



@mai.handle('inner <rating> [<rating_max>] {{maimai.help.inner}}')
async def _(msg: Bot.MessageSession):
    try:
        rating = float(msg.parsed_msg['<rating>'])
    except ValueError:
        return await msg.finish(msg.locale.t('error.range.notnumber'))
    rating_max = msg.parsed_msg.get('<rating_max>')
    if rating_max is not None:
        try:
            rating_max = float(rating_max)
        except ValueError:
            return await msg.finish(msg.locale.t('error.range.notnumber'))
        if rating > rating_max:
            return await msg.finish(msg.locale.t('error.range.invalid'))
        result_set = await inner_level_q(rating, rating_max)
        s = msg.locale.t("maimai.message.inner.range", rating=round(rating, 1), rating_max=round(rating_max, 1)) + "\n"
    else:
        result_set = await inner_level_q(rating)
        s = msg.locale.t("maimai.message.inner", rating=round(rating, 1)) + "\n"
    for elem in result_set:
        s += f"{elem[0]} {elem[1]}{' (DX)' if elem[5] == 'DX' else ''} {elem[3]} {elem[4]} ({elem[2]})\n"
    if len(result_set) == 0:
        return await msg.finish(msg.locale.t("maimai.message.music_not_found"))
    if len(result_set) > 200:
        return await msg.finish(msg.locale.t("maimai.message.too_much", length=len(result_set)))
    if len(result_set) <= 10:
        await msg.finish(s.strip())
    else:
        img = await msgchain2image([Plain(s)])
        await msg.finish([BImage(img)])

async def inner_level_q(ds1, ds2=None):
    result_set = []
    if ds2 is not None:
        music_data = (await total_list.get()).filter(ds=(ds1, ds2))
    else:
        music_data = (await total_list.get()).filter(ds=ds1)
    for music in sorted(music_data, key=lambda i: int(i['id'])):
        for i in music.diff:
            result_set.append((music['id'], music['title'], music['ds'][i], diff_label[i], music['level'][i], music['type']))
    return result_set



@mai.handle('search <keyword> {{maimai.help.search}}')
async def _(msg: Bot.MessageSession):
    name = msg.parsed_msg['<keyword>'].strip()
    if name == "":
        return
    res = (await total_list.get()).filter(title_search=name)
    if len(res) == 0:
        return await msg.finish(msg.locale.t("maimai.message.music_not_found"))
    if len(res) > 200:
        return await msg.finish(msg.locale.t("maimai.message.too_much", length=len(res)))
    else:
        search_result = msg.locale.t("maimai.message.search", keyword=name) + "\n"
        for music in sorted(res, key=lambda i: int(i['id'])):
            search_result += f"{music['id']} {music['title']}{' (DX)' if music['type'] == 'DX' else ''}\n"
        if len(res) <= 10:
            await msg.finish([Plain(search_result.strip())])
        else:
            img = await msgchain2image([Plain(search_result)])
            await msg.finish([BImage(img)])



@mai.handle('b40 [<username>] {{maimai.help.b40}}')
async def _(msg: Bot.MessageSession):
    username = msg.parsed_msg.get('<username>', None)
    if username is None and msg.target.senderFrom == "QQ":
        payload = {'qq': msg.session.sender}
    else:
        if username is None:
            await msg.finish(msg.locale.t("maimai.message.no_username"))
        payload = {'username': username}
    img, success = await generate(payload)
    if success == 400:
        await msg.finish(msg.locale.t("maimai.message.user_not_found"))
    elif success == 403:
        await msg.finish(msg.locale.t("maimai.message.forbidden"))
    else:
        if img:
            await msg.finish([BImage(img)])



@mai.handle('b50 [<username>] {{maimai.help.b50}}')
async def _(msg: Bot.MessageSession):
    username = msg.parsed_msg.get('<username>', None)
    if username is None and msg.target.senderFrom == "QQ":
        payload = {'qq': msg.session.sender, 'b50': True}
    else:
        if username is None:
            await msg.finish(msg.locale.t("maimai.message.no_username"))
        payload = {'username': username, 'b50': True}
    img, success = await generate50(payload)
    if success == 400:
        await msg.finish(msg.locale.t("maimai.message.user_not_found"))
    elif success == 403:
        await msg.finish(msg.locale.t("maimai.message.forbidden"))
    else:
        if img:
            await msg.finish([BImage(img)])



@mai.handle('random <diff> <level> [<type>] {{maimai.help.random.filter}}')
async def _(msg: Bot.MessageSession):
    diff = msg.parsed_msg['<diff>']
    level = msg.parsed_msg['<level>']
    tp = msg.parsed_msg.get('<type>', None)
    diff_index = get_label(diff)
    try:
        if tp == "dx":
            tp = ["DX"]
        elif tp in ["sd", "标准", "標準"]:
            tp = ["SD"]
        else:
            tp = ["SD", "DX"]

        if diff == "#":
            if level == "#":
                music_data = (await total_list.get()).filter(type=tp)
            else:
                music_data = (await total_list.get()).filter(level=level, type=tp)
        else:
            if level == "#":
                music_data = (await total_list.get()).filter(diff=[diff_index], type=tp)
            else:
                music_data = (await total_list.get()).filter(level=level, diff=[diff_index], type=tp)

        if len(music_data) == 0:
            rand_result = msg.locale.t("maimai.message.music_not_found")
        else:
            rand_result = song_txt(music_data.random())
        await msg.finish(rand_result)
    except Exception as e:
        Logger.error(e)
        await msg.finish(msg.locale.t("maimai.message.random.error"))


@mai.handle('random {{maimai.help.random}}')
async def _(msg: Bot.MessageSession):
    await msg.finish(song_txt((await total_list.get()).random()))



    
@mai.handle('song <id> [<diff>] {{maimai.help.song}}')
async def _(message: Bot.MessageSession):
    id = message.parsed_msg['<id>']
    diff = message.parsed_msg.get('<diff>', None)
    if diff is not None:
        try:
            diff_index = get_label(diff)
            name = id
            music = (await total_list.get()).by_id(name)
            chart = music['charts'][diff_index]
            ds = music['ds'][diff_index]
            level = music['level'][diff_index]
            file = f"https://www.diving-fish.com/covers/{get_cover_len4_id(music['id'])}.png"
            if len(chart['notes']) == 4:
                msg = message.locale.t("maimai.message.song.sd", diff=diff_label[diff_index], level=level, ds=ds, 
                                        tap=chart['notes'][0], hold=chart['notes'][1], slide=chart['notes'][2], _break=chart['notes'][3], 
                                        charter=chart['charter'])
            else:
                msg = message.locale.t("maimai.message.song.dx", diff=diff_label[diff_index], level=level, ds=ds, 
                                        tap=chart['notes'][0], hold=chart['notes'][1], slide=chart['notes'][2], touch=chart['notes'][3], _break=chart['notes'][4], 
                                        charter=chart['charter'])
            await message.finish([Plain(f"{music['id']} {music['title']} {' (DX)' if music['type'] == 'DX' else ''}\n"), BImage(f"{file}"), Plain(msg)])
        except Exception:
            await message.finish(message.locale.t("maimai.message.chart_not_found"))
    else:
        name = id
        music = (await total_list.get()).by_id(name)
        try:
            file = f"https://www.diving-fish.com/covers/{get_cover_len4_id(music['id'])}.png"
            await message.finish([Plain(f"{music['id']} {music['title']} {' (DX)' if music['type'] == 'DX' else ''}\n"),
                                  BImage(f"{file}"),
                                  Plain(message.locale.t("maimai.message.song", 
                                        artist=music['basic_info']['artist'], genre=music['basic_info']['genre'], 
                                        bpm=music['basic_info']['bpm'], version=music['basic_info']['from'], 
                                        level='/'.join(music['level'])))])
        except Exception:
            await message.finish(message.locale.t("maimai.message.music_not_found"))

            

#@mai.handle(['scoreline <difficulty+sid> <scoreline> {查找某首歌的分数线}',
#             'scoreline help {查看分数线帮助}'])
#async def _(msg: Bot.MessageSession):
#    r = "([绿黄红紫白])(id)?([0-9]+)"
#    arg1 = msg.parsed_msg.get('<difficulty+sid>')
#    args2 = msg.parsed_msg.get('<scoreline>')
#    argh = msg.parsed_msg.get('help', False)
#    if argh:
#        s = '''此功能为查找某首歌分数线设计。
#命令格式：maimai scoreline <difficulty+sid> <scoreline>
#例如：分数线 紫799 100
#命令将返回分数线允许的 TAP GREAT 容错以及 BREAK 50落等价的 TAP GREAT 数。
#以下为 TAP GREAT 的对应表：
#GREAT/GOOD/MISS
#TAP\t1/2.5/5
#HOLD\t2/5/10
#SLIDE\t3/7.5/15
#TOUCH\t1/2.5/5
#BREAK\t5/12.5/25(外加200落)'''
#        img = text_to_image(s)
#        if img:
#            await msg.finish([BImage(img)])
#    elif args2 is not None:
#        try:
#            grp = re.match(r, arg1).groups()
#            diff_index = diff_label_zhs.index(grp[0])
#            chart_id = grp[2]
#            line = float(arg1)
#            music = (await total_list.get()).by_id(chart_id)
#            chart: Dict[Any] = music['charts'][diff_index]
#            tap = int(chart['notes'][0])
#            slide = int(chart['notes'][2])
#            hold = int(chart['notes'][1])
#            touch = int(chart['notes'][3]) if len(chart['notes']) == 5 else 0
#            brk = int(chart['notes'][-1])
#            total_score = 500 * tap + slide * 1500 + hold * 1000 + touch * 500 + brk * 2500
#            break_bonus = 0.01 / brk
#            break_50_reduce = total_score * break_bonus / 4
#            reduce = 101 - line
#            if reduce <= 0 or reduce >= 101:
#                raise ValueError
#            await msg.finish(f'''{music['title']} {diff_label[diff_index]}
#分数线 {line}% 允许的最多 TAP GREAT 数量为 {(total_score * reduce / 10000):.2f}(每个-{10000 / total_score:.4f}%),
#BREAK 50落(一共{brk}个)等价于 {(break_50_reduce / 100):.3f} 个 TAP GREAT(-{break_50_reduce / total_score * 100:.4f}%)''')
#        except Exception:
#            await msg.finish("格式错误，输入“~maimai scoreline help”以查看帮助信息")