Add capability to filter diagnostic entities

AalianKhan · Nov 9, 2024 · 71a5442 · 71a5442
1 parent 44a20b0
commit 71a5442
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 0 deletions.
diff --git a/src/configurationDefaults.ts b/src/configurationDefaults.ts
@@ -18,6 +18,7 @@ export const configurationDefaults: StrategyDefaults = {
   domains: {
     _: {
       hide_config_entities: false,
+      hide_diagnostic_entities: false,
     },
     default: {
       title: "Miscellaneous",

diff --git a/src/mushroom-strategy.ts b/src/mushroom-strategy.ts
@@ -117,6 +117,10 @@ class MushroomStrategy extends HTMLTemplateElement {
                 Helper.strategyOptions.domains[domain ?? "_"].hide_config_entities
                 || Helper.strategyOptions.domains["_"].hide_config_entities;
 
+          let diagnosticEntityHidden =
+                Helper.strategyOptions.domains[domain ?? "_"].hide_diagnostic_entities
+                || Helper.strategyOptions.domains["_"].hide_diagnostic_entities;
+
           // Set the target for controller cards to entities without an area.
           if (area.area_id === "undisclosed") {
             target = {
@@ -188,6 +192,11 @@ class MushroomStrategy extends HTMLTemplateElement {
                 continue;
               }
 
+              // Don't include the diagnostic-entity if hidden in the strategy options.
+              if (entity.entity_category === "diagnostic" && diagnosticEntityHidden) {
+                continue;
+              }
+
               domainCards.push(new cardModule[className](entity, cardOptions).getCard());
             }
 

diff --git a/src/types/strategy/generic.ts b/src/types/strategy/generic.ts
@@ -38,11 +38,14 @@ export namespace generic {
    * @property {boolean} [hidden] True if the entity should be hidden from the dashboard.
    * @property {boolean} [hide_config_entities] True if the entity's categorie is "config" and should be hidden from the
    *                                            dashboard.
+   * @property {boolean} [hide_diagnostic_entities] True if the entity's categorie is "diagnostic" and should be hidden from the
+   *                                                dashboard.
    */
   export interface DomainConfig extends Partial<cards.ControllerCardConfig> {
     hidden?: boolean;
     order?: number;
     hide_config_entities?: boolean
+    hide_diagnostic_entities?: boolean
   }
 
   /**

diff --git a/src/views/AbstractView.ts b/src/views/AbstractView.ts
@@ -72,6 +72,9 @@ abstract class AbstractView {
     const configEntityHidden =
           Helper.strategyOptions.domains[this.#domain ?? "_"].hide_config_entities
           || Helper.strategyOptions.domains["_"].hide_config_entities;
+    const diagnosticEntityHidden =
+          Helper.strategyOptions.domains[this.#domain ?? "_"].hide_diagnostic_entities
+          || Helper.strategyOptions.domains["_"].hide_diagnostic_entities;
 
     // Create cards for each area.
     for (const area of Helper.areas) {
@@ -105,6 +108,10 @@ abstract class AbstractView {
           continue;
         }
 
+        if (entity.entity_category === "diagnostic" && diagnosticEntityHidden) {
+          continue;
+        }
+
         areaCards.push(new cardModule[className](entity, cardOptions).getCard());
       }