import socketio from titanembeds.utils import get_message_author, get_message_mentions, get_roles_list, get_attachments_list, get_embeds_list import time from email import utils as emailutils import discord class SocketIOInterface: def __init__(self, bot, redis_uri): self.io = socketio.AsyncRedisManager(redis_uri, write_only=True, channel='flask-socketio') self.bot = bot def format_datetime(self, datetimeobj): return emailutils.formatdate(time.mktime(datetimeobj.timetuple())) # https://stackoverflow.com/questions/3453177/convert-python-datetime-to-rfc-2822 def get_formatted_message(self, message): edit_ts = message.edited_at if not edit_ts: edit_ts = None else: edit_ts = self.format_datetime(edit_ts) msg = { "id": str(message.id), "channel_id": str(message.channel.id), "content": message.content, "author": get_message_author(message), "timestamp": self.format_datetime(message.created_at), "edited_timestamp": edit_ts, } if hasattr(message, "mentions"): msg["mentions"] = get_message_mentions(message.mentions) if hasattr(message, "attachments"): msg["attachments"] = get_attachments_list(message.attachments) if hasattr(message, "embeds"): msg["embeds"] = get_embeds_list(message.embeds) if hasattr(message, "author"): nickname = None if hasattr(message.author, 'nick') and message.author.nick: nickname = message.author.nick msg["author"]["nickname"] = nickname if hasattr(message, "mentions"): for mention in msg["mentions"]: mention["nickname"] = None member = message.guild.get_member(mention["id"]) if member: mention["nickname"] = member.nick return msg async def on_message(self, message): if message.guild: msg = self.get_formatted_message(message) await self.io.emit('MESSAGE_CREATE', data=msg, room=str("CHANNEL_"+str(message.channel.id)), namespace='/gateway') async def on_message_delete(self, message): if message.guild: msg = self.get_formatted_message(message) await self.io.emit('MESSAGE_DELETE', data=msg, room=str("CHANNEL_"+str(message.channel.id)), namespace='/gateway') async def on_message_update(self, message): if message.guild: msg = self.get_formatted_message(message) await self.io.emit('MESSAGE_UPDATE', data=msg, room=str("CHANNEL_"+str(message.channel.id)), namespace='/gateway') def get_formatted_user(self, user): userobj = { "avatar": user.avatar, "avatar_url": user.avatar_url, "color": str(user.color)[1:], "discriminator": user.discriminator, "game": None, "hoist-role": None, "id": str(user.id), "status": str(user.status), "username": user.name, "nick": None, } if userobj["color"] == "000000": userobj["color"] = None # if userobj["avatar_url"][len(userobj["avatar_url"])-15:] != ".jpg": # userobj["avatar_url"] = userobj["avatar_url"][:len(userobj["avatar_url"])-14] + ".jpg" if user.nick: userobj["nick"] = user.nick if hasattr(user, "activity") and user.activity: userobj["activity"] = { "name": user.activity.name } roles = sorted(user.roles, key=lambda k: k.position, reverse=True) for role in roles: if role.hoist: userobj["hoist-role"] = { "id": str(role.id), "name": role.name, "position": role.position, } break return userobj async def on_guild_member_add(self, member): user = self.get_formatted_user(member) await self.io.emit('GUILD_MEMBER_ADD', data=user, room=str("GUILD_"+str(member.guild.id)), namespace='/gateway') async def on_guild_member_remove(self, member): user = self.get_formatted_user(member) await self.io.emit('GUILD_MEMBER_REMOVE', data=user, room=str("GUILD_"+str(member.guild.id)), namespace='/gateway') async def on_guild_member_update(self, member): user = self.get_formatted_user(member) await self.io.emit('GUILD_MEMBER_UPDATE', data=user, room=str("GUILD_"+str(member.guild.id)), namespace='/gateway') def get_formatted_emojis(self, emojis): emotes = [] for emo in emojis: emotes.append({ "id": str(emo.id), "managed": emo.managed, "name": emo.name, "require_colons": emo.require_colons, "roles": get_roles_list(emo.roles), "url": emo.url, }) return emotes async def on_guild_emojis_update(self, emojis): emotes = self.get_formatted_emojis(emojis) await self.io.emit('GUILD_EMOJIS_UPDATE', data=emotes, room=str("GUILD_"+str(emojis[0].guild.id)), namespace='/gateway') def get_formatted_guild(self, guild): guil = { "id": str(guild.id), "name": guild.name, "icon": guild.icon, "icon_url": guild.icon_url, } return guil async def on_guild_update(self, guild): guildobj = self.get_formatted_guild(guild) await self.io.emit('GUILD_UPDATE', data=guildobj, room=str("GUILD_"+str(guild.id)), namespace='/gateway') def get_formatted_channel(self, channel): chan = { "id": str(channel.id), "guild_id": str(channel.guild.id), } return chan async def on_channel_delete(self, channel): if str(channel.type) != "text": return chan = self.get_formatted_channel(channel) await self.io.emit('CHANNEL_DELETE', data=chan, room=str("GUILD_"+str(channel.guild.id)), namespace='/gateway') async def on_channel_create(self, channel): if str(channel.type) != "text": return chan = self.get_formatted_channel(channel) await self.io.emit('CHANNEL_CREATE', data=chan, room=str("GUILD_"+str(channel.guild.id)), namespace='/gateway') async def on_channel_update(self, channel): if not isinstance(channel, discord.channel.TextChannel) and not isinstance(channel, discord.channel.CategoryChannel): return chan = self.get_formatted_channel(channel) await self.io.emit('CHANNEL_UPDATE', data=chan, room=str("GUILD_"+str(channel.guild.id)), namespace='/gateway') def get_formatted_role(self, role): rol = { "id": str(role.id), "guild_id": str(role.guild.id), "name": role.name, "color": role.color.value, "hoist": role.hoist, "position": role.position, "permissions": role.permissions.value, } return rol async def on_guild_role_create(self, role): rol = self.get_formatted_role(role) await self.io.emit('GUILD_ROLE_CREATE', data=rol, room=str("GUILD_"+str(role.guild.id)), namespace='/gateway') async def on_guild_role_update(self, role): rol = self.get_formatted_role(role) await self.io.emit('GUILD_ROLE_UPDATE', data=rol, room=str("GUILD_"+str(role.guild.id)), namespace='/gateway') async def on_guild_role_delete(self, role): rol = self.get_formatted_role(role) await self.io.emit('GUILD_ROLE_DELETE', data=rol, room=str("GUILD_"+str(role.guild.id)), namespace='/gateway')