Adding support for -ReplacePattern in Core Inline Transpiler (#84)

Author: James Brundage

Transpilers/Core/PipeScript.Inline.psx.ps1

@@ -23,6 +23,7 @@ $SourceSection,
 
 # A string containing the text contents of the file
 [Parameter(Mandatory,ParameterSetName='SourceTextAndPattern')]
+[Parameter(Mandatory,ParameterSetName='SourceTextReplace')]
 [string]
 $SourceText,
 
@@ -31,6 +32,25 @@ $SourceText,
 [regex]
 $SourcePattern,
 
+
+[Parameter(Mandatory,ParameterSetName='SourceTextReplace')]
+[Alias('Replace')]
+[ValidateScript({    
+    if ($_.GetGroupNames() -notcontains 'PS' -and 
+        $_.GetGroupNames() -notcontains 'PipeScript'
+    ) {
+        throw "Group Name PS or PipeScript required"
+    }
+    return $true
+})]
+[regex]
+$ReplacePattern,
+
+[Parameter(ParameterSetName='SourceTextReplace')]
+[Alias('Replacer')]
+[ScriptBlock]
+$ReplacementEvaluator,
+
 # If set, will not transpile script blocks.
 [Parameter(ParameterSetName='SourceTextAndPattern')]
 [Parameter(ParameterSetName='SourceSections')]
@@ -40,25 +60,29 @@ $NoTranspile,
 # The path to the source file.
 [Parameter(ParameterSetName='SourceTextAndPattern')]
 [Parameter(ParameterSetName='SourceSections')]
+[Parameter(ParameterSetName='SourceTextReplace')]
 [string]
 $SourceFile,
 
 # A Script Block that will be injected before each inline is run. 
 [Parameter(ParameterSetName='SourceTextAndPattern')]
 [Parameter(ParameterSetName='SourceSections')]
+[Parameter(ParameterSetName='SourceTextReplace')]
 [ScriptBlock]
 $Begin,
 
 # A Script Block that will be piped to after each output.
 [Parameter(ParameterSetName='SourceTextAndPattern')]
 [Parameter(ParameterSetName='SourceSections')]
+[Parameter(ParameterSetName='SourceTextReplace')]
 [Alias('Process')]
 [ScriptBlock]
 $ForeachObject,
 
 # A Script Block that will be injected after each inline script is run. 
 [Parameter(ParameterSetName='SourceTextAndPattern')]
 [Parameter(ParameterSetName='SourceSections')]
+[Parameter(ParameterSetName='SourceTextReplace')]
 [ScriptBlock]
 $End
 )
@@ -68,6 +92,82 @@ begin {
 }
 
 process {
+    if ($psCmdlet.ParameterSetName -eq 'SourceTextReplace') {
+        $fileText      = $SourceText
+        if (-not $PSBoundParameters["ReplacementEvaluator"]) {
+            $ReplacementEvaluator = {
+                param($match)
+
+                $pipeScriptText = 
+                    if ($Match.Groups["PipeScript"].Value) {
+                        $Match.Groups["PipeScript"].Value
+                    } elseif ($match.Groups["PS"].Value) {
+                        $Match.Groups["PS"].Value                        
+                    }
+
+                if (-not $pipeScriptText) {
+                    return
+                }
+
+                $InlineScriptBlock = [scriptblock]::Create($pipeScriptText)
+                if (-not $InlineScriptBlock) {                    
+                    return
+                }
+
+                if (-not $NoTranspile) {
+                    $TranspiledOutput = $InlineScriptBlock | .>Pipescript
+                    if ($TranspiledOutput -is [ScriptBlock]) {
+                        $InlineScriptBlock = $TranspiledOutput
+                    }
+                }
+                
+                $inlineAstString = $InlineScriptBlock.Ast.Extent.ToString()
+                if ($InlineScriptBlock.ParamBlock) {
+                    $inlineAstString = $inlineAstString.Replace($InlineScriptBlock.ParamBlock.Extent.ToString(), '')
+                }
+                $inlineAstString = $inlineAstString
+                $AddForeach =
+                    $(
+                        if ($ForeachObject) {
+                            '|' + [Environment]::NewLine
+                            @(foreach ($foreachStatement in $ForeachObject) {
+                                if ($foreachStatement.Ast.ProcessBlock -or $foreachStatement.Ast.BeginBlock) {
+                                    ". {$ForeachStatement}"
+                                } elseif ($foreachStatement.Ast.EndBlock.Statements -and 
+                                    $foreachStatement.Ast.EndBlock.Statements[0].PipelineElements[0].CommandElements -and
+                                    $foreachStatement.Ast.EndBlock.Statements[0].PipelineElements[0].CommandElements.Value -in 'Foreach-Object', '%') {
+                                    "$ForeachStatement"
+                                } else {
+                                    "Foreach-Object {$ForeachStatement}"
+                                }
+                            }) -join (' |' + [Environment]::NewLine)
+                        }
+                    )
+
+
+                $statements = @(
+                    if ($begin) {
+                        "$begin"
+                    }
+                    if ($AddForeach) {
+                        "@($inlineAstString)" + $AddForeach.Trim()
+                    } else {
+                        $inlineAstString
+                    }
+                    if ($end) {
+                        "$end"
+                    }
+                ) 
+
+                $codeToRun = [ScriptBlock]::Create($statements -join [Environment]::Newline)
+
+                "$(& $codeToRun)"
+            }
+        }
+
+        return $ReplacePattern.Replace($fileText, $ReplacementEvaluator)
+    }
+
     if ($psCmdlet.ParameterSetName -eq 'SourceTextAndPattern') {
 
         $fileText      = $SourceText