From e9974fcb008bc7330cd122b8546d80f36edf1f47 Mon Sep 17 00:00:00 2001
From: James Stringer <38541878+jamesstringerparsec@users.noreply.github.com>
Date: Sun, 12 Dec 2021 18:14:58 +0100
Subject: [PATCH] Add basic params checking

---
 CopyFilesToVM.ps1 | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/CopyFilesToVM.ps1 b/CopyFilesToVM.ps1
index 0f78d3d..8e6f209 100644
--- a/CopyFilesToVM.ps1
+++ b/CopyFilesToVM.ps1
@@ -19,6 +19,34 @@
 
 Import-Module $PSSCriptRoot\Add-VMGpuPartitionAdapterFiles.psm1
 
+Function Check-Params {
+
+$ExitReason = @()
+
+if (!(test-path $params.SourcePath)) {
+    $ExitReason += "ISO Path Invalid. Please enter a valid ISO Path in the SourcePath section of Params."
+    }
+if ($params.Username -eq $params.VMName ) {
+    $ExitReason += "Username cannot be the same as VMName."
+    }
+if (!($params.Username -match "^[a-zA-Z0-9]+$")) {
+    $ExitReason += "Username cannot contain special characters."
+    }
+if (!($params.VMName -match "^[a-zA-Z0-9]+$")) {
+    $ExitReason += "VMName cannot contain special characters."
+    }
+if (([Environment]::OSVersion.Version.Build -lt 22000) -and ($params.GPUName -ne "AUTO")) {
+    $ExitReason += "GPUName must be set to AUTO on Windows 10."
+    }
+If ($ExitReason.Count -gt 0) {
+    Write-Host "Script failed params check due to the following reasons:" -ForegroundColor DarkYellow
+    ForEach ($IndividualReason in $ExitReason) {
+        Write-Host "ERROR: $IndividualReason" -ForegroundColor RED
+        }
+    Exit
+    }
+}
+
 Function Setup-ParsecInstall {
 param(
 [string]$DriveLetter,
@@ -4244,6 +4272,8 @@ param(
     Assign-VMGPUPartitionAdapter -GPUName $GPUName -VMName $VMName -GPUResourceAllocationPercentage $GPUResourceAllocationPercentage
 }
 
+Check-Params @params
+
 New-GPUEnabledVM @params
 
 Start-VM -Name $params.VMName