From 3d7572843e313f5fba5e427bb086f807f52ac675 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Sun, 16 Jan 2022 09:38:29 -1000
Subject: [PATCH] Add switch platform to nexia (#64205)

---
 .coveragerc                              |  1 +
 homeassistant/components/nexia/const.py  |  8 ++-
 homeassistant/components/nexia/switch.py | 67 ++++++++++++++++++++++++
 tests/components/nexia/test_switch.py    | 11 ++++
 4 files changed, 86 insertions(+), 1 deletion(-)
 create mode 100644 homeassistant/components/nexia/switch.py
 create mode 100644 tests/components/nexia/test_switch.py

diff --git a/.coveragerc b/.coveragerc
index 1a9a2b47803..4be1a2f9dcf 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -733,6 +733,7 @@ omit =
     homeassistant/components/neurio_energy/sensor.py
     homeassistant/components/nexia/entity.py
     homeassistant/components/nexia/climate.py
+    homeassistant/components/nexia/switch.py
     homeassistant/components/nextcloud/*
     homeassistant/components/nfandroidtv/__init__.py
     homeassistant/components/nfandroidtv/notify.py
diff --git a/homeassistant/components/nexia/const.py b/homeassistant/components/nexia/const.py
index 2c6b5195bf8..4fa3cb022f8 100644
--- a/homeassistant/components/nexia/const.py
+++ b/homeassistant/components/nexia/const.py
@@ -1,7 +1,13 @@
 """Nexia constants."""
 from homeassistant.const import Platform
 
-PLATFORMS = [Platform.SENSOR, Platform.BINARY_SENSOR, Platform.CLIMATE, Platform.SCENE]
+PLATFORMS = [
+    Platform.SENSOR,
+    Platform.BINARY_SENSOR,
+    Platform.CLIMATE,
+    Platform.SCENE,
+    Platform.SWITCH,
+]
 
 ATTRIBUTION = "Data provided by mynexia.com"
 
diff --git a/homeassistant/components/nexia/switch.py b/homeassistant/components/nexia/switch.py
new file mode 100644
index 00000000000..09bc8a3852e
--- /dev/null
+++ b/homeassistant/components/nexia/switch.py
@@ -0,0 +1,67 @@
+"""Support for Nexia switches."""
+from __future__ import annotations
+
+from typing import Any
+
+from nexia.home import NexiaHome
+from nexia.thermostat import NexiaThermostat
+from nexia.zone import NexiaThermostatZone
+
+from homeassistant.components.switch import SwitchEntity
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from .const import DOMAIN
+from .coordinator import NexiaDataUpdateCoordinator
+from .entity import NexiaThermostatZoneEntity
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up switches for a Nexia device."""
+    coordinator: NexiaDataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
+    nexia_home: NexiaHome = coordinator.nexia_home
+    entities: list[NexiaHoldSwitch] = []
+    for thermostat_id in nexia_home.get_thermostat_ids():
+        thermostat: NexiaThermostat = nexia_home.get_thermostat_by_id(thermostat_id)
+        for zone_id in thermostat.get_zone_ids():
+            zone: NexiaThermostatZone = thermostat.get_zone_by_id(zone_id)
+            entities.append(NexiaHoldSwitch(coordinator, zone))
+
+    async_add_entities(entities)
+
+
+class NexiaHoldSwitch(NexiaThermostatZoneEntity, SwitchEntity):
+    """Provides Nexia hold switch support."""
+
+    def __init__(
+        self, coordinator: NexiaDataUpdateCoordinator, zone: NexiaThermostatZone
+    ) -> None:
+        """Initialize the hold mode switch."""
+        switch_name = f"{zone.get_name()} Hold"
+        zone_id = zone.zone_id
+        super().__init__(coordinator, zone, name=switch_name, unique_id=zone_id)
+
+    @property
+    def is_on(self) -> bool:
+        """Return if the zone is in hold mode."""
+        return self._zone.is_in_permanent_hold()
+
+    @property
+    def icon(self) -> str:
+        """Return the icon for the switch."""
+        return "mdi:timer-off" if self._zone.is_in_permanent_hold() else "mdi:timer"
+
+    def turn_on(self, **kwargs: Any) -> None:
+        """Enable permanent hold."""
+        self._zone.call_permanent_hold()
+        self._signal_zone_update()
+
+    def turn_off(self, **kwargs: Any) -> None:
+        """Disable permanent hold."""
+        self._zone.call_return_to_schedule()
+        self._signal_zone_update()
diff --git a/tests/components/nexia/test_switch.py b/tests/components/nexia/test_switch.py
new file mode 100644
index 00000000000..9b6661f0d3d
--- /dev/null
+++ b/tests/components/nexia/test_switch.py
@@ -0,0 +1,11 @@
+"""The switch tests for the nexia platform."""
+
+from homeassistant.const import STATE_ON
+
+from .util import async_init_integration
+
+
+async def test_hold_switch(hass):
+    """Test creation of the hold switch."""
+    await async_init_integration(hass)
+    assert hass.states.get("switch.nick_office_hold").state == STATE_ON