Add malicious url block tracker

jarvis/client.py

@@ -44,6 +44,7 @@ from nafftrack.client import StatsClient
 from pastypy import AsyncPaste as Paste
 
 from jarvis import const
+from jarvis.tracking import malicious_tracker
 from jarvis.utils import build_embed
 from jarvis.utils.embeds import warning_embed
 
@@ -605,6 +606,10 @@ class Jarvis(StatsClient):
                 embed = warning_embed(message.author, "Phishing URL")
                 await message.channel.send(embeds=embed)
                 await message.delete()
+                tracker = malicious_tracker.labels(
+                    guild_id=message.guild.id, guild_name=message.gulid.name
+                )
+                tracker.inc()
                 return True
         return False
 
@@ -637,6 +642,10 @@ class Jarvis(StatsClient):
                     embed = warning_embed(message.author, reasons)
                     await message.channel.send(embeds=embed)
                     await message.delete()
+                    tracker = malicious_tracker.labels(
+                        guild_id=message.guild.id, guild_name=message.gulid.name
+                    )
+                    tracker.inc()
                     return True
         return False
 

jarvis/tracking.py

@@ -0,0 +1,8 @@
+"""JARVIS Prometheus extra tracking."""
+from prometheus_client import Counter
+
+malicious_tracker = Counter(
+    "jarvis_malicious_blocked",
+    "Amount of malicious messages blocked",
+    labelnames=["guild_id", "guild_name"],
+)