Support for lock domain in esphome (#65280)

.coveragerc

@@ -319,6 +319,7 @@ omit =
     homeassistant/components/esphome/entry_data.py
     homeassistant/components/esphome/fan.py
     homeassistant/components/esphome/light.py
+    homeassistant/components/esphome/lock.py
     homeassistant/components/esphome/number.py
     homeassistant/components/esphome/select.py
     homeassistant/components/esphome/sensor.py

homeassistant/components/esphome/entry_data.py

@@ -19,6 +19,7 @@ from aioesphomeapi import (
     EntityState,
     FanInfo,
     LightInfo,
+    LockInfo,
     NumberInfo,
     SelectInfo,
     SensorInfo,
@@ -44,6 +45,7 @@ INFO_TYPE_TO_PLATFORM: dict[type[EntityInfo], str] = {
     CoverInfo: "cover",
     FanInfo: "fan",
     LightInfo: "light",
+    LockInfo: "lock",
     NumberInfo: "number",
     SelectInfo: "select",
     SensorInfo: "sensor",

homeassistant/components/esphome/lock.py

@@ -0,0 +1,87 @@
+"""Support for ESPHome locks."""
+from __future__ import annotations
+
+from typing import Any
+
+from aioesphomeapi import LockCommand, LockEntityState, LockInfo, LockState
+
+from homeassistant.components.lock import SUPPORT_OPEN, LockEntity
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import ATTR_CODE
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from . import EsphomeEntity, esphome_state_property, platform_async_setup_entry
+
+
+async def async_setup_entry(
+    hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+) -> None:
+    """Set up ESPHome switches based on a config entry."""
+    await platform_async_setup_entry(
+        hass,
+        entry,
+        async_add_entities,
+        component_key="lock",
+        info_type=LockInfo,
+        entity_type=EsphomeLock,
+        state_type=LockEntityState,
+    )
+
+
+# https://github.com/PyCQA/pylint/issues/3150 for all @esphome_state_property
+# pylint: disable=invalid-overridden-method
+
+
+class EsphomeLock(EsphomeEntity[LockInfo, LockEntityState], LockEntity):
+    """A lock implementation for ESPHome."""
+
+    @property
+    def assumed_state(self) -> bool:
+        """Return True if unable to access real state of the entity."""
+        return self._static_info.assumed_state
+
+    @property
+    def supported_features(self) -> int:
+        """Flag supported features."""
+        return SUPPORT_OPEN if self._static_info.supports_open else 0
+
+    @property
+    def code_format(self) -> str | None:
+        """Regex for code format or None if no code is required."""
+        if self._static_info.requires_code:
+            return self._static_info.code_format
+        return None
+
+    @esphome_state_property
+    def is_locked(self) -> bool | None:
+        """Return true if the lock is locked."""
+        return self._state.state == LockState.LOCKED
+
+    @esphome_state_property
+    def is_locking(self) -> bool | None:
+        """Return true if the lock is locking."""
+        return self._state.state == LockState.LOCKING
+
+    @esphome_state_property
+    def is_unlocking(self) -> bool | None:
+        """Return true if the lock is unlocking."""
+        return self._state.state == LockState.UNLOCKING
+
+    @esphome_state_property
+    def is_jammed(self) -> bool | None:
+        """Return true if the lock is jammed (incomplete locking)."""
+        return self._state.state == LockState.JAMMED
+
+    async def async_lock(self, **kwargs: Any) -> None:
+        """Lock the lock."""
+        await self._client.lock_command(self._static_info.key, LockCommand.LOCK)
+
+    async def async_unlock(self, **kwargs: Any) -> None:
+        """Unlock the lock."""
+        code = kwargs.get(ATTR_CODE, None)
+        await self._client.lock_command(self._static_info.key, LockCommand.UNLOCK, code)
+
+    async def async_open(self, **kwargs: Any) -> None:
+        """Open the door latch."""
+        await self._client.lock_command(self._static_info.key, LockCommand.OPEN)