mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 21:44:21 +00:00
511 lines
20 KiB
Python
511 lines
20 KiB
Python
import random
|
||
import re
|
||
import discord
|
||
import discord.opus
|
||
import discord.voice_client
|
||
import functools
|
||
import sys
|
||
import db
|
||
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
|
||
import cast
|
||
import logging
|
||
|
||
logging.basicConfig()
|
||
|
||
# Queue emojis
|
||
queue_emojis = [":one:",
|
||
":two:",
|
||
":three:",
|
||
":four:",
|
||
":five:",
|
||
":six:",
|
||
":seven:",
|
||
":eight:",
|
||
":nine:",
|
||
":keycap_ten:"]
|
||
|
||
# Init the event loop
|
||
loop = asyncio.get_event_loop()
|
||
|
||
# Init the config reader
|
||
config = configparser.ConfigParser()
|
||
config.read("config.ini")
|
||
|
||
|
||
class DurationError(Exception):
|
||
pass
|
||
|
||
|
||
class InfoNotRetrievedError(Exception):
|
||
pass
|
||
|
||
|
||
class FileNotDownloadedError(Exception):
|
||
pass
|
||
|
||
|
||
class AlreadyDownloadedError(Exception):
|
||
pass
|
||
|
||
|
||
class Video:
|
||
def __init__(self, url: str=None, file: str=None, info: dict=None):
|
||
self.url = url
|
||
if file is None and info is None:
|
||
self.file = str(hash(url)) + ".opus"
|
||
elif info is not None:
|
||
self.file = re.sub(r'[/\\?*"<>|!:]', "_", info["title"]) + ".opus"
|
||
else:
|
||
self.file = file
|
||
self.downloaded = False if file is None else True
|
||
self.info = info
|
||
|
||
def __str__(self):
|
||
if self.info is None or "title" not in self.info:
|
||
return f"`{self.file}`"
|
||
return f"_{self.info['title']}_"
|
||
|
||
def plain_text(self):
|
||
if self.info is None or "title" not in self.info:
|
||
return self.file
|
||
return self.info['title']
|
||
|
||
async def download(self, progress_hooks: typing.List["function"]=None):
|
||
# File already downloaded
|
||
if self.downloaded:
|
||
raise AlreadyDownloadedError()
|
||
# No progress hooks
|
||
if progress_hooks is None:
|
||
progress_hooks = []
|
||
# Check if under max duration
|
||
if self.info is not None and self.info.get("duration", 0) > int(config["YouTube"]["max_duration"]):
|
||
raise DurationError()
|
||
# Download the file
|
||
with youtube_dl.YoutubeDL({"noplaylist": True,
|
||
"format": "best",
|
||
"postprocessors": [{
|
||
"key": 'FFmpegExtractAudio',
|
||
"preferredcodec": 'opus'
|
||
}],
|
||
"outtmpl": f"./opusfiles/{self.file}",
|
||
"progress_hooks": progress_hooks,
|
||
"quiet": True}) as ytdl:
|
||
await loop.run_in_executor(executor, functools.partial(ytdl.download, [self.url]))
|
||
self.downloaded = True
|
||
|
||
async def create_player(self) -> discord.voice_client.ProcessPlayer:
|
||
# Check if the file has been downloaded
|
||
if not self.downloaded:
|
||
raise FileNotDownloadedError()
|
||
global voice_client
|
||
return voice_client.create_ffmpeg_player(f"./opusfiles/{self.file}")
|
||
|
||
|
||
# noinspection PyUnreachableCode
|
||
if __debug__:
|
||
version = "Dev"
|
||
commit_msg = "_in sviluppo_"
|
||
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()
|
||
commit_msg = str(subprocess.check_output(["git", "log", "-1", "--pretty=%B"]), 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.dll")
|
||
|
||
voice_client = None
|
||
voice_player = None
|
||
now_playing = None
|
||
voice_queue = []
|
||
|
||
# Init the executor
|
||
executor = concurrent.futures.ThreadPoolExecutor(max_workers=3)
|
||
|
||
# Init the Sentry client
|
||
sentry = raven.Client(config["Sentry"]["token"],
|
||
release=version,
|
||
install_logging_hook=False,
|
||
hook_libraries=[])
|
||
|
||
|
||
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 avviato e pronto a ricevere comandi!\n"
|
||
f"Ultimo aggiornamento: `{version}: {commit_msg}`")
|
||
await client.change_presence(game=None, status=discord.Status.online)
|
||
|
||
|
||
@client.event
|
||
async def on_message(message: discord.Message):
|
||
global voice_client
|
||
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
|
||
}
|
||
})
|
||
if message.content.startswith("!ping"):
|
||
await cmd_ping(channel=message.channel,
|
||
author=message.author,
|
||
params=["/ping"])
|
||
elif message.content.startswith("!cv"):
|
||
await cmd_cv(channel=message.channel,
|
||
author=message.author,
|
||
params=message.content.split(" "))
|
||
elif message.content.startswith("!play"):
|
||
await cmd_play(channel=message.channel,
|
||
author=message.author,
|
||
params=message.content.split(" "))
|
||
elif message.content.startswith("!skip"):
|
||
await cmd_skip(channel=message.channel,
|
||
author=message.author,
|
||
params=message.content.split(" "))
|
||
elif message.content.startswith("!remove"):
|
||
await cmd_remove(channel=message.channel,
|
||
author=message.author,
|
||
params=message.content.split(" "))
|
||
elif message.content.startswith("!queue"):
|
||
await cmd_queue(channel=message.channel,
|
||
author=message.author,
|
||
params=message.content.split(" "))
|
||
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: discord.Member = random.sample(list(message.server.members), 1)[0]
|
||
await client.send_message(message.channel, cast.cast(spell_name=spell, target_name=target.name,
|
||
platform="discord"))
|
||
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)
|
||
|
||
|
||
def command(func):
|
||
"""Decorator. Runs the function as a Discord command."""
|
||
async def new_func(channel: discord.Channel, author: discord.Member, params: typing.List[str], *args, **kwargs):
|
||
sentry.user_context({
|
||
"discord_id": author.id,
|
||
"username": f"{author.name}#{author.discriminator}"
|
||
})
|
||
try:
|
||
result = await func(channel=channel, author=author, params=params, *args, **kwargs)
|
||
except Exception:
|
||
ei = sys.exc_info()
|
||
try:
|
||
await client.send_message(channel,
|
||
f"☢ **ERRORE DURANTE L'ESECUZIONE DEL COMANDO {params[0]}**\n"
|
||
f"Il comando è stato ignorato.\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"```")
|
||
except Exception:
|
||
pass
|
||
sentry.captureException(exc_info=ei)
|
||
else:
|
||
return result
|
||
return new_func
|
||
|
||
|
||
def requires_cv(func):
|
||
"Decorator. Ensures the voice client is connected before running the command."
|
||
async def new_func(channel: discord.Channel, author: discord.Member, params: typing.List[str], *args, **kwargs):
|
||
global voice_client
|
||
if voice_client is None or not voice_client.is_connected():
|
||
await client.send_message(channel,
|
||
"⚠️ Non sono connesso alla cv!\n"
|
||
"Fammi entrare scrivendo `!cv` mentre sei in chat vocale.")
|
||
return
|
||
return await func(channel=channel, author=author, params=params, *args, **kwargs)
|
||
return new_func
|
||
|
||
|
||
def requires_rygdb(func, optional=False):
|
||
async def new_func(channel: discord.Channel, author: discord.Member, params: typing.List[str], *args, **kwargs):
|
||
session = await loop.run_in_executor(executor, db.Session)
|
||
dbuser = await loop.run_in_executor(executor,
|
||
session.query(db.Discord)
|
||
.filter_by(discord_id=author.id)
|
||
.join(db.Royal)
|
||
.first)
|
||
if not optional and dbuser is None:
|
||
await client.send_message(channel,
|
||
"⚠️ Devi essere registrato su Royalnet per poter utilizzare questo comando.")
|
||
return
|
||
await loop.run_in_executor(executor, session.close)
|
||
return await func(channel=channel, author=author, params=params, dbuser=dbuser, *args, **kwargs)
|
||
return new_func
|
||
|
||
|
||
@command
|
||
async def cmd_ping(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
|
||
await client.send_message(channel, f"Pong!")
|
||
|
||
|
||
@command
|
||
async def cmd_cv(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
|
||
if author.voice.voice_channel is None:
|
||
await client.send_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(author.voice.voice_channel)
|
||
else:
|
||
voice_client = await client.join_voice_channel(author.voice.voice_channel)
|
||
await client.send_message(channel, f"✅ Mi sono connesso in <#{author.voice.voice_channel.id}>.")
|
||
|
||
|
||
async def add_video_from_url(url):
|
||
# Retrieve info
|
||
with youtube_dl.YoutubeDL({"quiet": True,
|
||
"ignoreerrors": True,
|
||
"simulate": True}) as ytdl:
|
||
info = await loop.run_in_executor(executor,
|
||
functools.partial(ytdl.extract_info, url=url, download=False))
|
||
if "entries" in info:
|
||
# This is a playlist
|
||
for entry in info["entries"]:
|
||
voice_queue.append(Video(url=entry["webpage_url"], info=entry))
|
||
return
|
||
# This is a single video
|
||
voice_queue.append(Video(url=url, info=info))
|
||
|
||
|
||
async def add_video_from_file(file):
|
||
voice_queue.append(Video(file=file))
|
||
|
||
|
||
@command
|
||
@requires_cv
|
||
async def cmd_play(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
|
||
if len(params) < 2:
|
||
await client.send_message(channel, "⚠ Non hai specificato una canzone da riprodurre!\n"
|
||
"Sintassi: `!play <url|ricercayoutube|nomefile>`")
|
||
return
|
||
# Parse the parameter as URL
|
||
url = re.match(r"(?:https?://|ytsearch[0-9]*:).*", " ".join(params[1:]).strip("<>"))
|
||
if url is not None:
|
||
# This is a url
|
||
await add_video_from_url(url.group(0))
|
||
await client.send_message(channel, f"✅ Video aggiunto alla coda.")
|
||
return
|
||
# Parse the parameter as file
|
||
file_path = os.path.join(os.path.join(os.path.curdir, "opusfiles"), " ".join(params[1:]))
|
||
if os.path.exists(file_path):
|
||
# This is a file
|
||
await add_video_from_file(file=file_path)
|
||
await client.send_message(channel, f"✅ Video aggiunto alla coda.")
|
||
return
|
||
file_path += ".opus"
|
||
if os.path.exists(file_path):
|
||
# This is a file
|
||
await add_video_from_file(file=file_path)
|
||
await client.send_message(channel, f"✅ Video aggiunto alla coda.")
|
||
return
|
||
# Search the parameter on youtube
|
||
search = params[1]
|
||
# This is a search
|
||
await add_video_from_url(url=f"ytsearch:{search}")
|
||
await client.send_message(channel, f"✅ Video aggiunto alla coda.")
|
||
|
||
|
||
@command
|
||
@requires_cv
|
||
async def cmd_skip(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
|
||
global voice_player
|
||
if voice_player is None:
|
||
await client.send_message(channel, "⚠ Non c'è nessun video in riproduzione.")
|
||
return
|
||
voice_player.stop()
|
||
await client.send_message(channel, f"⏩ Video saltato.")
|
||
|
||
|
||
@command
|
||
@requires_cv
|
||
async def cmd_remove(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
|
||
if len(voice_queue) == 0:
|
||
await client.send_message(channel, "⚠ Non c'è nessun video in coda.")
|
||
return
|
||
if len(params) < 2:
|
||
index = len(voice_queue) - 1
|
||
else:
|
||
try:
|
||
index = int(params[1]) - 1
|
||
except ValueError:
|
||
await client.send_message(channel, "⚠ Il numero inserito non è valido.\n"
|
||
"Sintassi: `!remove [numerovideo]`")
|
||
return
|
||
if abs(index) >= len(voice_queue):
|
||
await client.send_message(channel, "⚠ Il numero inserito non corrisponde a nessun video nella playlist.\n"
|
||
"Sintassi: `!remove [numerovideo]`")
|
||
return
|
||
video = voice_queue.pop(index)
|
||
await client.send_message(channel, f":regional_indicator_x: {str(video)} è stato rimosso dalla coda.")
|
||
|
||
|
||
@command
|
||
async def cmd_queue(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
|
||
if len(voice_queue) == 0:
|
||
await client.send_message(channel, "**Video in coda:**\n"
|
||
"nessuno")
|
||
return
|
||
msg = "**Video in coda:**\n"
|
||
for index, video in enumerate(voice_queue[:10]):
|
||
msg += f"{queue_emojis[index]} {str(video)}\n"
|
||
if len(voice_queue) > 10:
|
||
msg += f"più altri {len(voice_queue) - 10} video!"
|
||
await client.send_message(channel, msg)
|
||
|
||
|
||
@command
|
||
@requires_cv
|
||
async def cmd_shuffle(channel: discord.Channel, author: discord.Member, params: typing.List[str]):
|
||
if len(voice_queue) == 0:
|
||
await client.send_message(channel, "⚠ Non ci sono video in coda!")
|
||
return
|
||
random.shuffle(voice_queue)
|
||
await client.send_message("♠️ ♦️ ♣️ ♥️ Shuffle completo!")
|
||
|
||
|
||
async def queue_predownload_videos():
|
||
while True:
|
||
for index, video in enumerate(voice_queue[:int(config["YouTube"]["predownload_videos"])].copy()):
|
||
if video.downloaded:
|
||
continue
|
||
try:
|
||
with async_timeout.timeout(int(config["YouTube"]["download_timeout"])):
|
||
await video.download()
|
||
except asyncio.TimeoutError:
|
||
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
|
||
f"⚠️ Il download di {str(video)} ha richiesto più di"
|
||
f" {config['YouTube']['download_timeout']} secondi, pertanto è stato rimosso"
|
||
f" dalla coda.")
|
||
del voice_queue[index]
|
||
continue
|
||
except DurationError:
|
||
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
|
||
f"⚠️ {str(video)} dura più di"
|
||
f" {str(int(config['YouTube']['max_duration']) // 60)}"
|
||
f" minuti, quindi è stato rimosso dalla coda.")
|
||
del voice_queue[index]
|
||
continue
|
||
except Exception as e:
|
||
await client.send_message(client.get_channel(config["Discord"]["main_channel"]),
|
||
f"⚠️ E' stato incontrato un errore durante il download di {str(video)},"
|
||
f" quindi è stato rimosso dalla coda.\n\n"
|
||
f"```python\n"
|
||
f"{str(e)}"
|
||
f"```")
|
||
del voice_queue[index]
|
||
continue
|
||
await asyncio.sleep(1)
|
||
|
||
|
||
async def queue_play_next_video():
|
||
await client.wait_until_ready()
|
||
global voice_client
|
||
global voice_player
|
||
global now_playing
|
||
while True:
|
||
if voice_client is None:
|
||
await asyncio.sleep(1)
|
||
continue
|
||
if voice_player is not None and not voice_player.is_done():
|
||
await asyncio.sleep(0.5)
|
||
continue
|
||
if len(voice_queue) == 0:
|
||
await asyncio.sleep(0.5)
|
||
if now_playing is not None:
|
||
await client.change_presence()
|
||
now_playing = None
|
||
continue
|
||
now_playing = voice_queue[0]
|
||
if not now_playing.downloaded:
|
||
await asyncio.sleep(0.5)
|
||
continue
|
||
voice_player = await now_playing.create_player()
|
||
voice_player.start()
|
||
await client.change_presence(game=discord.Game(name=now_playing.plain_text(), type=2))
|
||
await client.send_message(client.get_channel(config["Discord"]["main_channel"]), f":arrow_forward: Ora in riproduzione: {str(now_playing)}")
|
||
del voice_queue[0]
|
||
|
||
|
||
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(queue_predownload_videos())
|
||
asyncio.ensure_future(queue_play_next_video())
|
||
client.on_error = on_error
|
||
loop.run_until_complete(client.login(config["Discord"]["bot_token"], bot=True))
|
||
loop.run_until_complete(client.connect())
|
||
loop.run_until_complete(client.logout())
|
||
|
||
|
||
if __name__ == "__main__":
|
||
process()
|