tasks/PowerShell/AzureDtlCreateEnvironment/task.ps1

<##################################################################################################

    Description
    ===========

    Create a Lab Environment using the provided ARM template.

    Coming soon / planned work
    ==========================

    - N/A.    

##################################################################################################>

#
# Parameters to this script file.
#

[CmdletBinding()]
param(
    [string] $ConnectedServiceName,
    [string] $LabId,
    [string] $RepositoryId,
    [string] $TemplateId,
    [string] $EnvironmentName,
    [string] $ParameterFile,
    [string] $ParameterOverrides,
    [string] $TemplateOutputVariables,
    [string] $ExportEnvironmentTemplate,
    [string] $ExportEnvironmentTemplateLocation,
    [string] $EnvironmentTemplateLocationVariable,
    [string] $EnvironmentTemplateSasTokenVariable
)

###################################################################################################

#
# Required modules.
#

Import-Module Microsoft.TeamFoundation.DistributedTask.Task.Common
Import-Module Microsoft.TeamFoundation.DistributedTask.Task.Internal

###################################################################################################

#
# PowerShell configurations
#

# NOTE: Because the $ErrorActionPreference is "Stop", this script will stop on first failure.
#       This is necessary to ensure we capture errors inside the try-catch-finally block.
$ErrorActionPreference = "Stop"

# Ensure we set the working directory to that of the script.
Push-Location $PSScriptRoot

###################################################################################################

#
# Functions used in this script.
#

.".\task-funcs.ps1"

###################################################################################################

#
# Handle all errors in this script.
#

trap
{
    # NOTE: This trap will handle all errors. There should be no need to use a catch below in this
    #       script, unless you want to ignore a specific error.

    $message = $error[0].Exception.Message
    
    if ($message) {
        Write-Error "`n$message"
    }
}

###################################################################################################

#
# Main execution block.
#

[string] $environmentResourceId = ''
[string] $environmentResourceGroupId = ''

try
{
    Write-Host 'Starting Azure DevTest Labs Create Environment Task'

    Show-InputParameters

    $parameterSet = Get-ParameterSet -templateId $TemplateId -path $ParameterFile -overrides $ParameterOverrides

    Show-TemplateParameters -templateId $TemplateId -parameters $parameterSet
    
    $environmentResourceId = New-DevTestLabEnvironment -labId $LabId -templateId $TemplateId -environmentName $EnvironmentName -environmentParameterSet $parameterSet
    $environmentResourceGroupId = Get-DevTestLabEnvironmentResourceGroupId -environmentResourceId $environmentResourceId
    
    if ([System.Xml.XmlConvert]::ToBoolean($TemplateOutputVariables))
    {
        $environmentDeploymentOutput = [hashtable] (Get-DevTestLabEnvironmentOutput -environmentResourceId $environmentResourceId)

        Write-Host "Parse Output information."

        $environmentDeploymentOutput.Keys | ForEach-Object {

            if(Test-DevTestLabEnvironmentOutputIsSecret -templateId $TemplateId -key $_) {
                Write-Host "##vso[task.setvariable variable=$_;isSecret=true;isOutput=true;]$($environmentDeploymentOutput[$_])"
            } else {
                Write-Host "##vso[task.setvariable variable=$_;isSecret=false;isOutput=true;]$($environmentDeploymentOutput[$_])"
            }   
        }

        Write-Host "Completed Output information."

        if ([System.Xml.XmlConvert]::ToBoolean($ExportEnvironmentTemplate)) {

            $EnvironmentSasToken = $environmentDeploymentOutput["$EnvironmentTemplateSasTokenVariable"]
            $EnvironmentLocation = $environmentDeploymentOutput["$EnvironmentTemplateLocationVariable"]
            
            if (-not $EnvironmentLocation -or -not $EnvironmentSasToken) {
                Write-Host "Missing Environment Location or Environment SAS token as outputs."
            }

            Write-Host "Parse Environment information."

            $tempEnvLoc = $EnvironmentLocation.Split("/")
            $storageAccountName = $tempEnvLoc[2].Split(".")[0]
            $containerName = $tempEnvLoc[3]
            $dtlPrefix = "$($tempEnvLoc[4])/$($tempEnvLoc[5])"

            #Get Environment instance storage context
            $context = New-AzureStorageContext -StorageAccountName $storageAccountName -SasToken $EnvironmentSasToken

            $blobs = Get-AzureStorageBlob -Container $containerName -Context $context -Prefix $dtlPrefix

            New-Item -ItemType Directory -Force -Path $ExportEnvironmentTemplateLocation | Out-Null
            
            Write-Host "Downloading Azure templates"
            
            foreach ($blob in $blobs) {		            
                $shortName = $($blob.Name.TrimStart($dtlPrefix))

                if ($shortName.Contains("/")) {
                    New-Item -ItemType Directory -Force -Path "$ExportEnvironmentTemplateLocation\$($shortName.Substring(0,$shortName.IndexOf("/")))"
                }

                Get-AzureStorageBlobContent `
                    -Container $containerName -Blob $blob.Name -Destination "$ExportEnvironmentTemplateLocation\$shortName" `
		            -Context $context | Out-Null
      
            }
            Write-Host "Azure RM templates stored."

        }
    }
}
finally
{
    if (-not [string]::IsNullOrWhiteSpace($environmentResourceId))
    {
        Write-Host "##vso[task.setvariable variable=environmentResourceId;isSecret=false;isOutput=true;]$environmentResourceId"
    }

    if (-not [string]::IsNullOrWhiteSpace($environmentResourceGroupId))
    {
        Write-Host "##vso[task.setvariable variable=environmentResourceGroupId;isSecret=false;isOutput=true;]$environmentResourceGroupId"
    }

    Write-Host 'Completing Azure DevTest Labs Create Environment Task'
    Pop-Location
}