functions.ps1

Set-StrictMode -version latest;
$ErrorActionPreference = "Stop";
Function RunProc{
    [CmdletBinding()]
    Param($proc,$opts,
    [switch] $errok,
    [switch] $dry_run,
    [switch] $redirect_output,
    [switch] $no_wait,
    [ValidateSet("verbose","host","none")] 
	[String] $verbose_mode="host"
    )
    if ($no_wait -and $redirect_output){
    	throw "Cannot do no_wait and redirect_output";
    }
    if ($proc.ToUpper().EndsWith(".PL")){
        $path = (C:\Windows\system32\where.exe $proc | Out-String).Trim()
        $opts = "$path $opts";
        $proc="perl";       
    }
    $pinfo = New-Object System.Diagnostics.ProcessStartInfo;
    $pinfo.FileName = $proc;
    $working = Convert-Path .;
    $verbose_str = "Running: $proc $opts in $working";
    if ($verbose_mode -eq "host"){
    	Write-Host $verbose_str -Foreground Green;
    }elseif($verbose_mode -eq "verbose"){
		Write-Verbose $verbose_str;
    }
    $pinfo.WorkingDirectory = $working;
    $pinfo.UseShellExecute = $false;    

    $pinfo.Arguments = $opts;
    $p = New-Object System.Diagnostics.Process;
    $p.StartInfo = $pinfo;
    $oStdOutEvent = $null;
    $oStdErrEvent = $null;
    $oStdOutBuilder = $null;
    $oStdErrBuilder = $null;
    if ($dry_run){
    	return $null;
    }

    if ($redirect_output){
		$pinfo.RedirectStandardError = $true;
		$pinfo.RedirectStandardOutput = $true;
		$oStdOutBuilder = New-Object -TypeName System.Text.StringBuilder;
    	$oStdErrBuilder = New-Object -TypeName System.Text.StringBuilder;
		$sScripBlock = {
		        if (! [String]::IsNullOrWhitespace($EventArgs.Data)) {
		        	$Event.MessageData.AppendLine($EventArgs.Data);
		            #Write-Host $EventArgs.Data; #not sure how big the buffer is for this
		            #[console]::WriteLine($EventArgs.Data); #console.WriteLine does not work for remote powershell, but does display instant output
		        }
		    };		
		$oStdOutEvent = Register-ObjectEvent -InputObject $p -Action $sScripBlock -EventName 'OutputDataReceived' -MessageData $oStdOutBuilder;
    	$oStdErrEvent = Register-ObjectEvent -InputObject $p -Action $sScripBlock -EventName 'ErrorDataReceived'  -MessageData $oStdErrBuilder;
    }
    $p.Start() | Out-Null;
    if ($no_wait){
    	return $p;
    }
    if ($redirect_output){
 		$p.BeginOutputReadLine();
	   	$p.BeginErrorReadLine();
	}
	$p.WaitForExit();

    if ($redirect_output){
		Unregister-Event -SourceIdentifier $oStdOutEvent.Name;
    	Unregister-Event -SourceIdentifier $oStdErrEvent.Name;
    	$stdout = $oStdOutBuilder.ToString().Trim();
    	$stderr = $oStdErrBuilder.ToString().Trim();
    	if (! [String]::IsNullOrWhitespace($stdout)){
    		Write-Host $stdout;
    	}
    	if (! [String]::IsNullOrWhitespace($stderr)){
    		Write-Host $stderr;
    	}
	}
    if ($p.ExitCode -ne 0 -and -not $errok){
        throw "Process $proc $opts exited with non zero code: $($p.ExitCode) aborting!" ;
    }
    if ($errok){
        return $p.ExitCode;
    }
}
function out {
	Param(
		[Parameter(Mandatory = $true, ParameterSetName = "ByPath", Position = 0)]
  		$FilePath,
		[Parameter(Mandatory = $true,ValueFromPipeline=$true,ValueFromPipelineByPropertyName=$true, ParameterSetName = "ByPath", Position = 1)]
  		$InputObject,
  		[switch] $Append
  		)
		Begin {
			if (! $Append){
				Out-File -FilePath $FilePath -Encoding ASCII;
			}
		}
		Process {
				Out-File -Append -InputObject $InputObject -FilePath $FilePath	-Encoding ASCII
		}
}
Function confirm($str){
	return $PSCmdlet.ShouldContinue($str, "");
}
$global:last_time = Get-Date;
$global:PERF_FILE="";
Function TimerNow($name){
	$now = Get-Date;
	$diff = ($now -  $global:last_time).TotalSeconds.ToString("0.0");
	$str = "$(Get-Date) $name took $diff secs";
	Write-Host $str  -ForegroundColor Green;
	if ($PERF_FILE -ne ""){
		$str | out -FilePath $PERF_FILE -Append;
	}
	$global:last_time = $now;
}
Function WriteException($exp){
	write-host "Caught an exception:" -ForegroundColor Yellow -NoNewline
	write-host " $($exp.Exception.Message)" -ForegroundColor Red
	write-host "`tException Type: $($exp.Exception.GetType().FullName)"
	$stack = $exp.ScriptStackTrace;
	$stack = $stack.replace("`n","`n`t")
	write-host "`tStack Trace: $stack"
	if ($exp.Exception.InnerException){
		write-host "`tInnerException:" -ForegroundColor Yellow -NoNewline
		write-host " $($exp.Exception.InnerException.Message)" -ForegroundColor Red
	}
}