From d4e3d17393e4cd17fd475a8e9852d1ff552c5c61 Mon Sep 17 00:00:00 2001
From: Zevaryx <zevaryx@gmail.com>
Date: Wed, 2 Feb 2022 20:33:58 -0700
Subject: [PATCH] Re-add Mute tracking

---
 jarvis/cogs/admin/mute.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/jarvis/cogs/admin/mute.py b/jarvis/cogs/admin/mute.py
index ae1e0f8..73aa774 100644
--- a/jarvis/cogs/admin/mute.py
+++ b/jarvis/cogs/admin/mute.py
@@ -11,6 +11,7 @@ from dis_snek.models.snek.application_commands import (
     slash_option,
 )
 
+from jarvis.db.models import Mute
 from jarvis.utils import build_embed
 from jarvis.utils.permissions import admin_or_permissions
 
@@ -72,6 +73,14 @@ class MuteCog(Scale):
             return
 
         await user.timeout(communication_disabled_until=duration, reason=reason)
+        _ = Mute(
+            user=user.id,
+            reason=reason,
+            admin=ctx.author.id,
+            guild=ctx.guild.id,
+            duration=duration,
+            active=True,
+        ).save()
 
         embed = build_embed(
             title="User Muted",
@@ -113,3 +122,10 @@ class MuteCog(Scale):
         embed.set_thumbnail(url=user.avatar_url)
         embed.set_footer(text=f"{user.username}#{user.discriminator} | {user.id}")
         await ctx.send(embed=embed)
+        embed.set_author(
+            name=user.display_name,
+            icon_url=user.avatar_url,
+        )
+        embed.set_thumbnail(url=user.avatar_url)
+        embed.set_footer(text=f"{user.username}#{user.discriminator} | {user.id}")
+        await ctx.send(embed=embed)