2017-08-20 21:56:54 +02:00
|
|
|
import socketio
|
|
|
|
|
from titanembeds.utils import get_message_author, get_message_mentions
|
|
|
|
|
|
|
|
|
|
class SocketIOInterface:
|
|
|
|
|
def __init__(self, bot, redis_uri):
|
|
|
|
|
self.io = socketio.AsyncRedisManager(redis_uri, write_only=True, channel='flask-socketio')
|
|
|
|
|
self.bot = bot
|
|
|
|
|
|
|
|
|
|
async def on_message(self, message):
|
|
|
|
|
if message.server:
|
|
|
|
|
edit_ts = message.edited_timestamp
|
|
|
|
|
if not edit_ts:
|
|
|
|
|
edit_ts = None
|
|
|
|
|
else:
|
|
|
|
|
edit_ts = str(edit_ts)
|
|
|
|
|
msg = {
|
|
|
|
|
"id": message.id,
|
|
|
|
|
"channel_id": message.channel.id,
|
|
|
|
|
"content": message.content,
|
|
|
|
|
"author": get_message_author(message),
|
|
|
|
|
"timestamp": str(message.timestamp),
|
|
|
|
|
"edited_timestamp": edit_ts,
|
|
|
|
|
"mentions": get_message_mentions(message.mentions),
|
|
|
|
|
"attachments": message.attachments,
|
|
|
|
|
}
|
2017-08-21 06:26:51 +02:00
|
|
|
nickname = None
|
|
|
|
|
if hasattr(message.author, 'nick') and message.author.nick:
|
|
|
|
|
nickname = message.author.nick
|
|
|
|
|
msg["author"]["nickname"] = nickname
|
|
|
|
|
for mention in msg["mentions"]:
|
|
|
|
|
mention["nickname"] = None
|
|
|
|
|
member = message.server.get_member(mention["id"])
|
|
|
|
|
if member:
|
|
|
|
|
mention["nickname"] = member.nick
|
|
|
|
|
await self.io.emit('MESSAGE_CREATE', data=msg, room=str("CHANNEL_"+message.channel.id), namespace='/gateway')
|
