81 lines
3.0 KiB
Python
81 lines
3.0 KiB
Python
from datetime import datetime
|
|
from discord_webhook import DiscordWebhook, DiscordEmbed
|
|
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()
|