feat: Add mDNS/LLMNR properties Network API (#173)

Author: t3hk0d3

aiohasupervisor/models/__init__.py

@@ -99,6 +99,7 @@
     IPv4Config,
     IPv6,
     IPv6Config,
+    MulticastDnsMode,
     NetworkInfo,
     NetworkInterface,
     NetworkInterfaceConfig,
@@ -235,6 +236,7 @@
     "MountUsage",
     "MountsInfo",
     "MountsOptions",
+    "MulticastDnsMode",
     "NFSMountRequest",
     "NFSMountResponse",
     "NetworkInfo",

aiohasupervisor/models/network.py

@@ -49,6 +49,15 @@ class AuthMethod(StrEnum):
     WPA_PSK = "wpa-psk"
 
 
+class MulticastDnsMode(StrEnum):
+    """Multicast DNS working mode."""
+
+    DEFAULT = "default"
+    OFF = "off"
+    RESOLVE = "resolve"
+    ANNOUNCE = "announce"
+
+
 # --- OBJECTS ----
 
 
@@ -110,6 +119,8 @@ class NetworkInterface(ResponseData):
     ipv6: IPv6
     wifi: Wifi | None
     vlan: Vlan | None
+    mdns: MulticastDnsMode
+    llmnr: MulticastDnsMode
 
 
 @dataclass(frozen=True, slots=True)
@@ -169,6 +180,8 @@ class NetworkInterfaceConfig(Options):
     ipv4: IPv4Config | None = None
     ipv6: IPv6Config | None = None
     wifi: WifiConfig | None = None
+    mdns: MulticastDnsMode | None = None
+    llmnr: MulticastDnsMode | None = None
     enabled: bool | None = None
 
 
@@ -196,3 +209,5 @@ class VlanConfig(Options):
 
     ipv4: IPv4Config | None = None
     ipv6: IPv6Config | None = None
+    mdns: MulticastDnsMode | None = None
+    llmnr: MulticastDnsMode | None = None

tests/fixtures/network_info.json

@@ -24,7 +24,9 @@
           "ready": true
         },
         "wifi": null,
-        "vlan": null
+        "vlan": null,
+        "mdns": "default",
+        "llmnr": "default"
       }
     ],
     "docker": {

tests/fixtures/network_interface_info.json

@@ -22,6 +22,8 @@
       "ready": true
     },
     "wifi": null,
-    "vlan": null
+    "vlan": null,
+    "mdns": "announce",
+    "llmnr": "announce"
   }
 }

tests/test_network.py

@@ -10,6 +10,7 @@
 from aiohasupervisor.models import (
     InterfaceMethod,
     IPv4Config,
+    MulticastDnsMode,
     NetworkInterfaceConfig,
     VlanConfig,
 )
@@ -45,6 +46,8 @@ async def test_network_info(
     assert result.interfaces[0].ipv6.gateway is None
     assert result.interfaces[0].wifi is None
     assert result.interfaces[0].vlan is None
+    assert result.interfaces[0].mdns is MulticastDnsMode.DEFAULT
+    assert result.interfaces[0].llmnr is MulticastDnsMode.DEFAULT
 
     assert result.docker.interface == "hassio"
     assert result.docker.address.compressed == "172.30.32.0/23"
@@ -89,6 +92,8 @@ async def test_network_interface_info(
     assert result.ipv6.gateway is None
     assert result.wifi is None
     assert result.vlan is None
+    assert result.mdns is MulticastDnsMode.ANNOUNCE
+    assert result.llmnr is MulticastDnsMode.ANNOUNCE
 
 
 async def test_network_update_interface(
@@ -102,13 +107,25 @@ async def test_network_update_interface(
             address=[IPv4Interface("192.168.1.2/24")],
             gateway=IPv4Address("192.168.1.1"),
             nameservers=[IPv4Address("192.168.1.1")],
-        )
+        ),
+        mdns=MulticastDnsMode.OFF,
     )
     assert (
         await supervisor_client.network.update_interface("end0", config=config) is None
     )
-    assert responses.requests.keys() == {
-        ("POST", URL(f"{SUPERVISOR_URL}/network/interface/end0/update"))
+
+    request_key = ("POST", URL(f"{SUPERVISOR_URL}/network/interface/end0/update"))
+    assert responses.requests.keys() == {request_key}
+
+    assert len(responses.requests[request_key]) == 1
+    assert responses.requests[request_key][0].kwargs["json"] == {
+        "ipv4": {
+            "address": ["192.168.1.2/24"],
+            "gateway": "192.168.1.1",
+            "method": "static",
+            "nameservers": ["192.168.1.1"],
+        },
+        "mdns": "off",
     }
 
 
@@ -132,7 +149,11 @@ async def test_network_access_points(
 
 @pytest.mark.parametrize(
     "config",
-    [None, NetworkInterfaceConfig(ipv4=IPv4Config(method=InterfaceMethod.AUTO))],
+    [
+        None,
+        NetworkInterfaceConfig(ipv4=IPv4Config(method=InterfaceMethod.AUTO)),
+        NetworkInterfaceConfig(mdns=MulticastDnsMode.OFF),
+    ],
 )
 async def test_network_save_vlan(
     responses: aioresponses,