From 1385c659cd7062b9f7a9035fec3426e344386b4e Mon Sep 17 00:00:00 2001
From: David Rapan <david@rapan.cz>
Date: Thu, 19 Dec 2024 01:42:11 +0100
Subject: [PATCH] refactor: Add control code constants  for V5 HEARTBEAT

---
 custom_components/solarman/api.py   | 4 ++--
 custom_components/solarman/const.py | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/custom_components/solarman/api.py b/custom_components/solarman/api.py
index ef70a3f..0b58822 100644
--- a/custom_components/solarman/api.py
+++ b/custom_components/solarman/api.py
@@ -57,7 +57,7 @@ async def _heartbeat_response(self, request_frame):
         v5_frame = bytearray(
             self.v5_start
             + struct.pack("<H", 10)
-            + struct.pack("<H", 0x1710)
+            + CONTROL_CODE.HEARTBEAT_RESPONSE
             + request_frame[5:6]
             + self.v5_loggerserial
             + struct.pack("<H", 0x0100)
@@ -102,7 +102,7 @@ def _received_frame_is_valid(self, frame):
         if frame[5] != self.sequence_number:
             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"):
+        if frame.startswith(self.v5_start + CONTROL_CODE.HEARTBEAT):
             self.log.debug("[%s] V5_HEARTBEAT: %s", self.serial, frame.hex(" "))
             asyncio.ensure_future(self._heartbeat_response(frame))
             return False
diff --git a/custom_components/solarman/const.py b/custom_components/solarman/const.py
index 4234c8a..c1a15b8 100644
--- a/custom_components/solarman/const.py
+++ b/custom_components/solarman/const.py
@@ -76,6 +76,8 @@
 CONTROL_CODE = types.SimpleNamespace()
 CONTROL_CODE.REQUEST = struct.pack("<H", 0x4510)
 CONTROL_CODE.RESPONSE = struct.pack("<H", 0x1510)
+CONTROL_CODE.HEARTBEAT = struct.pack("<I", 0x47100001)
+CONTROL_CODE.HEARTBEAT_RESPONSE = struct.pack("<H", 0x1710)
 
 AUTO_RECONNECT = True