AzSKPreview

3.0.3

Framework/Core/SVT/SVTIaasBase.ps1

Set-StrictMode -Version Latest
class SVTIaasBase: SVTBase
{
    hidden [PSObject] $ResourceObject;
    hidden [PSObject[]] $vNetNics;
    hidden [PSObject[]] $vNetNicsOutput;

    SVTIaasBase([string] $subscriptionId, [string] $resourceGroupName, [string] $resourceName):
        Base($subscriptionId, $resourceGroupName, $resourceName)
    {
        $this.GetResourceObject();
        $this.GetvNetNics($this.ResourceObject.Subnets);
        $this.GetvnetNicsProperties($this.vNetNics);
    }

    SVTIaasBase([string] $subscriptionId, [SVTResource] $svtResource):
        Base($subscriptionId, $svtResource)
    {
        $this.GetResourceObject();
        $this.GetvNetNics($this.ResourceObject.Subnets);
        $this.GetvnetNicsProperties($this.vNetNics);
    }

    hidden [PSObject] GetResourceObject()
    {
        if (-not $this.ResourceObject) {
            $this.ResourceObject = Get-AzureRmVirtualNetwork -ResourceGroupName $this.ResourceContext.ResourceGroupName `
                                             -Name $this.ResourceContext.ResourceName
            if(-not $this.ResourceObject)
            {
                throw ([SuppressedException]::new(("Resource '{0}' not found under Resource Group '{1}'" -f ($this.ResourceContext.ResourceName), ($this.ResourceContext.ResourceGroupName)), [SuppressedExceptionType]::InvalidOperation))
            }
        }
        return $this.ResourceObject;
    }

    hidden [PSObject[]] GetvNetNics($VNetSubnets)
    {
        if (-not $this.vNetNics)
        {
            $VNetSubnets | ForEach-Object{
                Set-Variable -Name currentsubnet -Scope Local -Value $_
                if($null -ne $currentsubnet.IpConfigurations )
                {
                        $currentsubnet.IpConfigurations | ForEach-Object{
                        Set-Variable -Name currentipconfig -Scope Local -Value $_
                        if($currentipconfig.Id.Contains("Microsoft.Network/networkInterfaces"))
                        {
                                $currentipconfig = $currentipconfig.Id.ToLower()
                                $nicresourceid =  $currentipconfig.Substring(0,$currentipconfig.LastIndexOf("ipconfigurations")-1)
                                $nic = Get-AzureRmResource -ResourceId $nicresourceid
                                $this.vNetNics += $nic
                        }
                    }
                }
            }
        }
        return $this.vNetNics;
    }

    hidden [PSObject[]] GetvnetNicsProperties($vNetNics)
    {
        if(-not $this.vNetNicsOutput)
        {
            if($null -ne $vNetNics )
            {
                $tempVNetNICS = [array]($vNetNics)
                $tempVNetNICS | ForEach-Object{
                    Set-Variable -Name nic -Scope Local -Value $_
                    Set-Variable -Name nicproperties -Scope Local -Value $_.Properties

                    $out = ""| Select-Object NICName, VMName, VMId, PrimaryStatus, NetworkSecurityGroupName,NetworkSecurityGroupId, PublicIpAddress, PrivateIpAddress,  EnableIPForwarding, IpConfigurations
                    $out.NICName = $nic.Name
                    $out.IpConfigurations = $nicproperties.IpConfigurations
                    $out.EnableIPForwarding = $nicproperties.EnableIPForwarding
                    $PublicIpAddresses = @()
                    $PrivateIpAddresses = @()
                    $nicproperties.IpConfigurations | ForEach-Object{
                        Set-Variable -Name ipconfiguration -Scope Local -Value $_
                        if(($ipconfiguration | Get-Member -Name "Properties") -and ($ipconfiguration.Properties | Get-Member -Name "PublicIpAddress") -and $ipconfiguration.Properties.PublicIpAddress)
                        {
                            $IPResource = Get-AzureRmResource -ResourceId $ipconfiguration.Properties.PublicIpAddress.Id
                            $pubResourceName = Get-AzureRmPublicIpAddress -Name $IPResource.Name -ResourceGroupName $IPResource.ResourceGroupName
                            $PublicIpAddresses += $pubResourceName.IpAddress
                        }
                        $PrivateIpAddresses += $ipconfiguration.Properties.PrivateIpAddress
                    }
                    $out.PublicIpAddress = ([System.String]::Join(";",$PublicIpAddresses))
                    $out.PrivateIpAddress = ([System.String]::Join(";",$PrivateIpAddresses))

                    if(($nicproperties | Get-Member -Name "VirtualMachine") -and $nicproperties.VirtualMachine )
                    {
                        $vmresource = Get-AzureRmResource -ResourceId $nicproperties.VirtualMachine.Id
                        $out.VMName = $vmresource.Name
                    }
                    else {
                        $out.VMName = ""
                    }
                    if($null -ne ($nicproperties | Get-Member primary))
                    {
                        $out.PrimaryStatus = $nicproperties.primary
                    }

                    if(($nicproperties | Get-Member -Name "NetworkSecurityGroup") -and $nicproperties.NetworkSecurityGroup)
                    {
                        $nsgresource = Get-AzureRmResource -ResourceId $nicproperties.NetworkSecurityGroup.Id
                        $out.NetworkSecurityGroupName = $nsgresource.Name
                    }
                    $this.vNetNicsOutput += $out
                }
                $this.vNetNicsOutput = [array]($this.vNetNicsOutput)
            }
        }
        return $this.vNetNicsOutput;
    }
}