You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.6 KiB
53 lines
1.6 KiB
from disco.bot import Plugin
|
|
from disco.bot.command import CommandError
|
|
from disco.voice.client import Player, OpusItem, VoiceException
|
|
|
|
|
|
def download(url):
|
|
return OpusItem.from_raw_file('test.dca')
|
|
|
|
|
|
class MusicPlugin(Plugin):
|
|
def load(self):
|
|
super(MusicPlugin, self).load()
|
|
self.guilds = {}
|
|
|
|
@Plugin.command('join')
|
|
def on_join(self, event):
|
|
if event.guild.id in self.guilds:
|
|
return event.msg.reply("I'm already playing music here.")
|
|
|
|
state = event.guild.get_member(event.author).get_voice_state()
|
|
if not state:
|
|
return event.msg.reply('You must be connected to voice to use that command.')
|
|
|
|
try:
|
|
client = state.channel.connect()
|
|
except VoiceException as e:
|
|
return event.msg.reply('Failed to connect to voice: `{}`'.format(e))
|
|
|
|
self.guilds[event.guild.id] = Player(client)
|
|
self.guilds[event.guild.id].complete.wait()
|
|
del self.guilds[event.guild.id]
|
|
|
|
def get_player(self, guild_id):
|
|
if guild_id not in self.guilds:
|
|
raise CommandError("I'm not currently playing music here.")
|
|
return self.guilds.get(guild_id)
|
|
|
|
@Plugin.command('leave')
|
|
def on_leave(self, event):
|
|
player = self.get_player(event.guild.id)
|
|
player.disconnect()
|
|
|
|
@Plugin.command('play', '<url:str>')
|
|
def on_play(self, event, url):
|
|
self.get_player(event.guild.id).queue.put(download(url))
|
|
|
|
@Plugin.command('pause')
|
|
def on_pause(self, event):
|
|
self.get_player(event.guild.id).pause()
|
|
|
|
@Plugin.command('resume')
|
|
def on_resume(self, event):
|
|
self.get_player(event.guild.id).resume()
|
|
|