From 085b46c995c888751ca5a1ff73391b8f56591826 Mon Sep 17 00:00:00 2001
From: Guy Driesen <guydriesen@users.noreply.github.com>
Date: Wed, 27 Dec 2023 12:38:42 +0100
Subject: [PATCH] fix: Add Home Assistant `entity_category` for various Ubisys
 D1 entities (#20393)

---
 lib/extension/homeassistant.ts | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/lib/extension/homeassistant.ts b/lib/extension/homeassistant.ts
index c1555371ca..78bb9335e2 100644
--- a/lib/extension/homeassistant.ts
+++ b/lib/extension/homeassistant.ts
@@ -602,6 +602,11 @@ export default class HomeAssistant extends Extension {
                 battery_low: {entity_category: 'diagnostic', device_class: 'battery'},
                 button_lock: {entity_category: 'config', icon: 'mdi:lock'},
                 calibration: {entity_category: 'config', icon: 'mdi:progress-wrench'},
+                capabilities_configurable_curve: {entity_category: 'diagnostic', icon: 'mdi:tune'},
+                capabilities_forward_phase_control: {entity_category: 'diagnostic', icon: 'mdi:tune'},
+                capabilities_overload_detection: {entity_category: 'diagnostic', icon: 'mdi:tune'},
+                capabilities_reactance_discriminator: {entity_category: 'diagnostic', icon: 'mdi:tune'},
+                capabilities_reverse_phase_control: {entity_category: 'diagnostic', icon: 'mdi:tune'},
                 carbon_monoxide: {device_class: 'carbon_monoxide'},
                 card: {entity_category: 'config', icon: 'mdi:clipboard-check'},
                 child_lock: {entity_category: 'config', icon: 'mdi:account-lock'},
@@ -627,6 +632,11 @@ export default class HomeAssistant extends Extension {
                 presence: {device_class: 'presence'},
                 smoke: {device_class: 'smoke'},
                 sos: {device_class: 'safety'},
+                status_capacitive_load: {entity_category: 'diagnostic', icon: 'mdi:tune'},
+                status_forward_phase_control: {entity_category: 'diagnostic', icon: 'mdi:tune'},
+                status_inductive_load: {entity_category: 'diagnostic', icon: 'mdi:tune'},
+                status_overload: {entity_category: 'diagnostic', icon: 'mdi:tune'},
+                status_reverse_phase_control: {entity_category: 'diagnostic', icon: 'mdi:tune'},
                 tamper: {device_class: 'tamper'},
                 temperature_scale: {entity_category: 'config', icon: 'mdi:temperature-celsius'},
                 test: {entity_category: 'diagnostic', icon: 'mdi:test-tube'},
@@ -702,6 +712,10 @@ export default class HomeAssistant extends Extension {
                 auto_relock_time: {entity_category: 'config', icon: 'mdi:timer'},
                 away_preset_days: {entity_category: 'config', icon: 'mdi:timer'},
                 away_preset_temperature: {entity_category: 'config', icon: 'mdi:thermometer'},
+                ballast_maximum_level: {entity_category: 'config'},
+                ballast_minimum_level: {entity_category: 'config'},
+                ballast_physical_maximum_level: {entity_category: 'diagnostic'},
+                ballast_physical_minimum_level: {entity_category: 'diagnostic'},
                 battery: {device_class: 'battery', entity_category: 'diagnostic', state_class: 'measurement'},
                 battery2: {device_class: 'battery', entity_category: 'diagnostic', state_class: 'measurement'},
                 battery_voltage: {device_class: 'voltage', entity_category: 'diagnostic', state_class: 'measurement',
@@ -764,6 +778,7 @@ export default class HomeAssistant extends Extension {
                 max_temperature_limit: {entity_category: 'config', icon: 'mdi:thermometer-high'},
                 min_temperature_limit: {entity_category: 'config', icon: 'mdi:thermometer-low'},
                 min_temperature: {entity_category: 'config', icon: 'mdi:thermometer-low'},
+                minimum_on_level: {entity_category: 'config'},
                 measurement_poll_interval: {entity_category: 'config', icon: 'mdi:clock-out'},
                 occupancy_timeout: {entity_category: 'config', icon: 'mdi:timer'},
                 pm10: {device_class: 'pm10', state_class: 'measurement'},
@@ -985,6 +1000,7 @@ export default class HomeAssistant extends Extension {
             const settableText = firstExpose.type === 'text' && firstExpose.access & ACCESS_SET;
             const lookup: {[s: string]: KeyValue} = {
                 action: {icon: 'mdi:gesture-double-tap'},
+                level_config: {entity_category: 'diagnostic'},
                 programming_mode: {icon: 'mdi:calendar-clock'},
                 program: {value_template: `{{ value_json.${firstExpose.property}|default('',true) ` +
                     `| truncate(254, True, '', 0) }}`},