mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-08 20:28:55 -05:00
added sound effects support
!sfx - plays the specified sound file from the data/audio/sfx directory !audioset sfx - toggles sfx for server !list sfx - lists sound effects Gives responsibility of `!stop`ping a stuck player to the user, like the other audio commands do
This commit is contained in:
parent
71dc5ea751
commit
aa4afc3255
106
cogs/audio.py
106
cogs/audio.py
@ -49,6 +49,7 @@ class Audio:
|
|||||||
def __init__(self, bot):
|
def __init__(self, bot):
|
||||||
self.bot = bot
|
self.bot = bot
|
||||||
self.music_player = EmptyPlayer()
|
self.music_player = EmptyPlayer()
|
||||||
|
self.sfx_player = EmptyPlayer()
|
||||||
self.settings = fileIO("data/audio/settings.json", "load")
|
self.settings = fileIO("data/audio/settings.json", "load")
|
||||||
self.queue_mode = False
|
self.queue_mode = False
|
||||||
self.queue = []
|
self.queue = []
|
||||||
@ -209,6 +210,70 @@ class Audio:
|
|||||||
else:
|
else:
|
||||||
await self.bot.say("There are no valid playlists in the localtracks folder.\nIf you're the owner, see {}".format(help_link))
|
await self.bot.say("There are no valid playlists in the localtracks folder.\nIf you're the owner, see {}".format(help_link))
|
||||||
|
|
||||||
|
@commands.command(pass_context=True, no_pm=True)
|
||||||
|
async def sfx(self, ctx, name : str):
|
||||||
|
"""Plays a local sound file
|
||||||
|
|
||||||
|
For bot's owner:
|
||||||
|
audio files must be placed in the data/audio/sfx folder.
|
||||||
|
Supported files are mp3, flac, wav"""
|
||||||
|
#sound effects default enabled
|
||||||
|
server = ctx.message.server
|
||||||
|
if self.settings["SERVER_SFX_ON"].get(server.id,None) is None:
|
||||||
|
self.settings["SERVER_SFX_ON"][server.id] = True
|
||||||
|
if not self.settings["SERVER_SFX_ON"][server.id]:
|
||||||
|
await self.bot.say("Sound effects are not enabled on this server")
|
||||||
|
return
|
||||||
|
|
||||||
|
msg = ctx.message
|
||||||
|
localsfx = self.get_local_sfx()
|
||||||
|
if localsfx and ("data/audio/sfx/" not in name and "\\" not in name):
|
||||||
|
if name in localsfx.keys():
|
||||||
|
file = "data/audio/sfx/{}{}".format(name,localsfx[name])
|
||||||
|
|
||||||
|
if await self.check_voice(msg.author, ctx.message):
|
||||||
|
try:
|
||||||
|
if self.music_player.is_playing():
|
||||||
|
self.music_player.paused = True
|
||||||
|
self.music_player.pause()
|
||||||
|
|
||||||
|
if self.sfx_player.is_playing():
|
||||||
|
self.sfx_player.stop()
|
||||||
|
|
||||||
|
self.sfx_player = self.bot.voice.create_ffmpeg_player(file, options='''-filter:a "volume={}"'''.format(self.settings["VOLUME"]))
|
||||||
|
self.sfx_player.start()
|
||||||
|
while self.sfx_player.is_playing():
|
||||||
|
await asyncio.sleep(.5)
|
||||||
|
|
||||||
|
if not self.music_player.is_playing():
|
||||||
|
self.music_player.paused = False
|
||||||
|
self.music_player.resume()
|
||||||
|
except AttributeError:
|
||||||
|
#music_player not used yet. Still an EmptyPlayer.
|
||||||
|
#better to ask for forgiveness?
|
||||||
|
pass
|
||||||
|
except Exception as e:
|
||||||
|
print(e)
|
||||||
|
|
||||||
|
else:
|
||||||
|
await self.bot.say("There is no sound effect with that name.")
|
||||||
|
else:
|
||||||
|
await self.bot.say("There are no valid sound effects in the data/audio/sfx folder.")
|
||||||
|
|
||||||
|
def get_local_sfx(self):
|
||||||
|
filenames = {}
|
||||||
|
files = os.listdir("data/audio/sfx/")
|
||||||
|
#only reason why these are the only supported ext is cause I haven't tried any others
|
||||||
|
supported = [".mp3",".flac",".wav"]
|
||||||
|
for f in files:
|
||||||
|
item = os.path.splitext(f)
|
||||||
|
if item[1] in supported:
|
||||||
|
filenames[item[0]] = item[1]
|
||||||
|
if filenames != {}:
|
||||||
|
return filenames
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
|
||||||
@commands.command(pass_context=True, no_pm=True)
|
@commands.command(pass_context=True, no_pm=True)
|
||||||
async def loop(self, ctx):
|
async def loop(self, ctx):
|
||||||
"""Loops single song
|
"""Loops single song
|
||||||
@ -273,6 +338,7 @@ class Audio:
|
|||||||
self.playlist = []
|
self.playlist = []
|
||||||
self.current = -1
|
self.current = -1
|
||||||
if self.music_player.is_playing(): self.music_player.stop()
|
if self.music_player.is_playing(): self.music_player.stop()
|
||||||
|
if self.sfx_player.is_playing(): self.sfx_player.stop()
|
||||||
await asyncio.sleep(1)
|
await asyncio.sleep(1)
|
||||||
if self.bot.voice: await self.bot.voice.disconnect()
|
if self.bot.voice: await self.bot.voice.disconnect()
|
||||||
|
|
||||||
@ -397,6 +463,21 @@ class Audio:
|
|||||||
else:
|
else:
|
||||||
await self.bot.say("There are no local playlists.")
|
await self.bot.say("There are no local playlists.")
|
||||||
|
|
||||||
|
@_list.command(name="sfx", pass_context=True)
|
||||||
|
async def list_sfx(self, ctx):
|
||||||
|
msg = "Available local sound effects: \n\n```"
|
||||||
|
files = self.get_local_sfx()
|
||||||
|
if files:
|
||||||
|
for i, d in enumerate(files.keys()):
|
||||||
|
if i % 4 == 0 and i != 0:
|
||||||
|
msg = msg + d + "\n"
|
||||||
|
else:
|
||||||
|
msg = msg + d + "\t"
|
||||||
|
msg += "```"
|
||||||
|
await self.bot.send_message(ctx.message.author, msg)
|
||||||
|
else:
|
||||||
|
await self.bot.say("There are no local sound effects.")
|
||||||
|
|
||||||
@_list.command(name="queue", pass_context=True)
|
@_list.command(name="queue", pass_context=True)
|
||||||
async def list_queue(self, ctx):
|
async def list_queue(self, ctx):
|
||||||
queue_list = await self.queue_titles()
|
queue_list = await self.queue_titles()
|
||||||
@ -471,6 +552,22 @@ class Audio:
|
|||||||
else:
|
else:
|
||||||
await self.bot.say("Volume must be between 0 and 1. Example: 0.40")
|
await self.bot.say("Volume must be between 0 and 1. Example: 0.40")
|
||||||
|
|
||||||
|
@audioset.command(name="sfx", pass_context=True)
|
||||||
|
async def _sfx(self, ctx):
|
||||||
|
"""Enables/Disables sound effects usage in the server"""
|
||||||
|
#default on.
|
||||||
|
server = ctx.message.server
|
||||||
|
if self.settings["SERVER_SFX_ON"].get(server.id,None) is None:
|
||||||
|
self.settings["SERVER_SFX_ON"][server.id] = True
|
||||||
|
else:
|
||||||
|
self.settings["SERVER_SFX_ON"][server.id] = not self.settings["SERVER_SFX_ON"][server.id]
|
||||||
|
#for a toggle, settings should save here in case bot fails to send message
|
||||||
|
fileIO("data/audio/settings.json", "save", self.settings)
|
||||||
|
if self.settings["SERVER_SFX_ON"][server.id]:
|
||||||
|
await self.bot.say("Sound effects are now enabled on this server.")
|
||||||
|
else:
|
||||||
|
await self.bot.say("Sound effects are now disabled on this server.")
|
||||||
|
|
||||||
@audioset.command()
|
@audioset.command()
|
||||||
@checks.is_owner()
|
@checks.is_owner()
|
||||||
async def maxcache(self, size : int):
|
async def maxcache(self, size : int):
|
||||||
@ -538,6 +635,11 @@ class Audio:
|
|||||||
if self.downloader["ID"]:
|
if self.downloader["ID"]:
|
||||||
try:
|
try:
|
||||||
if self.music_player.is_playing(): self.music_player.stop()
|
if self.music_player.is_playing(): self.music_player.stop()
|
||||||
|
# sfx_player should only ever get stuck as much as music_player does
|
||||||
|
# when it happens to music_player, the reponsibility is put on the user to !stop
|
||||||
|
# so we'll do the same with sfx_player. a timeout could be placed here though.
|
||||||
|
while self.sfx_player.is_playing():
|
||||||
|
await asyncio.sleep(.5)
|
||||||
self.music_player = self.bot.voice.create_ffmpeg_player(path + self.downloader["ID"], options='''-filter:a "volume={}"'''.format(self.settings["VOLUME"]))
|
self.music_player = self.bot.voice.create_ffmpeg_player(path + self.downloader["ID"], options='''-filter:a "volume={}"'''.format(self.settings["VOLUME"]))
|
||||||
self.music_player.paused = False
|
self.music_player.paused = False
|
||||||
self.music_player.start()
|
self.music_player.start()
|
||||||
@ -822,7 +924,7 @@ class MaximumLength(Exception):
|
|||||||
return self.message
|
return self.message
|
||||||
|
|
||||||
def check_folders():
|
def check_folders():
|
||||||
folders = ("data/audio", "data/audio/cache", "data/audio/playlists", "data/audio/localtracks")
|
folders = ("data/audio", "data/audio/cache", "data/audio/playlists", "data/audio/localtracks", "data/audio/sfx")
|
||||||
for folder in folders:
|
for folder in folders:
|
||||||
if not os.path.exists(folder):
|
if not os.path.exists(folder):
|
||||||
print("Creating " + folder + " folder...")
|
print("Creating " + folder + " folder...")
|
||||||
@ -830,7 +932,7 @@ def check_folders():
|
|||||||
|
|
||||||
def check_files():
|
def check_files():
|
||||||
|
|
||||||
default = {"VOLUME" : 0.5, "MAX_LENGTH" : 3700, "QUEUE_MODE" : True, "MAX_CACHE" : 0, "SOUNDCLOUD_CLIENT_ID": None, "TITLE_STATUS" : True}
|
default = {"VOLUME" : 0.5, "MAX_LENGTH" : 3700, "QUEUE_MODE" : True, "MAX_CACHE" : 0, "SOUNDCLOUD_CLIENT_ID": None, "TITLE_STATUS" : True, "SERVER_SFX_ON" : {}}
|
||||||
settings_path = "data/audio/settings.json"
|
settings_path = "data/audio/settings.json"
|
||||||
|
|
||||||
if not os.path.isfile(settings_path):
|
if not os.path.isfile(settings_path):
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user