Added NSG builder support for nics

Author: Tanmay Jain

src/Farmer/Arm/Compute.fs

@@ -176,7 +176,7 @@ type NetworkInterfaceConfiguration = {
     EnableIpForwarding: bool option
     IpConfigs: IpConfiguration list
     VirtualNetwork: LinkedResource
-    NetworkSecurityGroup: ResourceId option
+    NetworkSecurityGroup: LinkedResource option
     Primary: bool
 } with
 
@@ -187,6 +187,7 @@ type NetworkInterfaceConfiguration = {
             enableIPForwarding = this.EnableIpForwarding |> Option.map box |> Option.defaultValue null
             networkSecurityGroup =
                 this.NetworkSecurityGroup
+                |> Option.map _.ResourceId
                 |> Option.map ResourceId.AsIdObject
                 |> Option.map box
                 |> Option.defaultValue null

src/Farmer/Arm/Network.fs

@@ -725,7 +725,7 @@ type NetworkInterface = {
     EnableIpForwarding: bool option
     IpConfigs: IpConfiguration list
     VirtualNetwork: LinkedResource
-    NetworkSecurityGroup: ResourceId option
+    NetworkSecurityGroup: LinkedResource option
     Primary: bool option
     Tags: Map<string, string>
 } with
@@ -752,8 +752,9 @@ type NetworkInterface = {
                         match linkedResource with
                         | Managed resId -> resId
                         | _ -> ()
-                if this.NetworkSecurityGroup.IsSome then
-                    this.NetworkSecurityGroup.Value
+                match this.NetworkSecurityGroup with
+                | Some(Managed id) -> id
+                | _ -> ()
             ]
 
             let props = {|
@@ -763,20 +764,18 @@ type NetworkInterface = {
                 ipConfigurations =
                     this.IpConfigs
                     |> List.mapi (fun index ipConfig -> ipConfig.ToArmJson(index, this.VirtualNetwork.ResourceId, true))
+                networkSecurityGroup =
+                    this.NetworkSecurityGroup
+                    |> Option.map (fun nsg -> {|
+                        id = nsg.ResourceId.ArmExpression.Eval()
+                    |})
+                    |> Option.defaultValue Unchecked.defaultof<_>
             |}
 
-            match this.NetworkSecurityGroup with
-            | None -> {|
+            {|
                 networkInterfaces.Create(this.Name, this.Location, dependsOn, this.Tags) with
                     properties = props
-              |}
-            | Some nsg -> {|
-                networkInterfaces.Create(this.Name, this.Location, dependsOn, this.Tags) with
-                    properties = {|
-                        props with
-                            networkSecurityGroup = {| id = nsg.Eval() |}
-                    |}
-              |}
+            |}
 
 type NetworkProfile = {
     Name: ResourceName

src/Farmer/Builders/Builders.NetworkInterface.fs

@@ -18,6 +18,7 @@ type NetworkInterfaceConfig = {
     SubnetName: string option
     SubnetPrefix: IPAddressCidr option
     LinkedSubnet: LinkedResource option
+    NetworkSecurityGroup: LinkedResource option
     PrivateIpAddress: AllocationMethod
     PrivateIpAddressVersion: AddressVersion
     Tags: Map<string, string>
@@ -56,7 +57,7 @@ type NetworkInterfaceConfig = {
                     IpConfigs = subnetIpConfigs
                     Primary = this.IsPrimary
                     VirtualNetwork = vnetId
-                    NetworkSecurityGroup = None
+                    NetworkSecurityGroup = this.NetworkSecurityGroup
                     Tags = this.Tags
                 }
 
@@ -102,7 +103,7 @@ type NetworkInterfaceConfig = {
                         IpConfigs = subnetIpConfigs
                         Primary = this.IsPrimary
                         VirtualNetwork = vnetId
-                        NetworkSecurityGroup = None
+                        NetworkSecurityGroup = this.NetworkSecurityGroup
                         Tags = this.Tags
                     }
                 | _ ->
@@ -120,6 +121,7 @@ type NetworkInterfaceBuilder() =
         SubnetName = None
         SubnetPrefix = None
         LinkedSubnet = None
+        NetworkSecurityGroup = None
         PrivateIpAddress = AllocationMethod.DynamicPrivateIp
         PrivateIpAddressVersion = IPv4
         Tags = Map.empty
@@ -195,6 +197,40 @@ type NetworkInterfaceBuilder() =
                     | _ -> IPv4
         }
 
+    /// Sets the network security group for network interface
+    [<CustomOperation "network_security_group">]
+    member _.NetworkSecurityGroup(state: NetworkInterfaceConfig, nsg: IArmResource) = {
+        state with
+            NetworkSecurityGroup = Some(Managed nsg.ResourceId)
+    }
+
+    member _.NetworkSecurityGroup(state: NetworkInterfaceConfig, nsg: ResourceId) = {
+        state with
+            NetworkSecurityGroup = Some(Managed nsg)
+    }
+
+    member _.NetworkSecurityGroup(state: NetworkInterfaceConfig, nsg: NsgConfig) = {
+        state with
+            NetworkSecurityGroup = Some(Managed (nsg :> IBuilder).ResourceId)
+    }
+
+    /// Links the network interface to an existing network security group.
+    [<CustomOperation "link_to_network_security_group">]
+    member _.LinkToNetworkSecurityGroup(state: NetworkInterfaceConfig, nsg: IArmResource) = {
+        state with
+            NetworkSecurityGroup = Some(Unmanaged(nsg.ResourceId))
+    }
+
+    member _.LinkToNetworkSecurityGroup(state: NetworkInterfaceConfig, nsg: ResourceId) = {
+        state with
+            NetworkSecurityGroup = Some(Unmanaged nsg)
+    }
+
+    member _.LinkToNetworkSecurityGroup(state: NetworkInterfaceConfig, nsg: NsgConfig) = {
+        state with
+            NetworkSecurityGroup = Some(Unmanaged (nsg :> IBuilder).ResourceId)
+    }
+
     interface ITaggable<NetworkInterfaceConfig> with
         member _.Add state tags = {
             state with

src/Farmer/Builders/Builders.Vm.fs

@@ -156,7 +156,7 @@ type VmConfig = {
                         Prefixes = [ this.SubnetPrefix ]
                         VirtualNetwork = Some(Managed vnet)
                         RouteTable = None
-                        NetworkSecurityGroup = nsgId |> Option.map Managed
+                        NetworkSecurityGroup = nsgId
                         Delegations = []
                         NatGateway = None
                         ServiceEndpoints = []
@@ -176,7 +176,9 @@ type VmConfig = {
         member this.ResourceId = this.ResourceId
 
         member this.BuildResources location =
-            let nsgId = this.NetworkSecurityGroup |> Option.map (fun nsg -> nsg.ResourceId)
+            let nsgId =
+                this.NetworkSecurityGroup |> Option.map (fun nsg -> (Managed nsg.ResourceId))
+
             let generatedNics = this.buildNics (location, nsgId)
 
             [

src/Farmer/Builders/Builders.VmScaleSet.fs

@@ -165,7 +165,8 @@ type VmScaleSetConfig = {
             match this.Vm with
             | None -> raiseFarmer "The 'vm_profile' must be set for the VM scale set."
             | Some vm ->
-                let nsgId = vm.NetworkSecurityGroup |> Option.map (fun lr -> lr.ResourceId)
+                let nsgId =
+                    vm.NetworkSecurityGroup |> Option.map (fun lr -> (Managed lr.ResourceId))
 
                 [
                     // The VM Scale Set

src/Tests/Network.fs

@@ -1174,6 +1174,82 @@ let tests =
                 "Incorrect subnet id for ipConfig"
         }
 
+        test "Creates basic network interface with existing vnet subnet and network security group" {
+            let deployment = arm {
+                location Location.EastUS
+
+                add_resources [
+                    nsg { name "my-nsg" }
+                    networkInterface {
+                        name "my-network-interface-with-nsg"
+                        link_to_subnet "test-subnet"
+                        link_to_vnet "test-vnet"
+                        network_security_group (networkSecurityGroups.resourceId "my-nsg")
+                    }
+                ]
+            }
+
+            let jobj = deployment.Template |> Writer.toJson |> JObject.Parse
+
+            let networkInterfaceWithNsg =
+                jobj.SelectToken "resources[?(@.type=='Microsoft.Network/networkInterfaces')]"
+
+            Expect.equal
+                (networkInterfaceWithNsg.["properties"].["networkSecurityGroup"].["id"]
+                    .ToString())
+                "[resourceId(\u0027Microsoft.Network/networkSecurityGroups\u0027, \u0027my-nsg\u0027)]"
+                "Incorrect networkSecurityGroup for networkInterface"
+
+            let networkInterfaceWithNsgDependencies =
+                jobj.SelectToken "resources[?(@.type=='Microsoft.Network/networkInterfaces')].dependsOn" :?> JArray
+
+            Expect.isNotNull networkInterfaceWithNsgDependencies "Missing dependency for networkInterface"
+
+            Expect.hasLength
+                networkInterfaceWithNsgDependencies
+                1
+                "Incorrect number of dependencies for networkInterface"
+
+            Expect.equal
+                (networkInterfaceWithNsgDependencies.[0].ToString())
+                "[resourceId(\u0027Microsoft.Network/networkSecurityGroups\u0027, \u0027my-nsg\u0027)]"
+                "Incorrect networkInterface dependencies"
+        }
+
+        test "Creates basic network interface with existing vnet subnet and existing network security group" {
+            let deployment = arm {
+                location Location.EastUS
+
+                add_resources [
+                    networkInterface {
+                        name "my-network-interface-with-existing-nsg"
+                        link_to_subnet "test-subnet"
+                        link_to_vnet "test-vnet"
+                        link_to_network_security_group (networkSecurityGroups.resourceId "my-nsg")
+                    }
+                ]
+            }
+
+            let jobj = deployment.Template |> Writer.toJson |> JObject.Parse
+
+            let networkInterfaceWithExistingNsg =
+                jobj.SelectToken "resources[?(@.type=='Microsoft.Network/networkInterfaces')]"
+
+            Expect.equal
+                (networkInterfaceWithExistingNsg.["properties"].["networkSecurityGroup"].["id"]
+                    .ToString())
+                "[resourceId(\u0027Microsoft.Network/networkSecurityGroups\u0027, \u0027my-nsg\u0027)]"
+                "Incorrect networkSecurityGroup for networkInterface"
+
+            let networkInterfaceWithExistingNsgDependencies =
+                jobj.SelectToken "resources[?(@.type=='Microsoft.Network/networkInterfaces')].dependsOn" :?> JArray
+
+            Expect.hasLength
+                networkInterfaceWithExistingNsgDependencies
+                0
+                "Incorrect number of dependencies for networkInterface"
+        }
+
         test "Creates basic network interface with existing vnet subnet and dynamic ip" {
             let deployment = arm {
                 location Location.EastUS