From f3d8051d9567698924034b15642a4a2a24fd271d Mon Sep 17 00:00:00 2001
From: OsikTech <osiktech+github@gmail.com>
Date: Wed, 17 Aug 2022 17:01:08 +0200
Subject: [PATCH] cleanup temp files

---
 WindowsAgentAIOInstall.ps1 | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/WindowsAgentAIOInstall.ps1 b/WindowsAgentAIOInstall.ps1
index 0780f10..d9fe6e9 100644
--- a/WindowsAgentAIOInstall.ps1
+++ b/WindowsAgentAIOInstall.ps1
@@ -31,7 +31,13 @@ If (!(Test-Path "C:\Program Files\Rustdesk\RustDesk.exe")) {
 
   cd $env:Temp
 
-  Invoke-WebRequest https://github.com/rustdesk/rustdesk/releases/download/$restdesk_version/rustdesk-$restdesk_version-windows_x64.zip -Outfile rustdesk.zip
+  If ([Environment]::Is64BitOperatingSystem) {
+    $os_arch = "x64"
+  } Else {
+    $os_arch = "x32"
+  }
+
+  Invoke-WebRequest https://github.com/rustdesk/rustdesk/releases/download/$restdesk_version/rustdesk-$restdesk_version-windows_$os_arch.zip -Outfile rustdesk.zip
 
   Expand-Archive rustdesk.zip
   cd rustdesk
@@ -66,6 +72,10 @@ $urlhandler_ps1 = @"
   Remove-Item "$env:ProgramFiles\RustDesk\urlhandler.ps1" > null
 
   Start-Sleep -s 20
+
+  # Cleanup Tempfiles
+  cd $env:Temp
+  Remove-Item $env:Temp\rustdesk -Recurse
 }
 
 # Write config