mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 19:44:20 +00:00
217 lines
No EOL
10 KiB
Python
217 lines
No EOL
10 KiB
Python
import datetime
|
||
import discord
|
||
import discord.opus
|
||
import functools
|
||
import db
|
||
import errors
|
||
import youtube_dl
|
||
|
||
# Init the event loop
|
||
import asyncio
|
||
loop = asyncio.get_event_loop()
|
||
|
||
# Init the config reader
|
||
import configparser
|
||
config = configparser.ConfigParser()
|
||
config.read("config.ini")
|
||
|
||
# Init the discord bot
|
||
client = discord.Client()
|
||
discord.opus.load_opus("libopus-0.dll")
|
||
voice_client = None
|
||
voice_player = None
|
||
voice_queue = asyncio.Queue()
|
||
|
||
|
||
class Video:
|
||
def __init__(self):
|
||
self.user = None
|
||
self.info = None
|
||
self.timestamp = None
|
||
|
||
@staticmethod
|
||
async def init(author, info, timestamp):
|
||
self = Video()
|
||
discord_user = await find_user(author)
|
||
self.user = discord_user.royal
|
||
self.info = info
|
||
self.timestamp = timestamp
|
||
return self
|
||
|
||
def add_to_db(self):
|
||
db.CVMusic.create_and_add(title=self.info["title"],
|
||
user=self.user,
|
||
timestamp=self.timestamp)
|
||
|
||
def create_embed(self):
|
||
embed = discord.Embed(type="rich",
|
||
title=self.info['title'] if 'title' in self.info else None,
|
||
url=self.info['webpage_url'] if 'webpage_url' in self.info else None,
|
||
colour=discord.Colour(13375518))
|
||
# Uploader
|
||
if "uploader" in self.info and self.info["uploader"] is not None:
|
||
embed.set_author(name=self.info["uploader"],
|
||
url=self.info["uploader_url"] if "uploader_url" in self.info else None)
|
||
# Thumbnail
|
||
if "thumbnail" in self.info:
|
||
embed.set_thumbnail(url=self.info["thumbnail"])
|
||
# Duration
|
||
embed.add_field(name="Durata", value=str(datetime.timedelta(seconds=self.info["duration"])))
|
||
# Views
|
||
if "view_count" in self.info and self.info["view_count"] is not None:
|
||
embed.add_field(name="Visualizzazioni", value="{:_}".format(self.info["view_count"]).replace("_", " "))
|
||
# Likes
|
||
if "like_count" in self.info and self.info["like_count"] is not None:
|
||
embed.add_field(name="Mi piace", value="{:_}".format(self.info["like_count"]).replace("_", " "))
|
||
# Dislikes
|
||
if "dislike_count" in self.info and self.info["dislike_count"] is not None:
|
||
embed.add_field(name="Non mi piace", value="{:_}".format(self.info["dislike_count"]).replace("_", " "))
|
||
return embed
|
||
|
||
async def download(self):
|
||
with youtube_dl.YoutubeDL({"noplaylist": True,
|
||
"format": "bestaudio",
|
||
"postprocessors": [{
|
||
"key": 'FFmpegExtractAudio',
|
||
"preferredcodec": 'opus'
|
||
}],
|
||
"outtmpl": "music.%(ext)s"}) as ytdl:
|
||
info = await loop.run_in_executor(None, functools.partial(ytdl.extract_info, self.info["webpage_url"]))
|
||
|
||
|
||
async def find_user(user: discord.User):
|
||
session = await loop.run_in_executor(None, db.Session)
|
||
user = await loop.run_in_executor(None, session.query(db.Discord).filter_by(discord_id=user.id).join(db.Royal).first)
|
||
return user
|
||
|
||
@client.event
|
||
async def on_message(message: discord.Message):
|
||
if message.content.startswith("!register"):
|
||
await client.send_typing(message.channel)
|
||
session = await loop.run_in_executor(None, db.Session())
|
||
try:
|
||
username = message.content.split(" ", 1)[1]
|
||
except IndexError:
|
||
await client.send_message(message.channel, "⚠️ Non hai specificato un username!\n"
|
||
"Sintassi corretta: `!register <username_ryg>`")
|
||
return
|
||
try:
|
||
d = db.Discord.create(session,
|
||
royal_username=username,
|
||
discord_user=message.author)
|
||
except errors.AlreadyExistingError:
|
||
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
|
||
session.add(d)
|
||
session.commit()
|
||
session.close()
|
||
await client.send_message(message.channel, "✅ Sincronizzazione completata!")
|
||
elif message.content.startswith("!cv") and discord.opus.is_loaded():
|
||
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
|
||
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("!madd"):
|
||
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.")
|
||
# 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: `!madd <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"ℹ️ Hai inviato una playlist al bot.\n"
|
||
f"L'elaborazione potrebbe richiedere un po' di tempo.")
|
||
# Extract the info from the url
|
||
try:
|
||
with youtube_dl.YoutubeDL({"quiet": True, "skip_download": True, "noplaylist": True, "format": "webm[abr>0]/bestaudio/best"}) as ytdl:
|
||
info = await loop.run_in_executor(None, functools.partial(ytdl.extract_info, url))
|
||
except youtube_dl.utils.DownloadError as e:
|
||
if "is not a valid URL" in str(e):
|
||
await client.send_message(message.channel, f"⚠️ Il link inserito non è valido.\n"
|
||
f"Se vuoi cercare un video su YouTube, usa `!msearch <query>`")
|
||
return
|
||
if "_type" not in info:
|
||
# If target is a single video
|
||
video = await Video.init(author=message.author, info=info, timestamp=datetime.datetime.now())
|
||
await client.send_message(message.channel, f"✅ Aggiunto alla coda:", embed=video.create_embed())
|
||
await voice_queue.put(video)
|
||
elif info["_type"] == "playlist":
|
||
# If target is a playlist
|
||
if len(info["entries"]) < 20:
|
||
for single_info in info["entries"]:
|
||
video = await Video.init(author=message.author, info=single_info, timestamp=datetime.datetime.now())
|
||
await client.send_message(message.channel, f"✅ Aggiunto alla coda:", embed=video.create_embed())
|
||
await voice_queue.put(video)
|
||
else:
|
||
await client.send_message(message.channel, f"ℹ La playlist contiene {len(info['entries'])} video.\n"
|
||
f"Sei sicuro di volerli aggiungere alla coda?\n"
|
||
f"Rispondi **sì** o **no**.\n"
|
||
f"_(Il bot potrebbe crashare.)_")
|
||
answer = await client.wait_for_message(timeout=60, author=message.author, channel=message.channel)
|
||
if "sì" in answer.content.lower() or "si" in answer.content.lower():
|
||
for single_info in info["entries"]:
|
||
video = await Video.init(author=message.author, info=single_info,
|
||
timestamp=datetime.datetime.now())
|
||
await client.send_message(message.channel, f"✅ Aggiunto alla coda:", embed=video.create_embed())
|
||
await voice_queue.put(video)
|
||
elif "no" in answer.content.lower():
|
||
await client.send_message(message.channel, f"ℹ Operazione annullata.")
|
||
return
|
||
elif message.content.startswith("!msearch"):
|
||
raise NotImplementedError()
|
||
elif message.content.startswith("!mskip"):
|
||
raise NotImplementedError()
|
||
elif message.content.startswith("!mpause"):
|
||
raise NotImplementedError()
|
||
elif message.content.startswith("!mresume"):
|
||
raise NotImplementedError()
|
||
elif message.content.startswith("!mcancel"):
|
||
raise NotImplementedError()
|
||
elif message.content.startswith("!mstop"):
|
||
raise NotImplementedError()
|
||
|
||
|
||
async def update_users_pipe(users_connection):
|
||
await client.wait_until_ready()
|
||
while True:
|
||
msg = await loop.run_in_executor(None, 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()
|
||
while True:
|
||
global voice_player
|
||
# Wait until there is nothing playing
|
||
if voice_player is None or voice_player.is_playing():
|
||
if voice_queue.qsize() == 0:
|
||
await asyncio.sleep(1)
|
||
continue
|
||
# Get the last video in the queue
|
||
video = await voice_queue.get()
|
||
# Download the video
|
||
await video.download()
|
||
# Play the video
|
||
voice_player = voice_client.create_ffmpeg_player(f"music.opus")
|
||
voice_player.start()
|
||
# Add the video to the db
|
||
await loop.run_in_executor(None, video.add_to_db)
|
||
|
||
|
||
def process(users_connection):
|
||
print("Discordbot starting...")
|
||
loop.create_task(update_users_pipe(users_connection))
|
||
loop.create_task(update_music_queue())
|
||
client.run(config["Discord"]["bot_token"]) |