Added nickname and IP address counting

Signed-off-by: Georg <georg@lysergic.dev>

1 files changed, 114 insertions, 21 deletions

diff --git a/plugin.py b/plugin.py
index d47440f..4d9a797 100644
--- a/plugin.py
+++ b/plugin.py
@@ -53,25 +53,60 @@ class SnoParser(callbacks.Plugin):
     """Parses the Server Notices from ErgoIRCd"""
     threaded = True
 
-    def redis_connect(self) -> redis.client.Redis:
+    def redis_connect_whois(self) -> redis.client.Redis:
         try:
-            redis_client = redis.Redis(
-                    host = self.registryValue('whois.redis.host'),
-                    port = self.registryValue('whois.redis.port'),
-                    password = self.registryValue('whois.redis.password'),
-                    username = self.registryValue('whois.redis.username'),
+            redis_client_whois = redis.Redis(
+                    host = self.registryValue('redis.host'),
+                    port = self.registryValue('redis.port'),
+                    password = self.registryValue('redis.password'),
+                    username = self.registryValue('redis.username'),
                     db = self.registryValue('whois.redis.db'),
-                    socket_timeout = int(self.registryValue('whois.redis.timeout'))
+                    socket_timeout = int(self.registryValue('redis.timeout'))
             )
-            ping = redis_client.ping()
+            ping = redis_client_whois.ping()
             if ping is True:
-                return redis_client
+                return redis_client_whois
         except redis.AuthenticationError:
             print("Could not authenticate to Redis backend.")
 
+    def redis_connect_nicks(self) -> redis.client.Redis:
+        try:
+            redis_client_nicks = redis.Redis(
+                    host = self.registryValue('redis.host'),
+                    port = self.registryValue('redis.port'),
+                    password = self.registryValue('redis.password'),
+                    username = self.registryValue('redis.username'),
+                    db = self.registryValue('redis.db1'),
+                    socket_timeout = int(self.registryValue('redis.timeout'))
+            )
+            ping = redis_client_nicks.ping()
+            if ping is True:
+                return redis_client_nicks
+        except redis.AuthenticationError:
+            print("Could not authenticate to Redis backend.")
+
+    def redis_connect_ips(self) -> redis.client.Redis:
+        try:
+            redis_client_ips = redis.Redis(
+                    host = self.registryValue('redis.host'),
+                    port = self.registryValue('redis.port'),
+                    password = self.registryValue('redis.password'),
+                    username = self.registryValue('redis.username'),
+                    db = self.registryValue('redis.db2'),
+                    socket_timeout = int(self.registryValue('redis.timeout'))
+            )
+            ping = redis_client_ips.ping()
+            if ping is True:
+                return redis_client_ips
+        except redis.AuthenticationError:
+            print("Could not authenticate to Redis backend.")
+
+
     def __init__(self, irc):
         super().__init__(irc)
-        self.redis_client = self.redis_connect()
+        self.redis_client_whois = self.redis_connect_whois()
+        self.redis_client_nicks = self.redis_connect_nicks()
+        self.redis_client_ips = self.redis_connect_ips()
 
     def whois_fresh(self, sourceip: str) -> dict:
         """Data from cache."""
@@ -96,17 +131,17 @@ class SnoParser(callbacks.Plugin):
     def whois_get_cache(self, key: str) -> str:
         """Data from Redis."""
 
-        k = self.redis_client.get(key)
+        k = self.redis_client_whois.get(key)
 
 #        self = SnoParser()
-#        val = self.redis_client.get(key)
+#        val = self.redis_client_whois.get(key)
         val = k
         return val
 
     def whois_set_cache(self, key: str, value: str) -> bool:
         """Data to Redis."""
 
-        state = self.redis_client.setex(key, timedelta(seconds=3600), value=value,)
+        state = self.redis_client_whois.setex(key, timedelta(seconds=3600), value=value,)
         return state
 
     def whois_run(self, sourceip: str) -> dict:
@@ -141,19 +176,76 @@ class SnoParser(callbacks.Plugin):
                     print(data)
             return data
 
-    def query(self, irc, msg, args, ipaddress):
+    def nick_run(self, nickname: str) -> dict:
+        """Tracks nicknames"""
+
+        data = self.redis_client_nicks.get(nickname)
+
+        if data is not None:
+            if self.registryValue('debug'):
+                print("SNOPARSER DEBUG - NICK_RUN, SEEN: TRUE")
+                print(nickname)
+                print(data)
+            self.redis_client_nicks.incrby(nickname,amount=1)
+            if data:
+                decoded = data.decode('utf-8')
+                return decoded
+            else:
+                return 0
+        else:
+            if self.registryValue('debug'):
+                print("SNOPARSER DEBUG _ NICK_RUN, SEEN: FALSE")
+                print(nickname)
+                print(data)
+            self.redis_client_nicks.set(nickname,value='1')
+            if data:
+                decoded = data.decode('utf-8')
+                return decoded
+            else:
+                return 0
+
+    def ip_run(self, ipaddress: str) -> dict:
+        """Tracks IP addresses"""
+
+        data = self.redis_client_ips.get(ipaddress)
+
+        if data is not None:
+            if self.registryValue('debug'):
+                print("SNOPARSER DEBUG - IP_RUN, SEEN: TRUE")
+                print(ipaddress)
+                print(data)
+            self.redis_client_ips.incrby(ipaddress,amount=1)
+            if data:
+                decoded = data.decode('utf-8')
+                return decoded
+            else:
+                return 0
+        else:
+            if self.registryValue('debug'):
+                print("SNOPARSER DEBUG _ IP_RUN, SEEN: FALSE")
+                print(ipaddress)
+                print(data)
+            self.redis_client_ips.set(ipaddress,value='1')
+            if data:
+                decoded = data.decode('utf-8')
+                return decoded
+            else:
+                return 0
+
+    def ipquery(self, irc, msg, args, ipaddress):
         """<IP address>
         Queries the cache for an address.
         """
 
         data = self.whois_get_cache(key=ipaddress)
         decoded = data.decode('utf-8')
-        ttl = self.redis_client.ttl(ipaddress)
+        ttl = self.redis_client_whois.ttl(ipaddress)
+        count = self.redis_client_ips.get(ipaddress)
 
         print('SnoParser manual query: ', data, ' ', ttl)
-        irc.reply(f'{decoded} - Remaining: {ttl}s')
+        irc.reply(f'{decoded} - Count: {count} - Remaining: {ttl}s')
         
-    query = wrap(query, ['anything'])
+    ipquery = wrap(ipquery, ['anything'])
 
 
     def doNotice(self, irc, msg):
@@ -173,13 +265,14 @@ class SnoParser(callbacks.Plugin):
                 else:
                     ip = couple.group(4)
                 realname = couple.group(5)
-                ip_seen = 0
-                nick_seen = 0
 
-                whoisout = self.whois_run(sourceip=ip)
+                ip_seen = self.ip_run(ipaddress=ip)
+                nick_seen = self.nick_run(nickname=nickname)
+                whois = self.whois_run(sourceip=ip)
+
                 DictFromSnotice = {'notice': 'connect', 'nickname': nickname, 'username': username, 'host': host, 'ip': ip, 'realname': realname, 'ipCount': ip_seen, 'nickCount': nick_seen}
                 #repl = f"\x02\x1FNOTICE: {DictFromSnotice['notice']} \x0F\x11\x0303==>>\x0F \x02Nick:\x0F {DictFromSnotice['nickname']} \x02Username:\x0F {DictFromSnotice['username']} \x02Hostname:\x0F {DictFromSnotice['host']} \x02IP:\x0F {DictFromSnotice['ip']} \x02Realname:\x0F {DictFromSnotice['realname']} \x02IPcount:\x0F {DictFromSnotice['ipCount']} \x02NickCount:\x0F {DictFromSnotice['nickCount']}"
-                repl = f"\x02\x1F{DictFromSnotice['notice']} \x0F\x11\x0303==>>\x0F \x02Nick:\x0F {DictFromSnotice['nickname']} \x02Username:\x0F {DictFromSnotice['username']} \x02Hostname:\x0F {DictFromSnotice['host']} \x02IP:\x0F {DictFromSnotice['ip']} {whoisout} \x02Realname:\x0F {DictFromSnotice['realname']} \x02IPcount:\x0F {DictFromSnotice['ipCount']} \x02NickCount:\x0F {DictFromSnotice['nickCount']}"
+                repl = f"\x02\x1F{DictFromSnotice['notice']} \x0F\x11\x0303==>>\x0F \x02Nick:\x0F {DictFromSnotice['nickname']} \x02Username:\x0F {DictFromSnotice['username']} \x02Hostname:\x0F {DictFromSnotice['host']} \x02IP:\x0F {DictFromSnotice['ip']} {whois} \x02Realname:\x0F {DictFromSnotice['realname']} \x02IPcount:\x0F {DictFromSnotice['ipCount']} \x02NickCount:\x0F {DictFromSnotice['nickCount']}"
 
                 self._sendSnotice(irc, msg, repl)
             if 'XLINE' in text and 'temporary' in text: