diff --git a/jarvis/client.py b/jarvis/client.py index 386349f..985c0df 100644 --- a/jarvis/client.py +++ b/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 diff --git a/jarvis/tracking.py b/jarvis/tracking.py new file mode 100644 index 0000000..4b4e1b0 --- /dev/null +++ b/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"], +)