From d354211e0c9b441764a013e42cb5acd11ff8c7e9 Mon Sep 17 00:00:00 2001
From: Zevaryx <zevaryx@gmail.com>
Date: Sat, 23 Apr 2022 14:40:57 -0600
Subject: [PATCH] Fix phishing sync

---
 jarvis/client.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/jarvis/client.py b/jarvis/client.py
index aa4be21..2cd3fe8 100644
--- a/jarvis/client.py
+++ b/jarvis/client.py
@@ -72,13 +72,21 @@ class Jarvis(Snake):
 
         self.logger.debug(f"Found {len(data)} changes to phishing domains")
 
+        add = 0
+        sub = 0
+
         for update in data:
             if update["type"] == "add":
-                if update["domain"] not in self.phishing_domains:
-                    self.phishing_domains.append(update["domain"])
+                for domain in update["domains"]:
+                    if domain not in self.phishing_domains:
+                        add += 1
+                        self.phishing_domains.append(domain)
             elif update["type"] == "delete":
-                if update["domain"] in self.phishing_domains:
-                    self.phishing_domains.remove(update["domain"])
+                for domain in update["domains"]:
+                    if domain in self.phishing_domains:
+                        sub -= 1
+                        self.phishing_domains.remove(domain)
+        self.logger.debug(f"{add} additions, {sub} removals")
 
     async def _prerun(self, ctx: Context, *args, **kwargs) -> None:
         name = ctx.invoked_name