From 3c86a5fefd04ae03cc1503aa552cb3b5e846017b Mon Sep 17 00:00:00 2001
From: Artur Borecki <me@pufereq.pl>
Date: Mon, 22 Sep 2025 21:13:32 +0200
Subject: [PATCH] feat(connector.py): add `on_message` handler function
 argument to init

---
 src/judas_client/connector.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/judas_client/connector.py b/src/judas_client/connector.py
index 4d5e774..13facef 100644
--- a/src/judas_client/connector.py
+++ b/src/judas_client/connector.py
@@ -4,6 +4,7 @@ from __future__ import annotations
 import logging as lg
 import socket
 import time
+from typing import Callable
 
 from judas_protocol import Message
 
@@ -17,6 +18,7 @@ class Connector:
         *,
         connect_timeout: float = 5.0,
         ack_timeout: float | None = None,
+        on_message: Callable[[Message], None],
     ) -> None:
         self.logger: lg.Logger = lg.getLogger(
             f"{__name__}.{self.__class__.__name__}"
@@ -35,6 +37,8 @@ class Connector:
 
         self.mac_address: str = mac_address
 
+        self.on_message: Callable[[Message], None] = on_message
+
     def _send_ack(self) -> None:
         self.logger.debug("[>] Sending ACK...")
         try: