1 changed files with 105 additions and 0 deletions
@ -0,0 +1,105 @@ |
|||||
|
import asyncio |
||||
|
import discord |
||||
|
|
||||
|
if not discord.opus.is_loaded(): |
||||
|
# the 'opus' library here is opus.dll on windows |
||||
|
# or libopus.so on linux in the current directory |
||||
|
# you should replace this with the location the |
||||
|
# opus library is located in. |
||||
|
discord.opus.load_opus('opus') |
||||
|
|
||||
|
class VoiceEntry: |
||||
|
def __init__(self, message, song): |
||||
|
self.requester = message.author |
||||
|
self.channel = message.channel |
||||
|
self.song = song |
||||
|
|
||||
|
class Bot(discord.Client): |
||||
|
def __init__(self): |
||||
|
super().__init__() |
||||
|
self.songs = asyncio.Queue() |
||||
|
self.play_next_song = asyncio.Event() |
||||
|
self.starter = None |
||||
|
self.player = None |
||||
|
self.current = None |
||||
|
|
||||
|
def toggle_next_song(self): |
||||
|
self.loop.call_soon_threadsafe(self.play_next_song.set) |
||||
|
|
||||
|
def can_control_song(self, author): |
||||
|
return author == self.starter or (self.current is not None and author == self.current.requester) |
||||
|
|
||||
|
def is_playing(self): |
||||
|
return self.player is not None and self.player.is_playing() |
||||
|
|
||||
|
@asyncio.coroutine |
||||
|
def on_message(self, message): |
||||
|
if message.author == self.user: |
||||
|
return |
||||
|
|
||||
|
if message.channel.is_private: |
||||
|
yield from self.send_message(message.channel, 'You cannot use this bot in private messages.') |
||||
|
|
||||
|
if message.content.startswith('$join'): |
||||
|
if self.is_voice_connected(): |
||||
|
yield from self.send_message(message.channel, 'Already connected to a voice channel') |
||||
|
channel_name = message.content[5:].strip() |
||||
|
check = lambda c: c.name == channel_name and c.type == discord.ChannelType.voice |
||||
|
|
||||
|
channel = discord.utils.find(check, message.server.channels) |
||||
|
if channel is None: |
||||
|
yield from self.send_message(message.channel, 'Cannot find a voice channel by that name.') |
||||
|
|
||||
|
yield from self.join_voice_channel(channel) |
||||
|
self.starter = message.author |
||||
|
|
||||
|
elif message.content.startswith('$leave'): |
||||
|
if not self.can_control_song(message.author): |
||||
|
return |
||||
|
self.starter = None |
||||
|
yield from self.voice.disconnect() |
||||
|
elif message.content.startswith('$pause'): |
||||
|
if not self.can_control_song(message.author): |
||||
|
fmt = 'Only the requester ({0.current.requester}) can control this song' |
||||
|
yield from self.send_message(message.channel, fmt.format(self)) |
||||
|
|
||||
|
if self.player.is_playing(): |
||||
|
self.player.pause() |
||||
|
elif message.content.startswith('$resume'): |
||||
|
if not self.can_control_song(message.author): |
||||
|
fmt = 'Only the requester ({0.current.requester}) can control this song' |
||||
|
yield from self.send_message(message.channel, fmt.format(self)) |
||||
|
|
||||
|
if self.player is not None and not self.is_playing(): |
||||
|
self.player.resume() |
||||
|
elif message.content.startswith('$next'): |
||||
|
filename = message.content[5:].strip() |
||||
|
yield from self.songs.put(VoiceEntry(message, filename)) |
||||
|
yield from self.send_message(message.channel, 'Successfully registered {}'.format(filename)) |
||||
|
elif message.content.startswith('$play'): |
||||
|
if self.player is not None and self.player.is_playing(): |
||||
|
yield from self.send_message(message.channel, 'Already playing a song') |
||||
|
return |
||||
|
while True: |
||||
|
if not self.is_voice_connected(): |
||||
|
yield from self.send_message(message.channel, 'Not connected to a voice channel') |
||||
|
return |
||||
|
|
||||
|
self.play_next_song.clear() |
||||
|
self.current = yield from self.songs.get() |
||||
|
self.player = self.voice.create_ffmpeg_player(self.current.song, after=self.toggle_next_song) |
||||
|
self.player.start() |
||||
|
fmt = 'Playing song "{0.song}" from {0.requester}' |
||||
|
yield from self.send_message(self.current.channel, fmt.format(self.current)) |
||||
|
yield from self.play_next_song.wait() |
||||
|
|
||||
|
@asyncio.coroutine |
||||
|
def on_ready(self): |
||||
|
print('Logged in as') |
||||
|
print(self.user.name) |
||||
|
print(self.user.id) |
||||
|
print('------') |
||||
|
|
||||
|
|
||||
|
bot = Bot() |
||||
|
bot.run('email', 'password') |
Loading…
Reference in new issue