From c1aa6783841f27f470b8a4e6d505798f0d1a44c8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mose=20M=C3=BCller?= <mosemueller@gmail.com>
Date: Wed, 27 Mar 2024 08:37:39 +0100
Subject: [PATCH] clients will now receive updates from socketio server and
 notify the observer

---
 src/pydase/client/client.py | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/src/pydase/client/client.py b/src/pydase/client/client.py
index bfc5ae3..3e82d5d 100644
--- a/src/pydase/client/client.py
+++ b/src/pydase/client/client.py
@@ -1,20 +1,34 @@
 import logging
 import time
+from typing import TYPE_CHECKING, TypedDict
 
 import socketio  # type: ignore
 
 from pydase.client.proxy_class_factory import ProxyClassFactory
+from pydase.utils.deserializer import loads
 from pydase.utils.serializer import SerializedObject
 
+if TYPE_CHECKING:
+    from pydase.client.proxy_class_factory import ProxyClass
+
 logger = logging.getLogger(__name__)
 
 
+class NotifyDataDict(TypedDict):
+    full_access_path: str
+    value: SerializedObject
+
+
+class NotifyDict(TypedDict):
+    data: NotifyDataDict
+
+
 class Client:
     def __init__(self, hostname: str, port: int):
         self.sio = socketio.Client()
         self.setup_events()
         self.proxy_class_factory = ProxyClassFactory(self.sio)
-        self.proxy = None
+        self.proxy: ProxyClass | None = None
         self.sio.connect(
             f"ws://{hostname}:{port}",
             socketio_path="/ws/socket.io",
@@ -29,5 +43,12 @@ class Client:
         def class_structure(data: SerializedObject) -> None:
             self.proxy = self.proxy_class_factory.create_proxy(data)
 
+        @self.sio.event
+        def notify(data: NotifyDict) -> None:
+            if self.proxy is not None:
+                self.proxy._notify_changed(
+                    data["data"]["full_access_path"], loads(data["data"]["value"])
+                )
+
     def disconnect(self) -> None:
         self.sio.disconnect()