From 1f148857e670ac190ee8a0f52c480974197be665 Mon Sep 17 00:00:00 2001
From: David Rapan <david@rapan.cz>
Date: Wed, 18 Dec 2024 20:33:36 +0100
Subject: [PATCH] feat: Implemented response to V5 HEARTBEAT

---
 custom_components/solarman/__init__.py |  2 +-
 custom_components/solarman/api.py      | 45 ++++++++++++++++++++++++--
 2 files changed, 44 insertions(+), 3 deletions(-)

diff --git a/custom_components/solarman/__init__.py b/custom_components/solarman/__init__.py
index 0778ede..db585de 100644
--- a/custom_components/solarman/__init__.py
+++ b/custom_components/solarman/__init__.py
@@ -21,7 +21,7 @@
 
 _LOGGER = logging.getLogger(__name__)
 
-PLATFORMS: list[Platform] = [Platform.SENSOR, Platform.BINARY_SENSOR, Platform.SWITCH, Platform.NUMBER, Platform.SELECT, Platform.DATETIME, Platform.TIME, Platform.BUTTON]
+PLATFORMS: list[Platform] = [Platform.SENSOR, Platform.BINARY_SENSOR, Platform.NUMBER, Platform.SWITCH, Platform.BUTTON, Platform.SELECT, Platform.DATETIME, Platform.TIME]
 
 CONFIG_SCHEMA = cv.empty_config_schema(DOMAIN)
 
diff --git a/custom_components/solarman/api.py b/custom_components/solarman/api.py
index 0606e50..775d09d 100644
--- a/custom_components/solarman/api.py
+++ b/custom_components/solarman/api.py
@@ -1,5 +1,6 @@
 import time
 import errno
+import struct
 import socket
 import logging
 import asyncio
@@ -13,7 +14,7 @@
 from .const import *
 from .common import *
 from .provider import *
-from .include.pysolarmanv5 import PySolarmanV5Async
+from .include.pysolarmanv5 import PySolarmanV5Async, NoSocketAvailableError
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -52,6 +53,45 @@ async def _tcp_send_receive_frame(self, mb_request_frame):
     async def _tcp_parse_response_adu(self, mb_request_frame):
         return parse_response_adu(await self._tcp_send_receive_frame(mb_request_frame), mb_request_frame)
 
+    async def _heartbeat_response(self):
+        v5_header = bytearray(
+            self.v5_start
+            + struct.pack("<H", 10)
+            + struct.pack("<H", 0x1710)
+            + struct.pack("<H", self._get_next_sequence_number())
+            + self.v5_loggerserial
+        )
+
+        v5_payload = bytearray(
+            struct.pack("<H", 0x0100)
+            + struct.pack("<I", int(time.time()))
+            + struct.pack("<I", 0)
+        )
+
+        v5_trailer = bytearray(self.v5_checksum + self.v5_end)
+
+        v5_frame = v5_header + v5_payload + v5_trailer
+
+        v5_frame[len(v5_frame) - 2] = self._calculate_v5_frame_checksum(v5_frame)
+
+        self.log.debug("[%s] V5_HEARTBEAT RESPONSE: %s", self.serial, v5_frame.hex(" "))
+        try:
+            self.writer.write(v5_frame)
+            await self.writer.drain()
+        except AttributeError as exc:
+            raise NoSocketAvailableError("Connection already closed") from exc
+        except NoSocketAvailableError:
+            raise
+        except TimeoutError:
+            raise
+        except OSError as exc:
+            if exc.errno == errno.EHOSTUNREACH:
+                raise TimeoutError from exc
+            raise
+        except Exception as exc:
+            self.log.exception("[%s] Send/Receive error: %s", self.serial, exc)
+            raise
+
     def _received_frame_is_valid(self, frame):
         if self._passthrough:
             return True
@@ -66,7 +106,8 @@ def _received_frame_is_valid(self, frame):
             self.log.debug("[%s] V5_SEQ_NO_MISMATCH: %s", self.serial, frame.hex(" "))
             return False
         if frame.startswith(self.v5_start + b"\x01\x00\x10\x47"):
-            self.log.debug("[%s] COUNTER: %s", self.serial, frame.hex(" "))
+            self.log.debug("[%s] V5_HEARTBEAT: %s", self.serial, frame.hex(" "))
+            execute_async(self._heartbeat_response())
             return False
         return True