Add wifi_setconfig method to configure WiFi access point (#990)

* Add wifi_setconfig method to configure WiFi access point

* fixes

* debug

* debug

* debug

* debug

* Revert "fixes"

This reverts commit 9b9f4da.

* remove workaround

Author: bdraco

aioshelly/rpc_device/device.py

@@ -45,6 +45,7 @@
     ShellyBLESetConfig,
     ShellyScript,
     ShellyScriptCode,
+    ShellyWiFiSetConfig,
     ShellyWsConfig,
     ShellyWsSetConfig,
 )
@@ -698,6 +699,27 @@ async def ble_getconfig(self) -> ShellyBLEConfig:
         """Get the BLE config with BLE.GetConfig."""
         return cast(ShellyBLEConfig, await self.call_rpc("BLE.GetConfig"))
 
+    async def wifi_setconfig(
+        self, *, ap_enable: bool | None = None
+    ) -> ShellyWiFiSetConfig:
+        """Configure WiFi settings with WiFi.SetConfig.
+
+        Args:
+            ap_enable: Whether to enable the WiFi AP
+
+        Returns:
+            Response dict, may contain "restart_required": bool
+
+        """
+        config: dict[str, Any] = {}
+        if ap_enable is not None:
+            config["ap"] = {"enable": ap_enable}
+
+        return cast(
+            ShellyWiFiSetConfig,
+            await self.call_rpc("WiFi.SetConfig", {"config": config}),
+        )
+
     async def ws_setconfig(
         self, enable: bool, server: str, ssl_ca: str = "*"
     ) -> ShellyWsSetConfig:

aioshelly/rpc_device/models.py

@@ -39,6 +39,12 @@ class ShellyBLESetConfig(TypedDict, total=False):
     restart_required: bool
 
 
+class ShellyWiFiSetConfig(TypedDict, total=False):
+    """Shelly WiFi Set Config."""
+
+    restart_required: bool
+
+
 class ShellyWsConfig(TypedDict, total=False):
     """Shelly Outbound Websocket Config."""
 

tests/rpc_device/test_device.py

@@ -537,6 +537,18 @@ async def test_ble_setconfig(rpc_device: RpcDevice) -> None:
     }
 
 
+@pytest.mark.asyncio
+async def test_wifi_setconfig(rpc_device: RpcDevice) -> None:
+    """Test RpcDevice wifi_setconfig method."""
+    await rpc_device.wifi_setconfig(ap_enable=False)
+
+    assert rpc_device.call_rpc_multiple.call_count == 1
+    assert rpc_device.call_rpc_multiple.call_args[0][0][0][0] == "WiFi.SetConfig"
+    assert rpc_device.call_rpc_multiple.call_args[0][0][0][1] == {
+        "config": {"ap": {"enable": False}}
+    }
+
+
 @pytest.mark.asyncio
 async def test_script_stop(rpc_device: RpcDevice) -> None:
     """Test RpcDevice script_stop method."""