mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 19:44:20 +00:00
434 lines
19 KiB
Python
434 lines
19 KiB
Python
import random
|
||
import re
|
||
import discord
|
||
import discord.opus
|
||
import discord.voice_client
|
||
import functools
|
||
import sys
|
||
import db
|
||
import errors
|
||
import youtube_dl
|
||
import concurrent.futures
|
||
import stagismo
|
||
import platform
|
||
import typing
|
||
import os
|
||
import asyncio
|
||
import configparser
|
||
import subprocess
|
||
import async_timeout
|
||
import raven
|
||
|
||
# Queue emojis
|
||
queue_emojis = [":one:", ":two:", ":three:", ":four:", ":five:", ":six:", ":seven:", ":eight:", ":nine:", ":ten:"]
|
||
|
||
# Init the event loop
|
||
loop = asyncio.get_event_loop()
|
||
|
||
# Init the config reader
|
||
config = configparser.ConfigParser()
|
||
config.read("config.ini")
|
||
|
||
|
||
async def find_user(user: discord.User):
|
||
session = await loop.run_in_executor(executor, db.Session)
|
||
user = await loop.run_in_executor(executor, session.query(db.Discord).filter_by(discord_id=user.id).join(db.Royal).first)
|
||
await loop.run_in_executor(executor, session.close)
|
||
return user
|
||
|
||
|
||
class DurationError(Exception):
|
||
pass
|
||
|
||
|
||
class Video:
|
||
def __init__(self):
|
||
self.enqueuer = None
|
||
self.filename = None
|
||
self.ytdl_url = None
|
||
|
||
@staticmethod
|
||
async def init(user_id: str, *, filename=None, ytdl_url=None):
|
||
if filename is None and ytdl_url is None:
|
||
raise Exception("Filename or url must be specified")
|
||
self = Video()
|
||
self.enqueuer = int(user_id)
|
||
self.filename = filename
|
||
self.ytdl_url = ytdl_url
|
||
return self
|
||
|
||
async def download(self):
|
||
# Retrieve info before downloading
|
||
with youtube_dl.YoutubeDL() as ytdl:
|
||
info = await loop.run_in_executor(executor, functools.partial(ytdl.extract_info, self.ytdl_url, download=False))
|
||
if "entries" in info:
|
||
info = info["entries"][0]
|
||
file_id = info.get("title", str(hash(self.ytdl_url)))
|
||
file_id = re.sub(r'[/\\?*"<>|]', "_", file_id)
|
||
# Set the filename to the downloaded video
|
||
self.filename = file_id
|
||
if os.path.exists(f"opusfiles/{file_id}.opus"):
|
||
return
|
||
if info.get("duration", 1) > int(config["YouTube"]["max_duration"]):
|
||
raise DurationError(f"File duration is over the limit "
|
||
f"set in the config ({config['YouTube']['max_duration']}).")
|
||
ytdl_args = {"noplaylist": True,
|
||
"format": "best",
|
||
"postprocessors": [{
|
||
"key": 'FFmpegExtractAudio',
|
||
"preferredcodec": 'opus'
|
||
}],
|
||
"outtmpl": f"opusfiles/{file_id}.opus",
|
||
"quiet": True}
|
||
if "youtu" in self.ytdl_url:
|
||
ytdl_args["username"] = config["YouTube"]["username"]
|
||
ytdl_args["password"] = config["YouTube"]["password"]
|
||
# Download the video
|
||
with youtube_dl.YoutubeDL(ytdl_args) as ytdl:
|
||
await loop.run_in_executor(executor, functools.partial(ytdl.download, [self.ytdl_url]))
|
||
|
||
async def add_to_db(self):
|
||
session = await loop.run_in_executor(executor, db.Session)
|
||
pm = db.PlayedMusic(enqueuer_id=self.enqueuer,
|
||
filename=self.filename)
|
||
session.add(pm)
|
||
await loop.run_in_executor(executor, session.commit)
|
||
await loop.run_in_executor(executor, session.close)
|
||
|
||
|
||
if __debug__:
|
||
version = "Dev"
|
||
else:
|
||
# Find the latest git tag
|
||
old_wd = os.getcwd()
|
||
try:
|
||
os.chdir(os.path.dirname(__file__))
|
||
version = str(subprocess.check_output(["git", "describe", "--tags"]), encoding="utf8").strip()
|
||
except Exception:
|
||
version = "❓"
|
||
finally:
|
||
os.chdir(old_wd)
|
||
|
||
# Init the discord bot
|
||
client = discord.Client()
|
||
if platform.system() == "Linux":
|
||
discord.opus.load_opus("/usr/lib/x86_64-linux-gnu/libopus.so")
|
||
elif platform.system() == "Windows":
|
||
discord.opus.load_opus("libopus-0.x64.dll")
|
||
|
||
voice_client: typing.Optional[discord.VoiceClient] = None
|
||
voice_player: typing.Optional[discord.voice_client.StreamPlayer] = None
|
||
voice_queue: typing.List[Video] = []
|
||
|
||
# Init the executor
|
||
executor = concurrent.futures.ThreadPoolExecutor(max_workers=3)
|
||
|
||
# Init the Sentry client
|
||
sentry = raven.Client(config["Sentry"]["token"],
|
||
release=version)
|
||
|
||
|
||
async def on_error(event, *args, **kwargs):
|
||
ei = sys.exc_info()
|
||
print("ERRORE CRITICO:\n" + repr(ei[1]) + "\n\n" + repr(ei))
|
||
try:
|
||
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
|
||
f"☢️ ERRORE CRITICO NELL'EVENTO `{event}`\n"
|
||
f"Il bot si è chiuso e si dovrebbe riavviare entro qualche minuto.\n"
|
||
f"Una segnalazione di errore è stata automaticamente mandata a Steffo.\n\n"
|
||
f"Dettagli dell'errore:\n"
|
||
f"```python\n"
|
||
f"{repr(ei[1])}\n"
|
||
f"```")
|
||
if voice_client is not None:
|
||
await voice_client.disconnect()
|
||
await client.change_presence(status=discord.Status.invisible)
|
||
await client.close()
|
||
except Exception as e:
|
||
print("ERRORE CRITICO PIU' CRITICO:\n" + repr(e) + "\n\n" + repr(sys.exc_info()))
|
||
loop.stop()
|
||
sentry.captureException(exc_info=ei)
|
||
os._exit(1)
|
||
pass
|
||
|
||
|
||
@client.event
|
||
async def on_ready():
|
||
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
|
||
f"ℹ Royal Bot {version} avviato e pronto a ricevere comandi!")
|
||
await client.change_presence(game=None, status=discord.Status.online)
|
||
|
||
|
||
@client.event
|
||
async def on_message(message: discord.Message):
|
||
global voice_queue
|
||
global voice_player
|
||
if not message.content.startswith("!"):
|
||
return
|
||
sentry.user_context({
|
||
"discord": {
|
||
"discord_id": message.author.id,
|
||
"name": message.author.name,
|
||
"discriminator": message.author.discriminator
|
||
}
|
||
})
|
||
await client.send_typing(message.channel)
|
||
session = await loop.run_in_executor(executor, db.Session)
|
||
user = session.query(db.Discord).filter_by(discord_id=message.author.id).one_or_none()
|
||
if user is None:
|
||
user = db.Discord(discord_id=message.author.id,
|
||
name=message.author.name,
|
||
discriminator=message.author.discriminator,
|
||
avatar_hex=message.author.avatar)
|
||
await loop.run_in_executor(executor, session.commit)
|
||
else:
|
||
sentry.user_context({
|
||
"discord": {
|
||
"discord_id": message.author.id,
|
||
"name": message.author.name,
|
||
"discriminator": message.author.discriminator
|
||
},
|
||
"royal": {
|
||
"user_id": user.royal_id
|
||
}
|
||
})
|
||
if message.content.startswith("!link"):
|
||
if user.royal_id is None:
|
||
await client.send_message(message.channel,
|
||
"⚠️ Il tuo account Discord è già collegato a un account RYG "
|
||
"o l'account RYG che hai specificato è già collegato a un account Discord.")
|
||
return
|
||
try:
|
||
username = message.content.split(" ", 1)[1]
|
||
except IndexError:
|
||
await client.send_message(message.channel, "⚠️ Non hai specificato un username!\n"
|
||
"Sintassi corretta: `!link <username_ryg>`")
|
||
return
|
||
royal = session.query(db.Royal).filter_by(username=username).one_or_none()
|
||
if royal is None:
|
||
await client.send_message(message.channel,
|
||
"⚠️ Non esiste nessun account RYG con questo nome.")
|
||
return
|
||
user.royal_id = royal.id
|
||
session.commit()
|
||
session.close()
|
||
await client.send_message(message.channel, "✅ Sincronizzazione completata!")
|
||
elif message.content.startswith("!cv"):
|
||
await client.send_typing(message.channel)
|
||
if message.author.voice.voice_channel is None:
|
||
await client.send_message(message.channel, "⚠ Non sei in nessun canale!")
|
||
return
|
||
global voice_client
|
||
if voice_client is not None and voice_client.is_connected():
|
||
await voice_client.move_to(message.author.voice.voice_channel)
|
||
else:
|
||
voice_client = await client.join_voice_channel(message.author.voice.voice_channel)
|
||
await client.send_message(message.channel, f"✅ Mi sono connesso in <#{message.author.voice.voice_channel.id}>.")
|
||
elif message.content.startswith("!play"):
|
||
await client.send_typing(message.channel)
|
||
# The bot should be in voice chat
|
||
if voice_client is None:
|
||
await client.send_message(message.channel, "⚠️ Non sono connesso alla cv!\n"
|
||
"Fammi entrare scrivendo `!cv` mentre sei in chat vocale.")
|
||
return
|
||
# Find the sent url
|
||
try:
|
||
url = message.content.split(" ", 1)[1]
|
||
except IndexError:
|
||
await client.send_message(message.channel, "⚠️ Non hai specificato un url!\n"
|
||
"Sintassi corretta: `!play <video>`")
|
||
return
|
||
# Se è una playlist, informa che potrebbe essere richiesto un po' di tempo
|
||
if "playlist" in url:
|
||
await client.send_message(message.channel, f"⚠ Le playlist non sono ancora supportate dal bot.\n"
|
||
f"Prova mettendo i video singoli a mano!")
|
||
# If target is a single video
|
||
video = await Video.init(user_id=message.author.id, ytdl_url=url)
|
||
await client.send_message(message.channel, f"✅ Aggiunto alla coda: <{url}>")
|
||
voice_queue.append(video)
|
||
elif message.content.startswith("!search"):
|
||
await client.send_typing(message.channel)
|
||
# The bot should be in voice chat
|
||
if voice_client is None:
|
||
await client.send_message(message.channel, "⚠️ Non sono connesso alla cv!\n"
|
||
"Fammi entrare scrivendo `!cv` mentre sei in chat vocale.")
|
||
return
|
||
# Find the sent text
|
||
try:
|
||
text = message.content.split(" ", 1)[1]
|
||
except IndexError:
|
||
await client.send_message(message.channel, "⚠️ Non hai specificato il titolo!\n"
|
||
"Sintassi corretta: `!search <titolo>`")
|
||
return
|
||
# If target is a single video
|
||
video = await Video.init(user_id=message.author.id, ytdl_url=f"ytsearch:{text}")
|
||
await client.send_message(message.channel, f"✅ Aggiunto alla coda: `ytsearch:{text}`")
|
||
voice_queue.append(video)
|
||
elif message.content.startswith("!file"):
|
||
await client.send_typing(message.channel)
|
||
# The bot should be in voice chat
|
||
if voice_client is None:
|
||
await client.send_message(message.channel, "⚠️ Non sono connesso alla cv!\n"
|
||
"Fammi entrare scrivendo `!cv` mentre sei in chat vocale.")
|
||
return
|
||
# Find the sent text
|
||
try:
|
||
text: str = message.content.split(" ", 1)[1]
|
||
except IndexError:
|
||
await client.send_message(message.channel, "⚠️ Non hai specificato il nome del file!\n"
|
||
"Sintassi corretta: `!file <nomefile>`")
|
||
return
|
||
# If target is a single video
|
||
video = await Video.init(user_id=message.author.id, filename=text)
|
||
await client.send_message(message.channel, f"✅ Aggiunto alla coda: `{text}`")
|
||
voice_queue.append(video)
|
||
elif message.content.startswith("!skip"):
|
||
global voice_player
|
||
voice_player.stop()
|
||
await client.send_message(message.channel, f"⏩ Video saltato.")
|
||
elif message.content.startswith("!pause"):
|
||
if voice_player is None or not voice_player.is_playing():
|
||
await client.send_message(message.channel, f"⚠️ Non è in corso la riproduzione di un video, "
|
||
f"pertanto non c'è niente da pausare.")
|
||
return
|
||
voice_player.pause()
|
||
await client.send_message(message.channel, f"⏸ Riproduzione messa in pausa.\n"
|
||
f"Riprendi con `!resume`.")
|
||
elif message.content.startswith("!resume"):
|
||
if voice_player is None or voice_player.is_playing():
|
||
await client.send_message(message.channel, f"⚠️ Non c'è nulla in pausa da riprendere!")
|
||
return
|
||
voice_player.resume()
|
||
await client.send_message(message.channel, f"▶️ Riproduzione ripresa.")
|
||
elif message.content.startswith("!cancel"):
|
||
if not len(voice_queue) > 0:
|
||
await client.send_message(message.channel, f"⚠ Non ci sono video da annullare.")
|
||
return
|
||
voice_queue.pop()
|
||
await client.send_message(message.channel, f"❌ L'ultimo video aggiunto alla playlist è stato rimosso.")
|
||
elif message.content.startswith("!stop"):
|
||
if voice_player is None:
|
||
await client.send_message(message.channel, f"⚠ Non c'è nulla da interrompere!")
|
||
return
|
||
voice_queue = []
|
||
voice_player.stop()
|
||
voice_player = None
|
||
await client.send_message(message.channel, f"⏹ Riproduzione interrotta e playlist svuotata.")
|
||
elif message.content.startswith("!disconnect"):
|
||
if voice_client is None:
|
||
await client.send_message(message.channel, "⚠ Il bot in questo momento non è in chat vocale.")
|
||
return
|
||
if voice_player is not None:
|
||
await client.send_message(message.channel, "⚠ Prima di disconnettere il bot, interrompi la riproduzione di "
|
||
"una canzone scrivendo `!stop`.")
|
||
return
|
||
await voice_client.disconnect()
|
||
voice_client = None
|
||
await client.send_message(message.channel, "✅ Mi sono disconnesso dalla chat vocale.")
|
||
elif message.content.startswith("!queue"):
|
||
msg = "Video in coda:\n"
|
||
for position in range(10) if len(voice_queue) > 10 else range(len(voice_queue)):
|
||
msg += f"{queue_emojis[position]} {'`' + voice_queue[position].filename + '`' if voice_queue[position].filename is not None else '<' + voice_queue[position].ytdl_url + '>'}\n"
|
||
await client.send_message(message.channel, msg)
|
||
elif message.content.startswith("!cast"):
|
||
try:
|
||
spell = message.content.split(" ", 1)[1]
|
||
except IndexError:
|
||
await client.send_message(message.channel, "⚠️ Non hai specificato nessun incantesimo!\n"
|
||
"Sintassi corretta: `!cast <nome_incantesimo>`")
|
||
return
|
||
target = random.sample(list(message.server.members), 1)[0]
|
||
# Seed the rng with the spell name
|
||
# so that spells always deal the same damage
|
||
random.seed(spell)
|
||
dmg_mod = random.randrange(-2, 3)
|
||
dmg_dice = random.randrange(1, 4)
|
||
dmg_max = random.sample([4, 6, 8, 10, 12, 20, 100], 1)[0]
|
||
# Reseed the rng with a random value
|
||
# so that the dice roll always deals a different damage
|
||
random.seed()
|
||
total = dmg_mod
|
||
for dice in range(0, dmg_dice):
|
||
total += random.randrange(1, dmg_max+1)
|
||
await client.send_message(message.channel,
|
||
f"❇️ Ho lanciato **{spell}** "
|
||
f"su **{target.nick if target.nick is not None else target.name}** "
|
||
f"per {dmg_dice}d{dmg_max}"
|
||
f"{'+' if dmg_mod > 0 else ''}{str(dmg_mod) if dmg_mod != 0 else ''}"
|
||
f"=**{total if total > 0 else 0}** danni!")
|
||
elif message.content.startswith("!smecds"):
|
||
ds = random.sample(stagismo.listona, 1)[0]
|
||
await client.send_message(message.channel, f"Secondo me, è colpa {ds}.", tts=True)
|
||
elif __debug__ and message.content.startswith("!exception"):
|
||
raise Exception("!exception was called")
|
||
|
||
|
||
async def update_users_pipe(users_connection):
|
||
await client.wait_until_ready()
|
||
while True:
|
||
msg = await loop.run_in_executor(executor, users_connection.recv)
|
||
if msg == "/cv":
|
||
discord_members = list(client.get_server(config["Discord"]["server_id"]).members)
|
||
users_connection.send(discord_members)
|
||
|
||
|
||
async def update_music_queue():
|
||
await client.wait_until_ready()
|
||
global voice_client
|
||
global voice_player
|
||
global voice_queue
|
||
while True:
|
||
if voice_client is None:
|
||
await asyncio.sleep(5)
|
||
continue
|
||
if voice_player is not None and not voice_player.is_done():
|
||
await asyncio.sleep(1)
|
||
continue
|
||
if len(voice_queue) == 0:
|
||
await client.change_presence()
|
||
await asyncio.sleep(1)
|
||
continue
|
||
video = voice_queue.pop(0)
|
||
if video.ytdl_url:
|
||
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
|
||
f"⬇️ E' iniziato il download di `{video.ytdl_url}`.")
|
||
try:
|
||
async with async_timeout.timeout(30):
|
||
await video.download()
|
||
except asyncio.TimeoutError:
|
||
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
|
||
f"⚠️ Il download della canzone ha richiesto più di 30 secondi ed è stato "
|
||
f"annullato. ")
|
||
continue
|
||
except DurationError:
|
||
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
|
||
f"⚠️ Il file supera il limite di durata impostato in config.ini "
|
||
f"(`{config['YouTube']['max_duration']}` secondi).")
|
||
continue
|
||
except Exception as e:
|
||
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
|
||
f"⚠️ C'è stato un errore durante il download di `{video.ytdl_url}`:\n"
|
||
f"```\n"
|
||
f"{e}\n"
|
||
f"```")
|
||
continue
|
||
voice_player = voice_client.create_ffmpeg_player(f"opusfiles/{video.filename}.opus")
|
||
voice_player.start()
|
||
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
|
||
f"▶ Ora in riproduzione in <#{voice_client.channel.id}>:\n"
|
||
f"`{video.filename}`")
|
||
await client.change_presence(game=discord.Game(name=video.filename, type=2))
|
||
await video.add_to_db()
|
||
|
||
|
||
def process(users_connection=None):
|
||
print("Discordbot starting...")
|
||
if users_connection is not None:
|
||
asyncio.ensure_future(update_users_pipe(users_connection))
|
||
asyncio.ensure_future(update_music_queue())
|
||
client.on_error = on_error
|
||
client.run(config["Discord"]["bot_token"])
|
||
|
||
|
||
if __name__ == "__main__":
|
||
process()
|