1
0
mirror of synced 2024-11-24 14:30:11 +01:00
bemaniutils/bemani/data/triggers.py

62 lines
2.8 KiB
Python

from datetime import datetime
from discord_webhook import DiscordWebhook, DiscordEmbed # type: ignore
from typing import Dict
from bemani.common.constants import GameConstants, BroadcastConstants
from bemani.data.config import Config
from bemani.data.types import Song
class Triggers:
"""
Class for broadcasting data to some outside service
"""
def __init__(self, config: Config) -> None:
self.config = config
def __gameconst_to_series(self, game: GameConstants) -> str:
return {
GameConstants.BISHI_BASHI: 'Bishi Bashi',
GameConstants.DANCE_EVOLUTION: 'Dance Evolution',
GameConstants.DDR: 'Dance Dance Revolution',
GameConstants.IIDX: 'Beatmania IIDX',
GameConstants.JUBEAT: 'Jubeat',
GameConstants.MGA: 'Metal Gear Arcade',
GameConstants.MUSECA: 'MÚSECA',
GameConstants.POPN_MUSIC: 'Pop\'n Music',
GameConstants.REFLEC_BEAT: 'Reflec Beat',
GameConstants.SDVX: 'Sound Voltex',
}.get(game, 'Unknown')
def has_broadcast_destination(self, game: GameConstants) -> bool:
# For now we only support discord
if self.config.webhooks.discord[game] is not None:
return True
# Nothing is hooked up for this game, so there is no destination.
return False
def broadcast_score(self, data: Dict[BroadcastConstants, str], game: GameConstants, song: Song) -> None:
# For now we only support discord
if self.config.webhooks.discord[game] is not None:
self.broadcast_score_discord(data, game, song)
def broadcast_score_discord(self, data: Dict[BroadcastConstants, str], game: GameConstants, song: Song) -> None:
if game in {GameConstants.IIDX, GameConstants.POPN_MUSIC}:
now = datetime.now()
webhook = DiscordWebhook(url=self.config.webhooks.discord[game])
scoreembed = DiscordEmbed(title=f'New {self.__gameconst_to_series(game)} Score!', color='fbba08')
scoreembed.set_footer(text=(now.strftime('Score was recorded on %m/%d/%y at %H:%M:%S')))
# lets give it an author
song_url = f"{self.config.server.uri}/{game.value}/topscores/{song.id}" if self.config.server.uri is not None else None
scoreembed.set_author(name=self.config.name, url=song_url)
for item, value in data.items():
inline = True
if item in {BroadcastConstants.DJ_NAME, BroadcastConstants.PLAYER_NAME, BroadcastConstants.SONG_NAME, BroadcastConstants.ARTIST_NAME, BroadcastConstants.PLAY_STATS_HEADER}:
inline = False
scoreembed.add_embed_field(name=item.value, value=value, inline=inline)
webhook.add_embed(scoreembed)
webhook.execute()