1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
import discord
import asyncio
import logging
import re
from urllib.parse import urlsplit
logging.basicConfig(level=logging.INFO)
client = discord.Client()
client.queue = asyncio.Queue(maxsize=5)
client.current_player = None
@client.event
@asyncio.coroutine
def on_ready():
logging.info('Logged in as\n\t{}\n\t{}'.format(client.user.name, client.user.id))
bot_name = ''
server_name = ''
reg = re.compile(r'^(?:!|\/){} (.*)$'.format(bot_name))
main_player_id = 0
@client.event
@asyncio.coroutine
def on_message(message):
global reg
global main_player_id
logging.debug('received message %s' % message)
if message.channel.is_private:
return
if message.server.name != server_name:
logging.info('wrong server %s' % message.server.name)
return
match = reg.search(message.content)
if not match:
logging.info('match failed')
return
command = match.group(1).split()[0]
if command == 'stop' and int(message.author.id) == main_player_id:
if client.current_player and client.current_player.is_playing():
client.current_player.stop()
return
if command == 'pause' and int(message.author.id) == main_player_id:
if client.current_player and client.current_player.is_playing():
client.current_player.pause()
return
if command == 'resume' and int(message.author.id) == main_player_id:
if client.current_player and not client.current_player.is_playing():
client.current_player.resume()
return
url = urlsplit(command, scheme='https')
if not (url.netloc and (url.path or (url.path is '/watch' and not url.query))):
yield from client.send_message(message.channel, 'format your commands right. fuck you.', tts=message.tts)
return
url = url.geturl()
logging.info(url)
if not client.is_voice_connected():
logging.info('connecting')
voice_chan = discord.utils.find(lambda x: x.name == 'General' and x.type is discord.ChannelType.voice,
message.server.channels)
if not voice_chan:
logging.error('no voice channel')
yield from client.join_voice_channel(voice_chan)
if client.current_player and client.current_player.is_playing():
client.current_player.stop()
client.current_player = yield from client.voice.create_ytdl_player(url)
client.current_player.start()
# enqueue_video(url, client, message.channel)
def enqueue_video(url, client, channel):
if not client.is_voice_connected():
logging.info
voice_chan = discord.utils.find(lambda x: x.name == 'General' and x.type is discord.ChannelType.voice,
channel.server.channels)
yield from client.join_voice_channel(voice_chan)
if not client.is_voice_connected():
yield from client.send_message(channel, 'go fuck yourself. voice isn\'t working.', tts=True)
return
if client.queue.full():
yield from client.send_message(channel, 'fuck you. wait for the other videos.', tts=True)
return
elem = yield from client.voice.create_ytdl_player(url)
client.queue.put(elem)
def run_video(client, loop):
logging.info('looping')
if client.current_player and client.current_player.is_playing():
return
client.current_player = yield from client.queue.get()
client.current_player.start()
loop.call_later(1, run_video, client, loop)
loop = asyncio.get_event_loop()
loop.call_soon(run_video, client, loop)
username = ''
password = ''
client.run(username, password)
|
