136 lines
5.4 KiB
Python
136 lines
5.4 KiB
Python
import discord
|
|
from discord.ext import bridge, commands
|
|
import wavelink
|
|
from bin.storage import Config
|
|
|
|
class Voice(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.queue = wavelink.Queue()
|
|
|
|
@commands.Cog.listener()
|
|
async def on_ready(self):
|
|
await wavelink.NodePool.create_node(bot=self.bot,
|
|
host=self.bot.config["wavelink"]["host"],
|
|
port=self.bot.config["wavelink"]["port"],
|
|
password=self.bot.config["wavelink"]["password"],)
|
|
|
|
@commands.Cog.listener()
|
|
async def on_track_end(self, player, track, reason):
|
|
if not reason == wavelink.TrackEndReason.STOPPED or self.queue.is_empty:
|
|
await player.play(self.queue.get())
|
|
|
|
@bridge.bridge_command(alises=["j"])
|
|
async def join(self, ctx, *, args: str=""):
|
|
await ctx.defer()
|
|
if ctx.author.voice is None and args == "":
|
|
await ctx.respond("You are not in a voice channel")
|
|
elif args != "":
|
|
channel = self.bot.get_channel(discord.utils.get(ctx.guild.channels, name=args).id)
|
|
await channel.connect(cls=wavelink.Player)
|
|
else:
|
|
await ctx.author.voice.channel.connect(cls=wavelink.Player)
|
|
|
|
@bridge.bridge_command(alises=["l"])
|
|
async def leave(self, ctx):
|
|
await ctx.defer()
|
|
if ctx.voice_client is None:
|
|
await ctx.respond("I am not in a voice channel")
|
|
else:
|
|
await ctx.voice_client.disconnect()
|
|
|
|
@bridge.bridge_command(aliases=["p"])
|
|
async def play(self, ctx, *, link: str=""):
|
|
await ctx.defer()
|
|
|
|
#check for overrides
|
|
# args = link.split(" ")
|
|
# print(args[6])
|
|
# for arg in range(len(args)-1):
|
|
# print(arg)
|
|
# print(args[arg])
|
|
# if args[arg] == "-channel" or args[arg] == "-c":
|
|
# channel = self.bot.get_channel(discord.utils.get(ctx.guild.channels, name=args[arg+1]).id)
|
|
# providedchannel = True
|
|
# if args[arg] == "-now" or args[arg] == "-n":
|
|
# queueoverride = True
|
|
# if args[arg] == "-earrape" or args[arg] == "-e":
|
|
# earape = True
|
|
providedchannel = False
|
|
queueoverride = False
|
|
earape = False
|
|
channel = ""
|
|
args = link.split(" -")
|
|
print(args)
|
|
if len(args) > 1:
|
|
for arg in range(len(args)):
|
|
if args[arg].startswith("channel") or args[arg].startswith("c"):
|
|
print(args[arg].split(" ")[1])
|
|
channel = self.bot.get_channel(discord.utils.get(ctx.guild.channels, name=args[arg].split(" ")[1]).id)
|
|
print(channel)
|
|
providedchannel = True
|
|
if args[arg].startswith("now") or args[arg].startswith("n"):
|
|
queueoverride = True
|
|
if args[arg].startswith("earrape") or args[arg].startswith("e"):
|
|
earape = True
|
|
track = await wavelink.YouTubeTrack.search(args[0], return_first=True)
|
|
|
|
|
|
if providedchannel and ctx.author.guild_permissions.administrator == False:
|
|
await ctx.respond("You do not have permission to specify a channel")
|
|
return
|
|
if queueoverride and ctx.author.guild_permissions.administrator == False:
|
|
await ctx.respond("You do not have permission to override the queue")
|
|
return
|
|
if earape and ctx.author.guild_permissions.administrator == False:
|
|
await ctx.respond("You do not have permission to earrape")
|
|
return
|
|
|
|
|
|
if ctx.author.voice is None and channel == "":
|
|
await ctx.respond("You are not in a voice channel, to specify a channel use `play <link> -channel <channel>`")
|
|
return
|
|
|
|
if ctx.voice_client is None:
|
|
if channel == "":
|
|
channel = ctx.author.voice.channel
|
|
await channel.connect(cls=wavelink.Player)
|
|
|
|
if link == "" and ctx.voice_client.is_paused():
|
|
await ctx.voice_client.resume()
|
|
return
|
|
|
|
if ctx.voice_client.is_paused():
|
|
ctx.voice_client.resume()
|
|
|
|
if (self.queue.is_empty and not ctx.voice_client.is_playing()) or queueoverride:
|
|
if earape:
|
|
ctx.voice_client.filters.set_filter(wavelink.Equalizer(wavelink.Filter.highpass(40, 1), wavelink.Filter.peaking(1000, 10)))
|
|
await ctx.voice_client.play(track)
|
|
await ctx.respond(f"Now playing: {track.title} by {track.author}\n {track.uri}")
|
|
else:
|
|
self.queue.put(item=track)
|
|
await ctx.respond(f"Added to queue: {track.title} by {track.author}\n {track.uri}")
|
|
|
|
@bridge.bridge_command(aliases=["stop","s"])
|
|
async def pause(self, ctx):
|
|
await ctx.defer()
|
|
if ctx.voice_client is None:
|
|
await ctx.respond("I am not in a voice channel")
|
|
else:
|
|
await ctx.voice_client.pause()
|
|
await ctx.respond("Paused")
|
|
|
|
@bridge.bridge_command(aliases=["next","n","sk"])
|
|
async def skip(self, ctx):
|
|
await ctx.defer()
|
|
if ctx.voice_client is None:
|
|
await ctx.respond("I am not in a voice channel")
|
|
else:
|
|
await ctx.voice_client.play(self.queue.get())
|
|
await ctx.respond("Skipped")
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Voice(bot))
|
|
|