You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
395 lines
16 KiB
395 lines
16 KiB
import asyncio
|
|
import contextlib
|
|
import enum
|
|
import logging
|
|
import wave
|
|
from functools import lru_cache
|
|
from inspect import cleandoc
|
|
from typing import Tuple
|
|
|
|
import discord
|
|
from d4dj_utils.master.chart_master import ChartDifficulty, ChartMaster
|
|
from d4dj_utils.master.common_enums import ChartSectionType
|
|
from d4dj_utils.master.music_master import MusicMaster
|
|
from discord.ext import commands
|
|
|
|
from main import asset_manager, masters
|
|
from miyu_bot.commands.common.argument_parsing import parse_arguments, ArgumentError, list_operator_for
|
|
from miyu_bot.commands.common.emoji import difficulty_emoji_ids
|
|
from miyu_bot.commands.common.formatting import format_info
|
|
from miyu_bot.commands.common.fuzzy_matching import romanize
|
|
from miyu_bot.commands.common.master_asset_manager import hash_master
|
|
from miyu_bot.commands.common.reaction_message import run_tabbed_message, run_paged_message
|
|
|
|
|
|
class Music(commands.Cog):
|
|
def __init__(self, bot: commands.Bot):
|
|
self.bot = bot
|
|
self.logger = logging.getLogger(__name__)
|
|
|
|
@property
|
|
def reaction_emojis(self):
|
|
return [self.bot.get_emoji(eid) for eid in difficulty_emoji_ids.values()]
|
|
|
|
difficulty_names = {
|
|
'expert': ChartDifficulty.Expert,
|
|
'hard': ChartDifficulty.Hard,
|
|
'normal': ChartDifficulty.Normal,
|
|
'easy': ChartDifficulty.Easy,
|
|
'exp': ChartDifficulty.Expert,
|
|
'hrd': ChartDifficulty.Hard,
|
|
'nrm': ChartDifficulty.Normal,
|
|
'esy': ChartDifficulty.Easy,
|
|
'ex': ChartDifficulty.Expert,
|
|
'hd': ChartDifficulty.Hard,
|
|
'nm': ChartDifficulty.Normal,
|
|
'es': ChartDifficulty.Easy,
|
|
}
|
|
|
|
@commands.command(name='song',
|
|
aliases=['music'],
|
|
description='Finds the song with the given name.',
|
|
help='!song grgr')
|
|
async def song(self, ctx: commands.Context, *, arg: commands.clean_content):
|
|
self.logger.info(f'Searching for song "{arg}".')
|
|
|
|
song = masters.music.get(arg, ctx)
|
|
|
|
if not song:
|
|
msg = f'Failed to find song "{arg}".'
|
|
await ctx.send(msg)
|
|
self.logger.info(msg)
|
|
return
|
|
self.logger.info(f'Found song "{song}" ({romanize(song.name)}).')
|
|
|
|
try:
|
|
thumb = discord.File(song.jacket_path, filename='jacket.png')
|
|
except FileNotFoundError:
|
|
# Just a fallback
|
|
thumb = discord.File(asset_manager.path / 'ondemand/stamp/stamp_10006.png', filename='jacket.png')
|
|
|
|
embed = discord.Embed(title=song.name)
|
|
embed.set_thumbnail(url=f'attachment://jacket.png')
|
|
|
|
artist_info = {
|
|
'Lyricist': song.lyricist,
|
|
'Composer': song.composer,
|
|
'Arranger': song.arranger,
|
|
'Unit': song.unit.name,
|
|
'Special Unit Name': song.special_unit_name,
|
|
}
|
|
|
|
music_info = {
|
|
'Category': song.category.name,
|
|
'Duration': self.format_duration(self.get_music_duration(song)),
|
|
'BPM': song.bpm,
|
|
'Section Trend': song.section_trend.name,
|
|
'Sort Order': song.default_order,
|
|
'Levels': ', '.join(c.display_level for c in song.charts.values()),
|
|
'Release Date': song.start_datetime,
|
|
}
|
|
|
|
embed.add_field(name='Artist',
|
|
value=format_info(artist_info),
|
|
inline=False)
|
|
embed.add_field(name='Info',
|
|
value=format_info(music_info),
|
|
inline=False)
|
|
|
|
await ctx.send(files=[thumb], embed=embed)
|
|
|
|
@commands.command(name='chart',
|
|
aliases=[],
|
|
description='Finds the chart with the given name.',
|
|
help='!chart grgr\n!chart grgr normal')
|
|
async def chart(self, ctx: commands.Context, *, arg: commands.clean_content):
|
|
self.logger.info(f'Searching for chart "{arg}".')
|
|
|
|
name, difficulty = self.parse_chart_args(arg)
|
|
song = masters.music.get(name, ctx)
|
|
|
|
if not song:
|
|
msg = f'Failed to find chart "{name}".'
|
|
await ctx.send(msg)
|
|
self.logger.info(msg)
|
|
return
|
|
self.logger.info(f'Found song "{song}" ({romanize(song.name)}).')
|
|
|
|
embeds, files = self.get_chart_embed_info(song)
|
|
|
|
# Difficulty enum easy-expert are 1-4, one more than the embed index
|
|
asyncio.ensure_future(run_tabbed_message(ctx, self.reaction_emojis, embeds, files, difficulty - 1))
|
|
|
|
@commands.command(name='sections',
|
|
aliases=['mixes'],
|
|
description='Finds the sections of the chart with the given name.',
|
|
help='!sections grgr')
|
|
async def sections(self, ctx: commands.Context, *, arg: commands.clean_content):
|
|
self.logger.info(f'Searching for chart sections "{arg}".')
|
|
|
|
name, difficulty = self.parse_chart_args(arg)
|
|
song = masters.music.get(name, ctx)
|
|
|
|
if not song:
|
|
msg = f'Failed to find chart "{name}".'
|
|
await ctx.send(msg)
|
|
self.logger.info(msg)
|
|
return
|
|
if not song.mix_info:
|
|
msg = f'Song "{song.name}" does not have mix enabled.'
|
|
await ctx.send(msg)
|
|
self.logger.info(msg)
|
|
return
|
|
self.logger.info(f'Found song "{song}" ({romanize(song.name)}).')
|
|
|
|
embeds, files = self.get_mix_embed_info(song)
|
|
|
|
asyncio.ensure_future(run_tabbed_message(ctx, self.reaction_emojis, embeds, files, difficulty - 1))
|
|
|
|
@commands.command(name='songs',
|
|
aliases=['songsearch', 'song_search'],
|
|
description='Finds songs matching the given name.',
|
|
brief='!songs lhg',
|
|
help=cleandoc('''
|
|
Named arguments:
|
|
- sort (<, =) [default|name|id|unit|level|difficulty|duration|date]
|
|
- [display|disp] = [default|name|id|unit|level|difficulty|duration|date]
|
|
- [difficulty|diff|level] ? <difficulty (11, 11.5, 11+, ...)>...
|
|
|
|
Extended examples:
|
|
- Songs in descending difficulty order
|
|
!songs sort<difficulty
|
|
- Songs with difficulty from 11+ to 13+
|
|
!songs diff>=11+ diff<=13+
|
|
- Songs with difficulty exactly 10 or 14, sorted alphabetically, displaying duration
|
|
!songs diff=10,14 sort=name disp=duration'''))
|
|
async def songs(self, ctx: commands.Context, *, arg: commands.clean_content = ''):
|
|
self.logger.info(f'Searching for songs "{arg}".' if arg else 'Listing songs.')
|
|
arguments = parse_arguments(arg)
|
|
songs = masters.music.get_sorted(arguments.text_argument, ctx)
|
|
|
|
try:
|
|
sort, sort_op = arguments.single('sort', MusicAttribute.DefaultOrder,
|
|
allowed_operators=['<', '>', '='], converter=music_attribute_aliases)
|
|
reverse_sort = sort_op == '<'
|
|
display, _ = arguments.single(['display', 'disp'], sort, allowed_operators=['='],
|
|
converter=music_attribute_aliases)
|
|
|
|
def difficulty_converter(d):
|
|
return int(d[:-1]) + 0.5 if d[-1] == '+' else int(d)
|
|
|
|
difficulty = arguments.repeatable(['difficulty', 'diff', 'level'], is_list=True,
|
|
converter=difficulty_converter)
|
|
arguments.require_all_arguments_used()
|
|
except ArgumentError as e:
|
|
await ctx.send(str(e))
|
|
return
|
|
|
|
for value, op in difficulty:
|
|
operator = list_operator_for(op)
|
|
songs = [song for song in songs if operator(song.charts[4].level, value)]
|
|
|
|
if not (arguments.text_argument and sort == MusicAttribute.DefaultOrder):
|
|
songs = sorted(songs, key=lambda s: sort.get_from_music(s))
|
|
if sort == MusicAttribute.DefaultOrder and songs and songs[0].id == 1:
|
|
songs = [*songs[1:], songs[0]]
|
|
if reverse_sort:
|
|
songs = reversed(songs)
|
|
|
|
listing = []
|
|
for song in songs:
|
|
display_prefix = display.get_formatted_from_music(song)
|
|
if display_prefix:
|
|
listing.append(
|
|
f'{display_prefix} : {song.name}{" (" + song.special_unit_name + ")" if song.special_unit_name else ""}')
|
|
else:
|
|
listing.append(f'{song.name}{" (" + song.special_unit_name + ")" if song.special_unit_name else ""}')
|
|
|
|
embed = discord.Embed(title=f'Song Search "{arg}"' if arg else 'Songs')
|
|
asyncio.ensure_future(run_paged_message(ctx, embed, listing))
|
|
|
|
def get_chart_embed_info(self, song):
|
|
embeds = []
|
|
|
|
try:
|
|
thumb = discord.File(song.jacket_path, filename='jacket.png')
|
|
except FileNotFoundError:
|
|
# fallback
|
|
thumb = discord.File(asset_manager.path / 'ondemand/stamp/stamp_10006.png', filename='jacket.png')
|
|
|
|
files = [thumb]
|
|
|
|
for difficulty in [ChartDifficulty.Easy, ChartDifficulty.Normal, ChartDifficulty.Hard, ChartDifficulty.Expert]:
|
|
chart = song.charts[difficulty]
|
|
chart_hash = hash_master(chart)
|
|
chart_path = chart.image_path
|
|
embed = discord.Embed(title=f'{song.name} [{chart.difficulty.name}]')
|
|
embed.set_thumbnail(url=f'attachment://jacket.png')
|
|
embed.set_image(
|
|
url=f'https://qwewqa.github.io/d4dj-dumps/music/charts/{chart_path.stem}_{chart_hash}{chart_path.suffix}'
|
|
)
|
|
|
|
chart_data = chart.load_chart_data()
|
|
note_counts = chart_data.get_note_counts()
|
|
|
|
embed.add_field(name='Info',
|
|
value=f'Level: {chart.display_level}\n'
|
|
f'Duration: {self.format_duration(self.get_music_duration(song))}\n'
|
|
f'Unit: {song.special_unit_name or song.unit.name}\n'
|
|
f'Category: {song.category.name}\n'
|
|
f'BPM: {song.bpm}',
|
|
inline=False)
|
|
embed.add_field(name='Combo',
|
|
value=f'Max Combo: {chart.note_counts[ChartSectionType.Full].count}\n'
|
|
f'Taps: {note_counts["tap"]} (dark: {note_counts["tap1"]}, light: {note_counts["tap2"]})\n'
|
|
f'Scratches: {note_counts["scratch"]} (left: {note_counts["scratch_left"]}, right: {note_counts["scratch_right"]})\n'
|
|
f'Stops: {note_counts["stop"]} (head: {note_counts["stop_start"]}, tail: {note_counts["stop_end"]})\n'
|
|
f'Long: {note_counts["long"]} (head: {note_counts["long_start"]}, tail: {note_counts["long_end"]})\n'
|
|
f'Slide: {note_counts["slide"]} (tick: {note_counts["slide_tick"]}, flick {note_counts["slide_flick"]})',
|
|
inline=True)
|
|
embed.add_field(name='Ratings',
|
|
value=f'NTS: {round(chart.trends[0] * 100, 2)}%\n'
|
|
f'DNG: {round(chart.trends[1] * 100, 2)}%\n'
|
|
f'SCR: {round(chart.trends[2] * 100, 2)}%\n'
|
|
f'EFT: {round(chart.trends[3] * 100, 2)}%\n'
|
|
f'TEC: {round(chart.trends[4] * 100, 2)}%\n',
|
|
inline=True)
|
|
embed.set_footer(text='1 column = 10 seconds')
|
|
|
|
embeds.append(embed)
|
|
|
|
return embeds, files
|
|
|
|
def get_mix_embed_info(self, song):
|
|
embeds = []
|
|
files = [discord.File(song.jacket_path, filename=f'jacket.png')]
|
|
|
|
for difficulty in [ChartDifficulty.Easy, ChartDifficulty.Normal, ChartDifficulty.Hard, ChartDifficulty.Expert]:
|
|
chart: ChartMaster = song.charts[difficulty]
|
|
embed = discord.Embed(title=f'Mix: {song.name} [{chart.difficulty.name}]')
|
|
embed.set_thumbnail(url=f'attachment://jacket.png')
|
|
embed.set_image(
|
|
url=f'https://qwewqa.github.io/d4dj-dumps/{chart.mix_path.relative_to(asset_manager.path).as_posix()}'
|
|
)
|
|
|
|
note_counts = chart.note_counts
|
|
mix_info = chart.mix_info
|
|
|
|
info = {
|
|
'Level': chart.display_level,
|
|
'Unit': song.unit.name,
|
|
'BPM': song.bpm,
|
|
'Section Trend': song.section_trend.name,
|
|
}
|
|
|
|
begin = {
|
|
'Time': f'{round(mix_info[ChartSectionType.Begin].duration, 2)}s',
|
|
'Combo': note_counts[ChartSectionType.Begin].count,
|
|
}
|
|
middle = {
|
|
'Time': f'{round(mix_info[ChartSectionType.Middle].duration, 2)}s',
|
|
'Combo': note_counts[ChartSectionType.Middle].count,
|
|
}
|
|
end = {
|
|
'Time': f'{round(mix_info[ChartSectionType.End].duration, 2)}s',
|
|
'Combo': note_counts[ChartSectionType.End].count,
|
|
}
|
|
|
|
embed.add_field(name='Info',
|
|
value=format_info(info),
|
|
inline=False)
|
|
embed.add_field(name='Begin',
|
|
value=format_info(begin),
|
|
inline=True)
|
|
embed.add_field(name='Middle',
|
|
value=format_info(middle),
|
|
inline=True)
|
|
embed.add_field(name='End',
|
|
value=format_info(end),
|
|
inline=True)
|
|
embed.set_footer(text='1 column = 10 seconds')
|
|
|
|
embeds.append(embed)
|
|
|
|
return embeds, files
|
|
|
|
def parse_chart_args(self, arg: str) -> Tuple[str, ChartDifficulty]:
|
|
split_args = arg.split()
|
|
|
|
difficulty = ChartDifficulty.Expert
|
|
if len(split_args) >= 2:
|
|
final_word = split_args[-1]
|
|
if final_word in self.difficulty_names:
|
|
difficulty = self.difficulty_names[final_word]
|
|
arg = ''.join(split_args[:-1])
|
|
return arg, difficulty
|
|
|
|
_music_durations = {}
|
|
|
|
@staticmethod
|
|
def get_music_duration(music: MusicMaster):
|
|
if music.id in Music._music_durations:
|
|
return Music._music_durations[music.id]
|
|
with contextlib.closing(wave.open(str(music.audio_path.with_name(music.audio_path.name + '.wav')), 'r')) as f:
|
|
frames = f.getnframes()
|
|
rate = f.getframerate()
|
|
duration = frames / float(rate)
|
|
Music._music_durations[music.id] = duration
|
|
return duration
|
|
|
|
@staticmethod
|
|
def format_duration(seconds):
|
|
minutes = int(seconds // 60)
|
|
seconds = round(seconds % 60, 2)
|
|
return f'{minutes}:{str(int(seconds)).zfill(2)}.{str(int(seconds % 1 * 100)).zfill(2)}'
|
|
|
|
|
|
class MusicAttribute(enum.Enum):
|
|
DefaultOrder = enum.auto()
|
|
Name = enum.auto()
|
|
Id = enum.auto()
|
|
Unit = enum.auto()
|
|
Level = enum.auto()
|
|
Duration = enum.auto()
|
|
Date = enum.auto()
|
|
|
|
def get_from_music(self, music: MusicMaster):
|
|
return {
|
|
self.DefaultOrder: -music.default_order,
|
|
self.Name: music.name,
|
|
self.Id: music.id,
|
|
self.Unit: music.unit.name if not music.special_unit_name else f'{music.unit.name} ({music.special_unit_name})',
|
|
self.Level: music.charts[4].display_level,
|
|
self.Duration: Music.get_music_duration(music),
|
|
self.Date: music.start_datetime
|
|
}[self]
|
|
|
|
def get_formatted_from_music(self, music: MusicMaster):
|
|
return {
|
|
self.DefaultOrder: None,
|
|
self.Name: None,
|
|
self.Id: str(music.id).zfill(7),
|
|
self.Unit: music.unit.name if not music.special_unit_name else f'{music.unit.name} ({music.special_unit_name})',
|
|
self.Level: music.charts[4].display_level,
|
|
self.Duration: Music.format_duration(Music.get_music_duration(music)),
|
|
self.Date: str(music.start_datetime.date()),
|
|
}[self]
|
|
|
|
|
|
music_attribute_aliases = {
|
|
'default': MusicAttribute.DefaultOrder,
|
|
'name': MusicAttribute.Name,
|
|
'id': MusicAttribute.Id,
|
|
'relevance': MusicAttribute.Name,
|
|
'unit': MusicAttribute.Unit,
|
|
'level': MusicAttribute.Level,
|
|
'difficulty': MusicAttribute.Level,
|
|
'diff': MusicAttribute.Level,
|
|
'duration': MusicAttribute.Duration,
|
|
'length': MusicAttribute.Duration,
|
|
'date': MusicAttribute.Date,
|
|
}
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Music(bot))
|
|
|