Dagger.psm1
|
. "$PSScriptRoot\Start-DaggerEngineSession.ps1" . "$PSScriptRoot\Stop-DaggerEngineSession.ps1" . "$PSScriptRoot\DSL.ps1" . "$PSScriptRoot\Invoke-DaggerQuery.ps1" . "$PSScriptRoot\Write-EngineOutput.ps1" . "$PSScriptRoot\Get-Weather.ps1" $Writer = $null #$Port = Start-DaggerEngineSession $bit = [Environment]::Is64BitOperatingSystem ? "amd64" : "386" $platform = switch ([Environment]::OSVersion.Platform) { "Win32NT" { "windows" } "Darwtin" { "darwin" } "Unix" { "linux" } } $ext = $platform -eq "windows" ? ".exe" : [string]::Empty $binaryName = 'dagger-engine-session_{0}_{1}' -f $platform, $bit $binaryPath = Join-Path -Path "$PSScriptRoot" -ChildPath bin -AdditionalChildPath "$binaryName$ext" #$binaryPath = 'C:\source\github\dagger\sdk\powershell\output\Dagger\bin\dagger-engine-session_windows_amd64.exe' $oldProcess = Get-Process -Name $binaryName -ErrorAction SilentlyContinue if ($oldProcess) { $oldProcess.Kill() } $startInfo = New-Object System.Diagnostics.ProcessStartInfo $startInfo.FileName = $binaryPath $startInfo.RedirectStandardError = $true $startInfo.RedirectStandardOutput = $true $startInfo.RedirectStandardInput = $true $startInfo.UseShellExecute = $false Write-Verbose "Started $binaryPath" $engineProcess = New-Object System.Diagnostics.Process $engineProcess.StartInfo = $startInfo $engineProcess.EnableRaisingEvents = $true $dataReceiveHander = [scriptblock] { param ([object]$source, [Diagnostics.DataReceivedEventArgs]$err) if (![string]::IsNullOrEmpty($err.Data)) { $message = "($($engineProcess.Id)) $($err.Data)" $message } } $errorEvents = Register-ObjectEvent -InputObject $engineProcess -EventName ErrorDataReceived -Action $dataReceiveHander $exitEvents = Register-ObjectEvent -InputObject $engineProcess -EventName Exited -Action $dataReceiveHander $startResult = $engineProcess.Start() $engineProcess.BeginErrorReadLine() Write-Verbose "Start result $startResult" $Writer = $engineProcess.StandardInput $Writer.WriteLine("ping") if ($startResult) { $i = 0 do { $output = $engineProcess.StandardOutput.ReadLine() Start-Sleep -Milliseconds 250 } until ($output -match "\d{4,5}" -or $i++ -ge $RetryLimit) } if ($output -notmatch "\d{4,5}") { Write-Error -Message "Failed to start `"$binaryPath`". No TCP Port available." -Category InvalidResult -RecommendedAction "Check that the user has access to run `"$binaryPath`"." } Write-Verbose "localhost:$output" $Port = $output $Uri = 'http://localhost:{0}/query' -f $Port Write-Verbose "Uri: $Uri" |