From 490c632597852290a6956a9a9ee603354462c7f3 Mon Sep 17 00:00:00 2001
From: Wes Carroll <wcarroll@oceanbeachlife.com>
Date: Tue, 19 Feb 2019 22:27:06 -0500
Subject: [PATCH] Set LocalSite Env var to reduce mapping API calls

---
 ZertoApiWrapper/Public/Connect-ZertoServer.ps1    | 2 ++
 ZertoApiWrapper/Public/Disconnect-ZertoServer.ps1 | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/ZertoApiWrapper/Public/Connect-ZertoServer.ps1 b/ZertoApiWrapper/Public/Connect-ZertoServer.ps1
index 653da69..2a3bea3 100644
--- a/ZertoApiWrapper/Public/Connect-ZertoServer.ps1
+++ b/ZertoApiWrapper/Public/Connect-ZertoServer.ps1
@@ -24,5 +24,7 @@ function Connect-ZertoServer {
     $results = Invoke-ZertoRestRequest -uri $uri -credential $credential -returnHeaders -body $body -method POST
     $zertoAuthorizationHeaders = @{"x-zerto-session" = $results.Headers['x-zerto-session'][0].ToString(); "Accept" = "application/json"}
     Set-Item Env:zertoAuthorizationHeaders -Value ($zertoAuthorizationHeaders | convertto-json -Compress)
+    $zertoLocalSiteInfo = Get-ZertoLocalSite
+    Set-Item Env:zertoLocalSiteInfo -Value ($zertoLocalSiteInfo | ConvertTo-Json -Compress)
     return $zertoAuthorizationHeaders
 }
diff --git a/ZertoApiWrapper/Public/Disconnect-ZertoServer.ps1 b/ZertoApiWrapper/Public/Disconnect-ZertoServer.ps1
index f4ebc3c..807739f 100644
--- a/ZertoApiWrapper/Public/Disconnect-ZertoServer.ps1
+++ b/ZertoApiWrapper/Public/Disconnect-ZertoServer.ps1
@@ -3,5 +3,6 @@ function Disconnect-ZertoServer {
     $uri = "session"
     Invoke-ZertoRestRequest -uri $uri -method DELETE
     Remove-Item Env:zertoAuthorizationHeaders
-    Remove-Item Env:zertoConnectionInformation  
+    Remove-Item Env:zertoConnectionInformation
+    Remove-Item Env:zertoLocalSiteInfo
 }