LittlePaimon/Guess_voice/__init__.py

import asyncio
from . import util
from typing import Union
from pathlib import Path
from nonebot import on_command
from nonebot.permission import SUPERUSER
from nonebot.params import CommandArg
from nonebot.exception import FinishedException
from nonebot.adapters.onebot.v11 import PrivateMessageEvent, GroupMessageEvent, MessageSegment, Bot, permission
from .handler import Guess, get_random_voice
from . import download_data
from ..utils.config import config


setting_time = config.paimon_guess_voice  # 游戏持续时间

dir_name = Path() / 'LittlePaimon' / 'LittlePaimon' / 'Guess_voice' / 'voice'
dir_name.mkdir(parents=True, exist_ok=True)


guess_game = on_command('原神猜语音', priority=12, block=True)
running_guess_game = on_command('我猜', aliases={'guess', 'ig'}, priority=12, permission=permission.GROUP, block=True)
ys_voice = on_command('原神语音', priority=12, block=True)
update_ys_voice = on_command('更新原神语音资源', priority=12, permission=SUPERUSER, block=True)


async def download_voice(bot: Bot, event: Union[PrivateMessageEvent, GroupMessageEvent]):
    if not dir_name.exists():
        dir_name.mkdir(parents=True, exist_ok=True)
        await bot.send(event, '资源尚未初始化，现在开始下载资源，这需要较长的时间，请耐心等待')
        await download_data.update_voice_data()
        await bot.send(event, '资源下载完成，请重新发送指令开始游戏')


@guess_game.handle()
async def guess_genshin_voice(bot: Bot, event: GroupMessageEvent, msg=CommandArg()):
    await download_voice(bot, event)
    keyword = str(msg).strip()
    guess = Guess(event.group_id, time=setting_time)

    hard_mode = False

    if keyword == '排行榜':
        await guess_game.finish(await guess.get_rank(bot, event))
    if keyword in ['中', '中国', '汉语', '中文', '中国话', 'Chinese', 'cn'] or not keyword:
        keyword = '中'
    elif keyword in ['日', '日本', '日语', '霓虹', '日本语', 'Japanese', 'jp']:
        keyword = '日'
    elif keyword in ['韩', '韩国', '韩语', '棒子', '南朝鲜', '南朝鲜语']:
        keyword = '韩'
    elif keyword in ['英', '英文', '英语', '洋文', 'English', 'en']:
        keyword = '英'
    elif keyword in ['2', '难', '困难', '地狱']:
        hard_mode = True
    else:
        await guess_game.finish(f'没有找到{keyword}的语音')
    if guess.is_start():
        await guess_game.finish('游戏正在进行中哦')
    guess.set_start()
    await guess_game.send(f'即将发送一段原神语音,将在{setting_time}秒后公布答案')
    await asyncio.sleep(1)
    try:
        if hard_mode:
            await guess_game.finish(await guess.start2())
        else:
            res = await guess.start(keyword.split())
            print(res)
            await guess_game.finish(res)
    except FinishedException:
        pass
    except Exception as e:
        guess.set_end()
        await guess_game.finish(str(e))


@running_guess_game.handle()
async def on_input_chara_name(event: GroupMessageEvent, msg=CommandArg()):
    msg = str(msg).strip()
    guess = Guess(event.group_id, time=setting_time)
    if guess.is_start():
        await guess.add_answer(event.user_id, msg)


@ys_voice.handle()
async def get_genshin_voice(bot: Bot, event: Union[PrivateMessageEvent, GroupMessageEvent], msg=CommandArg()):
    name = str(msg).strip()
    if name.startswith('日'):
        language = '日'
        name = name[1:]
    else:
        language = '中'
    await download_voice(bot, event)
    path = await get_random_voice(name, language)
    if not path:
        await ys_voice.finish(f'没有找到{name}的语音呢')
    await ys_voice.finish(MessageSegment.record(file=Path(path)))


@update_ys_voice.handle()
async def update_genshin_voice(bot: Bot, event: Union[PrivateMessageEvent, GroupMessageEvent]):
    await update_ys_voice.send('将在后台开始更新原神语音资源，请耐心等待资源下载完成后再使用原神语音')
    await download_data.update_voice_data()
    await update_ys_voice.finish('原神语音资源更新完成')