From 7aedcf8338f61b8548516592942fb3e68a5b4a7a Mon Sep 17 00:00:00 2001
From: Ben Claussen <46791633+benclaussen1@users.noreply.github.com>
Date: Mon, 28 Nov 2022 15:45:19 -0500
Subject: [PATCH] Add Site parameter and validation logic

---
 .../VirtualMachine/New-NetboxVirtualMachine.ps1             | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/Functions/Virtualization/VirtualMachine/New-NetboxVirtualMachine.ps1 b/Functions/Virtualization/VirtualMachine/New-NetboxVirtualMachine.ps1
index dbab718..46945de 100644
--- a/Functions/Virtualization/VirtualMachine/New-NetboxVirtualMachine.ps1
+++ b/Functions/Virtualization/VirtualMachine/New-NetboxVirtualMachine.ps1
@@ -22,6 +22,8 @@ function New-NetboxVirtualMachine {
         [string]$Name,
 
         [Parameter(Mandatory = $true)]
+        [uint16]$Site,
+
         [uint16]$Cluster,
 
         [uint16]$Tenant,
@@ -56,6 +58,10 @@ function New-NetboxVirtualMachine {
 
     #$PSBoundParameters.Status = ValidateVirtualizationChoice -ProvidedValue $Status -VirtualMachineStatus
 
+    if ($PSBoundParameters.ContainsKey('Cluster') -and (-not $PSBoundParameters.ContainsKey('Site'))) {
+        throw "You must specify a site ID with a cluster ID"
+    }
+
     $Segments = [System.Collections.ArrayList]::new(@('virtualization', 'virtual-machines'))
 
     $URIComponents = BuildURIComponents -URISegments $Segments -ParametersDictionary $PSBoundParameters