MyAnimeBot-DEV/myanimebot/discord.py

import discord
import pytz

import myanimebot.globals as globals # TODO Rename globals module
import myanimebot.utils as utils


def build_embed(feed : utils.Feed):
    ''' Build the embed message related to the anime's status '''

    # Get service
    if feed.service == utils.Service.MAL:
        service_name = 'MyAnimeList'
        profile_url = "{}{}".format(globals.MAL_PROFILE_URL, feed.user.name)
        icon_url = globals.MAL_ICON_URL
    elif feed.service == utils.Service.ANILIST:
        service_name = 'AniList'
        profile_url = "{}{}".format(globals.ANILIST_PROFILE_URL, feed.user.name)
        icon_url = globals.ANILIST_ICON_URL
    else:
        raise NotImplementedError('Unknown service {}'.format(feed.service))
    description = utils.build_description_string(feed)
    content = "[{}]({})\n```{}```".format(utils.filter_name(feed.media.name), feed.media.url, description)
    profile_url_label = "{}'s {}".format(feed.user.name, service_name)

    try:
        embed = discord.Embed(colour=int(feed.status.value, 16), url=feed.media.url, description=content, timestamp=feed.date_publication.astimezone(pytz.timezone("utc")))
        embed.set_thumbnail(url=feed.media.image)
        embed.set_author(name=profile_url_label, url=profile_url, icon_url=icon_url)
        embed.set_footer(text="MyAnimeBot", icon_url=globals.iconBot)

        return embed
    except Exception as e:
        globals.logger.error("Error when generating the message: " + str(e))
        return


async def send_embed_wrapper(asyncioloop, channelid, client, embed):
	''' Send an embed message to a channel '''

	channel = client.get_channel(int(channelid))
	
	try:
		await channel.send(embed=embed)
		globals.logger.info("Message sent in channel: {}".format(channelid))
	except Exception as e:
		globals.logger.debug("Impossible to send a message on '{}': {}".format(channelid, e)) 
		return

def in_allowed_role(user : discord.Member, server : int) -> bool :
	''' Check if a user has the permissions to configure the bot on a specific server '''

	targetRole = utils.get_allowed_role(server.id)
	globals.logger.debug ("Role target: " + str(targetRole))

	if user.guild_permissions.administrator:
		globals.logger.debug (str(user) + " is server admin on " + str(server) + "!")
		return True
	elif (targetRole is None):
		globals.logger.debug ("No group specified for " + str(server))
		return True
	else:
		for role in user.roles:
			if str(role.id) == str(targetRole):
				globals.logger.debug ("Permissions validated for " + str(user))
				return True

	return False