Start-IpsAwsDiagnosticsJob.ps1

<#
.SYNOPSIS
Start an Image Portability Service job to extract diagnostics from an AWS volume.

.DESCRIPTION
Start an Image Portability Service job to extract diagnostics from an AWS volume.

.PARAMETER CustomerId
Specifies the customer id of the Citrix customer running this command.

.PARAMETER SecureClientId
Specifies the client id of the Citrix customer's API client.

.PARAMETER SecureSecret
Specifies the client secret of the Citrix customer's API client.

.PARAMETER ResourceLocationId
Specifies the UUID of the resource location of the Citrix Connector Appliance in AWS.

.PARAMETER AwsCwSecretId
Specifies the credential wallet id for the credentials used to access AWS.

.PARAMETER AwsVolumeId
Specifies the id of the EC2 EBS volume that diagnostics will be extracted from.

.PARAMETER AwsRegion
Specifies the AWS region where the prepare will take place.

.PARAMETER AwsSubnetId
Specifies the subnet id in AWS where the prepare appliance will be created.

.PARAMETER AwsSecurityGroupIds
Specifies an array of security groups to attach to the network interface of the prepare appliance VM.

.PARAMETER SmbLocation
Specifies the SMB URI used to access the SMB server.

.PARAMETER SmbCwId
Specifies the credential wallet id for the credentials used to access the SMB server.

.PARAMETER CeLogs
If specified, logs will be collected from the compositing engine.

.PARAMETER WindowsEventLogs
If specified, Windows event logs will be colected from the image.

.PARAMETER IpsData
If specified, Image Portability Service specific data will collected from the image.

.PARAMETER Timeout
Specifies an optional time limit for the operation. If the job does not complete in less than this time it will fail with a timeout error. The default value is 7200.

.PARAMETER Tags
Specifies a hash table of string values to apply as labels to resources created by the diagnostics operation.

.PARAMETER Prefix
Specifies an optional prefix which will be prepended to the name of assets created by the diagnostics operation. The default value is 'ce'.

.PARAMETER DryRun
If specified, a test run is performed and any problems with the parameters specified are reported. No actual changes are made.

.PARAMETER Deployment
Specifies the service address to send the job request to. It defaults to api.layering.cloud.com. This can be used if necessary to send the request to a geo specific deployment such as api.eu.layering.cloud.com.

.PARAMETER LogFileDir
Specifies the path to the file to log to. The local directory is the default.

.PARAMETER LogFileName
Specifies the name of the file to log to.

.PARAMETER OverwriteLog
If specified the log file is overwritten otherwise it is appended to.

.PARAMETER Force
If specified then any existing diagnostics file is overwritten otherwise the operation will fail if the output file exists.

.INPUTS
None.

.OUTPUTS
PSCustomObject. A job description which can be consumed by the Wait-IpsJob cmdlet.

.EXAMPLE
PS> $DiagsParams = @{
        CustomerId = 'a7f4wb1example'
        SecureClientId = '7fed2a1e-1495-46b7-8fd3-5644764af395'
        SecureSecret = '9T.3Q~MGlnB6NNgpNUUWrcquVzODrdGK~eXampLe'
        ResourceLocationId = '47251663-6710-4f76-854a-2385e3fe002d'
        AwsCwSecretId = 'Aws-creds'
        AwsRegion = 'us-east-2'
        AwsSubnetId = 'subnet-d7880448508711ed0'
        AwsVolumeId = 'vol-0aa1d07ceb4c41301'
        SmbLocation = "smb://example.com/share/folder"
        SmbCwId = "smb-credential-wallet-id"
        CeLogs = $True
        WindowsEventLogs = $True
        IpsData = $True
        Prefix = 'acme'
        DryRun = $False
        Tags = @{
            MyTagName = "MyTagValue"
        }
        Timeout = 7200
        LogFileName = '.\DiagsAws.log'
    }
PS> Start-IpsAwsDiagnosticsJob @DiagsParams -Force -OverwriteLog -Verbose | Wait-IpsJob

.EXAMPLE
PS> $DiagsParams = @{
        CustomerId = 'a7f4wb1example'
        SecureClientId = '7fed2a1e-1495-46b7-8fd3-5644764af395'
        SecureSecret = '9T.3Q~MGlnB6NNgpNUUWrcquVzODrdGK~eXampLe'
        ResourceLocationId = '47251663-6710-4f76-854a-2385e3fe002d'
        AwsCwSecretId = 'Aws-creds'
        AwsRegion = 'us-east-2'
        AwsSubnetId = 'subnet-d7880448508711ed0'
        AwsVolumeId = 'vol-0aa1d07ceb4c41301'
        SmbLocation = "smb://example.com/share/folder"
        SmbCwId = "smb-credential-wallet-id"
        CeLogs = $False
        Prefix = 'acme'
        DryRun = $False
        Timeout = 7200
        LogFileName = '.\DiagsAws.log'
    }
PS> Start-IpsAwsDiagnosticsJob @DiagsParams -Force -OverwriteLog -Verbose | Wait-IpsJob
#>


Function Start-IpsAwsDiagnosticsJob
{
    Param(
        [Parameter(Mandatory = $true)]
        [string]$CustomerId,
        [Parameter(Mandatory = $true)]
        [string]$AwsCwSecretId,
        [Parameter(Mandatory = $false)]
        [string]$Deployment,
        [Parameter(Mandatory = $true)]
        [string]$ResourceLocationId,
        [Parameter(Mandatory = $true)]
        [string]$AwsRegion,
        [Parameter(Mandatory = $true)]
        [string]$AwsSubnetId,
        [Parameter(Mandatory = $false)]
        [string[]]$AwsSecurityGroupIds,
        [Parameter(Mandatory = $true)]
        [string]$AwsVolumeId,
        [Parameter(Mandatory = $true)]
        [string]$SmbLocation,
        [Parameter(Mandatory = $true)]
        [string]$SmbCwId,
        [Parameter(Mandatory = $false)]
        [bool]$UsePublicIP = $false,
        [Parameter(Mandatory = $false)]
        [string]$AssetsId,
        [Parameter(Mandatory = $false)]
        [HashTable]$Tags = @{},
        [Parameter(Mandatory = $false)]
        [int]$Timeout = 7200,
        [Parameter(Mandatory = $false)]
        [string]$Prefix = "ce",
        [Parameter(Mandatory = $false)]
        [bool]$CeLogs = $true,
        [Parameter(Mandatory = $false)]
        [bool]$WindowsEventLogs = $true,
        [Parameter(Mandatory = $false)]
        [bool]$IpsData = $true,
        [Parameter(Mandatory = $false)]
        [bool]$DryRun = $false,
        [Parameter(Mandatory = $false)]
        [string]$SecureClientId,
        [Parameter(Mandatory = $false)]
        [string]$SecureSecret,
        [Parameter(Mandatory = $false)]
        [string]$LogFileDir,
        [Parameter(Mandatory = $false)]
        [string]$LogFileName = 'DiagsAws.log',
        [Parameter(Mandatory = $false)]
        [switch]$OverwriteLog,
        [Parameter(Mandatory = $false)]
        [switch]$Force
    )

    Begin
    {
        Add-PSSnapin Citrix.*
    }
    Process
    {
        # Initialize Logger
        # Set parameter 'Verbose' by internal parameter 'VerbosePreference', since the option -Verbose is occupied by powershell cmdlet
        $Verbose = $VerbosePreference -eq 'Continue'
        LogInit $MyInvocation $LogFileDir $LogFileName $OverwriteLog $Verbose

        VersionCheck $Deployment $CustomerId

        try {
            # Authenticate to Citrix Cloud
            $parameters = AuthToCitrixCloud $CustomerId $SecureClientId $SecureSecret
            if ([string]::IsNullOrWhiteSpace($SecureClientId) -Or [string]::IsNullOrWhiteSpace($SecureSecret)) {
                $SecureClientId = $parameters.ApiKey
                $SecureSecret = $parameters.SecretKey
            }
        }
        catch {
            LogFatal "Failed to authenticate to Citrix Cloud"
        }

        try {
            Write-Host "***** Call Method: ImageDiagnosticsJob *****"
            $platformDiagsData = @{
                region = $AwsRegion
                volumeId = $AwsVolumeId
                subnetId = $AwsSubnetId
                securityGroupIds = $AwsSecurityGroupIds
                usePublicIP = $UsePublicIP
            }

            #Add default tags
            $Tags['ctx-user'] = ($env:UserName).ToLower()
            $diagsData = @{
                platform = "Aws"
                platformCredentialId = $AwsCwSecretId
                resourceLocationId = $ResourceLocationId
                outputStorageLocation = @{
                    credentialId = $SmbCwId
                    Location = $SmbLocation
                }
                CeLogs = $CeLogs
                WindowsEventLogs = $WindowsEventLogs
                IpsData = $IpsData
                tags = $Tags
                timeoutInSeconds = $Timeout
                prefix = $Prefix
                ceVmSku = $CeVmSku
                overwriteTargetFile = $Force.IsPresent
            }

            if ($AssetsId)
            {
                $diagsData['assetsId'] = $AssetsId
            }
            
            # Convert the object to JSON to use in the POST body (Note: Default depth is 2 when serializing)
            $json = ($diagsData + $platformDiagsData) | ConvertTo-Json -Depth 10
            LogIt "Aws Diagnostics POST body $json" $Verbose

            $query = @{
                "async" = $true
                "dryRun" = $DryRun
            }

            # Send the POST
            try {
                $response = Invoke-CCRestMethod -method 'Post' -deployment $Deployment -serviceRoute "images/`$diagnostics" -customerId $CustomerId -secureClientId $SecureClientId -secureSecret $SecureSecret -query $query -json $json
                $JobId = $response.id
                LogIt "Image diagnostics started with id $JobId"
            } catch {
                $JobId = "Job failed to start"
                throw "Failed to start diagnostics: $_"
            }
        }
        catch {
            LogFatal "Workflow failed: $_"
        }
        finally {
            $output = [PSCustomObject]@{
                CustomerId = $CustomerId
                Deployment = $Deployment
                JobId = $JobId
                LogFileDir = $LogFileDir
                LogFileName = $LogFileName
            }
            Write-Output $output

            # Clear credentials at end of pipeline
            if ($PSCmdlet.MyInvocation.PipelinePosition -eq $PSCmdlet.MyInvocation.PipelineLength) {
                Clear-XDCredentials
            }
        }
    }
}