From bb0e4e74aa139b428552e8f525ce63f20eae9152 Mon Sep 17 00:00:00 2001
From: Paramesh Babu <parameshbabu@users.noreply.github.com>
Date: Wed, 17 May 2023 11:39:59 -0700
Subject: [PATCH] mars switch based on provider (#121)

---
 tools/modules/AksEdgeDeploy/AksEdgeDeploy.psm1 | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tools/modules/AksEdgeDeploy/AksEdgeDeploy.psm1 b/tools/modules/AksEdgeDeploy/AksEdgeDeploy.psm1
index 17fbf5e..c7dc83a 100644
--- a/tools/modules/AksEdgeDeploy/AksEdgeDeploy.psm1
+++ b/tools/modules/AksEdgeDeploy/AksEdgeDeploy.psm1
@@ -639,7 +639,8 @@ function Invoke-AideLinuxVmShell {
     .EXAMPLE
         Invoke-AideLinuxVmShell
     #>
-    if ($aideSession.HostOS.IsServerSKU) {
+    $provider = (Get-ItemPropertyValue "HKLM:\SOFTWARE\Microsoft\AksEdge" -Name wssdProvider)
+    if ($provider -ne 'hcs') {
         $env:WSSD_CONFIG_PATH="c:\programdata\aksedge\protected\.wssd\cloudconfig"
         $LinuxVmTag="9f0ea5f3-5769-47e3-b504-2afacd1fef0f"
         $IdLine = & 'C:\Program Files\aksedge\nodectl' compute vm list --query "[?tags.keys(@).contains(@,'$LinuxVmTag')]" | Select-String -Pattern "ledge-id:"