Public/Recipes.ps1



# Recipes
Function Get-TMRecipe {
    [CmdletBinding()]
    param(
        [Parameter(Mandatory = $false)]
        [PSObject]$TMSession = 'Default',

        [Parameter(Mandatory = $false)]
        [String]$Name,

        [Parameter(Mandatory = $false)]
        [Switch]$ResetIDs,

        [Parameter(Mandatory = $false)]
        [String]$SaveCodePath,

        [Parameter(Mandatory = $false)]
        [Switch]$AsJson,

        [Parameter(Mandatory = $false)]
        [Switch]$Passthru

    )

    begin {
        # Get the session configuration
        $ExistingProgressPreference = $ProgressPreference
        $ProgressPreference = 'SilentlyContinue'
        Write-Verbose 'Checking for cached TMSession'
        $TMSession = Get-TMSession $TMSession
        Write-Debug 'TMSession:'
        Write-Debug ($TMSession | ConvertTo-Json -Depth 5)
        if (-not $TMSession) {
            throw "TMSession '$TMSession' not found. Use New-TMSession command before using features."
        }
    }
    process {

        # Format the uri
        $uri = "https://$($TMSession.TMServer)/tdstm/ws/cookbook/recipe/list?archived=n&context=All"

        # Honor SSL Settings
        $TMCertSettings = @{SkipCertificateCheck = $TMSession.AllowInsecureSSL }

        try {
            $response = Invoke-WebRequest -Method Get -Uri $uri -WebSession $TMSession.TMWebSession @TMCertSettings
        } catch {
            return $_
        }

        if ($response.StatusCode -in @(200, 204)) {
            $Result = ($response.Content | ConvertFrom-Json).data.list
        } else {
            return 'Unable to collect Recipes.'
        }

        ## Get each recipe's Source Code in the list
        for ($i = 0; $i -lt $Result.Count; $i++) {

            ## Get the Recipe in standard format
            $uri = "https://$($TMSession.TMServer)/tdstm/ws/cookbook/recipe/$($Result[$i].recipeId)"
            try {
                $response = Invoke-WebRequest -Method Get -Uri $uri -WebSession $TMSession.TMWebSession @TMCertSettings
            } catch {
                return $_
            }
            if ($response.StatusCode -in @(200, 204)) {
                $Result[$i] = ($response.Content | ConvertFrom-Json).data
            } else {
                return 'Unable to collect Recipe.'
            }

            ## Get the Recipe in JSON Format
            $uri = "https://$($TMSession.TMServer)/tdstm/ws/cookbook/recipe/$($Result[$i].recipeId)?format=json"
            try {
                $response = Invoke-WebRequest -Method Get -Uri $uri -WebSession $TMSession.TMWebSession @TMCertSettings
            } catch {
                return $_
            }

            if ($response.StatusCode -in @(200, 204)) {
                $RecipeData = ($response.Content | ConvertFrom-Json).data
                Add-Member -InputObject $Result[$i] -NotePropertyName 'sourceCodeJson' -NotePropertyValue $RecipeData.sourceCode
            } else {
                return 'Unable to collect Recipe.'
            }
        }

        if ($ResetIDs) {
            for ($i = 0; $i -lt $Result.Count; $i++) {

                ## Null the Recipe ID and Event
                $Result[$i].recipeId = $null
                if ($Result[$i].context.eventId) { $Result[$i].context.eventId = $null }
            }
        }

        ## Return the details
        if ($Name) {
            $Result = $Result | Where-Object { $_.name -eq $Name }
        } else {
            $Result = $Result
        }

        ## Save the Code Files to a folder
        if ($SaveCodePath) {

            ## Get a FieldToLabel map to build string replacements
            $LabelReplacements = [System.Collections.ArrayList]@()
            $FieldToLabelMap = Get-TMFieldToLabelMap -TMSession $TMSession

            ## Create a replacement token for each asset/label = field set
            foreach ($DomainClass in @('DEVICE', 'APPLICATION', 'DATABASE', 'STORAGE' )) {

                foreach ($FieldName in $FieldToLabelMap.$DomainClass.Keys) {
                    if ($FieldName -match 'custom\d+') {

                        # Write-Host $FieldName
                        [void]($LabelReplacements.Add([pscustomobject]@{
                                    DomainClass = $DomainClass
                                    FieldLabel  = $FieldToLabelMap.$DomainClass.$FieldName
                                    FieldName   = $FieldName
                                }
                            )
                        )
                    }
                }
            }

            ## Save Each of the Script Source Data
            foreach ($Item in $Result) {

                ## When using ResetIDs, custom field numbers will be converted to a "custom5|DEVICE|DSID-vCenter" Format
                if ($ResetIDs) {

                    ## Replace all 'customN' items with a codestring to replace the correct items in the offline package
                    $SourceCodeLines = $Item.sourceCode -split "`r`n" -split "`r" -split "`n"
                    $UpdatedSourceCodeLines = [System.Collections.ArrayList]::new()
                    foreach ($SourceCodeLine in $SourceCodeLines) {
                        $CurrentLine = $SourceCodeLine
                        $LabelReplacements | ForEach-Object {
                            if ($CurrentLine -match $_.FieldName) {
                                $CurrentLine = $CurrentLine -replace $_.FieldName, "customN|$($_.DomainClass)|$($_.FieldLabel)|"
                            }
                        }
                        [void]($UpdatedSourceCodeLines.Add($CurrentLine))
                    }
                    $Item.sourceCode = $UpdatedSourceCodeLines -join "`r`n"


                    ## Replace all customN in the JSON Data
                    $SourceCodeLines = ($Item.sourceCodeJson | ConvertTo-Json -Depth 10) -split "`r`n"
                    $UpdatedSourceCodeLines = [System.Collections.ArrayList]::new()
                    foreach ($SourceCodeLine in $SourceCodeLines) {
                        $CurrentLine = $SourceCodeLine
                        $LabelReplacements | ForEach-Object {
                            if ($CurrentLine -match $_.FieldName) {
                                $CurrentLine = $CurrentLine -replace $_.FieldName, "customN|$($_.DomainClass)|$($_.FieldLabel)|"
                            }
                        }
                        [void]($UpdatedSourceCodeLines.Add($CurrentLine))
                    }
                    $Item.sourceCodeJson = $UpdatedSourceCodeLines | ConvertFrom-Json
                }

                ## Get a FileName safe version of the Provider Name
                $SafeScriptName = $Item.name -replace '\:', '-' -replace '\/', '-' -replace '\|', '-' -replace '\\', '-'

                ## Create the Provider Recipe Folder path
                Test-FolderPath -FolderPath $SaveCodePath

                ##
                ## Save the Recipe (JSON)
                ##
                $RecipeConfigFileName = Join-Path $SaveCodePath "$SafeScriptName.json"
                $RecipeCodeFileName = Join-Path $SaveCodePath "$SafeScriptName.groovy"

                ## Save the Source Code
                $Item.sourceCode | Set-Content -Path $RecipeCodeFileName -Force
                $Item.PSObject.Properties.remove('sourceCode')
                $Item | ConvertTo-Json -Depth 10 | Set-Content -Path $RecipeConfigFileName -Force
            }
        }

        if ($Passthru -or !$SaveCodePath) {
            return $Result
        }
    }
    end {
        $ProgressPreference = $ExistingProgressPreference
    }
}


Function New-TMRecipe {
    param(
        [Parameter(Mandatory = $false)][PSObject]$TMSession = 'Default',
        [Parameter(Mandatory = $true)][PSObject]$Recipe,
        [Parameter(Mandatory = $false)][Switch]$Update,
        [Parameter(Mandatory = $false)][Switch]$PassThru
    )
    begin {
        # Get the session configuration
        Write-Verbose 'Checking for cached TMSession'
        $TMSession = Get-TMSession $TMSession
        Write-Debug 'TMSession:'
        Write-Debug ($TMSession | ConvertTo-Json -Depth 5)
        if (-not $TMSession) {
            throw "TMSession '$TMSession' not found. Use New-TMSession command before using features."
        }

        # Honor SSL Settings
        $TMCertSettings = @{SkipCertificateCheck = $TMSession.AllowInsecureSSL }

        # Use the TM session if a project id is not provided
        $ProjectId ??= $TMSession.UserContext.Project.id
    }

    process {

        ## Check for existing credential, or Create a shell
        $RecipeCheck = Get-TMRecipe -Name $Recipe.Name -TMSession $TMSession
        if ($RecipeCheck) {

            ## If Update is enabled, set the RecipeID for the update
            if ($Update) {
                $NewRecipeID = $RecipeCheck.recipeId
            } else {

                ## If Passthru is enabled, return the object
                if ($PassThru) {
                    return $RecipeCheck
                } else {
                    return
                }
            }
        } else {

            ## The Recipe needs to be created. Start by posting the name to get an ID
            $uri = 'https://'
            $uri += $TMSession.TMServer
            $uri += '/tdstm/ws/cookbook/recipe'

            ## Create a new Recipe
            $CreateNewRecipe = @{
                name        = $Recipe.name
                description = $Recipe.description
            } #| ConvertTo-Json

            ## Send the New recipe to the server
            Set-TMHeaderContentType -ContentType 'Form' -TMSession $TMSession
            $response = Invoke-WebRequest -Method Post -Uri $uri -WebSession $TMSession.TMWebSession -Body $CreateNewRecipe @TMCertSettings
            if ($response.StatusCode -eq 200) {

                ## Convert the response content
                $responseContent = $response.Content | ConvertFrom-Json

                ## If Successful
                if ($responseContent.status -eq 'success') {

                    ## Created a new recipe ID, update the Recipe Object and save it.
                    $NewRecipeID = $responseContent.data.recipeId

                } else {
                    throw "Unable to add Recipe. $($ResponseContent.errors))"
                }
            } else {
                throw 'Unable to add Recipe.'
            }
        }

        ##
        ## Update or Create the Recipe
        ##

        $RecipeSourceCode = $Recipe.sourceCode

        $NewRecipe = Get-TMRecipe -Name $Recipe.name -TMSession $TMSession
        if ($NewRecipe) {
            $Recipe = $NewRecipe
        }

        ## With the Existing or New RecipeID Update the Recipe Data
        $UpdatedRecipe = @{
            recipeId              = $NewRecipeID
            name                  = $Recipe.name
            description           = $Recipe.description
            createdBy             = $Recipe.createdBy
            lastUpdated           = $Recipe.lastUpdated
            versionNumber         = $Recipe.versionNumber
            releasedVersionNumber = $Recipe.releasedVersionNumber
            recipeVersionId       = $NewRecipe.recipeVersionId
            hasWIP                = $Recipe.hasWIP
            sourceCode            = $RecipeSourceCode
            changelog             = $Recipe.changelog
            clonedFrom            = $Recipe.clonedFrom
        }

        # Update the newly created recipe with the data
        $uri = 'https://'
        $uri += $TMSession.TMServer
        $uri += '/tdstm/ws/cookbook/recipe/' + $NewRecipeID

        ## Send the Recipe (New or Update)
        Set-TMHeaderContentType -ContentType 'Form' -TMSession $TMSession
        $response = Invoke-WebRequest -Method Post -Uri $uri -WebSession $TMSession.TMWebSession -Body $UpdatedRecipe @TMCertSettings
        if ($response.StatusCode -eq 200) {

            ## Convert the response content
            $responseContent = $response.Content | ConvertFrom-Json

            ## If Successful
            if ($responseContent.status -ne 'success') {
                throw "Unable to add Recipe: $($responseContent.errors)"
            }
        }

        ## If Passthru is enabled, return the Updated Recipe
        if ($PassThru) {
            return (Get-TMRecipe -Name $Recipe.name -TMSession $TMSession)
        }
    }
}


Function Read-TMRecipeScriptFile {
    param(
        [Parameter(Mandatory = $true)]$Path,
        [Switch][Parameter()]$UpdateFieldNames
    )

    ## Handle JSON files
    $File = Get-Item -Path $Path
    if ($File.Extension -eq '.json') {

        ## Read the Json file
        $TMRecipe = Get-Content -Path $Path | ConvertFrom-Json

        if ($UpdateFieldNames) {

            ## Convert any Custom Field Label Tokens from a template
            ## Get a FieldToLabel map to build string replacements
            $LabelReplacements = [System.Collections.ArrayList]@()
            $FieldToLabelMap = Get-TMFieldToLabelMap -TMSession $TMSession

            ## Create a replacement token for each asset/label = field set
            foreach ($DomainClass in @('DEVICE', 'APPLICATION', 'DATABASE', 'STORAGE' )) {

                foreach ($FieldName in $FieldToLabelMap.$DomainClass.Keys) {
                    if ($FieldName -match 'custom\d') {

                        # Write-Host $FieldName
                        [void]($LabelReplacements.Add([pscustomobject]@{
                                    DomainClass = $DomainClass
                                    FieldLabel  = $FieldToLabelMap.$DomainClass.$FieldName
                                    FieldName   = $FieldName
                                }
                            )
                        )
                    }
                }
            }
        }

        ## This might be a JSON+Groovy pair
        $RecipeGroovyPath = $Path -Replace '.json', '.groovy'
        if (Test-Path $RecipeGroovyPath) {

            ## Replace all 'customN' items with a codestring to replace the correct items in the offline package
            $SourceCodeLines = (Get-Content -Path $RecipeGroovyPath -Raw) -split "`r`n" -split "`r" -split "`n"
            $UpdatedSourceCodeLines = [System.Collections.ArrayList]::new()

            if (-Not $UpdateFieldNames) {
                $UpdatedSourceCodeLines = $SourceCodeLines
            } else {
                foreach ($SourceCodeLine in $SourceCodeLines) {
                    $CurrentLine = $SourceCodeLine
                    $LabelReplacements | ForEach-Object {
                        $ReplaceString = "customN\|$($_.DomainClass)\|$($_.FieldLabel)\|"
                        if ($CurrentLine -match $ReplaceString) {
                            $CurrentLine = $CurrentLine -replace $ReplaceString, $_.FieldName
                        }
                    }
                    [void]($UpdatedSourceCodeLines.Add($CurrentLine))
                }
            }

            Add-Member -InputObject $TMRecipe -NotePropertyName sourceCode -NotePropertyValue $($UpdatedSourceCodeLines -join "`r`n") -Force

        }

        ## Remove JSON metadata that is only stored offline
        if ($TMRecipe.PSObject.Properties.Name -contains 'sourceCodeJson') {
            $TMRecipe.PSObject.Properties.Remove('sourceCodeJson')
        }

        return $TMRecipe
    }

}

# Function Publish-TMRecipe {
# [CmdletBinding()]
# param(
# [Parameter(Mandatory = $false)]
# [PSObject]$TMSession = 'Default',

# [Parameter(Mandatory = $false)]
# [String]$Name,

# [Parameter(Mandatory = $false)]
# [Bool]$AllowInsecureSSL = $global:TMSessions[$TMSession].AllowInsecureSSL
# )

# ## Get Session Configuration
# $TMSession = $global:TMSessions[$TMSession]
# if (-not $TMSession) {
# Write-Host 'TMSession: [' -NoNewline
# Write-Host $TMSession -ForegroundColor Cyan
# Write-Host '] was not Found. Please use the New-TMSession command.'
# Throw 'TM Session Not Found. Use New-TMSession command before using features.'
# }

# #Honor SSL Settings
# $TMCertSettings = $TMSession.AllowInsecureSSL ? @{SkipCertificateCheck = $true } : @{SkipCertificateCheck = $false }

# ## Get the Existing Recipe to publish
# $ServerRecipe = Get-TMRecipe -Name $Name -TMSession $TMSession
# if (-Not $ServerRecipe) {
# throw 'Recipe not found'
# }

# if ($ServerRecipe.hasWIP -eq $False) {
# Write-Host 'This recipe does not have a Work In Progress, and is already released.'
# return
# }

# # Format the uri
# $Instance = $TMSession.TMServer.Replace('/tdstm', '').Replace('https://', '').Replace('http://', '')
# $uri = "https://$($TMSession.TMServer)/tdstm/ws/cookbook/recipe/release/" + $ServerRecipe.recipeId

# ## Set the Request to a Form post
# Set-TMHeaderContentType -ContentType 'FORM'
# $FormPostData = @{
# recipeId = $ServerRecipe.recipeId
# name = $ServerRecipe.name
# description = $ServerRecipe.description
# createdBy = $ServerRecipe.createdBy
# lastUpdated = $ServerRecipe.lastUpdated
# versionNumber = $ServerRecipe.versionNumber
# releasedVersionNumber = $ServerRecipe.releasedVersionNumber
# recipeVersionId = $ServerRecipe.recipeVersionId
# hasWip = $ServerRecipe.hasWip
# sourceCode = $ServerRecipe.sourceCode
# changelog = $ServerRecipe.changelog
# clonedFrom = $ServerRecipe.clonedFrom
# }

# ## Try the Request
# $response = Invoke-WebRequest -Method 'Post' -Uri $Uri -WebSession $TMSession.TMWebSession -Body $FormPostData -SkipCertificateCheck $TMSession.AllowInsecureSSL
# if ($response.StatusCode -eq 200) {
# $responseContent = $response.Content | ConvertFrom-Json
# if ($responseContent.status -ne 'success') {
# Write-Error ('Unable to Publish Recipe:' + $responseContent.errors)
# }
# }
# }

function Invoke-TMRecipe {
    <#
    .SYNOPSIS
    Invokes a TransitionManager Recipe and produces a runbook.
 
    .DESCRIPTION
    This function allows a user to invoke a TransitionManager Recipe,
    choosing the Recipe, Bundles and Events related.
 
    .PARAMETER Name
    The name of the Recipe to use to Generate Tasks from
 
    .PARAMETER EventName
    The name of the Event to create tasks in
 
    .PARAMETER UseWip
    Determines if the Work In Progress recipe is used to generate tasks
 
    .PARAMETER PublishTasks
    Determines if the resulting tasks are published in the runbook or not.
 
    .PARAMETER ActivityId
    When Provided, This function will include Write-Progress Activities, using this ID as the root
 
    .PARAMETER ParentActivityId
    When Provided, This function will include Write-Progress Activities, using this ID as the Parent
 
    .EXAMPLE
    Invoke-TMRecipe -Name 'Migration Recipe' -EventName 'Move 1' -UseWip $True -PublishTasks $True
 
    .OUTPUTS
    None
    #>


    [CmdletBinding()]
    param(
        [Parameter(Mandatory = $true)]
        [Alias('Recipe')]
        [String]$Name,

        [Parameter(Mandatory = $true)]
        [String]$EventName,

        [Parameter(Mandatory = $false)]
        [String[]]$Tags,

        [Parameter(Mandatory = $false)]
        [PSObject]$TMSession = 'Default',

        [Parameter(Mandatory = $false)]
        [Switch]$PassThru,

        [Parameter(Mandatory = $false)]
        [Switch]$UseWip,

        [Parameter(Mandatory = $false)]
        [Switch]$PublishTasks,

        [Parameter(Mandatory = $false)][Int16]$ActivityId,
        [Parameter(Mandatory = $false)][Int16]$ParentActivityId = -1

    )

    begin {

        ## Get Session Configuration
        $TMSession = Get-TMSession $TMSession
        if (-not $TMSession) {
            Write-Host 'TMSession: [' -NoNewline
            Write-Host $TMSession -ForegroundColor Cyan
            Write-Host '] was not Found. Please use the New-TMSession command.'
            Throw 'TM Session Not Found. Use New-TMSession command before using features.'
        }

        #Honor SSL Settings
        $TMCertSettings = @{SkipCertificateCheck = $TMSession.AllowInsecureSSL }
    }

    process {

        ## Get the Recipe indicated
        $Recipe = Get-TMRecipe -Name $Name -TMSession $TMSession
        if (!$Recipe) {
            throw ("Recipe [$Name] does not exist.")
        }

        ## Resolve usage of the UseWip switch
        $FinalUseWip = $UseWip.IsPresent
        if ($UseWip.IsPresent -and !$Recipe.hasWip) {

            ## Change to use the Released version
            Write-Verbose ('Recipe ' + $Recipe.name + ' does not have a WIP version. Creating Tasks with the released version instead.')
            $FinalUseWip = $False
        }

        ## Get the required Event
        $TMEvent = Get-TMEvent -Name $EventName
        if (!$TMEvent) {
            New-TMEvent -Name $EventName
        }

        ## Find any associated tags
        $IncludeTags = Get-TMTag -TMSession $TMSession | Where-Object {
            $_.name -in $Tags
        }

        ## Confirm the Recipe and Event
        Write-Verbose 'Recipe and Event are valid. Generating Tasks...'

        # Update the newly created recipe with the data
        $uri = 'https://'
        $uri += $TMSession.TMServer
        $uri += '/tdstm/ws/task/generateTasks'

        ## Create the Post Body
        $PostBody = [PSCustomObject]@{
            recipeId        = $Recipe.recipeId
            recipeVersionId = $Recipe.recipeVersionId
            deletePrevious  = $true
            useWIP          = $FinalUseWip
            autoPublish     = ($PublishTasks.IsPresent) ?? $False
            eventId         = $TMEvent.id
            tag             = $IncludeTags.id -as [array]
        } | ConvertTo-Json -Compress

        ## Set the right content type
        Set-TMHeaderContentType -ContentType 'JSON'
        Set-TMHeaderAccept -Accept 'JSON'

        ## Make the request
        try {
            $response = Invoke-WebRequest -Method Post -Uri $uri -Body $PostBody -WebSession $TMSession.TMWebSession @TMCertSettings
        } catch {
            throw $_
        }

        ## Ensure a useful response came back
        if ($response.StatusCode -eq 200) {
            $responseContent = $response.Content | ConvertFrom-Json

            if ($responseContent.status -ne 'success') {

                throw ('The recipe could not be run: ' + $responseContent.errors)
            }
        }
        ## Get the Job ID for the Task Generation
        $TaskGenerationJobId = $responseContent.data.jobId

        ## With the file uploaded, Initiate the Datascript on the server
        $uri = 'https://'
        $uri += $TMSession.TMServer
        $uri += "/tdstm/ws/progress/$TaskGenerationJobId"

        Set-TMHeaderContentType -ContentType JSON -TMSession $TMSession

        ## The Recipe Generation Job has started, monitor the status
        if ($ActivityId) {
            Write-Progress -Id ($ActivityId + 1) -ParentId $ActivityId -Activity 'Task Generation Started' -CurrentOperation 'Starting Datascript Processing' -PercentComplete 5
        } else {
            Write-Verbose 'TransitionManager Task Generation: Starting Generating Tasks'
        }

        ## Poll for the status of the Task Generation
        ## TODO: This should be converted to a function for polling the job engine. It's nearly dupilcated now in the Import Batch watching.
        $Completed = $false
        while ($Completed -eq $false) {

            ## Check the status of the TaskGeneration Job
            try {

                $response = Invoke-WebRequest -Method Get -Uri $uri -WebSession $TMSession.TMWebSession @TMCertSettings
            } catch {
                throw $_.Exception.Message
            }

            ## Check the status of the script
            if ($response.StatusCode -eq 200) {
                $responseContent = $response.Content | ConvertFrom-Json
                if ($responseContent.status -eq 'success') {
                    $TaskGenerationProgress = $responseContent.data

                    switch ($TaskGenerationProgress.status) {
                        'Queued' {

                            $CurrentOperation = 'Task Generation Queued'
                            $Status = 'Queued'
                            $ProgressString = 'Status - Queued: ' + $TaskGenerationProgress.percentComp + '%'
                            $PercentComplete = $TaskGenerationProgress.percentComp
                            $SleepSeconds = 2
                            Break

                        }
                        'Pending' {
                            $CurrentOperation = 'Task Generation Pending'
                            $Status = 'Pending'
                            $ProgressString = 'Status - Pending: ' + $TaskGenerationProgress.percentComp + '%'
                            $PercentComplete = $TaskGenerationProgress.percentComp
                            $SleepSeconds = 2
                            Break
                        }
                        'Processing' {
                            $CurrentOperation = 'Task Generation Running'
                            $Status = 'Generating Tasks'
                            $ProgressString = 'Status - Running: ' + $TaskGenerationProgress.percentComp + '%'
                            $PercentComplete = $TaskGenerationProgress.percentComp
                            $SleepSeconds = 2
                            Break
                        }
                        'COMPLETED' {
                            $TaskGenerationJobId = $TaskGenerationProgress.detail
                            $CurrentOperation = 'Task Generation Complete'
                            $Status = 'Task Generation Complete'
                            $SleepSeconds = 0
                            $PercentComplete = 99
                            $ProgressString = 'Status - Task Generation Complete Complete.'
                            $Completed = $true
                            Break
                        }
                        'Failed' {
                            $CurrentOperation = 'Failed'
                            $Status = $TaskGenerationProgress.status
                            Write-Host 'Task Generation Failed '$TaskGenerationProgress.detail
                            Throw $TaskGenerationProgress.detail
                        }
                        Default {
                            $CurrentOperation = 'State Unknown'
                            $Status = 'Unknown. Sleeping to try again.'
                            $ProgressString = 'Unknown Status: ' + $TaskGenerationProgress.status
                            $PercentComplete = 99
                            $SleepSeconds = 2
                            Break
                        }
                    }

                    ## Notify the user of the Datascript Progress
                    if ($ActivityId) {
                        Write-Progress -Id ($ActivityId + 1) `
                            -ParentId $ActivityId `
                            -Activity 'Task Generation' `
                            -CurrentOperation $CurrentOperation `
                            -Status $Status `
                            -PercentComplete $PercentComplete
                    } else {
                        Write-Host $ProgressString
                    }
                }
            }

            ## Sleep a few seconds to allow the process to complete
            Start-Sleep -Seconds $SleepSeconds
        }

        ## DEBUGGING
        Write-Host $TaskGenerationJob

    }
}