ping pong

utils/logger.py

@@ -68,6 +68,10 @@ class IRCLogger:
         # Join the IRC channel
         self.join_channel()
 
+        # Start a thread to handle incoming messages
+        receive_thread = threading.Thread(target=self.receive_messages)
+        receive_thread.start()
+
     def send_message(self, message):
         self.socket.send(bytes("PRIVMSG {} :{}\r\n".format(self.channel, message), "UTF-8"))
 
@@ -76,16 +80,26 @@ class IRCLogger:
         self.socket.send(bytes("USER {} 0 * :{}\r\n".format(self.nickname, self.nickname), "UTF-8"))
         self.socket.send(bytes("JOIN {}\r\n".format(self.channel), "UTF-8"))
 
+    def receive_messages(self):
+        while True:
+            data = self.socket.recv(4096).decode("UTF-8")
+            if data.startswith("PING"):
+                self.socket.send(bytes("PONG {}\r\n".format(data.split()[1]), "UTF-8"))
+            else:
+                self.logger.info(data)
+
     def log_message(self, message):
         self.logger.info(message)
         self.send_message(message)
 
-#if __name__ == "__main__":
-#    # Configure logging
-#    logging.basicConfig(level=logging.INFO)#
-#
-#    server = "irc.efnet.org"
-#    port = 6667
-#    nickname = "HFLogger"
-#    channel = "#hflogs"
+if __name__ == "__main__":
+    # Configure logging
+    logging.basicConfig(level=logging.INFO)
+
+    server = "irc.efnet.org"
+    port = 6667
+    nickname = "HFLogger"
+    channel = "#hflogs"
 
+#    logger = IRCLogger(server, port, nickname, channel)
+#    logger.log_message("This is a test log message from the IRC logger.")