From 509e15f8ef0ac4caa0177808b6bd63f28e5c45af Mon Sep 17 00:00:00 2001
From: Gal Szkolnik <code@lksz.me>
Date: Fri, 16 Jul 2021 09:23:58 -0400
Subject: [PATCH] Add support for preserving specific devices

---
 HomeAssistant/Refresh-HADeviceRegistry.ps1 | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/HomeAssistant/Refresh-HADeviceRegistry.ps1 b/HomeAssistant/Refresh-HADeviceRegistry.ps1
index 90340ee..181eea9 100644
--- a/HomeAssistant/Refresh-HADeviceRegistry.ps1
+++ b/HomeAssistant/Refresh-HADeviceRegistry.ps1
@@ -18,12 +18,20 @@ $Count = [ordered]@{
     })
 }
 
+# make sure the variable exist
+if( -not (Test-Path variable:HAPreserveEntities) ) {
+    $HAPreserveEntities = [ordered]@{
+            Platforms = @( 'zha' )
+            Names = @()
+        }
+}
+
 $entities = $(Get-Content ./core.entity_registry | ConvertFrom-Json -Depth 10)
 $Count.Entities.BeforeData = $entities.data.entities
 $Count.Entities.Before = $entities.data.entities.count
 $entities.data.entities = $entities.data.entities |
     Where-Object {
-        ( $_.platform -match 'zha' ) -or ($_.Original_name -in $HAPreserveEntities)
+        ( $_.platform -match $HAPreserveEntities.Platforms ) -or ($_.Original_name -in $HAPreserveEntities.Names)
     }
 $Count.Entities.AfterData = $entities.data.entities