Implement voiceloger

plugin/discord/cog.py

@@ -0,0 +1,106 @@
+import discord
+from discord.ext import commands as cmds
+from discord import app_commands as appcmds
+
+from meta import LionBot, LionCog, LionContext
+from meta.logger import log_wrap
+from utils.lib import utc_now
+
+from . import logger
+from ..data import VoiceLogData, VoiceLogGuild
+from .lib import ThreadedWebhook
+
+
+class VoiceLogCog(LionCog):
+    def __init__(self, bot: LionBot):
+        self.bot = bot
+        self.data = bot.db.load_registry(VoiceLogData())
+
+    async def cog_load(self):
+        await self.data.init()
+
+    @LionCog.listener("on_voice_state_update")
+    @log_wrap(action="Voice Log Update")
+    async def voicelog_update(
+        self,
+        member: discord.Member,
+        before: discord.VoiceState,
+        after: discord.VoiceState,
+    ):
+        if member.bot:
+            return
+
+        after_channel = after.channel
+        before_channel = before.channel
+        if after_channel == before.channel:
+            return
+
+        row = await VoiceLogGuild.fetch(member.guild.id)
+        if row is None or row.webhook_url is None:
+            return
+
+        hook = ThreadedWebhook.from_url(logger.webhook_url, client=self.bot)
+
+        embed = discord.Embed(timestamp=utc_now())
+
+        if before_channel is None:
+            embed.title = "User Joined Voice Channel"
+            embed.description = f"{member.mention} joined {after_channel.mention}"
+        elif after_channel is None:
+            embed.title = "User Left Voice Channel"
+            embed.description = f"{member.mention} left {before_channel.mention}"
+        else:
+            embed.title = "User Switched Voice Channel"
+            embed.description = f"{member.mention} moved from {before_channel.mention} to {after_channel.mention}"
+
+        await hook.send(embed=embed)
+
+    @cmds.hybrid_group(
+        name="voicelog", description="Base command group for the voice logging system"
+    )
+    @appcmds.default_permissions(manage_guild=True)
+    async def voicelog_group(self, ctx: LionContext): ...
+
+    @voicelog_group.command(
+        name="enable",
+        description="Enable voice activity logging and set the webhook url to use.",
+    )
+    async def voicelog_enable(self, ctx: LionContext, webhook_url: str):
+        if not ctx.guild:
+            return
+        if not ctx.author.guild_permissions.manage_guild:
+            return
+
+        webhook = ThreadedWebhook.from_url(webhook_url, client=self.bot)
+        try:
+            embed = discord.Embed(
+                title="Testing",
+                description="Testing logging webhook, feel free to delete.",
+            )
+            await webhook.send(embed=embed, wait=True)
+            existing = await VoiceLogGuild.fetch(ctx.guild.id)
+            if existing:
+                await existing.update(webhook_url=webhook_url)
+            else:
+                await VoiceLogGuild.create(
+                    guildid=ctx.guild.id, webhook_url=webhook_url
+                )
+
+            await ctx.reply("Voice logging enabled!")
+        except discord.HTTPException:
+            await ctx.error_reply("Could not post to the given webhook!")
+
+    @voicelog_group.command(
+        name="disable", description="Disable voice activity logging in this server."
+    )
+    async def voicelog_disable(self, ctx: LionContext):
+        if not ctx.guild:
+            return
+        if not ctx.author.guild_permissions.manage_guild:
+            return
+
+        await self.data.voicelog_guilds.update_where(guildid=ctx.guild.id).set(
+            webhook_url=None
+        )
+
+        await ctx.reply("Voice Activity logging disabled.")