Add floor

Author: piitaya

src/common/entity/compute_entity_name.ts

@@ -5,6 +5,7 @@ import { stripPrefixFromEntityName } from "./strip_prefix_from_entity_name";
 import { computeStateName } from "./compute_state_name";
 import { computeDeviceName } from "./compute_device_name";
 import { computeAreaName } from "./compute_area_name";
+import { computeFloorName } from "./compute_floor_name";
 
 export const computeEntityFullName = (
   stateObj: HassEntity,
@@ -85,3 +86,22 @@ export const computeEntityAreaName = (
 
   return area ? computeAreaName(area) : undefined;
 };
+
+export const computeEntityFloorName = (
+  stateObj: HassEntity,
+  entities: HomeAssistant["entities"],
+  devices: HomeAssistant["devices"],
+  areas: HomeAssistant["areas"],
+  floors: HomeAssistant["floors"]
+): string | undefined => {
+  const entry = entities[stateObj.entity_id] as
+    | EntityRegistryDisplayEntry
+    | undefined;
+  const device = entry?.device_id ? devices[entry?.device_id] : undefined;
+
+  const areaId = entry?.area_id || device?.area_id;
+  const area = areaId ? areas[areaId] : undefined;
+  const floor = area?.floor_id ? floors[area?.floor_id] : undefined;
+
+  return floor ? computeFloorName(floor) : undefined;
+};

src/common/entity/compute_floor_name.ts

@@ -0,0 +1,5 @@
+import { FloorRegistryEntry } from "../../data/floor_registry";
+
+export const computeFloorName = (
+  floor: FloorRegistryEntry
+): string | undefined => floor.name?.trim();

src/components/entity/ha-entity-picker.ts

@@ -9,6 +9,7 @@ import { computeDomain } from "../../common/entity/compute_domain";
 import {
   computeEntityAreaName,
   computeEntityDeviceName,
+  computeEntityFloorName,
   computeEntityFullName,
   computeEntityName,
 } from "../../common/entity/compute_entity_name";
@@ -343,6 +344,13 @@ export class HaEntityPicker extends LitElement {
       hass.devices,
       hass.areas
     );
+    const floorName = computeEntityFloorName(
+      stateObj,
+      hass.entities,
+      hass.devices,
+      hass.areas,
+      hass.floors
+    );
     const deviceName = computeEntityDeviceName(
       stateObj,
       hass.entities,
@@ -361,6 +369,7 @@ export class HaEntityPicker extends LitElement {
     const entityContext = [
       entityName !== deviceName ? deviceName : undefined,
       areaName,
+      floorName,
     ]
       .filter(Boolean)
       .join(" ⸱ ");
@@ -373,6 +382,7 @@ export class HaEntityPicker extends LitElement {
         displayedName ?? "",
         areaName ?? "",
         deviceName ?? "",
+        floorName ?? "",
       ].filter(Boolean),
       entity_name: entityName,
       entity_context: entityContext,

src/dialogs/more-info/ha-more-info-dialog.ts

@@ -22,6 +22,7 @@ import { computeDomain } from "../../common/entity/compute_domain";
 import {
   computeEntityAreaName,
   computeEntityDeviceName,
+  computeEntityFloorName,
   computeEntityName,
 } from "../../common/entity/compute_entity_name";
 import { shouldHandleRequestSelectedEvent } from "../../common/mwc/handle-request-selected-event";
@@ -301,6 +302,16 @@ export class MoreInfoDialog extends LitElement {
         )
       : "";
 
+    const floorName = stateObj
+      ? computeEntityFloorName(
+          stateObj,
+          this.hass.entities,
+          this.hass.devices,
+          this.hass.areas,
+          this.hass.floors
+        )
+      : "";
+
     const deviceName = stateObj
       ? computeEntityDeviceName(stateObj, this.hass.entities, this.hass.devices)
       : "";
@@ -309,7 +320,11 @@ export class MoreInfoDialog extends LitElement {
 
     const subtitle = this._childView?.viewTitle
       ? undefined
-      : [entityName !== deviceName ? deviceName : undefined, areaName] // Do not include device name if it's the same as entity name
+      : [
+          entityName !== deviceName ? deviceName : undefined,
+          areaName,
+          floorName,
+        ] // Do not include device name if it's the same as entity name
           .filter(Boolean)
           .join(" ⸱ ");