record server name in cliexit table (#64)

Author: jesopo

beryllia/database/cliconn.py

@@ -179,6 +179,7 @@ async def add(
         hostname: str,
         ip: Optional[Union[IPv4Address, IPv6Address]],
         reason: str,
+        server: str,
     ) -> int:
 
         query = """
@@ -192,9 +193,10 @@ async def add(
                 search_host,
                 ip,
                 reason,
+                server,
                 ts
             )
-            VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, NOW()::TIMESTAMP)
+            VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, NOW()::TIMESTAMP)
             RETURNING id
         """
         args = [
@@ -207,6 +209,7 @@ async def add(
             str(self.to_search(hostname, SearchType.HOST)),
             ip,
             reason,
+            server,
         ]
         async with self.pool.acquire() as conn:
             return await conn.fetchval(query, *args)

beryllia/parse/snote.py

@@ -135,7 +135,7 @@ async def _handle_cliexit(self, server: str, match: Match) -> None:
             cliconn_id, _ = self._cliconns.pop(nickname)
 
         await self._database.cliexit.add(
-            cliconn_id, nickname, username, hostname, ip, reason
+            cliconn_id, nickname, username, hostname, ip, reason, server
         )
 
         if not nickname in self._kline_waiting_exit:

make-database.sql

@@ -122,6 +122,7 @@ CREATE TABLE cliexit (
     search_host  VARCHAR(64)  NOT NULL,
     ip           INET,
     reason       VARCHAR(260) NOT NULL,
+    server       VARCHAR(92),
     ts           TIMESTAMP    NOT NULL
 );
 -- for retention period bulk deletion