Remove Legacy Works With Nest (#96111)

* Remove Legacy Works With Nest

* Simplify nest configuration

* Cleanup legacy nest config entries

homeassistant/components/nest/sensor.py

@@ -1,20 +1,104 @@
-"""Support for Nest sensors that dispatches between API versions."""
+"""Support for Google Nest SDM sensors."""
+from __future__ import annotations
 
+import logging
+
+from google_nest_sdm.device import Device
+from google_nest_sdm.device_manager import DeviceManager
+from google_nest_sdm.device_traits import HumidityTrait, TemperatureTrait
+
+from homeassistant.components.sensor import (
+    SensorDeviceClass,
+    SensorEntity,
+    SensorStateClass,
+)
 from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import PERCENTAGE, UnitOfTemperature
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from .const import DATA_SDM
-from .legacy.sensor import async_setup_legacy_entry
-from .sensor_sdm import async_setup_sdm_entry
+from .const import DATA_DEVICE_MANAGER, DOMAIN
+from .device_info import NestDeviceInfo
+
+_LOGGER = logging.getLogger(__name__)
+
+
+DEVICE_TYPE_MAP = {
+    "sdm.devices.types.CAMERA": "Camera",
+    "sdm.devices.types.DISPLAY": "Display",
+    "sdm.devices.types.DOORBELL": "Doorbell",
+    "sdm.devices.types.THERMOSTAT": "Thermostat",
+}
 
 
 async def async_setup_entry(
     hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
 ) -> None:
     """Set up the sensors."""
-    if DATA_SDM not in entry.data:
-        await async_setup_legacy_entry(hass, entry, async_add_entities)
-        return
 
-    await async_setup_sdm_entry(hass, entry, async_add_entities)
+    device_manager: DeviceManager = hass.data[DOMAIN][entry.entry_id][
+        DATA_DEVICE_MANAGER
+    ]
+    entities: list[SensorEntity] = []
+    for device in device_manager.devices.values():
+        if TemperatureTrait.NAME in device.traits:
+            entities.append(TemperatureSensor(device))
+        if HumidityTrait.NAME in device.traits:
+            entities.append(HumiditySensor(device))
+    async_add_entities(entities)
+
+
+class SensorBase(SensorEntity):
+    """Representation of a dynamically updated Sensor."""
+
+    _attr_should_poll = False
+    _attr_state_class = SensorStateClass.MEASUREMENT
+    _attr_has_entity_name = True
+
+    def __init__(self, device: Device) -> None:
+        """Initialize the sensor."""
+        self._device = device
+        self._device_info = NestDeviceInfo(device)
+        self._attr_unique_id = f"{device.name}-{self.device_class}"
+        self._attr_device_info = self._device_info.device_info
+
+    @property
+    def available(self) -> bool:
+        """Return the device availability."""
+        return self._device_info.available
+
+    async def async_added_to_hass(self) -> None:
+        """Run when entity is added to register update signal handler."""
+        self.async_on_remove(
+            self._device.add_update_listener(self.async_write_ha_state)
+        )
+
+
+class TemperatureSensor(SensorBase):
+    """Representation of a Temperature Sensor."""
+
+    _attr_device_class = SensorDeviceClass.TEMPERATURE
+    _attr_native_unit_of_measurement = UnitOfTemperature.CELSIUS
+
+    @property
+    def native_value(self) -> float:
+        """Return the state of the sensor."""
+        trait: TemperatureTrait = self._device.traits[TemperatureTrait.NAME]
+        # Round for display purposes because the API returns 5 decimal places.
+        # This can be removed if the SDM API issue is fixed, or a frontend
+        # display fix is added for all integrations.
+        return float(round(trait.ambient_temperature_celsius, 1))
+
+
+class HumiditySensor(SensorBase):
+    """Representation of a Humidity Sensor."""
+
+    _attr_device_class = SensorDeviceClass.HUMIDITY
+    _attr_native_unit_of_measurement = PERCENTAGE
+
+    @property
+    def native_value(self) -> int:
+        """Return the state of the sensor."""
+        trait: HumidityTrait = self._device.traits[HumidityTrait.NAME]
+        # Cast without loss of precision because the API always returns an integer.
+        return int(trait.ambient_humidity_percent)