Add GetDeploymentBySlot for networking cmdlets

Author: Ed Munoz

src/ServiceManagement/Network/Commands.Network/Commands.Network.csproj

@@ -1,4 +1,4 @@
-<?xml version="1.0" encoding="utf-8"?>
+<?xml version="1.0" encoding="utf-8"?>
 <Project ToolsVersion="4.0" DefaultTargets="Build" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
   <PropertyGroup>
     <Configuration Condition=" '$(Configuration)' == '' ">Debug</Configuration>
@@ -199,6 +199,8 @@
     <Compile Include="NetworkSecurityGroup\Utilities\NetworkSecurityGroupConfigurationBaseCmdlet.cs" />
     <Compile Include="..\..\Compute\Commands.ServiceManagement\Model\IPersistentVM.cs" />
     <Compile Include="..\..\Compute\Commands.ServiceManagement\Model\PersistentVM.cs" />
+    <Compile Include="..\..\Compute\Commands.ServiceManagement\Model\PersistentVMRoleContext.cs" />
+    <Compile Include="..\..\Compute\Commands.ServiceManagement\Model\ServiceOperationContext.cs" />
     <Compile Include="Properties\Resources.Designer.cs">
       <AutoGen>True</AutoGen>
       <DesignTime>True</DesignTime>

src/ServiceManagement/Network/Commands.Network/NetworkClient.cs

@@ -14,26 +14,27 @@
 
 namespace Microsoft.Azure.Commands.Network
 {
+    using Gateway.Model;
+    using Hyak.Common;
+    using Microsoft.Azure.Common.Authentication;
+    using Microsoft.Azure.Common.Authentication.Models;
+    using Microsoft.WindowsAzure.Commands.ServiceManagement.Model;
+    using Microsoft.WindowsAzure.Management.Compute;
+    using NetworkSecurityGroup.Model;
+    using Routes.Model;
     using System;
     using System.Collections.Generic;
     using System.Linq;
     using System.Management.Automation;
-    using Gateway.Model;
-    using NetworkSecurityGroup.Model;
-    using Routes.Model;
+    using System.Security.Cryptography.X509Certificates;
     using WindowsAzure.Commands.Common.Storage;
     using WindowsAzure.Commands.Utilities.Common;
     using WindowsAzure.Management;
     using WindowsAzure.Management.Network;
     using WindowsAzure.Management.Network.Models;
     using WindowsAzure.Storage.Auth;
-    using Microsoft.Azure.Common.Authentication.Models;
-    using Microsoft.Azure.Common.Authentication;
-    using Hyak.Common;
-    using System.Security.Cryptography.X509Certificates;
-    using PowerShellAppGwModel = ApplicationGateway.Model;
-    using Microsoft.WindowsAzure.Management.Compute;
     using ComputeModels = Microsoft.WindowsAzure.Management.Compute.Models;
+    using PowerShellAppGwModel = ApplicationGateway.Model;
 
     public class NetworkClient
     {
@@ -836,5 +837,46 @@ public void SetNetworkSecurityGroupForSubnet(string networkSecurityGroupName, st
 
             client.NetworkSecurityGroups.AddToSubnet(virtualNetworkName, subnetName, parameters);
         }
+
+        public string GetDeploymentBySlot(string serviceName, string slot)
+        {
+            if (string.IsNullOrEmpty(serviceName))
+            {
+                throw new ArgumentNullException(serviceName);
+            }
+
+            if (string.IsNullOrEmpty(slot))
+            {
+                throw new ArgumentNullException(slot);
+            }
+
+            var slotType = (ComputeModels.DeploymentSlot)Enum.Parse(typeof(ComputeModels.DeploymentSlot), slot, true);
+
+            return this.computeClient.Deployments.GetBySlot(
+                        serviceName,
+                        slotType).Name;
+        }
+
+        public string GetDeploymentName(IPersistentVM vm, string slot, string serviceName)
+        {
+            string deploymentName = null;
+            var vmRoleContext = vm as PersistentVMRoleContext;
+            if (vmRoleContext != null)
+            {
+                deploymentName = vmRoleContext.DeploymentName;
+            }
+
+            if (string.IsNullOrEmpty(slot) && string.IsNullOrEmpty(deploymentName))
+            {
+                slot = DeploymentSlotType.Production;
+            }
+
+            if (string.IsNullOrEmpty(deploymentName))
+            {
+                deploymentName = this.GetDeploymentBySlot(serviceName, slot);
+            }
+
+            return deploymentName;
+        }
     }
 }