RunQueue.ps1

# Run this every x minutes from Task Scheduler

$ErrorActionPreference = "Stop"

. (Join-Path $PsScriptRoot "settings.ps1")

if (Test-Path -Path "C:\demo\*\BcContainerHelper.psm1") {
    $module = Get-Item -Path "C:\demo\*\BcContainerHelper.psm1"
    Import-module $module.FullName -DisableNameChecking
} else {
    Import-Module -name bccontainerhelper -DisableNameChecking
}

$storageContext = New-AzStorageContext -ConnectionString $storageConnectionString
$queuename = $publicDnsName.Split('.')[0]

# Create Queue (if it doesn't exist)
$storageQueue = Get-AzStorageQueue -Context $storageContext -Name $queuename -ErrorAction Ignore
if (!($storageQueue)) {
    New-AzStorageQueue -Name $queuename -Context $storageContext -ErrorAction Ignore | Out-Null
}
$storageQueue = Get-AzStorageQueue -Name $queuename -Context $storageContext

# Create table for replies (if it doesn't exist)
$storageTable = Get-AzStorageTable –Name "QueueStatus" –Context $storageContext -ErrorAction Ignore
if (!($storageTable)) {
    New-AzStorageTable –Name "QueueStatus" –Context $storageContext -ErrorAction Ignore | Out-Null
}
$storageTable = Get-AzStorageTable –Name "QueueStatus" –Context $storageContext

# Transcript container on azure storage
New-AzStorageContainer -Context $storageContext -Name "transcript" -Permission Blob -ErrorAction SilentlyContinue | Out-Null

# Folder where command scripts are placed
$commandFolder = (Join-Path $PsScriptRoot "request").ToLowerInvariant()

# Folder for temp storage of transcripts
$transcriptFolder = Join-Path $PsScriptRoot "transcript"
if (!(Test-Path -Path $transcriptFolder -PathType Container)) {
    New-Item -Path $transcriptFolder -ItemType Directory | Out-Null
}

while ($true) {
    # Get message
    $timeUntilMessageReappearsInQueue = [TimeSpan]::FromMinutes(10)
    $message = $storageQueue.CloudQueue.GetMessage($timeUntilMessageReappearsInQueue)
    if (!($message)) {
        break
    }

    $transcriptname = ([Guid]::NewGuid().ToString()+".txt")
    $transcriptfilename = Join-Path $transcriptFolder $transcriptname
    $transcripting = $false

    $maxAttempts = 1
    $cmd = ""

    $ht = @{ "Queue" = $queuename
        "Cmd" = ""
        "Id" = ""
        "Logstr" = ""
        "Dequeue" = $message.DequeueCount
        "Status" = "Begin"
        "transcript" = ""
    }

    Start-Transcript -Path $transcriptfilename
    $transcripting = $true
    Write-Host '---------------------------'
    Write-Host $message.AsString
    Write-Host '---------------------------'

    try {
        $json = $message.AsString | ConvertFrom-Json
    
        $cmd = $json.cmd
        $id = ""
        $LogStr = "$cmd"
    
        $Parameters = @{}
        $json | Get-Member -MemberType NoteProperty | ForEach-Object {
            $key = $_.Name
            $value = $json."$key"
            if ($key -eq "id") {
                $id = $value
            }
            elseif ($key -eq "maxAttempts") {
                $maxAttempts = $value
            }
            elseif ($key -ne "cmd") {
                $LogStr += " -$key '$value'"
                $Parameters += @{ $key = $value }
            }
        }
        
        $ht.Cmd = $cmd
        $ht.id = $id
        $ht.Logstr = $LogStr

        if ($message.DequeueCount -gt $maxAttempts) {
            throw "No more attempts"
        }
        
        Add-AzTableRow -table $storageTable.CloudTable -partitionKey $queuename -rowKey ([string]::Format("{0:D19}", [DateTime]::MaxValue.Ticks - [DateTime]::UtcNow.Ticks)) -property $ht | Out-Null

        $script = Get-Item -Path (Join-Path $commandFolder "$cmd.ps1") -ErrorAction Ignore
        if (($script) -and ($script.FullName.ToLowerInvariant().StartsWith($commandFolder))) {
            . $script @Parameters
            $ht.Status = "Success"
        } else {
            $ht.Status = "Error"
            Write-Host "Illegal request"
        }

        $storageQueue.CloudQueue.DeleteMessage($message)

    } catch {
        $ht.Status = "Exception"
        Write-Host "Exception performing RunQueue (Cmd=$cmd)"
        Write-Host $_.Exception.Message
        Write-Host $_.ScriptStackTrace

        if ($message.DequeueCount -ge $maxAttempts) {
            $storageQueue.CloudQueue.DeleteMessage($message)
        }
    } finally {
    }

    if ($transcripting) {
        $transcripting = $false
        Stop-Transcript

        Set-AzStorageBlobContent -File $transcriptfilename -Context $storageContext -Container "transcript" -Blob $transcriptname -Force | Out-Null
        $ht.transcript = "$($StorageContext.BlobEndPoint)transcript/$transcriptname"
    }
    Add-AzTableRow -table $storageTable.CloudTable -partitionKey $queuename -rowKey ([string]::Format("{0:D19}", [DateTime]::MaxValue.Ticks - [DateTime]::UtcNow.Ticks)) -property $ht | Out-Null

}