Tasks/Invoke-WhiskeyPowerShell.ps1
|
function Invoke-WhiskeyPowerShell { [Whiskey.Task("PowerShell",SupportsClean=$true,SupportsInitialize=$true)] [CmdletBinding()] param( [Parameter(Mandatory=$true)] [Whiskey.Context] $TaskContext, [Parameter(Mandatory=$true)] [hashtable] $TaskParameter ) Set-StrictMode -Version 'Latest' Use-CallerPreference -Cmdlet $PSCmdlet -SessionState $ExecutionContext.SessionState if( -not ($TaskParameter.ContainsKey('Path')) ) { Stop-WhiskeyTask -TaskContext $TaskContext -Message ('Property ''Path'' is mandatory. It should be one or more paths, which should be a list of PowerShell Scripts to run, e.g. Build: - PowerShell: Path: - myscript.ps1 - myotherscript.ps1 WorkingDirectory: bin') return } $path = $TaskParameter['Path'] | Resolve-WhiskeyTaskPath -TaskContext $TaskContext -PropertyName 'Path' $workingDirectory = (Get-Location).ProviderPath $argument = $TaskParameter['Argument'] if( -not $argument ) { $argument = @{ } } $moduleRoot = Join-Path -Path $PSScriptRoot -ChildPath '..' -Resolve foreach( $scriptPath in $path ) { if( -not (Test-Path -Path $WorkingDirectory -PathType Container) ) { Stop-WhiskeyTask -TaskContext $TaskContext -Message ('Can''t run PowerShell script ''{0}'': working directory ''{1}'' doesn''t exist.' -f $scriptPath,$WorkingDirectory) continue } $scriptCommand = Get-Command -Name $scriptPath -ErrorAction Ignore if( -not $scriptCommand ) { Stop-WhiskeyTask -TaskContext $TaskContext -Message ('Can''t run PowerShell script ''{0}'': it has a syntax error.' -f $scriptPath) continue } $passTaskContext = $scriptCommand.Parameters.ContainsKey('TaskContext') if( (Get-Member -InputObject $argument -Name 'Keys') ) { $scriptCommand.Parameters.Values | Where-Object { $_.ParameterType -eq [switch] } | Where-Object { $argument.ContainsKey($_.Name) } | ForEach-Object { $argument[$_.Name] = $argument[$_.Name] | ConvertFrom-WhiskeyYamlScalar } } $resultPath = Join-Path -Path $TaskContext.OutputDirectory -ChildPath ('PowerShell-{0}-RunResult-{1}' -f ($scriptPath | Split-Path -Leaf),([IO.Path]::GetRandomFileName())) $serializableContext = $TaskContext | ConvertFrom-WhiskeyContext $job = Start-Job -ScriptBlock { $workingDirectory = $using:WorkingDirectory $scriptPath = $using:ScriptPath $argument = $using:argument $serializedContext = $using:serializableContext $moduleRoot = $using:moduleRoot $resultPath = $using:resultPath $passTaskContext = $using:passTaskContext Invoke-Command -ScriptBlock { $VerbosePreference = 'SilentlyContinue'; & (Join-Path -Path $moduleRoot -ChildPath 'Import-Whiskey.ps1' -Resolve -ErrorAction Stop) } [Whiskey.Context]$context = $serializedContext | ConvertTo-WhiskeyContext $VerbosePreference = $using:VerbosePreference $contextArgument = @{ } if( $passTaskContext ) { $contextArgument['TaskContext'] = $context } Set-Location $workingDirectory $Global:LASTEXITCODE = 0 & $scriptPath @contextArgument @argument $result = @{ 'ExitCode' = $Global:LASTEXITCODE 'Successful' = $? } $result | ConvertTo-Json | Set-Content -Path $resultPath } do { $job | Receive-Job } while( -not ($job | Wait-Job -Timeout 1) ) $job | Receive-Job if( (Test-Path -Path $resultPath -PathType Leaf) ) { $runResult = Get-Content -Path $resultPath -Raw | ConvertFrom-Json } else { Stop-WhiskeyTask -TaskContext $TaskContext -Message ('PowerShell script ''{0}'' threw a terminating exception.' -F $scriptPath) return } if( $runResult.ExitCode -ne 0 ) { Stop-WhiskeyTask -TaskContext $TaskContext -Message ('PowerShell script ''{0}'' failed, exited with code {1}.' -F $scriptPath,$runResult.ExitCode) return } elseif( $runResult.Successful -eq $false ) { Stop-WhiskeyTask -TaskContext $TaskContext -Message ('PowerShell script ''{0}'' threw a terminating exception.' -F $scriptPath) return } } } |