From 150bfd513d9a08583529311bf77b281354a707bd Mon Sep 17 00:00:00 2001
From: Woyken <13166802+Woyken@users.noreply.github.com>
Date: Wed, 6 Mar 2024 11:11:21 +0200
Subject: [PATCH] Few more logs to help figuring out websocket failures

---
 src/pyelectroluxocp/__init__.py        | 2 +-
 src/pyelectroluxocp/oneAppApi.py       | 7 +++++++
 src/pyelectroluxocp/webSocketClient.py | 3 +++
 3 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/pyelectroluxocp/__init__.py b/src/pyelectroluxocp/__init__.py
index 429d974..f17f773 100644
--- a/src/pyelectroluxocp/__init__.py
+++ b/src/pyelectroluxocp/__init__.py
@@ -1,6 +1,6 @@
 """Python package wrapper around Electrolux OneApp (OCP) api."""
 
 __all__ = ["OneAppApi"]
-__version__ = "0.0.14"
+__version__ = "0.0.15"
 
 from .oneAppApi import OneAppApi
diff --git a/src/pyelectroluxocp/oneAppApi.py b/src/pyelectroluxocp/oneAppApi.py
index 3305ea8..16ac91a 100644
--- a/src/pyelectroluxocp/oneAppApi.py
+++ b/src/pyelectroluxocp/oneAppApi.py
@@ -53,6 +53,7 @@ async def get_client_cred_token(self):
 
     async def connect_websocket(self, appliance_ids: list[str]):
         """Start websocket connection, listen to events"""
+        _LOGGER.debug("connect_websocket(), appliance_ids: %s", dumps(appliance_ids))
 
         async def get_websocket_headers():
             token = await self._get_formatted_user_token()
@@ -79,6 +80,7 @@ async def get_websocket_headers():
 
     async def disconnect_websocket(self):
         """Stop websocket connection"""
+        _LOGGER.debug("disconnect_websocket()")
         ws_client = await self._get_websocket_client()
         await ws_client.disconnect()
 
@@ -195,6 +197,10 @@ async def watch_for_appliance_state_updates(
         callback: Callable[[dict[str, Dict[str, Any]]], None],
     ):
         """Fetch current appliance state and watch for state changes"""
+        _LOGGER.debug(
+            "watch_for_appliance_state_updates(), appliance_ids: %s",
+            dumps(appliance_ids),
+        )
 
         def handle_websocket_response(responseData: WebSocketResponse):
             _LOGGER.debug(
@@ -305,6 +311,7 @@ async def debug_dump_everything_to_files(
 
     async def close(self) -> None:
         """Dispose session and dependencies"""
+        _LOGGER.debug("close()")
         if self._gigya_client:
             await self._gigya_client.close()
         if self._ws_client:
diff --git a/src/pyelectroluxocp/webSocketClient.py b/src/pyelectroluxocp/webSocketClient.py
index 9b4ab4b..dfe3863 100644
--- a/src/pyelectroluxocp/webSocketClient.py
+++ b/src/pyelectroluxocp/webSocketClient.py
@@ -50,6 +50,7 @@ def remove_disconnected_event_handler(self, handler: Callable[[], None]):
     async def connect(
         self, get_headers: Callable[[], Coroutine[Any, Any, dict[str, Any]]]
     ):
+        _LOGGER.debug("connect()")
         await self.disconnect()
         await self._connect(get_headers)
 
@@ -114,6 +115,8 @@ async def disconnect(self):
             await self.websocket.close()
 
     async def close(self) -> None:
+        _LOGGER.debug("close()")
+        self.retry = False
         if self.websocket is not None:
             await self.websocket.close()
         if self._client_session and self._close_session: