fix: Fix ID selector to include state only entities

src/editor/components/idSelector/IdSelector.ts

@@ -57,15 +57,15 @@ export default class IdSelector {
 
     #createTargetData(): TargetData {
         const entityRegistry = haServer.getEntityRegistry();
-        const entities = haServer.getEntities();
+        const states = haServer.getEntities();
         const devices = haServer.getDevices();
         const areas = haServer.getAreas();
         const floors = haServer.getFloors();
         const labels = haServer.getLabels();
 
         if (this.#filter.length === 0) {
             return {
-                entities,
+                entities: states,
                 devices,
                 areas,
                 floors,
@@ -87,25 +87,22 @@ export default class IdSelector {
                 supported_features: supportedFeatures,
             } = filter;
 
-            for (const entity of entityRegistry) {
-                // Skip disabled entities
-                if (entity.disabled_by !== null) {
-                    continue;
-                }
+            for (const state of states) {
+                const entity = entityRegistry.find(
+                    (e) => e.entity_id === state.entity_id,
+                );
 
                 // Skip entities that are not part of the integration
-                if (integration && entity.platform !== integration) {
+                if (integration && entity?.platform !== integration) {
                     continue;
                 }
 
                 // Skip entities that are not part of the domain
-                const entityDomain = entity.entity_id.split('.')[0];
+                const entityDomain = state.entity_id.split('.')[0];
                 if (domain?.length && !domain.includes(entityDomain)) {
                     continue;
                 }
 
-                const state = haServer.getEntity(entity.entity_id);
-
                 // Skip entities that are not part of the device class
                 if (
                     deviceClass?.length &&
@@ -126,6 +123,13 @@ export default class IdSelector {
                     continue;
                 }
 
+                filteredEntities.push(state);
+
+                // Skip the rest of the checks if the entity is not part of the registry
+                if (!entity) {
+                    continue;
+                }
+
                 // Add devices that the entity is part of
                 let device: HassDevice | undefined;
                 if (entity.device_id) {
@@ -164,8 +168,6 @@ export default class IdSelector {
                         }
                     }
                 }
-
-                filteredEntities.push(state);
             }
         }