Include - Allowing Including URLs (Fixes #481)

Author: James Brundage

Transpilers/Include.psx.ps1

@@ -16,16 +16,23 @@
     {
         [Include('*-*.ps1')]$psScriptRoot
     } | .>PipeScript
+.EXAMPLE
+    {
+        [Include('https://pssvg.start-automating.com/Examples/PowerShellChevron.svg')]$PSChevron
+    } | .>PipeScript
 #>
-[ValidateScript({
-    if ($_ -is [Management.Automation.Language.CommandAst]) {
-        return $_.CommandsElements[0].Value -in 'include','includes'
+[ValidateScript({    
+    $validating = $_    
+    if ($validating -is [Management.Automation.Language.CommandAst]) {
+        return $validating.CommandElements[0].Value -in 'include','includes'
     }
 })]
 [Alias('Includes')]
 param(
 # The File Path to Include
-[Parameter(Mandatory,Position=0)]
+[Parameter(Mandatory,ParameterSetName='VariableAST',Position=0)]
+[Parameter(ParameterSetName='CommandAst',Position=0)]
+[Alias('FullName','Uri','Url')]
 [string]
 $FilePath,
 
@@ -42,11 +49,16 @@ $Passthru,
 [string[]]
 $Exclude = '\.[^\.]+\.ps1$',
 
+# The variable that include will be applied to.
+# If including files with wildcards, this will be the base path.
+# Otherwise, this variable will be assigned to the included value.
 [Parameter(Mandatory,ParameterSetName='VariableAST', ValueFromPipeline)]
 [Management.Automation.Language.VariableExpressionast]
 $VariableAst,
 
 
+# The CommandAST.
+# This is provided by the transpiler when include is used as a keyword.
 [Parameter(Mandatory,ParameterSetName='CommandAst',ValueFromPipeline)]
 [Management.Automation.Language.CommandAst]
 $CommandAst
@@ -74,14 +86,22 @@ process {
                 $ExecutionContext.SessionState.PSVariable.Set($paramName, $mySentence.Parameters[$paramName])
             }
         }
+
+        if ($mySentence.ArgumentList -and -not $FilePath) {
+            $FilePath = $mySentence.ArgumentList[0]
+        }
     }
 
 # 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
+    if ($FilePath -match '^https?://') {
+        $includingUrl = $FilePath -as [uri] 
+    } else {
+        if (-not $FilePath.Contains('*')) {
+            Write-Error "Could not resolve $($FilePath).  Must be a command, path, or URI."
+            return
+        }
     }
 }
 
@@ -167,6 +187,52 @@ if ($Passthru) { [Environment]::NewLine + "`${$($includingCommand.Name)}"}
     } elseif ($includingCommand) {
         IncludeFileContents $includingCommand.Source    
     }
+    elseif ($includingUrl) {
+        $webResponse = Invoke-WebRequest -Uri $includingUrl
+        if ($webResponse.Content -is [string]) {
+            $restResponse = Invoke-RestMethod -Uri $includingUrl
+            if (
+                (
+                    ($restResponse -is [PSObject]) -or 
+                    ($restResponse -is [Object[]])
+                ) -and (
+                    -not ($webResponse.Content -as [xml])
+                ) -and (
+                    $restResponse -isnot [string]
+                )
+            ) {
+                [ScriptBlock]::Create((@(
+                    "@'"
+                    $($restResponse | ConvertTo-Json -Depth 100)
+                    "'@ |"
+                    "    ConvertFrom-JSON"
+                ) -join [Environment]::NewLine))
+            }             
+            else {
+                if ($restResponse -is [string] -and $(
+                    $restScript = try { [scriptblock]::Create($restResponse) } catch { $null }
+                    $restScript
+                )) {
+                    $restScript
+                }
+                else {
+                    [ScriptBlock]::Create((@(
+                        "$(if ($webResponse.Content -as [xml]) { '[xml]'})@'"
+                        $(if ($restResponse -is [xml]) { $restResponse.OuterXML } else { $restResponse })
+                        "'@"                        
+                    ) -join [Environment]::NewLine))
+                }
+            }
+        } elseif ($webResponse.Content -is [byte[]]) {
+            [ScriptBlock]::Create(
+                "@'" + [Environment]::NewLine + 
+                [Convert]::ToBase64String(
+                    $webResponse.Content,
+                    'InsertLineBreaks'
+                ) + [Environment]::NewLine + 
+                "'@")
+        }
+    }
 
 if ($psCmdlet.ParameterSetName -eq 'ScriptBlock' -or 
     $VariableAst.VariablePath -match '^null$') {
@@ -176,9 +242,13 @@ if ($psCmdlet.ParameterSetName -eq 'ScriptBlock' -or
         $includedScript
     }
 
-} elseif ($VariableAst.VariablePath -and $includingCommand) {    
+} elseif ($VariableAst.VariablePath -and $IncludedScript) {    
     [ScriptBlock]::Create("$($VariableAst) = $IncludedScript")
-} elseif ($VariableAst.VariablePath -notmatch '^null$') {
+} 
+elseif ($CommandAst) {
+    $IncludedScript
+}
+elseif ($VariableAst.VariablePath -notmatch '^null$') {
     [ScriptBlock]::Create(@"
 :ToIncludeFiles foreach (`$file in (Get-ChildItem -Path "$($VariableAst)" -Filter "$FilePath" -Recurse)) {
     if (`$file.Extension -ne '.ps1')      { continue }  # Skip if the extension is not .ps1