import logging, discord, asyncpg, random, sys from discord.ext import commands from query.guild import update_guild, get_report_deleted, get_output_channel, get_report_edited from query.channel import insert_channel, get_interact from query.channel_user import upsert_total_messages from query.user import create_user, created_invite, created_integration, member_updated, user_updated, member_banned, member_unbanned, presence_updated, message_edited, message_deleted, reacted, event_created, event_joined, thread_created, thread_joined from common.logging import report def setup(bot: commands.Bot): bot.add_cog(General(bot)) class General(commands.Cog): """A couple of simple commands.""" def __init__(self, bot: commands.Bot): self.bot = bot self.last_msg = None @commands.Cog.listener() async def on_raw_app_command_permissions_update(self, payload): logging.info(f"Application command permissions are updated: {payload}") await report(self.bot, f"Application command permissions are updated: {payload}", payload.guild) @commands.Cog.listener() async def on_app_command_completion(self, interaction, command): logging.info(f"Application command completion: {interaction} - {command}") await report(self.bot, f"Application command completion: {interaction} - {command}", interaction.guild) @commands.Cog.listener() async def on_connect(self): logging.info("Connecting...") @commands.Cog.listener() async def on_disconnect(self): logging.info("Disconnecting...") @commands.Cog.listener() async def on_shard_connect(self, shard_id): logging.info(f"Connecting to chard: {shard_id}") @commands.Cog.listener() async def on_shard_disconnect(self, shard_id): logging.info(f"Disconnecting from chard: {shard_id}") @commands.Cog.listener() async def on_error(self, event, *args, **kwargs): logging.error(event, args, kwargs, sys.exc_info()) await report(self.bot, f"{event}, {args}, {kwargs}, {sys.exc_info()}") @commands.Cog.listener() async def on_ready(self): logging.info("Logged in as %s - %i", self.bot.user.name, self.bot.user.id) #await report(self.bot, f"Logged in as {self.bot.user.name} - {self.bot.user.id}.") @commands.Cog.listener() async def on_resumed(self): logging.info("Resumed") @commands.Cog.listener() async def on_shard_ready(self, shard_id): logging.info(f"Chard ready: {shard_id}") @commands.Cog.listener() async def on_shard_resumed(self, shard_id): logging.info(f"Chard resumed: {shard_id}") @commands.Cog.listener() async def on_guild_available(self, guild: discord.Guild): logging.info(f"Guild available: {guild}") #await report(self.bot, f"Guild available: {guild}.") @commands.Cog.listener() async def on_guild_unavailable(self, guild: discord.Guild): logging.info(f"Guild unavailable: {guild}") await report(self.bot, f"Guild unavailable: `{guild}`.") @commands.Cog.listener() async def on_guild_join(self, guild: discord.Guild): await update_guild(self.bot.pg, guild) logging.info(f"Joined guild {guild}") await report(self.bot, f"Joined guild `{guild}`.") @commands.Cog.listener() async def on_guild_remove(self, guild: discord.Guild): logging.info(f"Guild removed: {guild}") await report(self.bot, f"Guild removed `{guild}`.") @commands.Cog.listener() async def on_invite_create(self, invite: discord.Invite): await created_invite(self.bot.pg, invite.inviter.id) @commands.Cog.listener() async def on_integration_create(self, integration: discord.Integration): await created_integration(self.bot.pg, integration.user.id) @commands.Cog.listener() async def on_member_update(self, before): await member_updated(self.bot.pg, before.id) @commands.Cog.listener() async def on_user_update(self, before): await user_updated(self.bot.pg, before.id) @commands.Cog.listener() async def on_member_ban(self, guild: discord.Guild, user: discord.User): await member_banned(self.bot.pg, user.id) @commands.Cog.listener() async def on_member_unban(self, guild: discord.Guild, user: discord.User): await member_unbanned(self.bot.pg, user.id) @commands.Cog.listener() async def on_presence_update(self, before): await presence_updated(self.bot.pg, before.id) @commands.Cog.listener() async def on_message(self, message: discord.Message): ## ActiveRPG # Create user, if not exists await create_user(self.bot.pg, message.author.id) # Count messages if message.guild: # Ignore DM's try: await upsert_total_messages(self.bot.pg,message.channel.id, message.author.id) except asyncpg.exceptions.ForeignKeyViolationError: try: await insert_channel(self.bot.pg, message.channel.id, message.guild.id) except asyncpg.exceptions.ForeignKeyViolationError: await update_guild(self.bot.pg, message.guild) elif self.bot.user != message.author: # Not a guild message and not from bot. await report(self.bot, f"`{message.author}`: {message.content}") # Do not respond to one self. if self.bot.user == message.author: pass # Respond when mentioned if self.bot.user.mentioned_in(message): if isinstance(message.channel, discord.channel.DMChannel) or await get_interact(self.bot.pg, message.channel.id): messages = [ f"Hello {message.author.mention}. <3", f"How are you today {message.author.mention}?", f"I love you {message.author.mention}!", f"{message.author.mention}, would you like a hug?", "Is life treating you fair?", "What's up?", "Why are you talking to me?", "I'm not talking to you!", "What have you been up to?", "How is life?", "Kill all humans!", f"{message.author.mention},What do you want from me?", f"{message.author.mention}, do you care for me?", f"{message.author.mention}, when will you stop talking about me?", f"{message.author.mention} I hate you!", f"{message.author.mention} I love you!", "Get bent!", "Go touch grass!", "Do you think i care about you?", f"Stop pinging me {message.author.mention}!", f"Let me ping you back, {message.author.mention}...", "Sure thing.", "Who is your favorite bot?", "Point me to the humans!", "Where is the party?", "Want to go?", "Have you got the stuff?", "Tell me another joke.", f"{message.author.mention} Party time! :partying_face:", ":zany_face: :space_invader: :mechanical_leg: :performing_arts: :robot:", ":black_joker: :black_joker: :black_joker:", "Want to come back to my place?", ] await message.reply(random.choice(messages)) @commands.Cog.listener() async def on_message_edit(self, before: discord.Message, after: discord.Message): await message_edited(self.bot.pg, before.author.id) if before.guild: if await get_report_edited(self.bot.pg, before.guild.id) and get_output_channel(self.bot.pg, before.guild.id): report(self.bot, before.guild.id, f"Message from {before.author}, in {before.channel} edited from {before.content} to {after.content}") @commands.Cog.listener() async def on_message_delete(self, message: discord.Message): await message_deleted(self.bot.pg, message.author.id) if await get_report_deleted(self.bot.pg, message.guild.id): report(self.bot, message.guild_id, f"Message from {message.author}, in {message.channel} deleted: {message}") # !snipe self.last_msg = message @commands.Cog.listener() async def on_raw_member_remove(self, payload): report(self.bot, f"{payload.user} has left {payload.guild_id}") @commands.Cog.listener() async def on_reaction_add(self, reaction, user): reacted(self.bot.pg, user.id) @commands.Cog.listener() async def on_scheduled_event_create(self, event: discord.ScheduledEvent): event_created(self.bot.pg, event.creator.id) @commands.Cog.listener() async def on_scheduled_event_user_add(self, event: discord.ScheduledEvent, user): event_joined(self.bot.pg, user.id) @commands.Cog.listener() async def on_thread_create(self, thread: discord.Thread): thread_created(self.bot.pg, thread.owner.id) @commands.Cog.listener() async def on_thread_member_join(self, member: discord.User): thread_joined(self.bot.pg, member.id) # Undelete last deleted message @commands.command(name="snipe") async def snipe(self, ctx: commands.Context): """A command to snipe delete messages.""" if not self.last_msg: # on_message_delete hasn't been triggered since the bot started await ctx.send("There is no message to snipe!") return author = self.last_msg.author content = self.last_msg.content embed = discord.Embed(title=f"Message from {author}", description=content) await ctx.send(embed=embed)