From 218114d8fee9c50f4a40211c15ec4ddd993e5e28 Mon Sep 17 00:00:00 2001
From: Ben Claussen <claussen@neonet.org>
Date: Thu, 22 Jul 2021 10:48:41 -0400
Subject: [PATCH] Add Get/Set timeout functions

---
 Functions/Setup/Get-NetboxTimeout.ps1 | 13 +++++++++++++
 Functions/Setup/Set-NetboxTimeout.ps1 | 17 +++++++++++++++++
 NetboxPS.psproj                       |  2 ++
 3 files changed, 32 insertions(+)
 create mode 100644 Functions/Setup/Get-NetboxTimeout.ps1
 create mode 100644 Functions/Setup/Set-NetboxTimeout.ps1

diff --git a/Functions/Setup/Get-NetboxTimeout.ps1 b/Functions/Setup/Get-NetboxTimeout.ps1
new file mode 100644
index 0000000..7547e33
--- /dev/null
+++ b/Functions/Setup/Get-NetboxTimeout.ps1
@@ -0,0 +1,13 @@
+
+function Get-NetboxTimeout {
+    [CmdletBinding()]
+    [OutputType([uint16])]
+    param ()
+    
+    Write-Verbose "Getting Netbox Timeout"
+    if ($null -eq $script:NetboxConfig.Timeout) {
+        throw "Netbox Timeout is not set! You may set it with Set-NetboxTimeout -TimeoutSeconds [uint16]"
+    }
+    
+    $script:NetboxConfig.Timeout
+}
\ No newline at end of file
diff --git a/Functions/Setup/Set-NetboxTimeout.ps1 b/Functions/Setup/Set-NetboxTimeout.ps1
new file mode 100644
index 0000000..cee71d2
--- /dev/null
+++ b/Functions/Setup/Set-NetboxTimeout.ps1
@@ -0,0 +1,17 @@
+
+function Set-NetboxTimeout {
+    [CmdletBinding(ConfirmImpact = 'Low',
+                   SupportsShouldProcess = $true)]
+    [OutputType([uint16])]
+    param
+    (
+        [Parameter(Mandatory = $false)]
+        [ValidateRange(1, 65535)]
+        [uint16]$TimeoutSeconds = 30
+    )
+    
+    if ($PSCmdlet.ShouldProcess('Netbox Timeout', 'Set')) {
+        $script:NetboxConfig.Timeout = $TimeoutSeconds
+        $script:NetboxConfig.Timeout
+    }
+}
\ No newline at end of file
diff --git a/NetboxPS.psproj b/NetboxPS.psproj
index 60a4dff..f78ceb1 100644
--- a/NetboxPS.psproj
+++ b/NetboxPS.psproj
@@ -112,6 +112,8 @@
     <File Build="0" Shared="True" ReferenceFunction="Invoke-Set-NetboxCipherSSL_ps1" ExportFunctions="True">Functions\Setup\Set-NetboxCipherSSL.ps1</File>
     <File Build="0" Shared="True" ReferenceFunction="Invoke-Set-NetboxInvokeParams_ps1" ExportFunctions="True">Functions\Setup\Set-NetboxInvokeParams.ps1</File>
     <File Build="0" Shared="True" ReferenceFunction="Invoke-Set-NetboxUnstrustedSSL_ps1" ExportFunctions="True">Functions\Setup\Set-NetboxUnstrustedSSL.ps1</File>
+    <File Build="0" Shared="True" ReferenceFunction="Invoke-Set-NetboxTimeout_ps1" ExportFunctions="True">Functions\Setup\Set-NetboxTimeout.ps1</File>
+    <File Build="0" Shared="True" ReferenceFunction="Invoke-Get-NetboxTimeout_ps1" ExportFunctions="True">Functions\Setup\Get-NetboxTimeout.ps1</File>
   </Files>
   <StartupScript>R:\Netbox\NetboxPS\Test-Module.ps1</StartupScript>
 </Project>
\ No newline at end of file