def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.sentry_dsn = kwargs.get('sentry_dsn')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
self.voice_sessions_ids = dict()
if self.shard_id is not None:
self.shard = [self.shard_id, self.shard_count]
else:
self.shard = [0, 1]
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.sentry_dsn = kwargs.get('sentry_dsn')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
self.voice_sessions_ids = dict()
if self.shard_id is not None:
self.shard = [self.shard_id, self.shard_count]
else:
self.shard = [0, 1]
class Mee6(discord.Client):
"""A modified discord.Client class
This mod dispatches most events to the different plugins.
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
def run(self, *args):
self.loop.run_until_complete(self.start(*args))
async def on_ready(self):
"""Called when the bot is ready.
Connects to the database
Dispatched all the ready events
"""
log.info('Connected to the database')
if hasattr(self, 'shard_id'):
msg = 'Chat Shard {}/{} restarted'.format(self.shard_id,
self.shard_count)
else:
msg = 'Mee6 Chat restarted'
self.stats.event(msg, 'Server count: {}'.format(len(self.servers)))
await self.add_all_servers()
for plugin in self.plugins:
self.loop.create_task(plugin.on_ready())
async def add_all_servers(self):
"""Syncing all the servers to the DB"""
log.debug('Syncing servers and db')
for server in self.servers:
self.stats.set('mee6.servers', server.id)
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.sadd('servers', server.id)
if server.name:
await self.db.redis.set('server:{}:name'.format(server.id),
server.name)
if server.icon:
await self.db.redis.set('server:{}:icon'.format(server.id),
server.icon)
async def on_server_join(self, server):
"""Called when joining a new server"""
self.stats.set('mee6.servers', server.id)
self.stats.incr('mee6.server_join')
log.info('Joined {} server : {} !'.format(server.owner.name,
server.name))
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.sadd('servers', server.id)
await self.db.redis.set('server:{}:name'.format(server.id),
server.name)
if server.icon:
await self.db.redis.set('server:{}:icon'.format(server.id),
server.icon)
# Dispatching to global plugins
for plugin in self.plugins:
if plugin.is_global:
self.loop.create_task(plugin.on_server_join(server))
async def on_server_remove(self, server):
"""Called when leaving or kicked from a server
Removes the server from the db.
"""
log.info('Leaving {} server : {} !'.format(server.owner.name,
server.name))
log.debug('Removing server {}\'s id from the db'.format(server.id))
await self.db.redis.srem('servers', server.id)
async def get_plugins(self, server):
plugins = await self.plugin_manager.get_all(server)
return plugins
async def send_message(self, *args, **kwargs):
self.stats.incr('mee6.sent_messages')
return await super().send_message(*args, **kwargs)
async def on_message(self, message):
self.stats.incr('mee6.recv_messages')
if message.channel.is_private:
return
server = message.server
if message.content == "!shard?":
if hasattr(self, 'shard_id'):
await self.send_message(
message.channel,
"shard {}/{}".format(self.shard_id + 1, self.shard_count))
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin._on_message(message))
async def on_message_edit(self, before, after):
if before.channel.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_edit(before, after))
async def on_message_delete(self, message):
if message.channel.is_private:
return
server = message.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_delete(message))
async def on_channel_create(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_create(channel))
async def on_channel_update(self, before, after):
if before.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_update(before, after))
async def on_channel_delete(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_delete(channel))
async def on_member_join(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_join(member))
async def on_member_remove(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_remove(member))
async def on_member_update(self, before, after):
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_update(before, after))
async def on_server_update(self, before, after):
server = after
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_update(before, after))
async def on_server_role_create(self, server, role):
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_role_create(server, role))
async def on_server_role_delete(self, server, role):
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_role_delete(server, role))
async def on_server_role_update(self, before, after):
server = None
for s in self.servers:
if after.id in map(lambda r: r.id, s.roles):
server = s
break
if server is None:
return
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_role_update(before, after))
async def on_voice_state_update(self, before, after):
if after is None and before is None:
return
elif after is None:
server = before.server
elif before is None:
server = after.server
else:
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_voice_state_update(before, after))
async def on_member_ban(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_ban(member))
async def on_member_unban(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_unban(member))
async def on_typing(self, channel, user, when):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_typing(channel, user, when))
class Mee6(discord.Client):
"""A modified discord.Client class
This mod dispatched most events to the different plugins.
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
async def on_ready(self):
"""Called when the bot is ready.
Connects to the database
Dispatched all the ready events
"""
log.info('Connected to the database')
if hasattr(self, 'shard_id'):
msg = 'Chat Shard {}/{} restarted'.format(self.shard_id, self.shard_count)
else:
msg = 'Mee6 Chat restarted'
self.stats.event(msg, 'Server count: {}'.format(len(self.servers)))
await self.add_all_servers()
for plugin in self.plugins:
self.loop.create_task(plugin.on_ready())
async def add_all_servers(self):
"""Syncing all the servers to the DB"""
log.debug('Syncing servers and db')
for server in self.servers:
self.stats.set('mee6.servers', server.id)
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.sadd('servers', server.id)
if server.name:
await self.db.redis.set('server:{}:name'.format(server.id), server.name)
if server.icon:
await self.db.redis.set('server:{}:icon'.format(server.id), server.icon)
async def on_server_join(self, server):
"""Called when joining a new server"""
self.stats.set('mee6.servers', server.id)
self.stats.incr('mee6.server_join')
log.info('Joined {} server : {} !'.format(server.owner.name, server.name))
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.sadd('servers', server.id)
await self.db.redis.set('server:{}:name'.format(server.id), server.name)
if server.icon:
await self.db.redis.set('server:{}:icon'.format(server.id), server.icon)
# Dispatching to global plugins
for plugin in self.plugins:
if plugin.is_global:
self.loop.create_task(plugin.on_server_join(server))
async def on_server_remove(self, server):
"""Called when leaving or kicked from a server
Removes the server from the db.
"""
log.info('Leaving {} server : {} !'.format(server.owner.name, server.name))
log.debug('Removing server {}\'s id from the db'.format(server.id))
await self.db.redis.srem('servers', server.id)
async def get_plugins(self, server):
plugins = await self.plugin_manager.get_all(server)
return plugins
async def send_message(self, *args, **kwargs):
self.stats.incr('mee6.sent_messages')
return await super().send_message(*args, **kwargs)
async def on_message(self, message):
self.stats.incr('mee6.recv_messages')
if message.channel.is_private:
return
server = message.server
if message.content == "!shard?":
if hasattr(self, 'shard_id'):
await self.send_message(message.channel, "shard {}/{}".format(self.shard_id+1,
self.shard_count))
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message(message))
async def on_message_edit(self, before, after):
if before.channel.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_edit(before, after))
async def on_message_delete(self, message):
if message.channel.is_private:
return
server = message.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_delete(message))
async def on_channel_create(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_create(channel))
async def on_channel_update(self, before, after):
if before.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_update(before, after))
async def on_channel_delete(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_delete(channel))
async def on_member_join(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_join(member))
async def on_member_remove(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_remove(member))
async def on_member_update(self, before, after):
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_update(before, after))
async def on_server_update(self, before, after):
server = after
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_update(before, after))
async def on_server_role_create(self, server, role):
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_role_create(server, role))
async def on_server_role_delete(self, server, role):
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_role_delete(server, role))
async def on_server_role_update(self, before, after):
server = None
for s in self.servers:
if after.id in map(lambda r:r.id, s.roles):
server = s
break
if server is None:
return
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_role_update(before, after))
async def on_voice_state_update(self, before, after):
if after is None:
server = before.server
elif before is None:
server = after.server
else:
return
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_voice_state_update(before, after))
async def on_member_ban(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_ban(member))
async def on_member_unban(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_unban(member))
async def on_typing(self, channel, user, when):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_typing(channel, user, when))
class Mee6(discord.Client):
"""A modified discord.Client class
This mod dispatches most events to the different plugins.
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.sentry_dsn = kwargs.get('sentry_dsn')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
def run(self, *args):
self.loop.run_until_complete(self.start(*args))
async def ping(self):
PING_INTERVAL = 3
key = 'mee6.gateway.{}-{}'.format(self.shard_id,
self.shard_count)
while True:
if self.is_logged_in:
self.stats.check(key, 'OK')
await self.db.redis.setex(key, PING_INTERVAL + 2, '1')
else:
self.stats.check(key, 'Critical')
await asyncio.sleep(PING_INTERVAL)
async def on_ready(self):
"""Called when the bot is ready.
Connects to the database
Dispatched all the ready events
"""
log.info('Connected to the database')
await self.add_all_servers()
for plugin in self.plugins:
self.loop.create_task(plugin.on_ready())
self.loop.create_task(self.ping())
async def add_all_servers(self):
"""Syncing all the servers to the DB"""
log.debug('Syncing servers and db')
for server in self.servers:
self.stats.set('mee6.servers', server.id)
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.sadd('servers', server.id)
if server.name:
await self.db.redis.set(
'server:{}:name'.format(server.id),
server.name
)
if server.icon:
await self.db.redis.set(
'server:{}:icon'.format(server.id),
server.icon
)
async def on_server_join(self, server):
"""Called when joining a new server"""
self.stats.set('mee6.servers', server.id)
self.stats.incr('mee6.server_join')
await self.db.redis.sadd('servers', server.id)
log.info('Joined {} server : {} !'.format(
server.owner.name,
server.name
))
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.set('server:{}:name'.format(server.id), server.name)
if server.icon:
await self.db.redis.set(
'server:{}:icon'.format(server.id),
server.icon
)
# Dispatching to global plugins
for plugin in self.plugins:
if plugin.is_global:
self.loop.create_task(plugin.on_server_join(server))
async def on_server_remove(self, server):
"""Called when leaving or kicked from a server
Removes the server from the db.
"""
log.info('Leaving {} server : {} !'.format(
server.owner.name,
server.name
))
log.debug('Removing server {}\'s id from the db'.format(
server.id
))
await self.db.redis.srem('servers', server.id)
async def get_plugins(self, server):
plugins = await self.plugin_manager.get_all(server)
return plugins
async def send_message(self, *args, **kwargs):
self.stats.incr('mee6.sent_messages')
return await super().send_message(*args, **kwargs)
async def on_message(self, message):
self.stats.incr('mee6.recv_messages')
if message.channel.is_private:
return
if message.author.__class__ != discord.Member:
return
server = message.server
if message.content == "!shard?":
if hasattr(self, 'shard_id'):
await self.send_message(
message.channel,
"shard {}/{}".format(self.shard_id+1, self.shard_count)
)
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin._on_message(message))
async def on_message_edit(self, before, after):
if before.channel.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_edit(before, after))
async def on_message_delete(self, message):
if message.channel.is_private:
return
server = message.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_delete(message))
async def on_channel_create(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_create(channel))
async def on_channel_update(self, before, after):
if before.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_update(before, after))
async def on_channel_delete(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_delete(channel))
async def on_member_join(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_join(member))
async def on_member_remove(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_remove(member))
async def on_member_update(self, before, after):
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_update(before, after))
async def on_server_update(self, before, after):
server = after
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_update(before, after))
class Mee6(discord.Client):
"""A modified discord.Client class
This mod dispatches most events to the different plugins.
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.sentry_dsn = kwargs.get('sentry_dsn')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
def run(self, *args):
self.loop.run_until_complete(self.start(*args))
async def ping(self):
PING_INTERVAL = 3
key = 'mee6.gateway.{}-{}'.format(self.shard_id,
self.shard_count)
while True:
if self.is_logged_in:
self.stats.check(key, 'OK')
await self.db.redis.setex(key, PING_INTERVAL + 2, '1')
else:
self.stats.check(key, 'Critical')
await asyncio.sleep(PING_INTERVAL)
async def on_ready(self):
"""Called when the bot is ready.
Connects to the database
Dispatched all the ready events
"""
log.info('Connected to the database')
await self.add_all_servers()
for plugin in self.plugins:
self.loop.create_task(plugin.on_ready())
self.loop.create_task(self.ping())
async def add_all_servers(self):
"""Syncing all the servers to the DB"""
log.debug('Syncing servers and db')
for server in self.servers:
self.stats.set('mee6.servers', server.id)
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.sadd('servers', server.id)
if server.name:
await self.db.redis.set(
'server:{}:name'.format(server.id),
server.name
)
if server.icon:
await self.db.redis.set(
'server:{}:icon'.format(server.id),
server.icon
)
async def on_server_join(self, server):
"""Called when joining a new server"""
self.stats.set('mee6.servers', server.id)
self.stats.incr('mee6.server_join')
await self.db.redis.sadd('servers', server.id)
log.info('Joined {} server : {} !'.format(
server.owner.name,
server.name
))
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.set('server:{}:name'.format(server.id), server.name)
if server.icon:
await self.db.redis.set(
'server:{}:icon'.format(server.id),
server.icon
)
# Dispatching to global plugins
for plugin in self.plugins:
if plugin.is_global:
self.loop.create_task(plugin.on_server_join(server))
async def on_server_remove(self, server):
"""Called when leaving or kicked from a server
Removes the server from the db.
"""
log.info('Leaving {} server : {} !'.format(
server.owner.name,
server.name
))
log.debug('Removing server {}\'s id from the db'.format(
server.id
))
await self.db.redis.srem('servers', server.id)
async def get_plugins(self, server):
plugins = await self.plugin_manager.get_all(server)
return plugins
async def send_message(self, *args, **kwargs):
self.stats.incr('mee6.sent_messages')
return await super().send_message(*args, **kwargs)
async def on_message(self, message):
self.stats.incr('mee6.recv_messages')
if message.channel.is_private:
return
if message.author.__class__ != discord.Member:
return
server = message.server
if message.content == "!shard?":
if hasattr(self, 'shard_id'):
await self.send_message(
message.channel,
"shard {}/{}".format(self.shard_id, self.shard_count)
)
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin._on_message(message))
async def on_message_edit(self, before, after):
if before.channel.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_edit(before, after))
async def on_message_delete(self, message):
if message.channel.is_private:
return
server = message.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_delete(message))
async def on_channel_create(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_create(channel))
async def on_channel_update(self, before, after):
if before.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_update(before, after))
async def on_channel_delete(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_delete(channel))
async def on_member_join(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_join(member))
async def on_member_remove(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_remove(member))
async def on_member_update(self, before, after):
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_update(before, after))
async def on_server_update(self, before, after):
server = after
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_update(before, after))
class Mee6(discord.Client):
"""A modified discord.Client class
This mod dispatches most events to the different plugins.
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.sentry_dsn = kwargs.get('sentry_dsn')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
self.voice_sessions_ids = dict()
if self.shard_id is not None:
self.shard = [self.shard_id, self.shard_count]
else:
self.shard = [0, 1]
def run(self, *args):
console_coro = self.loop.create_server(make_console(self), '127.0.0.1',
8000)
self.loop.run_until_complete(console_coro)
self.loop.run_until_complete(self.start(*args))
async def console(reader, writer):
loop = asyncio.get_event_loop()
data = await reader.read(100)
message = data.decode()
result = eval(message)
writer.write(result)
await writer.drain()
writer.close()
def send_monitoring_message(self, msg):
self.loop.create_task(self._send_monitoring_message(msg))
async def _send_monitoring_message(self, msg):
url = os.getenv('MONITORING_WH')
if not url:
return
headers = {}
headers['Content-Type'] = 'application/json'
data = json.dumps({'content': msg})
async with aiohttp.ClientSession() as session:
async with session.post(url, headers=headers, data=data) as d:
pass
async def on_ready(self):
"""Called when the bot is ready.
Connects to the database
Dispatched all the ready events
"""
log.info('Connected to the database')
await self.add_all_servers()
for plugin in self.plugins:
self.loop.create_task(plugin.on_ready())
async def add_all_servers(self):
"""Syncing all the servers to the DB"""
log.debug('Syncing servers and db')
for server in self.servers:
self.stats.set('mee6.servers', server.id)
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.sadd('servers', server.id)
if server.name:
await self.db.redis.set('server:{}:name'.format(server.id),
server.name)
if server.icon:
await self.db.redis.set('server:{}:icon'.format(server.id),
server.icon)
async def on_server_join(self, server):
"""Called when joining a new server"""
self.stats.set('mee6.servers', server.id)
self.stats.incr('mee6.server_join')
await self.db.redis.sadd('servers', server.id)
log.info('Joined {} server : {} !'.format(server.owner.name,
server.name))
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.set('server:{}:name'.format(server.id),
server.name)
if server.icon:
await self.db.redis.set('server:{}:icon'.format(server.id),
server.icon)
# Dispatching to global plugins
for plugin in self.plugins:
if plugin.is_global:
self.loop.create_task(plugin.on_server_join(server))
async def on_server_remove(self, server):
"""Called when leaving or kicked from a server
Removes the server from the db.
"""
log.info('Leaving {} server : {} !'.format(server.owner.name,
server.name))
log.debug('Removing server {}\'s id from the db'.format(server.id))
await self.db.redis.srem('servers', server.id)
async def get_plugins(self, server):
plugins = await self.plugin_manager.get_all(server)
return plugins
async def send_message(self, *args, **kwargs):
self.stats.incr('mee6.sent_messages')
return await super().send_message(*args, **kwargs)
async def on_message(self, message):
self.stats.incr('mee6.recv_messages')
if message.channel.is_private:
return
wh_key = 'channel.{}.last_message_whid'.format(message.channel.id)
if message.author.__class__ != discord.Member:
return
server = message.server
if message.content == "!shard?":
if hasattr(self, 'shard_id'):
await self.send_message(
message.channel,
"shard {}/{}".format(self.shard_id, self.shard_count))
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin._on_message(message))
async def on_message_edit(self, before, after):
if before.channel.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_edit(before, after))
async def on_message_delete(self, message):
if message.channel.is_private:
return
server = message.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_delete(message))
async def on_channel_create(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_create(channel))
async def on_channel_update(self, before, after):
if before.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_update(before, after))
async def on_channel_delete(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_delete(channel))
async def on_member_join(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_join(member))
async def on_member_remove(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_remove(member))
async def on_member_update(self, before, after):
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_update(before, after))
async def on_server_update(self, before, after):
server = after
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_update(before, after))
class Mee6(discord.Client):
"""A modified discord.Client class
This mod dispatches most events to the different plugins.
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.sentry_dsn = kwargs.get('sentry_dsn')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
self.voice_sessions_ids = dict()
if self.shard_id is not None:
self.shard = [self.shard_id, self.shard_count]
else:
self.shard = [0, 1]
def run(self, *args):
console_coro = self.loop.create_server(make_console(self), '127.0.0.1',
8000)
self.loop.run_until_complete(console_coro)
self.loop.create_task(self.connect_schwifty())
self.loop.run_until_complete(self.start(*args))
async def connect_schwifty(self):
self.schwifty = await SchwiftyWebsocket.create(self.shard, self)
while not self.is_closed:
try:
await self.schwifty.poll_event()
except (ConnectionClosed, asyncio.TimeoutError) as e:
await asyncio.sleep(1)
self.schwifty = await SchwiftyWebsocket.create(
self.shard, self)
async def console(reader, writer):
loop = asyncio.get_event_loop()
data = await reader.read(100)
message = data.decode()
result = eval(message)
writer.write(result)
await writer.drain()
writer.close()
def send_monitoring_message(self, msg):
self.loop.create_task(self._send_monitoring_message(msg))
async def _send_monitoring_message(self, msg):
url = os.getenv('MONITORING_WH')
if not url:
return
headers = {}
headers['Content-Type'] = 'application/json'
data = json.dumps({'content': msg})
async with aiohttp.ClientSession() as session:
async with session.post(url, headers=headers, data=data) as d:
pass
async def on_socket_raw_receive(self, payload):
tags = {'op': str(payload['op']), 't': payload.get('t') or "NONE"}
self.stats.incr('discord.event', tags=tags)
if not payload['op'] == 0:
return
if payload['t'] == 'VOICE_STATE_UPDATE':
d = payload['d']
if str(d['user_id']) != self.user.id:
return
guild_id = d['guild_id']
self.voice_sessions_ids[guild_id] = d['session_id']
if payload['t'] == 'VOICE_SERVER_UPDATE':
d = payload['d']
guild_id = d['guild_id']
d['session_id'] = self.voice_sessions_ids.get(guild_id)
if d.get('endpoint'):
await self.schwifty.voice_update(d)
if payload['t'] == 'READY':
if not hasattr(self, 'shard_id'):
return
msg = "**[READY]** shard {}/{}".format(self.shard_id,
self.shard_count)
self.send_monitoring_message(msg)
if payload['t'] == 'RESUMED':
if not hasattr(self, 'shard_id'):
return
msg = "**[RESUMED]** shard {}/{}".format(self.shard_id,
self.shard_count)
self.send_monitoring_message(msg)
async def dispatch_schwifty_event(self, t, d):
listener_name = 'on_schwifty_' + t.lower()
listener = getattr(self, listener_name, None)
if listener:
await listener(**d)
async def on_ready(self):
"""Called when the bot is ready.
Connects to the database
Dispatched all the ready events
"""
log.info('Connected to the database')
await self.add_all_servers()
for plugin in self.plugins:
self.loop.create_task(plugin.on_ready())
async def add_all_servers(self):
"""Syncing all the servers to the DB"""
log.debug('Syncing servers and db')
for server in self.servers:
self.stats.set('mee6.servers', server.id)
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.sadd('servers', server.id)
if server.name:
await self.db.redis.set('server:{}:name'.format(server.id),
server.name)
if server.icon:
await self.db.redis.set('server:{}:icon'.format(server.id),
server.icon)
async def on_server_join(self, server):
"""Called when joining a new server"""
self.stats.set('mee6.servers', server.id)
self.stats.incr('mee6.server_join')
await self.db.redis.sadd('servers', server.id)
log.info('Joined {} server : {} !'.format(server.owner.name,
server.name))
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.set('server:{}:name'.format(server.id),
server.name)
if server.icon:
await self.db.redis.set('server:{}:icon'.format(server.id),
server.icon)
# Dispatching to global plugins
for plugin in self.plugins:
if plugin.is_global:
self.loop.create_task(plugin.on_server_join(server))
async def on_server_remove(self, server):
"""Called when leaving or kicked from a server
Removes the server from the db.
"""
log.info('Leaving {} server : {} !'.format(server.owner.name,
server.name))
log.debug('Removing server {}\'s id from the db'.format(server.id))
await self.db.redis.srem('servers', server.id)
async def get_plugins(self, server):
plugins = await self.plugin_manager.get_all(server)
return plugins
async def send_message(self, *args, **kwargs):
self.stats.incr('mee6.sent_messages')
return await super().send_message(*args, **kwargs)
async def on_message(self, message):
self.stats.incr('mee6.recv_messages')
if message.channel.is_private:
return
wh_key = 'channel.{}.last_message_whid'.format(message.channel.id)
if message.webhook_id:
self.db.redis.set(wh_key, message.webhook_id)
else:
self.db.redis.delete(wh_key)
if message.author.__class__ != discord.Member:
return
server = message.server
if message.content == "!shard?":
if hasattr(self, 'shard_id'):
await self.send_message(
message.channel,
"shard {}/{}".format(self.shard_id, self.shard_count))
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin._on_message(message))
async def on_message_edit(self, before, after):
if before.channel.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_edit(before, after))
async def on_message_delete(self, message):
if message.channel.is_private:
return
server = message.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_delete(message))
async def on_channel_create(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_create(channel))
async def on_channel_update(self, before, after):
if before.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_update(before, after))
async def on_channel_delete(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_delete(channel))
async def on_member_join(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_join(member))
async def on_member_remove(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_remove(member))
async def on_member_update(self, before, after):
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_update(before, after))
async def on_server_update(self, before, after):
server = after
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_update(before, after))
async def on_schwifty_playing(self, guild_id, url):
server = discord.Object(id=str(guild_id))
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_schwifty_playing(guild_id, url))
async def on_schwifty_finished_playing(self, guild_id):
server = discord.Object(id=str(guild_id))
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(
plugin.on_schwifty_finished_playing(guild_id))
class Mee6(discord.Client):
'''A modified discord.Client class
This mod dispatches most events to the different plugins.
'''
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.sentry_dsn = kwargs.get('sentry_dsn')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
self.voice_sessions_ids = dict()
if self.shard_id is not None:
self.shard = [self.shard_id, self.shard_count]
else:
self.shard = [0, 1]
def run(self, *args):
console_coro = self.loop.create_server(make_console(self), '127.0.0.1',
8000)
self.loop.run_until_complete(console_coro)
# These are related to music player I think. Disabled currently since I couldn't get it working properly.
#self.loop.create_task(self.connect_schwifty())
self.loop.run_until_complete(self.start(*args))
# These are related to music player I think. Disabled currently since I couldn't get it working properly.
# async def connect_schwifty(self):
# self.schwifty = await SchwiftyWebsocket.create(
# self.shard,
# self
# )
#
# while not self.is_closed:
# try:
# await self.schwifty.poll_event()
# except (ConnectionClosed, asyncio.TimeoutError) as e:
# await asyncio.sleep(1)
# self.schwifty = await SchwiftyWebsocket.create(
# self.shard,
# self
# )
async def console(reader, writer):
loop = asyncio.get_event_loop()
data = await reader.read(100)
message = data.decode()
result = eval(message)
writer.write(result)
await writer.drain()
writer.close()
#These two functions log the bots status to some discord servers webhook. TODO need to adjust that.
def send_monitoring_message(self, msg):
self.loop.create_task(self._send_monitoring_message(msg))
async def _send_monitoring_message(self, msg):
url = os.getenv('MONITORING_WH')
if not url:
return
headers = {}
headers['Content-Type'] = 'application/json'
data = json.dumps({'content': msg})
async with aiohttp.ClientSession() as session:
async with session.post(url, headers=headers, data=data) as d:
pass
async def on_voice_state_update(self, member):
#TODO stats?
if str(member.id) != self.user.id:
return
guild_id = member.guild_id
self.voice_sessions_ids[guild_id] = member.server.id
#TODO This function's parameters have been changed on the new discord.py version. It is no longer sent as parsed.
# async def on_socket_raw_receive(self, payload):
# tags = {'op': str(payload['op']),
# 't': payload.get('t') or "NONE"}
# self.stats.incr('discord.event', tags=tags)
#
# if not payload['op'] == 0:
# return
#
#
# if payload['t'] == 'VOICE_SERVER_UPDATE':
# d = payload['d']
# guild_id = d['guild_id']
# d['session_id'] = self.voice_sessions_ids.get(guild_id)
# if d.get('endpoint'):
# await self.schwifty.voice_update(d)
#
# if payload['t'] == 'READY':
# if not hasattr(self, 'shard_id'):
# return
#
# msg = "**[READY]** shard {}/{}".format(
# self.shard_id,
# self.shard_count
# )
# self.send_monitoring_message(msg)
#
# if payload['t'] == 'RESUMED':
# if not hasattr(self, 'shard_id'):
# return
#
# msg = "**[RESUMED]** shard {}/{}".format(self.shard_id,
# self.shard_count)
# self.send_monitoring_message(msg)
async def dispatch_schwifty_event(self, t, d):
listener_name = 'on_schwifty_' + t.lower()
listener = getattr(self, listener_name, None)
if listener:
await listener(**d)
async def on_ready(self):
'''
Called when the bot is ready.Connects to the database
Dispatched all the ready events
'''
log.info('Connected to the database')
await self.add_all_servers()
for plugin in self.plugins:
self.loop.create_task(plugin.on_ready())
async def add_all_servers(self):
'''
Syncing all the servers to the DB
'''
log.debug('Syncing servers and db')
for server in self.servers:
self.stats.set('mee6.servers', server.id)
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.sadd('servers', server.id)
if server.name:
await self.db.redis.set('server:{}:name'.format(server.id),
server.name)
if server.icon:
await self.db.redis.set('server:{}:icon'.format(server.id),
server.icon)
async def on_server_join(self, server):
'''
Called when joining a new server
'''
self.stats.set('mee6.servers', server.id)
self.stats.incr('mee6.server_join')
await self.db.redis.sadd('servers', server.id)
log.info('Joined {} server : {} !'.format(server.owner.name,
server.name))
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.set('server:{}:name'.format(server.id),
server.name)
if server.icon:
await self.db.redis.set('server:{}:icon'.format(server.id),
server.icon)
# Dispatching to global plugins
for plugin in self.plugins:
if plugin.is_global:
self.loop.create_task(plugin.on_server_join(server))
async def on_server_remove(self, server):
'''
Called when leaving or kicked from a server Removes the server from the db.
'''
log.info('Leaving {} server : {} !'.format(server.owner.name,
server.name))
log.debug('Removing server {}\'s id from the db'.format(server.id))
await self.db.redis.srem('servers', server.id)
async def get_plugins(self, server):
plugins = await self.plugin_manager.get_all(server)
return plugins
async def send_message(self, *args, **kwargs):
self.stats.incr('mee6.sent_messages')
return await super().send_message(*args, **kwargs)
async def on_message(self, message):
self.stats.incr('mee6.recv_messages')
if message.channel.is_private:
return
#Not sure about this discord.py docs doesn't list this among the variables passed with new messages so commented out for now.
#This probably changed recently since webhook messages have a seperate event on the discord.py
#wh_key = 'channel.{}.last_message_whid'.format(message.channel.id)
#log.info(message)
#if message.webhook_id:
# self.db.redis.set(wh_key, message.webhook_id)
#else:
# self.db.redis.delete(wh_key)
if message.author.__class__ != discord.Member:
return
server = message.server
if message.content == "!shard?":
if hasattr(self, 'shard_id'):
await self.send_message(
message.channel,
"shard {}/{}".format(self.shard_id, self.shard_count))
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin._on_message(message))
async def on_message_edit(self, before, after):
if before.channel.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_edit(before, after))
async def on_message_delete(self, message):
if message.channel.is_private:
return
server = message.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_delete(message))
async def on_channel_create(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_create(channel))
async def on_channel_update(self, before, after):
if before.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_update(before, after))
async def on_channel_delete(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_delete(channel))
async def on_member_join(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_join(member))
async def on_member_remove(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_remove(member))
async def on_member_update(self, before, after):
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_update(before, after))
async def on_server_update(self, before, after):
server = after
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_update(before, after))
class Mee6(discord.Client):
"""A modified discord.Client class
This mod dispatches most events to the different plugins.
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.redis_url = kwargs.get('redis_url')
self.mongo_url = kwargs.get('mongo_url')
self.dd_agent_url = kwargs.get('dd_agent_url')
self.sentry_dsn = kwargs.get('sentry_dsn')
self.db = Db(self.redis_url, self.mongo_url, self.loop)
self.plugin_manager = PluginManager(self)
self.plugin_manager.load_all()
self.last_messages = []
self.stats = DDAgent(self.dd_agent_url)
self.voice_sessions_ids = dict()
if self.shard_id is not None:
self.shard = [self.shard_id, self.shard_count]
else:
self.shard = [0, 1]
def run(self, *args):
console_coro = self.loop.create_server(make_console(self),
'127.0.0.1', 8000)
self.loop.run_until_complete(console_coro)
self.loop.create_task(self.connect_schwifty())
self.loop.run_until_complete(self.start(*args))
async def connect_schwifty(self):
self.schwifty = await SchwiftyWebsocket.create(
self.shard,
self
)
while not self.is_closed:
try:
await self.schwifty.poll_event()
except (ConnectionClosed, asyncio.TimeoutError) as e:
await asyncio.sleep(1)
self.schwifty = await SchwiftyWebsocket.create(
self.shard,
self
)
async def console(reader, writer):
loop = asyncio.get_event_loop()
data = await reader.read(100)
message = data.decode()
result = eval(message)
writer.write(result)
await writer.drain()
writer.close()
def send_monitoring_message(self, msg):
self.loop.create_task(self._send_monitoring_message(msg))
async def _send_monitoring_message(self, msg):
url = os.getenv('MONITORING_WH')
if not url:
return
headers = {}
headers['Content-Type'] = 'application/json'
data = json.dumps({'content': msg})
async with aiohttp.ClientSession() as session:
async with session.post(url, headers=headers, data=data) as d:
pass
async def on_socket_raw_receive(self, payload):
tags = {'op': str(payload['op']),
't': payload.get('t') or "NONE"}
self.stats.incr('discord.event', tags=tags)
if not payload['op'] == 0:
return
if payload['t'] == 'VOICE_STATE_UPDATE':
d = payload['d']
if str(d['user_id']) != self.user.id:
return
guild_id = d['guild_id']
self.voice_sessions_ids[guild_id] = d['session_id']
if payload['t'] == 'VOICE_SERVER_UPDATE':
d = payload['d']
guild_id = d['guild_id']
d['session_id'] = self.voice_sessions_ids.get(guild_id)
if d.get('endpoint'):
await self.schwifty.voice_update(d)
if payload['t'] == 'READY':
if not hasattr(self, 'shard_id'):
return
msg = "**[READY]** shard {}/{}".format(
self.shard_id,
self.shard_count
)
self.send_monitoring_message(msg)
if payload['t'] == 'RESUMED':
if not hasattr(self, 'shard_id'):
return
msg = "**[RESUMED]** shard {}/{}".format(self.shard_id,
self.shard_count)
self.send_monitoring_message(msg)
async def dispatch_schwifty_event(self, t, d):
listener_name = 'on_schwifty_' + t.lower()
listener = getattr(self, listener_name, None)
if listener:
await listener(**d)
async def on_ready(self):
"""Called when the bot is ready.
Connects to the database
Dispatched all the ready events
"""
log.info('Connected to the database')
await self.add_all_servers()
for plugin in self.plugins:
self.loop.create_task(plugin.on_ready())
async def add_all_servers(self):
"""Syncing all the servers to the DB"""
log.debug('Syncing servers and db')
for server in self.servers:
self.stats.set('mee6.servers', server.id)
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.sadd('servers', server.id)
if server.name:
await self.db.redis.set(
'server:{}:name'.format(server.id),
server.name
)
if server.icon:
await self.db.redis.set(
'server:{}:icon'.format(server.id),
server.icon
)
async def on_server_join(self, server):
"""Called when joining a new server"""
self.stats.set('mee6.servers', server.id)
self.stats.incr('mee6.server_join')
await self.db.redis.sadd('servers', server.id)
log.info('Joined {} server : {} !'.format(
server.owner.name,
server.name
))
log.debug('Adding server {}\'s id to db'.format(server.id))
await self.db.redis.set('server:{}:name'.format(server.id), server.name)
if server.icon:
await self.db.redis.set(
'server:{}:icon'.format(server.id),
server.icon
)
# Dispatching to global plugins
for plugin in self.plugins:
if plugin.is_global:
self.loop.create_task(plugin.on_server_join(server))
async def on_server_remove(self, server):
"""Called when leaving or kicked from a server
Removes the server from the db.
"""
log.info('Leaving {} server : {} !'.format(
server.owner.name,
server.name
))
log.debug('Removing server {}\'s id from the db'.format(
server.id
))
await self.db.redis.srem('servers', server.id)
async def get_plugins(self, server):
plugins = await self.plugin_manager.get_all(server)
return plugins
async def send_message(self, *args, **kwargs):
self.stats.incr('mee6.sent_messages')
return await super().send_message(*args, **kwargs)
async def on_message(self, message):
self.stats.incr('mee6.recv_messages')
if message.channel.is_private:
return
wh_key = 'channel.{}.last_message_whid'.format(message.channel.id)
if message.webhook_id:
self.db.redis.set(wh_key, message.webhook_id)
else:
self.db.redis.delete(wh_key)
if message.author.__class__ != discord.Member:
return
server = message.server
if message.content == "!shard?":
if hasattr(self, 'shard_id'):
await self.send_message(
message.channel,
"shard {}/{}".format(self.shard_id, self.shard_count)
)
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin._on_message(message))
async def on_message_edit(self, before, after):
if before.channel.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_edit(before, after))
async def on_message_delete(self, message):
if message.channel.is_private:
return
server = message.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_message_delete(message))
async def on_channel_create(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_create(channel))
async def on_channel_update(self, before, after):
if before.is_private:
return
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_update(before, after))
async def on_channel_delete(self, channel):
if channel.is_private:
return
server = channel.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_channel_delete(channel))
async def on_member_join(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_join(member))
async def on_member_remove(self, member):
server = member.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_remove(member))
async def on_member_update(self, before, after):
server = after.server
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_member_update(before, after))
async def on_server_update(self, before, after):
server = after
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_server_update(before, after))
async def on_schwifty_playing(self, guild_id, url):
server = discord.Object(id=str(guild_id))
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_schwifty_playing(guild_id, url))
async def on_schwifty_finished_playing(self, guild_id):
server = discord.Object(id=str(guild_id))
enabled_plugins = await self.get_plugins(server)
for plugin in enabled_plugins:
self.loop.create_task(plugin.on_schwifty_finished_playing(guild_id))