allow filtering by entity_category (#17)

itsteddyyo · Sep 16, 2024 · 404d769 · 404d769
2 parents cda9286 + 8f1f835
commit 404d769
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/util/filter.ts b/src/util/filter.ts
@@ -137,4 +137,8 @@ export const filterValue: Record<FilterType, (entity: EntityRegistryEntry, hass:
             const hiddenBy = entity.hidden_by;
             return compare(comparator, hiddenBy, value);
         },
+        entity_category: (entity, hass, value, comparator) => {
+            const entityCategory = entity.entity_category;
+            return compare(comparator, entityCategory, value);
+        },
     };
diff --git a/src/util/types.ts b/src/util/types.ts
@@ -132,6 +132,16 @@ export enum FilterType {
      * ```
      */
     hidden_by = "hidden_by",
+    /**
+     * Filter on the category of the entity.
+     * @example
+     * ```yaml
+     * - type: entity_category
+     *   comparator: equal
+     *   value: diagnostic
+     * ```
+     */
+    entity_category = "entity_category",
 }
 
 export enum Comparator {