"""Support for interacting with Linode nodes.""" from __future__ import annotations import logging from typing import Any import voluptuous as vol from homeassistant.components.switch import PLATFORM_SCHEMA, SwitchEntity from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from . import ( ATTR_CREATED, ATTR_IPV4_ADDRESS, ATTR_IPV6_ADDRESS, ATTR_MEMORY, ATTR_NODE_ID, ATTR_NODE_NAME, ATTR_REGION, ATTR_VCPUS, CONF_NODES, DATA_LINODE, ) _LOGGER = logging.getLogger(__name__) DEFAULT_NAME = "Node" PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( {vol.Required(CONF_NODES): vol.All(cv.ensure_list, [cv.string])} ) def setup_platform( hass: HomeAssistant, config: ConfigType, add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None, ) -> None: """Set up the Linode Node switch.""" linode = hass.data[DATA_LINODE] nodes = config[CONF_NODES] dev = [] for node in nodes: if (node_id := linode.get_node_id(node)) is None: _LOGGER.error("Node %s is not available", node) return dev.append(LinodeSwitch(linode, node_id)) add_entities(dev, True) class LinodeSwitch(SwitchEntity): """Representation of a Linode Node switch.""" def __init__(self, li, node_id): # pylint: disable=invalid-name """Initialize a new Linode sensor.""" self._linode = li self._node_id = node_id self.data = None self._attr_extra_state_attributes = {} def turn_on(self, **kwargs: Any) -> None: """Boot-up the Node.""" if self.data.status != "running": self.data.boot() def turn_off(self, **kwargs: Any) -> None: """Shutdown the nodes.""" if self.data.status == "running": self.data.shutdown() def update(self) -> None: """Get the latest data from the device and update the data.""" self._linode.update() if self._linode.data is not None: for node in self._linode.data: if node.id == self._node_id: self.data = node if self.data is not None: self._attr_is_on = self.data.status == "running" self._attr_extra_state_attributes = { ATTR_CREATED: self.data.created, ATTR_NODE_ID: self.data.id, ATTR_NODE_NAME: self.data.label, ATTR_IPV4_ADDRESS: self.data.ipv4, ATTR_IPV6_ADDRESS: self.data.ipv6, ATTR_MEMORY: self.data.specs.memory, ATTR_REGION: self.data.region.country, ATTR_VCPUS: self.data.specs.vcpus, } self._attr_name = self.data.label