cmdg_Starboard.py

from Uncute_Rina import *
from import_modules import *

messageIdMarkedForDeletion = []

class Starboard(commands.Cog):
    def __init__(self, client: Bot):
        global RinaDB
        self.client: Bot = client
        RinaDB = client.RinaDB

    async def updateStat(self, star_message: discord.Message, starboard_emoji: discord.Emoji, downvote_init_value: int):
        # find original message
        text = star_message.embeds[0].fields[0].value ## "[Jump!]({msgLink})"
        link = text.split("(")[1]
        # Initial attempt to use [:-1] to remove the final ")" character doesn't work if there are unknown
        # files in the original starboard message because rina mentions them in the starboard msg after the
        # [Jump] link, adding "\n[...]" so ye.
        link = link.split(")",1)[0]
        #      0    1      2           3          4: guild_id          5: channel_id         6: message_id
        #    https:/ /discord.com / channels / 985931648094834798 / 1006682505149169694 / 1014887159485968455
        guild_id, channel_id, message_id = [int(i) for i in link.split("/")[4:]]
        try:
            ch = self.client.get_channel(channel_id)
            original_message = await ch.fetch_message(message_id)
        except discord.NotFound:
            # if original message removed, remove starboard message
            await log_to_guild(self.client, star_message.guild, f"{starboard_emoji} :x: Starboard message {star_message.id} was removed (from {message_id}) (original message could not be found)")
            messageIdMarkedForDeletion.append(star_message.id)
            await star_message.delete()
            return

        star_reacters = []
        reactionTotal = 0
        # get message's starboard-reacters
        for reaction in original_message.reactions:
            if reaction.is_custom_emoji():
                if getattr(reaction.emoji, "id", None) == starboard_emoji.id:
                    async for user in reaction.users():
                        if user.id not in star_reacters:
                            star_reacters.append(user.id)

        # get starboard's starboard-reacters
        for reaction in star_message.reactions:
            if reaction.is_custom_emoji():
                if getattr(reaction.emoji, "id", None) == starboard_emoji.id:
                    async for user in reaction.users():
                        if user.id not in star_reacters:
                            star_reacters.append(user.id)

        star_stat = len(star_reacters)
        if self.client.user.id in star_reacters:
            star_stat -= 1

        for reaction in star_message.reactions:
            if reaction.emoji == '❌':
                reactionTotal = star_stat + reaction.count - reaction.me # stars (exc. rina) + x'es - rina's x
                star_stat -= reaction.count - reaction.me

        #if more x'es than stars, and more than [15] reactions, remove message
        if star_stat < 0 and reactionTotal >= downvote_init_value:
            await log_to_guild(self.client, star_message.guild, f"{starboard_emoji} :x: Starboard message {star_message.id} was removed (from {message_id}) (too many downvotes! Score: {star_stat}, Votes: {reactionTotal})")
            messageIdMarkedForDeletion.append(star_message.id)
            await star_message.delete()
            return

        # update message to new star value
        parts = star_message.content.split("**")
        parts[1] = str(star_stat)
        new_content = '**'.join(parts)
        # update embed message to keep most accurate nickname
        embeds = star_message.embeds
        if isinstance(original_message.author, discord.Member):
            name = original_message.author.nick or original_message.author.name
        else:
            name = original_message.author.name
        embeds[0].set_author(
            name=f"{name}",
            url=f"https://original.poster/{original_message.author.id}/",
            icon_url=original_message.author.display_avatar.url
        )
        await star_message.edit(content=new_content, embeds=embeds)

    @commands.Cog.listener()
    async def on_raw_reaction_add(self, payload: discord.RawReactionActionEvent):
        if payload.guild_id in [None, self.client.staff_server_id]:
            return
        _star_channel, star_minimum, channel_blacklist, starboard_emoji_id, downvote_init_value = await self.client.get_guild_info(
            payload.guild_id, "starboardChannel", "starboardCountMinimum", "starboardBlacklistedChannels", "starboardEmoji", "starboardDownvoteInitValue")
        if payload.member.id == self.client.user.id or \
                (getattr(payload.emoji, "id",   None) != starboard_emoji_id and
                 getattr(payload.emoji, "name", None) != "❌"):
            # only run starboard code if the reactions tracked are actually starboard emojis (or the downvote emoji)
            return

        #get the message id from payload.message_id through the channel (with payload.channel_id) (oof lengthy process)
        try:
            ch = self.client.get_channel(payload.channel_id)
            message = await ch.fetch_message(payload.message_id)
        except discord.errors.NotFound:
            await log_to_guild(
                self.client, self.client.get_guild(payload.guild_id),
                f'**:warning: Warning: **Couldn\'t find channel {payload.channel_id} (<#{payload.channel_id}>) or message {payload.message_id}!\n'
                f'Potentially broken link: https://discord.com/channels/{payload.guild_id}/{payload.channel_id}/{payload.message_id}\n'
                f'This is likely caused by someone removing a PluralKit message by reacting with the :x: emoji.')
            return

        
        # print(repr(message.guild.id), repr(self.client.staff_server_id), message.guild.id == self.client.staff_server_id)
        star_channel = self.client.get_channel(_star_channel)
        starboard_emoji = self.client.get_emoji(starboard_emoji_id)

        if message.channel.id == star_channel.id:
            await self.updateStat(message, starboard_emoji, downvote_init_value)
            return

        for reaction in message.reactions:
            if getattr(reaction.emoji, "id", None) == starboard_emoji_id:
                if reaction.me:
                    # check if this message is already in the starboard. If so, update it
                    async for star_message in star_channel.history(limit=200):
                        for embed in star_message.embeds:
                            if embed.footer.text == str(message.id):
                                await self.updateStat(star_message, starboard_emoji, downvote_init_value)
                                return
                    return
                elif reaction.count == star_minimum:
                    if message.author == self.client.user:
                        #can't starboard Rina's message
                        return
                    if message.channel.id in channel_blacklist:
                        return

                    try:
                        # Try to add the initial starboard emoji to starboarded message
                        # to prevent duplicate entries in starboard.
                        await message.add_reaction(starboard_emoji)
                    except discord.errors.Forbidden:
                        # If "Reaction blocked", then maybe message author blocked Rina.
                        # Thus, I can't track if Rina added it to starboard already or not.
                        await log_to_guild(self.client, self.client.get_guild(payload.guild_id), 
                                     f'**:warning: Warning: **Couldn\'t add starboard emoji to {message.jump_url}. They might have blocked Rina...')
                        return

                    msgLink = f"https://discord.com/channels/{message.guild.id}/{message.channel.id}/{message.id}"
                    embed = discord.Embed(
                            color=discord.Colour.from_rgb(r=255, g=172, b=51),
                            title='',
                            description=f"{message.content}",
                            timestamp=datetime.now()
                        )
                    embed.add_field(name="Source", value=f"[Jump!]({msgLink})")
                    embed.set_footer(text=f"{message.id}")
                    if isinstance(message.author, discord.Member):
                        name = message.author.nick or message.author.name
                    else:
                        name = message.author.name
                    embed.set_author(
                            name=f"{name}",
                            url=f"https://original.poster/{message.author.id}/",
                            icon_url=message.author.display_avatar.url
                    )
                    embed_list = []
                    for attachment in message.attachments:
                        try:
                            if attachment.content_type.split("/")[0] == "image": #is image or GIF
                                if len(embed_list) == 0:
                                    embed.set_image(url=attachment.url)
                                    embed_list = [embed]
                                else:
                                    # can only set one image per embed... But you can add multiple embeds :]
                                    embed = discord.Embed(
                                        color = discord.Colour.from_rgb(r=255, g=172, b=51),
                                    )
                                    embed.set_image(url=attachment.url)
                                    embed_list.append(embed)
                            else:
                                if len(embed_list) == 0:
                                    embed.set_field_at(0,
                                                       name = embed.fields[0].name,
                                                       value = embed.fields[0].value + f"\n\n(⚠️ +1 Unknown attachment ({attachment.content_type}))")
                                else:
                                    embed_list[0].set_field_at(0,
                                                               name=embed_list[0].fields[0].name,
                                                               value=embed_list[0].fields[0].value + f"\n\n(⚠️ +1 Unknown attachment ({attachment.content_type}))")
                        except AttributeError:
                            # if it is neither an image, video, application, or recognised file type:
                            if len(embed_list) == 0:
                                embed.set_field_at(0,
                                                   name=embed.fields[0].name,
                                                   value=embed.fields[0].value + f"\n\n(💔 +1 Unrecognized attachment type)")
                            else:
                                embed_list[0].set_field_at(0,
                                                           name=embed_list[0].fields[0].name,
                                                           value=embed_list[0].fields[0].value + f"\n\n(💔 +1 Unrecognized attachment type)")
                    if len(embed_list) == 0:
                        embed_list.append(embed)

                    msg = await star_channel.send(
                            f"💫 **{reaction.count}** | <#{message.channel.id}>",
                            embeds=embed_list,
                            allowed_mentions=discord.AllowedMentions.none(),
                        )
                    await log_to_guild(self.client, star_channel.guild, 
                                 f"{starboard_emoji} Starboard message {msg.jump_url} was created from {message.jump_url}. "
                                 f"Content: \"\"\"{message.content}\"\"\" and attachments: {[x.url for x in message.attachments]}")
                    # add new starboard msg
                    await msg.add_reaction(starboard_emoji)
                    await msg.add_reaction("❌")
                    # add star reaction to original message to prevent message from being re-added to the starboard

    @commands.Cog.listener()
    async def on_raw_reaction_remove(self, payload: discord.RawReactionActionEvent):
        if payload.guild_id in [None, self.client.staff_server_id]:
            return
        _star_channel, starboard_emoji_id, downvote_init_value = await self.client.get_guild_info(payload.guild_id, "starboardChannel", "starboardEmoji", "starboardDownvoteInitValue")
        if getattr(payload.emoji, "id", None) != starboard_emoji_id and \
            getattr(payload.emoji, "name", None) != "❌":
            # only run starboard code if the reactions tracked are actually starboard emojis (or the downvote emoji)
            return
        #get the message id from payload.message_id through the channel (with payload.channel_id) (oof lengthy process)
        ch = self.client.get_channel(payload.channel_id)
        message = await ch.fetch_message(payload.message_id)

        
        star_channel = self.client.get_channel(_star_channel)
        starboard_emoji = self.client.get_emoji(starboard_emoji_id)

        if message.channel.id == star_channel.id:
            await self.updateStat(message, starboard_emoji, downvote_init_value)
            return

        for reaction in message.reactions:
            if getattr(reaction.emoji, "id", None) == starboard_emoji_id:
                if reaction.me:
                    # check if this message is already in the starboard. If so, update it
                    async for star_message in star_channel.history(limit=500):
                        for embed in star_message.embeds:
                            if embed.footer.text == str(message.id):
                                await self.updateStat(star_message, starboard_emoji, downvote_init_value)
                                return

    @commands.Cog.listener()
    async def on_raw_message_delete(self, message_payload: discord.RawMessageDeleteEvent):
        if message_payload.guild_id in [None, self.client.staff_server_id]:
            return
        _star_channel, starboard_emoji_id = await self.client.get_guild_info(message_payload.guild_id, "starboardChannel", "starboardEmoji")
        star_channel = self.client.get_channel(_star_channel)
        starboard_emoji = self.client.get_emoji(starboard_emoji_id)

        if message_payload.message_id in messageIdMarkedForDeletion: #global variable
            # this prevents having two 'message deleted' logs for manual deletion of starboard message
            messageIdMarkedForDeletion.remove(message_payload.message_id)
            return
        if message_payload.channel_id == star_channel.id:
            # check if the deleted message is a starboard message; if so, log it at starboard message deletion
            await log_to_guild(self.client, star_channel.guild, 
                         f"{starboard_emoji} :x: Starboard message was removed (from {message_payload.message_id}) "
                         f"(Starboard message was deleted manually).")
            return
        elif message_payload.channel_id != star_channel.id:
            # check if this message's is in the starboard. If so, delete it
            async for star_message in star_channel.history(limit=300):
                for embed in star_message.embeds:
                    if embed.footer.text == str(message_payload.message_id):
                        try:
                            image = star_message.embeds[0].image.url
                        except AttributeError:
                            image = ""
                        try:
                            msg_link = str(message_payload.message_id)+"  |  "+(await self.client.get_channel(message_payload.channel_id).fetch_message(message_payload.message_id)).jump_url
                        except discord.NotFound:
                            msg_link = str(message_payload.message_id)+" (couldn't get jump link)"
                        await log_to_guild(self.client, star_channel.guild, 
                                     f"{starboard_emoji} :x: Starboard message {star_message.id} was removed (from {msg_link}) "
                                     f"(original message was removed (this starboard message's linked id matched the removed message's)). "
                                     f"Content: \"\"\"{star_message.embeds[0].description}\"\"\" and attachment: {image}")
                        messageIdMarkedForDeletion.append(star_message.id)
                        await star_message.delete()
                        return

async def setup(client):
    await client.add_cog(Starboard(client))