Project initHEADmain

1 files changed, 205 insertions, 0 deletions

diff --git a/cogs/Info.py b/cogs/Info.py
new file mode 100644
index 0000000..deb0e74
--- /dev/null
+++ b/cogs/Info.py
@@ -0,0 +1,205 @@
+from discord.ext import commands, tasks
+import datetime
+import discord
+import importlib
+import utils
+importlib.reload(utils)
+
+
+class DiscordInfo(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+        # https://discordpy.readthedocs.io/en/latest/ext/tasks/
+        global cached_guild_invites
+        cached_guild_invites = {}
+        self.initial_setup.start()
+        global extension_name
+
+    @tasks.loop(count=1)
+    async def initial_setup(self):
+        global cached_guild_invites
+        for guild in self.bot.guilds:
+            try:
+                if guild.me.guild_permissions.manage_guild:
+                    cached_guild_invites[guild.id] = await self.bot.get_guild(guild.id).invites()
+
+            except discord.Forbidden as e:
+                print(f"[Info Forbidden Error] Guild: {guild.name} | ID: {guild.id} | {e}")
+
+            except discord.HTTPException as e:
+                print(f"HTTPException error during initial invite info: {e}")
+
+    @commands.group(invoke_without_command=True)
+    async def info(self, ctx):
+        """Get information about users or the Discord server."""
+        prefix = await self.bot.get_prefix(ctx.message)
+        emb = await utils.embed(ctx, f"Commands for {prefix[2]}info", "info you can get information about a user or server.")
+        emb = await utils.field(emb, f"{prefix[2]}info server", "Provides information about the Discord server.")
+        emb = await utils.field(emb, f"{prefix[2]}info user [username]", "Provides information about the given user.")
+        await ctx.send(embed=emb)
+
+    @info.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[]))
+    async def user(self, ctx, *, member: discord.Member):
+        """Tells you some info about the member."""
+        emb = await utils.embed(ctx, "", "", thumbnail=member.avatar.url)
+        emb = await utils.author(emb, member.name)
+        emb = await utils.field(emb, "Roles:", value=str(len(member.roles)-1))
+        emb = await utils.field(emb, "Created At:", value=member.created_at.strftime("%d %B %Y - %H:%M:%S"))
+        emb = await utils.field(emb, "Joined:", value=member.joined_at.strftime("%d %B %Y - %H:%M:%S"))
+        emb = await utils.field(emb, "Top Role:", value=member.top_role)
+        emb = await utils.field(emb, "Bot:", value=member.bot)
+        # emb = await utils.thumb(emb, member.avatar_url)
+        await ctx.send(embed=emb)
+
+    @info.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[]))
+    async def server(self, ctx):
+        """Tells you some info about the guild."""
+
+        emb = await utils.embed(ctx, "", "")
+        emb = await utils.field(emb, "Owner:", ctx.guild.owner)
+        emb = await utils.field(emb, "Created:", ctx.guild.created_at.strftime("%d %B %Y"))
+        emb = await utils.field(emb, "Stats:", value=f"{len(ctx.guild.categories)} categories\n"
+                                                     f"{len(ctx.guild.text_channels)} text channels\n"
+                                                     f"{len(ctx.guild.voice_channels)} voice channels.\n"
+                                                     f"{ctx.guild.member_count} members.\n"
+                                                     f"{len(ctx.guild.roles)} roles.\n"
+                                                     f"{len(ctx.guild.emojis)} emojis.\n")
+        await ctx.send(embed=emb)
+
+    @info.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[]))
+    async def bot(self, ctx):
+        """Tells you some info about the bot."""
+
+        emb = await utils.embed(ctx, "Synthy Info - bot", f"Currently serving {len(self.bot.guilds)} guilds.")
+        await ctx.send(embed=emb)
+        print(self.bot.guilds)
+
+    @commands.Cog.listener()
+    async def on_member_join(self, member):
+        global cached_guild_invites
+        current_guild_invites = []
+
+        print('Member Join Event Start')
+
+        # Obtain the current invites
+        try:
+            if member.guild.me.guild_permissions.manage_guild:
+                guild_invites = await self.bot.get_guild(member.guild.id).invites()
+            else:
+                return
+        except discord.Forbidden as e:
+            print(f"Failed to get initial invite info: {e}")
+            return
+        except discord.HTTPException as e:
+            print(f"HTTPException error during initial invite info: {e}")
+            return
+
+        for invite in guild_invites:
+            current_guild_invites.append(invite)
+
+        # Determine the difference
+        for cachedInvite, invite in [(cachedInvite, invite) for cachedInvite in cached_guild_invites[member.guild.id] for invite in current_guild_invites]:
+            print(f"Comparing {invite.code}/{invite.uses} to {cachedInvite.code}/{cachedInvite.uses}")
+            if invite.code == cachedInvite.code and invite.uses > cachedInvite.uses:
+                str_invite_used = str(invite.code)
+                str_invite_owner = str(invite.inviter.name)
+                str_invite_uses = str(invite.uses)
+                break
+
+        channel_id = await utils.sql('SELECT channel_id FROM "database1".synthy.invitedetails WHERE guild_id = %s', (member.guild.id,))
+        print(f"channel_id {channel_id}")
+        if not channel_id:
+            print(f"channel_id []")
+            return
+        else:
+            print(f"channel_id B {channel_id}")
+            obj_channel = self.bot.get_channel(channel_id[0]["channel_id"])
+
+            user_joined = f"**User Joined:** {member.display_name} ({member.mention})"
+            invite_used = f"**Invite Used:** {str_invite_used}"
+            invite_owner = f"**Invite Owner:** {str_invite_owner}"
+            invite_uses = f"**Total times used:** {str_invite_uses}"
+            joined_details = f"\n{user_joined}\n{invite_used}\n{invite_owner}\n{invite_uses}"
+
+            emb = await utils.notice("New User Joined!", joined_details)
+            await obj_channel.send(embed=emb)
+
+        cached_guild_invites[member.guild.id] = current_guild_invites
+
+    @commands.Cog.listener()
+    async def on_member_remove(self, member):
+        channel_id = await utils.sql('SELECT channel_id FROM "database1".synthy.invitedetails WHERE guild_id = %s', (member.guild.id,))
+        if channel_id == ():
+            return
+
+        obj_channel = self.bot.get_channel(channel_id[0]["channel_id"])
+
+        # print(obj_channel)
+        # print(member.display_name)
+
+        if channel_id is not None:
+            emb = await utils.notice("User has left!", f"The user {member.name} has left the server.")
+            await obj_channel.send(embed=emb)
+
+    @commands.Cog.listener()
+    async def on_guild_join(self):
+        global cached_guild_invites
+        cached_guild_invites = {}
+        for guild in self.bot.guilds:
+            try:
+                cached_guild_invites[guild.id] = await self.bot.get_guild(guild.id).invites()
+            except Exception as e:
+                print(f"[INFO]on_guild_join:{guild.id}: {e}")
+
+    @commands.Cog.listener()
+    async def on_guild_remove(self):
+        global cached_guild_invites
+        cached_guild_invites = {}
+        for guild in self.bot.guilds:
+            try:
+                cached_guild_invites[guild.id] = await self.bot.get_guild(guild.id).invites()
+            except Exception as e:
+                print(f"[INFO]on_guild_remove:{guild.id}: {e}")
+
+    @commands.group(invoke_without_command=True)
+    async def details(self, ctx):
+        """Configure join/quit notices."""
+        # Get guilds prefix
+        prefix = await self.bot.get_prefix(ctx.message)
+        chan_id = await utils.sql('SELECT channel_id FROM "database1".synthy.invitedetails WHERE guild_id = %s;', (ctx.guild.id,))
+
+        if chan_id:
+            obj_channel = self.bot.get_channel(chan_id[0]["channel_id"])
+            cur_channel = f"\nThe channel I am using is #{obj_channel.name}."
+        else:
+            cur_channel = ""
+
+        # Create message
+        emb = await utils.embed(ctx, f"Help for `{prefix[2]}details`",
+                                     "The details feature allows me to post extra info when a user joins to a " +
+                                     "channel, like who joined, what invite was used and who owns that invite.")
+        emb = await utils.field(emb, f"{prefix[2]}details set",
+                                     f"Use this command in the channel you want me to post in.{cur_channel}")
+        emb = await utils.field(emb, f"{prefix[2]}details clear",
+                                     "Use this command to stop me posting join and leave messages.")
+        await ctx.send(embed=emb)
+
+    @details.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[]))
+    async def set(self, ctx):
+        await utils.sql('INSERT INTO "database1".synthy.invitedetails (guild_id, channel_id) VALUES (%s, %s) ON CONFLICT (guild_id) DO UPDATE SET channel_id = %s;', (ctx.guild.id, ctx.channel.id, ctx.channel.id,))
+        await ctx.send(f"I will put extra details into #{ctx.channel.name}.")
+
+    @details.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[]))
+    async def clear(self, ctx):
+        await utils.sql('DELETE FROM "database1".synthy.invitedetails WHERE guild_id = %s;', (ctx.guild.id,))
+        await ctx.send(f"Channel has been cleared.")
+
+
+def setup(bot):
+    print("INFO: Loading [Info]... ", end="")
+    bot.add_cog(DiscordInfo(bot))
+    print("Done!")
+
+
+def teardown():
+    print("INFO: Unloading [Info]")