From 39e934c7f34755d654d91ce08caeb40f064fd458 Mon Sep 17 00:00:00 2001
From: suaveolent <suaveolent@users.noreply.github.com>
Date: Thu, 14 Mar 2024 15:47:00 +0100
Subject: [PATCH] implement identify dtu and inverter

---
 README.md                 | 10 +++++++---
 hoymiles_wifi/__main__.py | 28 ++++++++++++++++++++++++++++
 hoymiles_wifi/hoymiles.py |  2 +-
 3 files changed, 36 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index f4baabc..db08460 100644
--- a/README.md
+++ b/README.md
@@ -43,6 +43,8 @@ commands:
     get-information-data,
     get-version-info,
     heartbeat,
+    identify-dtu,
+    identify-inverters
 
 The `--as-json` option is optional and allows formatting the output as JSON.
 ```
@@ -73,9 +75,11 @@ else:
 - `async_set_power_limit(power_limit)`: Set the power limit of the inverter (0-100%)
 - `async_set_wifi(wifi_ssid, wifi_password)`: Configure the wifi network
 - `async_firmware_update()`: Update to latest firmware
-- `async_restart`: Restart the inverter
-- `async_turn_on`: Turn the inverter on
-- `async_turn_off`: Turn the inverter off
+- `async_restart_dtu`: Restart the DTU
+- `async_turn_on_inverter`: Turn the inverter on
+- `async_turn_off_inverter`: Turn the inverter off
+- `async_get_information_data`: Retrieve information data
+- `async_heartbeat`: Request a heartbeat message from the DTU
 
 ## Note
 
diff --git a/hoymiles_wifi/__main__.py b/hoymiles_wifi/__main__.py
index b99cd38..d2cecc9 100644
--- a/hoymiles_wifi/__main__.py
+++ b/hoymiles_wifi/__main__.py
@@ -14,8 +14,11 @@ from hoymiles_wifi.const import DTU_FIRMWARE_URL_00_01_11, MAX_POWER_LIMIT
 from hoymiles_wifi.dtu import DTU
 from hoymiles_wifi.hoymiles import (
     generate_dtu_version_string,
+    generate_inverter_serial_number,
     generate_sw_version_string,
     generate_version_string,
+    get_dtu_model_name,
+    get_inverter_model_name,
 )
 from hoymiles_wifi.protobuf import (
     AppGetHistPower_pb2,
@@ -267,6 +270,27 @@ async def async_heatbeat(dtu: DTU) -> APPHeartbeatPB_pb2.APPHeartbeatResDTO | No
     return await dtu.async_heartbeat()
 
 
+async def async_identify_dtu(dtu: DTU) -> str:
+    """Identify the DTU asynchronously."""
+
+    real_data = await async_get_real_data_new(dtu)
+    return get_dtu_model_name(real_data.device_serial_number)
+
+
+async def async_identify_inverters(dtu: DTU) -> list[str]:
+    """Identify the DTU asynchronously."""
+
+    inverter_models = []
+    real_data = await async_get_real_data_new(dtu)
+    if real_data:
+        for sgs_data in real_data.sgs_data:
+            serial_number = generate_inverter_serial_number(sgs_data.serial_number)
+            inverter_model = get_inverter_model_name(serial_number)
+            inverter_models.append(inverter_model)
+
+    return inverter_models
+
+
 def print_invalid_command(command: str) -> None:
     """Print an invalid command message."""
 
@@ -306,6 +330,8 @@ async def main() -> None:
             "get-information-data",
             "get-version-info",
             "heartbeat",
+            "identify-dtu",
+            "identify-inverters",
         ],
         help="Command to execute",
     )
@@ -331,6 +357,8 @@ async def main() -> None:
         "get-information-data": async_get_information_data,
         "get-version-info": async_get_version_info,
         "heartbeat": async_heatbeat,
+        "identify-dtu": async_identify_dtu,
+        "identify-inverters": async_identify_inverters,
     }
 
     command_func = switch.get(args.command, print_invalid_command)
diff --git a/hoymiles_wifi/hoymiles.py b/hoymiles_wifi/hoymiles.py
index 70be4f1..8ebe122 100644
--- a/hoymiles_wifi/hoymiles.py
+++ b/hoymiles_wifi/hoymiles.py
@@ -245,7 +245,7 @@ def get_inverter_power(serial_bytes: bytes) -> InverterPower:
     return power
 
 
-def get_hw_model_name(serial_number: str) -> str:
+def get_inverter_model_name(serial_number: str) -> str:
     """Get hardware model name."""
 
     if serial_number == "22069994886948":