Transpilers/Include.psx.ps1
|
<# .SYNOPSIS Includes Files .DESCRIPTION Includes Files or Functions into a Script. .Example { [Include("Invoke-PipeScript")]$null } | .>PipeScript .Example { [Include("Invoke-PipeScript")] param() } | .>PipeScript .EXAMPLE { [Include('*-*.ps1')]$psScriptRoot } | .>PipeScript #> param( # The File Path to Include [Parameter(Mandatory,Position=0)] [string] $FilePath, # If set, will include the content as a byte array [switch] $AsByte, [Parameter(Mandatory,ParameterSetName='VariableAST', ValueFromPipeline)] [Management.Automation.Language.VariableExpressionast] $VariableAst ) # Determine the command we would be including (relative to the current path) $includingCommand = $ExecutionContext.SessionState.InvokeCommand.GetCommand($FilePath, 'All') if (-not $includingCommand) { # if we could not determine the command, we may need to error out. if (-not $FilePath.Contains('*')) { Write-Error "Could not resolve $($FilePath)" return } } function IncludeFileContents { param($FilePath) process { if ($AsByte) { [ScriptBlock]::Create( "@'" + [Environment]::NewLine + [Convert]::ToBase64String( [IO.File]::ReadAllBytes($FilePath), 'InsertLineBreaks' ) + [Environment]::NewLine + "'@") } else { [ScriptBlock]::Create( "@'" + [Environment]::NewLine + ([IO.File]::ReadAllLines($FilePath) -replace "^@'", "@''" -replace "^'@", "''@" -join [Environment]::NewLine) + [Environment]::NewLine + "'@" + @' -split "[\r\n]{1,2}" -replace "^@''", "@'" -replace "^''@", "'@" -join [Environment]::NewLine '@ ) } } } $includedScript = if ($includingCommand -is [Management.Automation.CmdletInfo]) { Write-Error "Cannot Include Cmdlets" return } elseif ($includingCommand -is [Management.Automation.FunctionInfo]) { if ($VariableAst -and $VariableAst.VariablePath -notmatch '^null$') { # If we're including a function as a variable, define it as a ScriptBlock [ScriptBlock]::Create(@" { $($includingCommand.ScriptBlock) } "@) } else { # If we're including a function, define it inline [ScriptBlock]::Create(@" function $($includingCommand.Name) { $($includingCommand.ScriptBlock) } "@) } } elseif ($includingCommand.ScriptBlock) { # If we're including a command with a ScriptBlock, assign it to a variable [ScriptBlock]::Create(@" `${$($includingCommand.Name)} = { $($includingCommand.ScriptBlock) } "@) } elseif ($includingCommand.Source -match '\.ps1{0,}\.(?<ext>[^.]+$)') { $transpiledFile = Invoke-PipeScript -CommandInfo $includingCommand if (-not $transpiledFile) { Write-Error "Could not transpile $($includingCommand.Source)" return } IncludeFileContents $transpiledFile.Fullname } elseif ($includingCommand.Source -match '\.ps$') { [ScriptBlock]::Create(@" `${$($includingCommand.Name)} = { $([ScriptBlock]::Create([IO.File]::ReadAllText($includingCommand.Source)) | .<PipeScript>) } "@) } elseif ($includingCommand) { IncludeFileContents $includingCommand.Source } if ($psCmdlet.ParameterSetName -eq 'ScriptBlock' -or $VariableAst.VariablePath -match '^null$') { if ($ScriptBlock -and $ScriptBlock.ToString().Length) { [ScriptBlock]::Create("$ScriptBlock" + [Environment]::NewLine + $includedScript) } else { $includedScript } } elseif ($VariableAst.VariablePath -and $includingCommand) { [ScriptBlock]::Create("$($VariableAst) = $IncludedScript") } elseif ($VariableAst.VariablePath -notmatch '^null$') { [ScriptBlock]::Create(@" foreach (`$file in (Get-ChildItem -Path "$($VariableAst)" -Filter "$FilePath" -Recurse)) { if (`$file.Extension -ne '.ps1') { continue } . `$file.FullName } "@) } |