2017-11-11 17:55:13 +00:00
|
|
|
|
import random
|
2018-02-26 15:38:47 +00:00
|
|
|
|
import re
|
2017-10-27 11:38:32 +00:00
|
|
|
|
import discord
|
2017-10-31 22:50:05 +00:00
|
|
|
|
import discord.opus
|
2018-02-25 18:50:57 +00:00
|
|
|
|
import discord.voice_client
|
2017-11-01 18:23:11 +00:00
|
|
|
|
import functools
|
2017-11-09 19:34:18 +00:00
|
|
|
|
import sys
|
2017-10-27 11:38:32 +00:00
|
|
|
|
import db
|
2017-11-05 21:50:37 +00:00
|
|
|
|
import youtube_dl
|
2018-01-15 12:42:40 +00:00
|
|
|
|
import concurrent.futures
|
2018-02-25 18:50:57 +00:00
|
|
|
|
import typing
|
2018-02-25 19:47:12 +00:00
|
|
|
|
import os
|
2018-02-26 09:48:17 +00:00
|
|
|
|
import asyncio
|
|
|
|
|
import configparser
|
2018-04-09 20:55:48 +00:00
|
|
|
|
import async_timeout
|
2018-04-12 16:04:13 +00:00
|
|
|
|
import raven
|
2018-05-31 19:43:43 +00:00
|
|
|
|
import logging
|
2018-07-25 19:10:43 +00:00
|
|
|
|
import errors
|
2018-07-26 17:26:03 +00:00
|
|
|
|
import datetime
|
2018-08-08 15:48:57 +00:00
|
|
|
|
import sqlalchemy.exc
|
2018-09-17 22:26:01 +00:00
|
|
|
|
import coloredlogs
|
2018-05-31 19:43:43 +00:00
|
|
|
|
|
2018-09-18 22:02:39 +00:00
|
|
|
|
logging.getLogger().disabled = True
|
2018-08-16 17:46:07 +00:00
|
|
|
|
logger = logging.getLogger(__name__)
|
2018-09-18 22:13:41 +00:00
|
|
|
|
os.environ["COLOREDLOGS_LOG_FORMAT"] = "%(asctime)s %(levelname)s %(name)s %(message)s"
|
2018-09-17 22:26:01 +00:00
|
|
|
|
coloredlogs.install(level="DEBUG", logger=logger)
|
2017-10-27 11:38:32 +00:00
|
|
|
|
|
2018-02-26 15:38:47 +00:00
|
|
|
|
# Queue emojis
|
2018-05-25 17:58:30 +00:00
|
|
|
|
queue_emojis = [":one:",
|
|
|
|
|
":two:",
|
|
|
|
|
":three:",
|
|
|
|
|
":four:",
|
|
|
|
|
":five:",
|
|
|
|
|
":six:",
|
|
|
|
|
":seven:",
|
|
|
|
|
":eight:",
|
|
|
|
|
":nine:",
|
|
|
|
|
":keycap_ten:"]
|
2018-02-26 15:38:47 +00:00
|
|
|
|
|
2018-10-10 17:27:24 +00:00
|
|
|
|
# Zalgo text
|
2018-10-10 19:37:18 +00:00
|
|
|
|
zalgo_up = ['̍', '̎', '̄', '̅', '̿', '̑', '̆', '̐', '͒', '͗', '͑', '̇', '̈', '̊', '͂', '̓', '̈́', '͊',
|
|
|
|
|
'͋', '͌', '̃', '̂', '̌', '͐', '́', '̋', '̏', '̽', '̉', 'ͣ', 'ͤ', 'ͥ', 'ͦ', 'ͧ', 'ͨ', 'ͩ',
|
|
|
|
|
'ͪ', 'ͫ', 'ͬ', 'ͭ', 'ͮ', 'ͯ', '̾', '͛', '͆', '̚', ]
|
|
|
|
|
zalgo_down = ['̖', '̗', '̘', '̙', '̜', '̝', '̞', '̟', '̠', '̤', '̥', '̦', '̩', '̪', '̫', '̬', '̭', '̮',
|
|
|
|
|
'̯', '̰', '̱', '̲', '̳', '̹', '̺', '̻', '̼', 'ͅ', '͇', '͈', '͉', '͍', '͎', '͓', '͔', '͕',
|
|
|
|
|
'͖', '͙', '͚', '', ]
|
|
|
|
|
zalgo_middle = ['̕', '̛', '̀', '́', '͘', '̡', '̢', '̧', '̨', '̴', '̵', '̶', '͜', '͝', '͞', '͟', '͠', '͢',
|
|
|
|
|
'̸', '̷', '͡', ]
|
2018-10-10 17:27:24 +00:00
|
|
|
|
|
2017-10-27 11:38:32 +00:00
|
|
|
|
# Init the event loop
|
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
|
|
|
|
|
|
# Init the config reader
|
|
|
|
|
config = configparser.ConfigParser()
|
|
|
|
|
config.read("config.ini")
|
|
|
|
|
|
2018-08-05 21:24:41 +00:00
|
|
|
|
# Radio messages
|
2018-10-10 17:48:05 +00:00
|
|
|
|
radio_messages = ["https://www.youtube.com/watch?v=3-yeK1Ck4yk",
|
|
|
|
|
"https://youtu.be/YcR7du_A1Vc",
|
|
|
|
|
"https://clyp.it/byg3i52l"]
|
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
song_special_messages = {
|
|
|
|
|
"despacito": ":arrow_forward: this is so sad. alexa play {song}",
|
|
|
|
|
"faded": ":arrow_forward: Basta Garf, lasciami ascoltare {song}",
|
|
|
|
|
"ligma": ":arrow_forward: What is ligma? {song}!",
|
|
|
|
|
"sugma": ":arrow_forward: What is sugma? {song}!",
|
|
|
|
|
"sugondese": ":arrow_forward: What is sugondese? {song}!",
|
|
|
|
|
"bofa": ":arrow_forward: What is bofa? {song}!",
|
2018-08-16 17:33:38 +00:00
|
|
|
|
"updog": ":arrow_forward: What's up, dog? {song}!",
|
2018-08-08 12:09:57 +00:00
|
|
|
|
"sayo-nara": ":arrow_forward: I gently open the door. {song} awaits me inside.",
|
|
|
|
|
"monika": ":arrow_forward: Just Monika. Just Monika. Just {song}.",
|
2018-11-06 22:11:35 +00:00
|
|
|
|
"country road": ":arrow_forward: Take me home, to {song}, the place I belong!",
|
2018-08-08 12:09:57 +00:00
|
|
|
|
"never gonna give you up": ":arrow_forward: Rickrolling in 2018. Enjoy {song}!",
|
|
|
|
|
"september": ":arrow_forward: Do you remember? {song}.",
|
|
|
|
|
"homestuck": ":arrow_forward: > Enter song name. {song}",
|
|
|
|
|
"undertale": ":arrow_forward: Howdy! I'm Flowey! Listen to this friendly song: {song}",
|
|
|
|
|
"pumped up kicks": ":arrow_forward: Non metterti mica in testa strane idee ascoltando {song}...",
|
|
|
|
|
"jesus": ":arrow_forward: Respawn in 3 giorni. Intanto, ascolta {song}.",
|
|
|
|
|
"through The fire And flames": ":arrow_forward: Fai {song} su osu!, se ne sei capace!",
|
|
|
|
|
"slow clap": ":arrow_forward: :clap: :clap: :clap: {song} :clap: :clap: :clap:",
|
|
|
|
|
"pub scrubs": ":arrow_forward: MAV COME BACK WE MISS {song}!",
|
|
|
|
|
"alleluia": ":arrow_forward: Wah. Waaaah. Waluigi tiime: {song}",
|
|
|
|
|
"wah": ":arrow_forward: Wah. Waaaah. Waluigi tiime: {song}",
|
|
|
|
|
"waluigi": ":arrow_forward: Wah. Waaaah. Waluigi tiime: {song}",
|
|
|
|
|
"nyan cat": ":arrow_forward: Meow! :3 {song}",
|
|
|
|
|
"dragonborn": ":arrow_forward: FUS RO {song}!",
|
|
|
|
|
"dovahkiin": ":arrow_forward: FUS RO {song}!",
|
|
|
|
|
"initial d": ":arrow_forward: Guarda mamma sto driftando sul balcone di Balu grazie a {song}!",
|
|
|
|
|
"persona": ":arrow_forward: You'll never see {song} comiiiiing!",
|
|
|
|
|
"flamingo": ":arrow_forward: How many {song} do you have to eat?",
|
|
|
|
|
"linkin park": ":arrow_forward: Crawling in my {song}!",
|
2018-11-06 22:11:35 +00:00
|
|
|
|
"magicite": "⚠️ Warning: {song} contiene numerosi bug. E' ora in riproduzione.",
|
2018-08-08 12:09:57 +00:00
|
|
|
|
"papers please": ":arrow_forward: Glory to Arstotzka! {song}!",
|
|
|
|
|
"we are number one": ":arrow_forward: Now paying respect to Robbie Rotten: {song}",
|
2018-11-06 22:11:35 +00:00
|
|
|
|
"jump up superstar": ":arrow_forward: Is {song} the Tengen Toppa Guren Lagann opening?",
|
|
|
|
|
"the world revolving": ":arrow_forward: CHAOS! CHAOS! I CAN DO {song}!",
|
|
|
|
|
"deltarune": ":arrow_forward: You hug Ralsei. A strange music starts playing: {song}",
|
|
|
|
|
"song of unhealing": ":arrow_forward: BEN {song}",
|
|
|
|
|
"police academy": ":arrow_forward: {song} - freedom.png",
|
|
|
|
|
"super smash bros. ultimate": ":arrow_forward: Re-awaken the undying light with {song}!"
|
2018-08-08 12:09:57 +00:00
|
|
|
|
}
|
2018-08-05 21:24:41 +00:00
|
|
|
|
|
2018-08-07 23:30:43 +00:00
|
|
|
|
# FFmpeg settings
|
|
|
|
|
ffmpeg_settings = {}
|
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
# Init the executor
|
|
|
|
|
executor = concurrent.futures.ThreadPoolExecutor(max_workers=3)
|
|
|
|
|
|
|
|
|
|
# Init the Sentry client
|
|
|
|
|
sentry = raven.Client(config["Sentry"]["token"],
|
2018-09-13 16:38:21 +00:00
|
|
|
|
release=raven.fetch_git_sha(os.path.dirname(__file__)),
|
|
|
|
|
install_logging_hook=False,
|
2018-08-08 12:09:57 +00:00
|
|
|
|
hook_libraries=[])
|
|
|
|
|
|
2018-05-27 13:23:33 +00:00
|
|
|
|
|
2018-02-25 19:47:12 +00:00
|
|
|
|
class DurationError(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
2018-05-27 13:23:33 +00:00
|
|
|
|
|
|
|
|
|
class InfoNotRetrievedError(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class FileNotDownloadedError(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class AlreadyDownloadedError(Exception):
|
2018-05-26 08:50:54 +00:00
|
|
|
|
pass
|
2018-02-26 09:48:17 +00:00
|
|
|
|
|
2018-05-26 08:50:54 +00:00
|
|
|
|
|
2018-08-07 23:30:43 +00:00
|
|
|
|
class InvalidConfigError(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
2018-05-26 08:50:54 +00:00
|
|
|
|
class Video:
|
2018-08-08 12:09:57 +00:00
|
|
|
|
def __init__(self, url: str = None, file: str = None, info: dict = None, enqueuer: discord.Member = None):
|
2018-05-26 08:50:54 +00:00
|
|
|
|
self.url = url
|
2018-05-27 13:23:33 +00:00
|
|
|
|
if file is None and info is None:
|
|
|
|
|
self.file = str(hash(url)) + ".opus"
|
|
|
|
|
elif info is not None:
|
2018-11-08 22:39:32 +00:00
|
|
|
|
self.file = "./opusfiles/" + re.sub(r'[/\\?*"<>|!:]', "_", info["title"]) + ".opus"
|
2018-05-27 13:23:33 +00:00
|
|
|
|
else:
|
|
|
|
|
self.file = file
|
|
|
|
|
self.downloaded = False if file is None else True
|
|
|
|
|
self.info = info
|
2018-07-25 22:08:53 +00:00
|
|
|
|
self.enqueuer = enqueuer
|
2018-08-08 12:09:57 +00:00
|
|
|
|
self.duration = None
|
2018-05-26 08:50:54 +00:00
|
|
|
|
|
|
|
|
|
def __str__(self):
|
2018-05-27 13:23:33 +00:00
|
|
|
|
if self.info is None or "title" not in self.info:
|
|
|
|
|
return f"`{self.file}`"
|
|
|
|
|
return f"_{self.info['title']}_"
|
|
|
|
|
|
2018-09-17 22:19:35 +00:00
|
|
|
|
def __repr__(self):
|
|
|
|
|
return f"<discordbot.Video {str(self)}>"
|
|
|
|
|
|
2018-05-29 21:50:47 +00:00
|
|
|
|
def plain_text(self):
|
|
|
|
|
if self.info is None or "title" not in self.info:
|
|
|
|
|
return self.file
|
|
|
|
|
return self.info['title']
|
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
async def download(self, progress_hooks: typing.List["function"] = None):
|
2018-05-27 13:23:33 +00:00
|
|
|
|
# File already downloaded
|
|
|
|
|
if self.downloaded:
|
|
|
|
|
raise AlreadyDownloadedError()
|
|
|
|
|
# No progress hooks
|
|
|
|
|
if progress_hooks is None:
|
|
|
|
|
progress_hooks = []
|
|
|
|
|
# Check if under max duration
|
2018-08-08 12:09:57 +00:00
|
|
|
|
self.duration = datetime.timedelta(seconds=self.info.get("duration", 0))
|
|
|
|
|
if self.info is not None and self.duration.total_seconds() > int(config["YouTube"]["max_duration"]):
|
2018-05-27 13:23:33 +00:00
|
|
|
|
raise DurationError()
|
|
|
|
|
# Download the file
|
2018-09-17 22:19:35 +00:00
|
|
|
|
logger.info(f"Now downloading {repr(self)}.")
|
2018-05-27 13:23:33 +00:00
|
|
|
|
with youtube_dl.YoutubeDL({"noplaylist": True,
|
|
|
|
|
"format": "best",
|
|
|
|
|
"postprocessors": [{
|
|
|
|
|
"key": 'FFmpegExtractAudio',
|
2018-08-18 15:36:44 +00:00
|
|
|
|
"preferredcodec": 'opus'
|
2018-05-27 13:23:33 +00:00
|
|
|
|
}],
|
2018-11-09 22:11:21 +00:00
|
|
|
|
"outtmpl": self.file,
|
2018-05-27 13:23:33 +00:00
|
|
|
|
"progress_hooks": progress_hooks,
|
|
|
|
|
"quiet": True}) as ytdl:
|
|
|
|
|
await loop.run_in_executor(executor, functools.partial(ytdl.download, [self.url]))
|
2018-09-17 22:19:35 +00:00
|
|
|
|
logger.info(f"Download of {repr(self)} complete.")
|
|
|
|
|
self.downloaded = True
|
2018-05-27 13:23:33 +00:00
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
def create_audio_source(self) -> discord.PCMVolumeTransformer:
|
2018-05-27 13:23:33 +00:00
|
|
|
|
# Check if the file has been downloaded
|
|
|
|
|
if not self.downloaded:
|
|
|
|
|
raise FileNotDownloadedError()
|
2018-11-08 22:39:32 +00:00
|
|
|
|
return discord.PCMVolumeTransformer(discord.FFmpegPCMAudio(f"{self.file}", **ffmpeg_settings))
|
2018-08-07 23:30:43 +00:00
|
|
|
|
|
|
|
|
|
|
2018-10-10 17:27:24 +00:00
|
|
|
|
class SecretVideo(Video):
|
|
|
|
|
|
|
|
|
|
def __str__(self):
|
|
|
|
|
final_string = ""
|
|
|
|
|
for letter in self.file:
|
|
|
|
|
final_string += random.sample(zalgo_up, 1)[0]
|
|
|
|
|
final_string += random.sample(zalgo_middle, 1)[0]
|
|
|
|
|
final_string += random.sample(zalgo_down, 1)[0]
|
|
|
|
|
final_string += letter
|
|
|
|
|
return final_string
|
|
|
|
|
|
|
|
|
|
def plain_text(self):
|
|
|
|
|
final_string = ""
|
|
|
|
|
for letter in self.file:
|
|
|
|
|
final_string += random.sample(zalgo_up, 1)[0]
|
|
|
|
|
final_string += random.sample(zalgo_middle, 1)[0]
|
|
|
|
|
final_string += random.sample(zalgo_down, 1)[0]
|
|
|
|
|
final_string += letter
|
|
|
|
|
return final_string
|
|
|
|
|
|
|
|
|
|
def create_audio_source(self) -> discord.PCMVolumeTransformer:
|
|
|
|
|
# Check if the file has been downloaded
|
|
|
|
|
if not self.downloaded:
|
|
|
|
|
raise FileNotDownloadedError()
|
2018-11-03 17:42:56 +00:00
|
|
|
|
return discord.PCMVolumeTransformer(discord.FFmpegPCMAudio(f"./opusfiles/{self.file}", **ffmpeg_settings))
|
2018-10-10 17:27:24 +00:00
|
|
|
|
|
|
|
|
|
|
2018-11-07 19:50:55 +00:00
|
|
|
|
def escape(message: str):
|
|
|
|
|
return message.replace("<", "<").replace(">", ">")
|
|
|
|
|
|
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
def command(func):
|
|
|
|
|
"""Decorator. Runs the function as a Discord command."""
|
|
|
|
|
|
|
|
|
|
async def new_func(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str], *args,
|
|
|
|
|
**kwargs):
|
|
|
|
|
if author is not None:
|
|
|
|
|
sentry.user_context({
|
|
|
|
|
"discord_id": author.id,
|
|
|
|
|
"username": f"{author.name}#{author.discriminator}"
|
|
|
|
|
})
|
|
|
|
|
else:
|
|
|
|
|
sentry.user_context({
|
|
|
|
|
"source": "Telegram"
|
|
|
|
|
})
|
|
|
|
|
try:
|
|
|
|
|
result = await func(self, channel=channel, author=author, params=params, *args, **kwargs)
|
|
|
|
|
except Exception:
|
|
|
|
|
ei = sys.exc_info()
|
2018-09-17 22:07:00 +00:00
|
|
|
|
# noinspection PyBroadException
|
2018-08-08 12:09:57 +00:00
|
|
|
|
try:
|
|
|
|
|
await channel.send(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_connected_voice_client(func):
|
|
|
|
|
"""Decorator. Ensures the voice client is connected before running the command."""
|
|
|
|
|
|
|
|
|
|
async def new_func(self: "RoyalDiscordBot", channel: discord.TextChannel, author: discord.Member,
|
|
|
|
|
params: typing.List[str], *args, **kwargs):
|
|
|
|
|
for voice_client in self.voice_clients:
|
|
|
|
|
if voice_client.channel in self.main_guild.channels and voice_client.is_connected():
|
|
|
|
|
break
|
|
|
|
|
else:
|
|
|
|
|
await channel.send("⚠️ Non sono connesso alla cv!\n"
|
|
|
|
|
"Fammi entrare scrivendo `!cv` mentre sei in chat vocale.")
|
|
|
|
|
return
|
|
|
|
|
return await func(self, channel=channel, author=author, params=params, *args, **kwargs)
|
|
|
|
|
|
|
|
|
|
return new_func
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def requires_rygdb(func, optional=False):
|
|
|
|
|
async def new_func(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str], *args,
|
|
|
|
|
**kwargs):
|
|
|
|
|
session = db.Session()
|
|
|
|
|
dbuser = await loop.run_in_executor(executor,
|
|
|
|
|
session.query(db.Discord)
|
|
|
|
|
.filter_by(discord_id=author.id)
|
|
|
|
|
.join(db.Royal)
|
|
|
|
|
.first)
|
|
|
|
|
await loop.run_in_executor(executor, session.close)
|
|
|
|
|
if not optional and dbuser is None:
|
|
|
|
|
await channel.send("⚠️ Devi essere registrato su Royalnet per poter utilizzare questo comando.")
|
|
|
|
|
return
|
|
|
|
|
return await func(self, channel=channel, author=author, params=params, dbuser=dbuser, *args, **kwargs)
|
|
|
|
|
|
|
|
|
|
return new_func
|
|
|
|
|
|
|
|
|
|
|
2018-08-07 23:30:43 +00:00
|
|
|
|
class RoyalDiscordBot(discord.Client):
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
|
|
super().__init__(*args, **kwargs)
|
|
|
|
|
self.main_channel: typing.Optional[discord.TextChannel] = None
|
|
|
|
|
self.main_guild: typing.Optional[discord.Guild] = None
|
2018-08-08 12:09:57 +00:00
|
|
|
|
self.commands = {
|
|
|
|
|
"!ping": self.cmd_ping,
|
|
|
|
|
"!cv": self.cmd_cv,
|
|
|
|
|
"!summon": self.cmd_cv,
|
|
|
|
|
"!play": self.cmd_play,
|
|
|
|
|
"!p": self.cmd_play,
|
|
|
|
|
"!search": self.cmd_play,
|
|
|
|
|
"!file": self.cmd_play,
|
|
|
|
|
"!skip": self.cmd_skip,
|
|
|
|
|
"!s": self.cmd_skip,
|
2018-11-06 22:11:35 +00:00
|
|
|
|
"!next": self.cmd_skip,
|
2018-08-08 12:09:57 +00:00
|
|
|
|
"!remove": self.cmd_remove,
|
2018-10-02 22:18:41 +00:00
|
|
|
|
"!r": self.cmd_remove,
|
2018-08-08 12:09:57 +00:00
|
|
|
|
"!cancel": self.cmd_remove,
|
|
|
|
|
"!queue": self.cmd_queue,
|
|
|
|
|
"!q": self.cmd_queue,
|
|
|
|
|
"!shuffle": self.cmd_shuffle,
|
|
|
|
|
"!clear": self.cmd_clear,
|
|
|
|
|
"!register": self.cmd_register,
|
|
|
|
|
"!forceplay": self.cmd_forceplay,
|
|
|
|
|
"!fp": self.cmd_forceplay,
|
|
|
|
|
"!radiomessages": self.cmd_radiomessages,
|
|
|
|
|
"!yes": self.null,
|
2018-08-17 13:09:03 +00:00
|
|
|
|
"!no": self.null,
|
|
|
|
|
"!pause": self.cmd_pause,
|
2018-11-08 22:41:41 +00:00
|
|
|
|
"!resume": self.cmd_resume,
|
|
|
|
|
"!m": self.cmd_m
|
2018-08-08 12:09:57 +00:00
|
|
|
|
}
|
|
|
|
|
self.video_queue: typing.List[Video] = []
|
|
|
|
|
self.now_playing = None
|
2018-09-08 11:03:30 +00:00
|
|
|
|
self.radio_messages = True
|
2018-08-08 12:09:57 +00:00
|
|
|
|
self.next_radio_message_in = int(config["Discord"]["radio_messages_every"])
|
2018-09-14 23:05:19 +00:00
|
|
|
|
self.inactivity_timer = 0
|
2018-08-08 12:09:57 +00:00
|
|
|
|
asyncio.ensure_future(self.queue_predownload_videos())
|
|
|
|
|
asyncio.ensure_future(self.queue_play_next_video())
|
2018-09-17 22:07:00 +00:00
|
|
|
|
asyncio.ensure_future(self.inactivity_countdown())
|
2018-11-18 16:38:02 +00:00
|
|
|
|
asyncio.ensure_future(self.activity_task())
|
2018-08-07 23:30:43 +00:00
|
|
|
|
|
|
|
|
|
async def on_ready(self):
|
|
|
|
|
# Get the main channel
|
|
|
|
|
self.main_channel = self.get_channel(int(config["Discord"]["main_channel"]))
|
|
|
|
|
if not isinstance(self.main_channel, discord.TextChannel):
|
|
|
|
|
raise InvalidConfigError("The main channel is not a TextChannel!")
|
|
|
|
|
# Get the main guild
|
|
|
|
|
self.main_guild = self.get_guild(int(config["Discord"]["server_id"]))
|
|
|
|
|
if not isinstance(self.main_guild, discord.Guild):
|
|
|
|
|
raise InvalidConfigError("The main guild does not exist!")
|
|
|
|
|
await self.change_presence(status=discord.Status.online, activity=None)
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.info("Bot is ready!")
|
2018-08-07 23:30:43 +00:00
|
|
|
|
|
|
|
|
|
async def on_message(self, message: discord.Message):
|
|
|
|
|
if message.channel != self.main_channel or message.author.bot:
|
|
|
|
|
return
|
|
|
|
|
sentry.user_context({
|
|
|
|
|
"discord": {
|
|
|
|
|
"discord_id": message.author.id,
|
|
|
|
|
"name": message.author.name,
|
|
|
|
|
"discriminator": message.author.discriminator
|
|
|
|
|
}
|
|
|
|
|
})
|
|
|
|
|
if not message.content.startswith("!"):
|
|
|
|
|
await message.channel.send(f":warning: In questa chat sono consentiti solo comandi per il bot.\n"
|
|
|
|
|
f"Riinvia il tuo messaggio in un altro canale!")
|
|
|
|
|
await message.delete()
|
|
|
|
|
return
|
|
|
|
|
data = message.content.split(" ")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
if data[0] not in self.commands:
|
2018-08-07 23:30:43 +00:00
|
|
|
|
await message.channel.send(":warning: Comando non riconosciuto.")
|
|
|
|
|
return
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Received command: {message.content}")
|
2018-09-05 17:48:34 +00:00
|
|
|
|
sentry.extra_context({
|
|
|
|
|
"command": data[0],
|
|
|
|
|
"message": message
|
|
|
|
|
})
|
2018-09-14 23:05:19 +00:00
|
|
|
|
self.inactivity_timer = 3600
|
2018-08-08 12:09:57 +00:00
|
|
|
|
await self.commands[data[0]](channel=message.channel,
|
|
|
|
|
author=message.author,
|
|
|
|
|
params=data)
|
2017-11-06 15:54:36 +00:00
|
|
|
|
|
2018-08-07 23:30:43 +00:00
|
|
|
|
async def on_error(self, event_method, *args, **kwargs):
|
|
|
|
|
ei = sys.exc_info()
|
2018-08-16 17:46:07 +00:00
|
|
|
|
logger.error(f"Critical error: {repr(ei[1])}")
|
2018-09-13 16:38:21 +00:00
|
|
|
|
# noinspection PyBroadException
|
2018-08-07 23:30:43 +00:00
|
|
|
|
try:
|
2018-09-14 23:05:19 +00:00
|
|
|
|
await self.main_channel.send(f"☢️ **ERRORE CRITICO NELL'EVENTO** `{event_method}`!\n"
|
2018-08-07 23:30:43 +00:00
|
|
|
|
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"```")
|
|
|
|
|
await self.change_presence(status=discord.Status.invisible)
|
|
|
|
|
await self.close()
|
2018-09-13 16:38:21 +00:00
|
|
|
|
except Exception:
|
|
|
|
|
logger.error(f"Double critical error: {sys.exc_info()}")
|
2018-08-07 23:30:43 +00:00
|
|
|
|
loop.stop()
|
|
|
|
|
sentry.captureException(exc_info=ei)
|
|
|
|
|
exit(1)
|
|
|
|
|
|
|
|
|
|
async def feed_pipe(self, connection):
|
|
|
|
|
await self.wait_until_ready()
|
|
|
|
|
while True:
|
|
|
|
|
msg = await loop.run_in_executor(executor, connection.recv)
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Received from the Telegram-Discord pipe: {msg}")
|
2018-08-07 23:30:43 +00:00
|
|
|
|
if msg == "get cv":
|
|
|
|
|
discord_members = list(self.main_guild.members)
|
2018-08-28 13:48:12 +00:00
|
|
|
|
channels = {0: None}
|
|
|
|
|
members_in_channels = {0: []}
|
|
|
|
|
message = ""
|
|
|
|
|
# Find all the channels
|
|
|
|
|
for member in discord_members:
|
2018-08-28 13:49:29 +00:00
|
|
|
|
if member.voice is not None:
|
2018-08-28 13:48:12 +00:00
|
|
|
|
channel = members_in_channels.get(member.voice.channel.id)
|
|
|
|
|
if channel is None:
|
|
|
|
|
members_in_channels[member.voice.channel.id] = list()
|
|
|
|
|
channel = members_in_channels[member.voice.channel.id]
|
|
|
|
|
channels[member.voice.channel.id] = member.voice.channel
|
|
|
|
|
channel.append(member)
|
|
|
|
|
else:
|
|
|
|
|
members_in_channels[0].append(member)
|
|
|
|
|
# Edit the message, sorted by channel
|
2018-09-18 23:04:48 +00:00
|
|
|
|
for channel in sorted(channels, key=lambda c: -c):
|
2018-08-28 13:48:12 +00:00
|
|
|
|
members_in_channels[channel].sort(key=lambda x: x.nick if x.nick is not None else x.name)
|
|
|
|
|
if channel == 0:
|
2018-09-18 23:04:48 +00:00
|
|
|
|
message += "<b>Non in chat vocale:</b>\n"
|
2018-08-28 13:48:12 +00:00
|
|
|
|
else:
|
2018-11-07 19:50:55 +00:00
|
|
|
|
message += f"<b>In #{escape(channels[channel].name)}:</b>\n"
|
2018-08-28 13:48:12 +00:00
|
|
|
|
for member in members_in_channels[channel]:
|
2018-09-18 23:04:48 +00:00
|
|
|
|
# Ignore not-connected non-notable members
|
|
|
|
|
if channel == 0 and len(member.roles) < 2:
|
|
|
|
|
continue
|
|
|
|
|
# Ignore offline members
|
2018-08-28 13:51:06 +00:00
|
|
|
|
if member.status == discord.Status.offline and member.voice is None:
|
2018-08-28 13:48:12 +00:00
|
|
|
|
continue
|
|
|
|
|
# Online status emoji
|
|
|
|
|
if member.bot:
|
|
|
|
|
message += "🤖 "
|
|
|
|
|
elif member.status == discord.Status.online:
|
|
|
|
|
message += "🔵 "
|
|
|
|
|
elif member.status == discord.Status.idle:
|
|
|
|
|
message += "⚫️ "
|
|
|
|
|
elif member.status == discord.Status.dnd:
|
|
|
|
|
message += "🔴 "
|
|
|
|
|
elif member.status == discord.Status.offline:
|
|
|
|
|
message += "⚪️ "
|
|
|
|
|
# Voice
|
|
|
|
|
if channel != 0:
|
|
|
|
|
# Voice status
|
|
|
|
|
if member.voice.self_deaf:
|
|
|
|
|
message += f"🔇 "
|
|
|
|
|
elif member.voice.self_mute:
|
|
|
|
|
message += f"🔈 "
|
|
|
|
|
else:
|
|
|
|
|
message += f"🔊 "
|
|
|
|
|
# Nickname
|
|
|
|
|
if member.nick is not None:
|
2018-11-07 19:50:55 +00:00
|
|
|
|
message += escape(member.nick)
|
2018-08-28 13:48:12 +00:00
|
|
|
|
else:
|
2018-11-07 19:50:55 +00:00
|
|
|
|
message += escape(member.name)
|
2018-08-28 13:48:12 +00:00
|
|
|
|
# Game or stream
|
|
|
|
|
if member.activity is not None:
|
2018-09-18 23:04:48 +00:00
|
|
|
|
if member.activity.type == discord.ActivityType.playing:
|
2018-11-07 19:50:55 +00:00
|
|
|
|
message += f" | 🎮 {escape(member.activity.name)}"
|
2018-09-18 23:04:48 +00:00
|
|
|
|
# Rich presence
|
2018-09-19 16:20:56 +00:00
|
|
|
|
try:
|
2018-11-06 22:11:35 +00:00
|
|
|
|
if member.activity.state is not None:
|
2018-11-07 19:50:55 +00:00
|
|
|
|
message += f" ({escape(member.activity.state)})"
|
2018-09-19 16:20:56 +00:00
|
|
|
|
except AttributeError:
|
|
|
|
|
try:
|
2018-11-06 22:11:35 +00:00
|
|
|
|
if member.activity.details is not None:
|
2018-11-07 19:50:55 +00:00
|
|
|
|
message += f" ({escape(member.activity.details)})"
|
2018-09-19 16:20:56 +00:00
|
|
|
|
except AttributeError:
|
|
|
|
|
pass
|
2018-09-18 23:04:48 +00:00
|
|
|
|
elif member.activity.type == discord.ActivityType.streaming:
|
2018-11-07 19:50:55 +00:00
|
|
|
|
message += f" | 📡 [{escape(member.activity.name)}]({escape(member.activity.url)})"
|
2018-09-18 23:04:48 +00:00
|
|
|
|
elif member.activity.type == discord.ActivityType.listening:
|
2018-11-07 19:50:55 +00:00
|
|
|
|
message += f" | 🎧 {escape(member.activity.name)}"
|
2018-08-28 13:48:12 +00:00
|
|
|
|
elif member.activity.type == discord.ActivityType.watching:
|
2018-11-07 19:50:55 +00:00
|
|
|
|
message += f" | 📺 {escape(member.activity.name)}"
|
2018-08-28 13:48:12 +00:00
|
|
|
|
message += "\n"
|
|
|
|
|
message += "\n"
|
|
|
|
|
connection.send(message)
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Answered successfully cvlist request.")
|
2018-08-07 23:30:43 +00:00
|
|
|
|
elif msg.startswith("!"):
|
|
|
|
|
data = msg.split(" ")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
if data[0] not in self.commands:
|
2018-08-07 23:30:43 +00:00
|
|
|
|
connection.send("error")
|
|
|
|
|
continue
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Received command: {msg}")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
await self.main_channel.send(f"{msg}\n"
|
|
|
|
|
f"_(da Telegram)_")
|
2018-09-05 17:48:34 +00:00
|
|
|
|
await self.commands[data[0]](channel=self.main_channel,
|
2018-08-08 12:09:57 +00:00
|
|
|
|
author=None,
|
|
|
|
|
params=data)
|
2018-08-07 23:30:43 +00:00
|
|
|
|
connection.send("success")
|
2018-02-25 18:50:57 +00:00
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
async def queue_predownload_videos(self):
|
|
|
|
|
while True:
|
|
|
|
|
for index, video in enumerate(self.video_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:
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.warning(f"Video download took more than {config['YouTube']['download_timeout']}s:"
|
|
|
|
|
f" {video.plain_text()}")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
await self.main_channel.send(f"⚠️ Il download di {str(video)} ha richiesto più di"
|
|
|
|
|
f" {config['YouTube']['download_timeout']} secondi, pertanto è stato"
|
|
|
|
|
f" rimosso dalla coda.")
|
|
|
|
|
del self.video_queue[index]
|
|
|
|
|
continue
|
|
|
|
|
except DurationError:
|
|
|
|
|
await self.main_channel.send(f"⚠️ {str(video)} dura più di"
|
|
|
|
|
f" {str(int(config['YouTube']['max_duration']) // 60)}"
|
|
|
|
|
f" minuti, quindi è stato rimosso dalla coda.")
|
|
|
|
|
del self.video_queue[index]
|
|
|
|
|
continue
|
|
|
|
|
except Exception as e:
|
2018-09-18 23:04:48 +00:00
|
|
|
|
sentry.user_context({
|
|
|
|
|
"discord": {
|
|
|
|
|
"discord_id": video.enqueuer.id,
|
|
|
|
|
"name": video.enqueuer.name,
|
|
|
|
|
"discriminator": video.enqueuer.discriminator
|
|
|
|
|
}
|
|
|
|
|
})
|
|
|
|
|
sentry.extra_context({
|
|
|
|
|
"video": video.plain_text()
|
|
|
|
|
})
|
|
|
|
|
sentry.captureException()
|
|
|
|
|
logger.error(f"Video download error: {str(e)}")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
await self.main_channel.send(f"⚠️ E' stato incontrato un errore durante il download di "
|
|
|
|
|
f"{str(video)}, quindi è stato rimosso dalla coda.\n\n"
|
|
|
|
|
f"**Dettagli sull'errore:**\n"
|
|
|
|
|
f"```python\n"
|
|
|
|
|
f"{str(e)}"
|
|
|
|
|
f"```")
|
|
|
|
|
del self.video_queue[index]
|
|
|
|
|
continue
|
|
|
|
|
await asyncio.sleep(1)
|
2018-02-25 18:50:57 +00:00
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
async def queue_play_next_video(self):
|
|
|
|
|
await self.wait_until_ready()
|
|
|
|
|
while True:
|
|
|
|
|
# Fun things will happen with multiple voice clients!
|
|
|
|
|
for voice_client in self.voice_clients:
|
2018-10-02 22:18:41 +00:00
|
|
|
|
if not voice_client.is_connected() or voice_client.is_playing() or voice_client.is_paused():
|
2018-08-08 12:09:57 +00:00
|
|
|
|
continue
|
|
|
|
|
if len(self.video_queue) == 0:
|
|
|
|
|
self.now_playing = None
|
|
|
|
|
continue
|
|
|
|
|
now_playing = self.video_queue[0]
|
|
|
|
|
try:
|
|
|
|
|
audio_source = now_playing.create_audio_source()
|
|
|
|
|
except FileNotDownloadedError:
|
|
|
|
|
continue
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.info(f"Started playing {repr(now_playing)}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
voice_client.play(audio_source)
|
|
|
|
|
del self.video_queue[0]
|
|
|
|
|
activity = discord.Activity(name=now_playing.plain_text(),
|
2018-11-06 22:11:35 +00:00
|
|
|
|
type=discord.ActivityType.listening)
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Updated bot presence to {now_playing.plain_text()}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
await self.change_presence(status=discord.Status.online, activity=activity)
|
|
|
|
|
if now_playing.enqueuer is not None:
|
2018-08-08 15:48:57 +00:00
|
|
|
|
try:
|
|
|
|
|
session = db.Session()
|
|
|
|
|
enqueuer = await loop.run_in_executor(executor, session.query(db.Discord)
|
|
|
|
|
.filter_by(discord_id=now_playing.enqueuer.id)
|
|
|
|
|
.one_or_none)
|
|
|
|
|
played_music = db.PlayedMusic(enqueuer=enqueuer,
|
|
|
|
|
filename=now_playing.plain_text(),
|
|
|
|
|
timestamp=datetime.datetime.now())
|
|
|
|
|
session.add(played_music)
|
|
|
|
|
await loop.run_in_executor(executor, session.commit)
|
|
|
|
|
await loop.run_in_executor(executor, session.close)
|
|
|
|
|
except sqlalchemy.exc.OperationalError:
|
|
|
|
|
pass
|
2018-08-08 12:09:57 +00:00
|
|
|
|
for key in song_special_messages:
|
|
|
|
|
if key in now_playing.file.lower():
|
|
|
|
|
await self.main_channel.send(song_special_messages[key].format(song=str(now_playing)))
|
|
|
|
|
break
|
|
|
|
|
else:
|
|
|
|
|
await self.main_channel.send(f":arrow_forward: Ora in riproduzione: {str(now_playing)}")
|
|
|
|
|
await asyncio.sleep(1)
|
|
|
|
|
|
2018-09-14 23:05:19 +00:00
|
|
|
|
async def inactivity_countdown(self):
|
|
|
|
|
while True:
|
|
|
|
|
await asyncio.sleep(1)
|
|
|
|
|
if self.inactivity_timer > 0:
|
|
|
|
|
self.inactivity_timer -= 1
|
|
|
|
|
continue
|
|
|
|
|
for voice_client in self.voice_clients:
|
|
|
|
|
if voice_client.is_connected():
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.info("Disconnecting due to inactivity.")
|
2018-09-17 22:07:00 +00:00
|
|
|
|
await voice_client.disconnect()
|
2018-11-06 22:11:35 +00:00
|
|
|
|
await self.change_presence(status=discord.Status.online, activity=None)
|
2018-09-17 22:07:00 +00:00
|
|
|
|
await self.main_channel.send("💤 Mi sono disconnesso dalla cv per inattività.")
|
2018-09-14 23:05:19 +00:00
|
|
|
|
|
2018-11-18 16:38:02 +00:00
|
|
|
|
async def create_activityreport(self):
|
|
|
|
|
logger.debug("Fetching Discord users...")
|
|
|
|
|
discord_users = list(self.main_guild.members)
|
|
|
|
|
online_members_count = 0
|
|
|
|
|
ingame_members_count = 0
|
|
|
|
|
cv_count = 0
|
|
|
|
|
cv_members_count = 0
|
|
|
|
|
non_empty_channels = []
|
|
|
|
|
for member in discord_users:
|
|
|
|
|
if member.bot:
|
|
|
|
|
continue
|
|
|
|
|
if member.voice is not None:
|
|
|
|
|
cv_count += 1
|
|
|
|
|
if member.voice.channel.id not in non_empty_channels:
|
|
|
|
|
non_empty_channels.append(member.voice.channel.id)
|
|
|
|
|
if len(member.roles) >= 2:
|
|
|
|
|
if member.voice is not None:
|
|
|
|
|
cv_members_count += 1
|
|
|
|
|
if member.status != discord.Status.offline and member.status != discord.Status.idle:
|
|
|
|
|
online_members_count += 1
|
|
|
|
|
if member.activity is not None and member.activity.type == discord.ActivityType.playing:
|
|
|
|
|
ingame_members_count += 1
|
|
|
|
|
logger.debug("Creating and committing db.ActivityReport...")
|
|
|
|
|
session = db.Session()
|
|
|
|
|
activityreport = db.ActivityReport(timestamp=datetime.datetime.now(),
|
|
|
|
|
discord_members_online=online_members_count,
|
|
|
|
|
discord_members_ingame=ingame_members_count,
|
|
|
|
|
discord_cv=cv_count,
|
|
|
|
|
discord_members_cv=cv_members_count,
|
|
|
|
|
discord_channels_used=len(non_empty_channels))
|
|
|
|
|
session.add(activityreport)
|
|
|
|
|
await loop.run_in_executor(executor, session.commit)
|
|
|
|
|
await loop.run_in_executor(executor, session.close)
|
|
|
|
|
logger.info("ActivityReport created.")
|
|
|
|
|
|
|
|
|
|
async def activity_task(self):
|
2018-11-18 16:42:49 +00:00
|
|
|
|
await self.wait_until_ready()
|
2018-11-18 16:38:02 +00:00
|
|
|
|
time_to_wait = config["Discord"]["activityreport_sample_time"]
|
|
|
|
|
while True:
|
|
|
|
|
await self.create_activityreport()
|
|
|
|
|
logger.debug(f"Waiting {time_to_wait} seconds before the next record.")
|
|
|
|
|
await asyncio.sleep(time_to_wait)
|
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
async def add_video_from_url(self, url, index: typing.Optional[int] = None, enqueuer: discord.Member = None):
|
|
|
|
|
# Retrieve info
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Retrieving info for {url}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
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 info is None:
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"No video found at {url}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
await self.main_channel.send(f"⚠ Non è stato trovato nessun video all'URL `{url}`,"
|
|
|
|
|
f" pertanto non è stato aggiunto alla coda.")
|
|
|
|
|
return
|
|
|
|
|
if "entries" in info:
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Playlist detected at {url}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
for entry in info["entries"]:
|
|
|
|
|
if index is not None:
|
|
|
|
|
self.video_queue.insert(index, Video(url=entry["webpage_url"], info=entry, enqueuer=enqueuer))
|
|
|
|
|
else:
|
|
|
|
|
self.video_queue.append(Video(url=entry["webpage_url"], info=entry, enqueuer=enqueuer))
|
|
|
|
|
return
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Single video detected at {url}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
if index is not None:
|
|
|
|
|
self.video_queue.insert(index, Video(url=url, info=info, enqueuer=enqueuer))
|
2018-05-25 17:58:30 +00:00
|
|
|
|
else:
|
2018-08-08 12:09:57 +00:00
|
|
|
|
self.video_queue.append(Video(url=url, info=info, enqueuer=enqueuer))
|
2018-05-25 17:58:30 +00:00
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
async def add_video_from_file(self, file, index: typing.Optional[int] = None, enqueuer: discord.Member = None):
|
|
|
|
|
if index is not None:
|
|
|
|
|
self.video_queue.insert(index, Video(file=file, enqueuer=enqueuer))
|
|
|
|
|
else:
|
|
|
|
|
self.video_queue.append(Video(file=file, enqueuer=enqueuer))
|
2018-05-25 17:58:30 +00:00
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
@command
|
|
|
|
|
async def null(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
pass
|
2018-05-25 17:58:30 +00:00
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
@command
|
|
|
|
|
async def cmd_ping(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
await channel.send(f"Pong!")
|
2018-05-25 17:58:30 +00:00
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
@command
|
|
|
|
|
async def cmd_register(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
2018-07-31 21:41:05 +00:00
|
|
|
|
session = db.Session()
|
2018-08-08 12:09:57 +00:00
|
|
|
|
if len(params) < 1:
|
|
|
|
|
await channel.send("⚠️ Non hai specificato un username!\n"
|
|
|
|
|
"Sintassi corretta: `!register <username_ryg>`")
|
2018-05-26 08:50:54 +00:00
|
|
|
|
return
|
2018-08-08 12:09:57 +00:00
|
|
|
|
try:
|
|
|
|
|
# noinspection PyTypeChecker
|
|
|
|
|
d = db.Discord.create(session,
|
|
|
|
|
royal_username=params[0],
|
|
|
|
|
discord_user=author)
|
|
|
|
|
except errors.AlreadyExistingError:
|
|
|
|
|
await channel.send("⚠ 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 channel.send("✅ Sincronizzazione completata!")
|
|
|
|
|
|
|
|
|
|
@command
|
|
|
|
|
async def cmd_cv(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
"""Summon the bot in the author's voice channel."""
|
|
|
|
|
if author is None:
|
|
|
|
|
await channel.send("⚠ Questo comando richiede un autore.")
|
|
|
|
|
return
|
|
|
|
|
if author.voice is None:
|
|
|
|
|
await channel.send("⚠ Non sei in nessun canale!")
|
|
|
|
|
return
|
|
|
|
|
if author.voice.channel == self.main_guild.afk_channel:
|
|
|
|
|
await channel.send("⚠ Non posso connettermi al canale AFK!")
|
|
|
|
|
return
|
|
|
|
|
if author.voice.channel.bitrate < 64000:
|
|
|
|
|
await channel.send("ℹ️ Sei in un canale con un bitrate ridotto.\n"
|
|
|
|
|
"L'utilizzo del bot in quel canale ignorerà il limite di bitrate e potrebbe causare lag"
|
|
|
|
|
" o eccessivo consumo di dati.\n"
|
|
|
|
|
"Se vuoi procedere comunque, scrivi `!yes`.")
|
|
|
|
|
try:
|
|
|
|
|
await self.wait_for("message", check=lambda m: m.content == "!yes", timeout=10.0)
|
|
|
|
|
except asyncio.TimeoutError:
|
|
|
|
|
return
|
|
|
|
|
# Check if there's already a connected client
|
|
|
|
|
for voice_client in self.voice_clients:
|
|
|
|
|
if voice_client.channel in self.main_guild.channels and voice_client.is_connected():
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.info(f"Moving to {author.voice.channel.name}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
await voice_client.move_to(author.voice.channel)
|
|
|
|
|
await channel.send(f"⤵️ Mi sono spostato in <#{author.voice.channel.id}>.")
|
|
|
|
|
break
|
|
|
|
|
else:
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.info(f"Connecting to {author.voice.channel.name}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
await author.voice.channel.connect()
|
|
|
|
|
await channel.send(f"⤵️ Mi sono connesso in <#{author.voice.channel.id}>.")
|
|
|
|
|
|
|
|
|
|
@command
|
|
|
|
|
@requires_connected_voice_client
|
|
|
|
|
async def cmd_play(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
if len(params) < 2:
|
|
|
|
|
await channel.send("⚠ Non hai specificato una canzone da riprodurre!\n"
|
|
|
|
|
"Sintassi: `!play <url|ricercayoutube|nomefile>`")
|
|
|
|
|
return
|
|
|
|
|
channel.typing()
|
|
|
|
|
# If the radio messages are enabled...
|
|
|
|
|
if self.radio_messages:
|
|
|
|
|
self.next_radio_message_in -= 1
|
|
|
|
|
if self.next_radio_message_in <= 0:
|
2018-11-03 17:42:56 +00:00
|
|
|
|
radio_message = random.sample(radio_messages, 1)[0]
|
2018-08-08 12:09:57 +00:00
|
|
|
|
self.next_radio_message_in = int(config["Discord"]["radio_messages_every"])
|
|
|
|
|
await self.add_video_from_url(radio_message)
|
|
|
|
|
await channel.send(f"📻 Aggiunto un messaggio radio, disattiva con `!radiomessages off`.")
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.info(f"Radio message added to the queue.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
# 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 self.add_video_from_url(url.group(0), enqueuer=author)
|
|
|
|
|
await channel.send(f"✅ Video aggiunto alla coda.")
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Added {url} to the queue as URL.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
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 self.add_video_from_file(file=file_path, enqueuer=author)
|
|
|
|
|
await channel.send(f"✅ Video aggiunto alla coda.")
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Added {file_path} to the queue as file.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
return
|
|
|
|
|
file_path += ".opus"
|
|
|
|
|
if os.path.exists(file_path):
|
|
|
|
|
# This is a file
|
|
|
|
|
await self.add_video_from_file(file=file_path, enqueuer=author)
|
|
|
|
|
await channel.send(f"✅ Video aggiunto alla coda.")
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Added {file_path} to the queue as file.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
return
|
|
|
|
|
# Search the parameter on youtube
|
|
|
|
|
search = " ".join(params[1:])
|
|
|
|
|
# This is a search
|
|
|
|
|
await self.add_video_from_url(url=f"ytsearch:{search}", enqueuer=author)
|
|
|
|
|
await channel.send(f"✅ Video aggiunto alla coda.")
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Added ytsearch:{search} to the queue as YouTube search.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
|
2018-11-08 22:41:41 +00:00
|
|
|
|
@command
|
|
|
|
|
async def cmd_m(self, channel, author, params):
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/1.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/2.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/3.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/4.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/5.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/6.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/7.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/8.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/9.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/a.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/b.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/c.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/d.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/e.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/f.opus")
|
|
|
|
|
await self.add_video_from_file("./opusfiles/m/g.opus")
|
|
|
|
|
await channel.send(f"?")
|
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
@command
|
|
|
|
|
@requires_connected_voice_client
|
|
|
|
|
async def cmd_skip(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
for voice_client in self.voice_clients:
|
|
|
|
|
if voice_client.is_playing():
|
|
|
|
|
voice_client.stop()
|
|
|
|
|
await channel.send(f"⏩ Video saltato.")
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"A song was skipped.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
break
|
|
|
|
|
else:
|
|
|
|
|
await channel.send("⚠ Non c'è nessun video in riproduzione.")
|
2018-08-01 23:11:43 +00:00
|
|
|
|
|
2018-08-08 12:09:57 +00:00
|
|
|
|
@command
|
|
|
|
|
@requires_connected_voice_client
|
|
|
|
|
async def cmd_remove(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
if len(self.video_queue) == 0:
|
|
|
|
|
await channel.send("⚠ Non c'è nessun video in coda.")
|
|
|
|
|
return
|
|
|
|
|
if len(params) == 1:
|
|
|
|
|
index = len(self.video_queue) - 1
|
|
|
|
|
else:
|
|
|
|
|
try:
|
|
|
|
|
index = int(params[1]) - 1
|
|
|
|
|
except ValueError:
|
|
|
|
|
await channel.send("⚠ Il numero inserito non è valido.\n"
|
|
|
|
|
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
|
|
|
|
|
return
|
|
|
|
|
if len(params) < 3:
|
|
|
|
|
if abs(index) >= len(self.video_queue):
|
|
|
|
|
await channel.send("⚠ Il numero inserito non corrisponde a nessun video nella playlist.\n"
|
|
|
|
|
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
|
|
|
|
|
return
|
|
|
|
|
video = self.video_queue.pop(index)
|
|
|
|
|
await channel.send(f":regional_indicator_x: {str(video)} è stato rimosso dalla coda.")
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Removed from queue: {video.plain_text()}")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
return
|
|
|
|
|
try:
|
|
|
|
|
start = int(params[1]) - 1
|
|
|
|
|
except ValueError:
|
|
|
|
|
await channel.send("⚠ Il numero iniziale inserito non è valido.\n"
|
|
|
|
|
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
|
|
|
|
|
return
|
|
|
|
|
if start >= len(self.video_queue):
|
|
|
|
|
await channel.send("⚠ Il numero iniziale inserito non corrisponde a nessun video nella"
|
|
|
|
|
" playlist.\n"
|
|
|
|
|
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
|
|
|
|
|
return
|
|
|
|
|
try:
|
|
|
|
|
end = int(params[2]) - 2
|
|
|
|
|
except ValueError:
|
|
|
|
|
await channel.send("⚠ Il numero finale inserito non è valido.\n"
|
|
|
|
|
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
|
|
|
|
|
return
|
|
|
|
|
if end >= len(self.video_queue):
|
|
|
|
|
await channel.send("⚠ Il numero finale inserito non corrisponde a nessun video nella"
|
|
|
|
|
" playlist.\n"
|
|
|
|
|
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
|
|
|
|
|
return
|
|
|
|
|
if start > end:
|
|
|
|
|
await channel.send("⚠ Il numero iniziale è maggiore del numero finale.\n"
|
|
|
|
|
"Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
|
|
|
|
|
return
|
|
|
|
|
del self.video_queue[start:end]
|
|
|
|
|
await channel.send(f":regional_indicator_x: {end - start} video rimossi dalla coda.")
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"Removed from queue {end - start} videos.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
|
|
|
|
|
@command
|
|
|
|
|
async def cmd_queue(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
if len(self.video_queue) == 0:
|
|
|
|
|
await channel.send("**Video in coda:**\n"
|
|
|
|
|
"nessuno")
|
|
|
|
|
return
|
|
|
|
|
msg = "**Video in coda:**\n"
|
|
|
|
|
for index, video in enumerate(self.video_queue[:10]):
|
|
|
|
|
msg += f"{queue_emojis[index]} {str(video)}\n"
|
|
|
|
|
if len(self.video_queue) > 10:
|
|
|
|
|
msg += f"più altri {len(self.video_queue) - 10} video!"
|
|
|
|
|
await channel.send(msg)
|
|
|
|
|
|
|
|
|
|
@command
|
|
|
|
|
@requires_connected_voice_client
|
|
|
|
|
async def cmd_shuffle(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
if len(self.video_queue) == 0:
|
|
|
|
|
await channel.send("⚠ Non ci sono video in coda!")
|
|
|
|
|
return
|
|
|
|
|
random.shuffle(self.video_queue)
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.info(f"The queue was shuffled by {author.name}#{author.discriminator}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
await channel.send("♠️ ♦️ ♣️ ♥️ Shuffle completo!")
|
|
|
|
|
|
|
|
|
|
@command
|
|
|
|
|
@requires_connected_voice_client
|
|
|
|
|
async def cmd_clear(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
if len(self.video_queue) == 0:
|
|
|
|
|
await channel.send("⚠ Non ci sono video in coda!")
|
|
|
|
|
return
|
|
|
|
|
self.video_queue = []
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.info(f"The queue was cleared by {author.name}#{author.discriminator}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
await channel.send(":regional_indicator_x: Tutti i video in coda rimossi.")
|
|
|
|
|
|
|
|
|
|
@command
|
|
|
|
|
@requires_connected_voice_client
|
|
|
|
|
async def cmd_forceplay(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
if len(params) < 2:
|
|
|
|
|
await channel.send("⚠ Non hai specificato una canzone da riprodurre!\n"
|
|
|
|
|
"Sintassi: `!forceplay <url|ricercayoutube|nomefile>`")
|
|
|
|
|
return
|
|
|
|
|
for voice_client in self.voice_clients:
|
|
|
|
|
if voice_client.is_playing():
|
|
|
|
|
voice_client.stop()
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.info(f"Video play was forced by {author.name}#{author.discriminator}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
# 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
|
2018-09-18 23:04:48 +00:00
|
|
|
|
await self.add_video_from_url(url.group(0), enqueuer=author)
|
|
|
|
|
await channel.send(f"✅ Video aggiunto alla coda.")
|
|
|
|
|
logger.debug(f"Forced {url} as URL.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
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
|
2018-09-18 23:04:48 +00:00
|
|
|
|
await self.add_video_from_file(file=file_path, enqueuer=author)
|
|
|
|
|
await channel.send(f"✅ Video aggiunto alla coda.")
|
|
|
|
|
logger.debug(f"Forced {file_path} as file.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
return
|
|
|
|
|
file_path += ".opus"
|
|
|
|
|
if os.path.exists(file_path):
|
|
|
|
|
# This is a file
|
2018-09-18 23:04:48 +00:00
|
|
|
|
await self.add_video_from_file(file=file_path, enqueuer=author)
|
|
|
|
|
await channel.send(f"✅ Video aggiunto alla coda.")
|
|
|
|
|
logger.debug(f"Forced {file_path} as file.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
return
|
|
|
|
|
# Search the parameter on youtube
|
|
|
|
|
search = " ".join(params[1:])
|
|
|
|
|
# This is a search
|
2018-09-18 23:04:48 +00:00
|
|
|
|
await self.add_video_from_url(url=f"ytsearch:{search}", enqueuer=author)
|
|
|
|
|
await channel.send(f"✅ Video aggiunto alla coda.")
|
|
|
|
|
logger.debug(f"Forced ytsearch:{search} as YouTube search.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
|
|
|
|
|
@command
|
|
|
|
|
async def cmd_radiomessages(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
if len(params) < 2:
|
|
|
|
|
self.radio_messages = not self.radio_messages
|
|
|
|
|
else:
|
|
|
|
|
if params[1].lower() == "on":
|
|
|
|
|
self.radio_messages = True
|
|
|
|
|
elif params[1].lower() == "off":
|
|
|
|
|
self.radio_messages = False
|
|
|
|
|
else:
|
|
|
|
|
await channel.send("⚠ Sintassi del comando non valida.\n"
|
|
|
|
|
"Sintassi: `!radiomessages [on|off]`")
|
|
|
|
|
return
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.info(f"Radio messages status toggled to {self.radio_messages}.")
|
2018-08-08 12:09:57 +00:00
|
|
|
|
await channel.send(f"📻 Messaggi radio **{'attivati' if self.radio_messages else 'disattivati'}**.")
|
2018-06-13 22:10:57 +00:00
|
|
|
|
|
2018-08-16 17:33:38 +00:00
|
|
|
|
@command
|
|
|
|
|
@requires_connected_voice_client
|
|
|
|
|
async def cmd_pause(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
2018-08-17 13:09:03 +00:00
|
|
|
|
for voice_client in self.voice_clients:
|
|
|
|
|
if voice_client.is_playing():
|
|
|
|
|
voice_client.pause()
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"The audio stream was paused.")
|
2018-08-17 13:09:03 +00:00
|
|
|
|
await channel.send(f"⏸ Riproduzione messa in pausa.\n"
|
|
|
|
|
f"Riprendi con `!resume`.")
|
|
|
|
|
|
|
|
|
|
@command
|
|
|
|
|
@requires_connected_voice_client
|
|
|
|
|
async def cmd_resume(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
|
|
|
|
|
for voice_client in self.voice_clients:
|
2018-10-04 10:46:15 +00:00
|
|
|
|
if voice_client.is_paused():
|
2018-08-17 13:09:03 +00:00
|
|
|
|
voice_client.resume()
|
2018-09-18 23:04:48 +00:00
|
|
|
|
logger.debug(f"The audio stream was resumed.")
|
2018-08-17 13:09:03 +00:00
|
|
|
|
await channel.send(f"⏯ Riproduzione ripresa.")
|
2018-08-16 17:33:38 +00:00
|
|
|
|
|
|
|
|
|
|
2018-01-15 12:42:40 +00:00
|
|
|
|
def process(users_connection=None):
|
2018-08-16 17:46:07 +00:00
|
|
|
|
logger.info("Initializing the bot...")
|
2018-08-07 23:30:43 +00:00
|
|
|
|
bot = RoyalDiscordBot()
|
2018-01-15 12:42:40 +00:00
|
|
|
|
if users_connection is not None:
|
2018-08-16 17:46:07 +00:00
|
|
|
|
logger.info("Initializing Telegram-Discord connection...")
|
2018-08-07 23:30:43 +00:00
|
|
|
|
asyncio.ensure_future(bot.feed_pipe(users_connection))
|
2018-08-16 17:46:07 +00:00
|
|
|
|
logger.info("Logging in...")
|
2018-08-07 23:30:43 +00:00
|
|
|
|
loop.run_until_complete(bot.login(config["Discord"]["bot_token"], bot=True))
|
2018-08-16 17:46:07 +00:00
|
|
|
|
logger.info("Connecting...")
|
2018-09-17 22:07:00 +00:00
|
|
|
|
loop.run_until_complete(bot.connect())
|
|
|
|
|
logger.info("Now stopping...")
|
|
|
|
|
loop.run_until_complete(bot.logout())
|
2018-01-15 12:42:40 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
2018-01-25 14:30:07 +00:00
|
|
|
|
process()
|