Support for Event entity messages (#853)

Co-authored-by: Jesse Hills <3060199+jesserockz@users.noreply.github.com>

aioesphomeapi/api.proto

@@ -1721,6 +1721,33 @@ message TimeCommandRequest {
   uint32 second = 4;
 }
 
+// ==================== EVENT ====================
+message ListEntitiesEventResponse {
+  option (id) = 107;
+  option (source) = SOURCE_SERVER;
+  option (ifdef) = "USE_EVENT";
+
+  string object_id = 1;
+  fixed32 key = 2;
+  string name = 3;
+  string unique_id = 4;
+
+  string icon = 5;
+  bool disabled_by_default = 6;
+  EntityCategory entity_category = 7;
+  string device_class = 8;
+
+  repeated string event_types = 9;
+}
+message EventResponse {
+  option (id) = 108;
+  option (source) = SOURCE_SERVER;
+  option (ifdef) = "USE_EVENT";
+
+  fixed32 key = 1;
+  string event_type = 2;
+}
+
 // ==================== VALVE ====================
 message ListEntitiesValveResponse {
   option (id) = 109;

aioesphomeapi/core.py

@@ -46,6 +46,7 @@ from .api_pb2 import (  # type: ignore
     DeviceInfoResponse,
     DisconnectRequest,
     DisconnectResponse,
+    EventResponse,
     ExecuteServiceRequest,
     FanCommandRequest,
     FanStateResponse,
@@ -66,6 +67,7 @@ from .api_pb2 import (  # type: ignore
     ListEntitiesDateResponse,
     ListEntitiesDateTimeResponse,
     ListEntitiesDoneResponse,
+    ListEntitiesEventResponse,
     ListEntitiesFanResponse,
     ListEntitiesLightResponse,
     ListEntitiesLockResponse,
@@ -374,6 +376,8 @@ MESSAGE_TYPE_TO_PROTO = {
     104: TimeStateResponse,
     105: TimeCommandRequest,
     106: VoiceAssistantAudio,
+    107: ListEntitiesEventResponse,
+    108: EventResponse,
     109: ListEntitiesValveResponse,
     110: ValveStateResponse,
     111: ValveCommandRequest,

aioesphomeapi/model.py

@@ -262,6 +262,18 @@ class CoverState(EntityState):
         return self.position == 0.0
 
 
+# ==================== EVENT ==================
+@_frozen_dataclass_decorator
+class EventInfo(EntityInfo):
+    device_class: str = ""
+    event_types: list[str] = converter_field(default_factory=list, converter=list)
+
+
+@_frozen_dataclass_decorator
+class Event(EntityState):
+    event_type: str = ""
+
+
 # ==================== FAN ====================
 @_frozen_dataclass_decorator
 class FanInfo(EntityInfo):
@@ -907,6 +919,7 @@ COMPONENT_TYPE_TO_INFO: dict[str, type[EntityInfo]] = {
     "text": TextInfo,
     "time": TimeInfo,
     "valve": ValveInfo,
+    "event": EventInfo,
 }
 
 
@@ -1269,6 +1282,7 @@ _TYPE_TO_NAME = {
     TextInfo: "text_info",
     TimeInfo: "time",
     ValveInfo: "valve",
+    EventInfo: "event",
 }
 
 

aioesphomeapi/model_conversions.py

@@ -9,6 +9,7 @@ from .api_pb2 import (  # type: ignore
     CoverStateResponse,
     DateStateResponse,
     DateTimeStateResponse,
+    EventResponse,
     FanStateResponse,
     LightStateResponse,
     ListEntitiesAlarmControlPanelResponse,
@@ -19,6 +20,7 @@ from .api_pb2 import (  # type: ignore
     ListEntitiesCoverResponse,
     ListEntitiesDateResponse,
     ListEntitiesDateTimeResponse,
+    ListEntitiesEventResponse,
     ListEntitiesFanResponse,
     ListEntitiesLightResponse,
     ListEntitiesLockResponse,
@@ -62,6 +64,8 @@ from .model import (
     DateTimeState,
     EntityInfo,
     EntityState,
+    Event,
+    EventInfo,
     FanInfo,
     FanState,
     LightInfo,
@@ -97,6 +101,7 @@ SUBSCRIBE_STATES_RESPONSE_TYPES: dict[Any, type[EntityState]] = {
     CoverStateResponse: CoverState,
     DateStateResponse: DateState,
     DateTimeStateResponse: DateTimeState,
+    EventResponse: Event,
     FanStateResponse: FanState,
     LightStateResponse: LightState,
     LockStateResponse: LockEntityState,
@@ -121,6 +126,7 @@ LIST_ENTITIES_SERVICES_RESPONSE_TYPES: dict[Any, type[EntityInfo] | None] = {
     ListEntitiesCoverResponse: CoverInfo,
     ListEntitiesDateResponse: DateInfo,
     ListEntitiesDateTimeResponse: DateTimeInfo,
+    ListEntitiesEventResponse: EventInfo,
     ListEntitiesFanResponse: FanInfo,
     ListEntitiesLightResponse: LightInfo,
     ListEntitiesLockResponse: LockInfo,

tests/test_model.py

@@ -16,6 +16,7 @@ from aioesphomeapi.api_pb2 import (
     DateStateResponse,
     DateTimeStateResponse,
     DeviceInfoResponse,
+    EventResponse,
     FanStateResponse,
     HomeassistantServiceMap,
     HomeassistantServiceResponse,
@@ -27,6 +28,7 @@ from aioesphomeapi.api_pb2 import (
     ListEntitiesCoverResponse,
     ListEntitiesDateResponse,
     ListEntitiesDateTimeResponse,
+    ListEntitiesEventResponse,
     ListEntitiesFanResponse,
     ListEntitiesLightResponse,
     ListEntitiesLockResponse,
@@ -82,6 +84,8 @@ from aioesphomeapi.model import (
     DateTimeInfo,
     DateTimeState,
     DeviceInfo,
+    Event,
+    EventInfo,
     FanInfo,
     FanState,
     HomeassistantServiceCall,
@@ -280,6 +284,8 @@ def test_api_version_ord():
         (TimeState, TimeStateResponse),
         (DateTimeInfo, ListEntitiesDateTimeResponse),
         (DateTimeState, DateTimeStateResponse),
+        (EventInfo, ListEntitiesEventResponse),
+        (Event, EventResponse),
     ],
 )
 def test_basic_pb_conversions(model, pb):