poshy-ecks

0.5.22

src/New-ViewerInvocationFunction.ps1

#!/usr/bin/env pwsh
$ErrorActionPreference = "Stop"
Set-StrictMode -Version Latest


<#
.SYNOPSIS
    Creates a new wrapper function that invokes a binary with the given arguments.
.PARAMETER FunctionName
    The name of the function to create.
.PARAMETER Bin
    The path to the binary to invoke.
.PARAMETER AcceptsInputPiped
    Whether the function accepts input from the pipeline.
.PARAMETER Force
    Whether to overwrite an existing function.
.PARAMETER PassThru
    Whether to return the function object.
.PARAMETER BinArgs
    The arguments to pass to the binary.
.OUTPUTS
    A new wrapper function that invokes a binary with the given arguments.
#>
function New-ViewerInvocationFunction {
    param(
        [Parameter(Mandatory = $true, Position = 0, HelpMessage = "The name of the function to create.")]
        [ValidateNotNullOrEmpty()]
        [string] $FunctionName,

        [Parameter(Mandatory = $true, Position = 1, ValueFromPipelineByPropertyName = $true)]
        [ValidateNotNullOrEmpty()]
        [string] $Bin,

        [Parameter(Mandatory = $false, Position = 2, ValueFromPipelineByPropertyName = $true)]
        [bool] $AcceptsInputPiped = $false,

        [Parameter(Mandatory = $false, Position = 3)]
        [switch] $Force,

        [Parameter(Mandatory = $false, Position = 4)]
        [switch] $PassThru,

        [Parameter(Mandatory = $false, Position = 5, ValueFromPipelineByPropertyName = $true)]
        [string[]] $BinArgs = @()
    )
    if ($AcceptsInputPiped) {
        $result = New-Item -Path Function:\ -Name "Global:$FunctionName" -Value {
            [CmdletBinding(DefaultParameterSetName = "Path")]
            param(
                [Parameter(Mandatory = $false, Position = 0, ValueFromPipeline = $false, ValueFromPipelineByPropertyName=$true, ParameterSetName = "Path")]
                [ValidateNotNullOrEmpty()]
                [string[]] $Path,

                [Parameter(Mandatory = $true, Position = 1, ValueFromPipeline = $false, ValueFromPipelineByPropertyName=$true, ParameterSetName = "LiteralPath")]
                [ValidateNotNullOrEmpty()]
                [string[]] $LiteralPath,

                [Parameter(Mandatory = $true, Position = 0, ValueFromPipeline = $true, ParameterSetName = "InputObject")]
                $InputObject
            )
            Begin {
                Write-Debug "${FunctionName}[AcceptsInputPiped=true]: `$MyInvocation.ExpectingInput: $($MyInvocation.ExpectingInput)"
                Write-Debug "${FunctionName}[AcceptsInputPiped=true]: `$PSCmdlet.ParameterSetName: $($PSCmdlet.ParameterSetName)"
                Write-Debug "${FunctionName}[AcceptsInputPiped=true]: `$Path: $($Path)"
                Write-Debug "${FunctionName}[AcceptsInputPiped=true]: `$LiteralPath: $($LiteralPath)"
                Write-Debug "${FunctionName}[AcceptsInputPiped=true]: `$Bin: $($Bin)"
                if ($DebugPreference -eq "Continue") {
                    Write-Debug "${FunctionName}[AcceptsInputPiped=true]: `$BinArgs: $($BinArgs | ConvertTo-Json)"
                }
                if ($MyInvocation.ExpectingInput -or $PSCmdlet.ParameterSetName -eq "InputObject") {
                    $scriptCmd = {
                        &$Bin @BinArgs
                    }
                } elseif ($PSCmdlet.ParameterSetName -eq "Path") {
                    $scriptCmd = {
                        Get-Item -Path $Path -Force | ForEach-Object {
                            &$Bin @BinArgs $_.FullName
                        }
                    }
                } elseif ($PSCmdlet.ParameterSetName -eq "LiteralPath") {
                    $scriptCmd = {
                        Get-Item -LiteralPath $LiteralPath -Force | ForEach-Object {
                            &$Bin @BinArgs $_.FullName
                        }
                    }
                } else {
                    Write-Error "No input provided."
                    return
                }
                $steppablePipeline = $scriptCmd.GetSteppablePipeline($myInvocation.CommandOrigin)
                $cleaned = $false
                try {
                    $steppablePipeline.Begin($PSCmdlet)
                }
                catch {
                    if (-not $cleaned) {
                        if ($null -ne $steppablePipeline) {
                            $steppablePipeline.Clean()
                        }
                        $cleaned = $true
                    }
                }
            }
            Process {
                try {
                    if ($MyInvocation.ExpectingInput -or $PSCmdlet.ParameterSetName -eq "InputObject") {
                        $steppablePipeline.Process($_)
                    } else {
                        $steppablePipeline.Process()
                    }
                }
                catch {
                    if (-not $cleaned) {
                        if ($null -ne $steppablePipeline) {
                            $steppablePipeline.Clean()
                        }
                        $cleaned = $true
                    }
                }
            }
            End {
                try {
                    $steppablePipeline.End()
                } finally {
                    if (-not $cleaned) {
                        if ($null -ne $steppablePipeline) {
                            $steppablePipeline.Clean()
                        }
                        $cleaned = $true
                    }
                }
            }
        }.GetNewClosure() -Options AllScope -Force:$Force
        if ($PassThru) {
            $result
        }
    } else {
        $result = New-Item -Path Function:\ -Name "Global:$FunctionName" -Value {
            [CmdletBinding(DefaultParameterSetName = "Path")]
            param(
                [Parameter(Mandatory = $true, Position = 0, ValueFromPipeline = $false, ValueFromPipelineByPropertyName=$true, ParameterSetName = "Path")]
                [ValidateNotNullOrEmpty()]
                [string[]] $Path,

                [Parameter(Mandatory = $true, Position = 0, ValueFromPipeline = $false, ValueFromPipelineByPropertyName=$true, ParameterSetName = "LiteralPath")]
                [ValidateNotNullOrEmpty()]
                [string[]] $LiteralPath
            )
            Write-Debug "${FunctionName}[AcceptsInputPiped=false]: `$MyInvocation.ExpectingInput: $($MyInvocation.ExpectingInput)"
            Write-Debug "${FunctionName}[AcceptsInputPiped=false]: `$PSCmdlet.ParameterSetName: $($PSCmdlet.ParameterSetName)"
            Write-Debug "${FunctionName}[AcceptsInputPiped=false]: `$Path: $($Path)"
            Write-Debug "${FunctionName}[AcceptsInputPiped=false]: `$LiteralPath: $($LiteralPath)"
            Write-Debug "${FunctionName}[AcceptsInputPiped=false]: `$Bin: $($Bin)"
            if ($DebugPreference -eq "Continue") {
                Write-Debug "${FunctionName}[AcceptsInputPiped=false]: `$BinArgs: $($BinArgs | ConvertTo-Json)"
            }
            if ($PSCmdlet.ParameterSetName -eq "Path") {
                Get-Item -Path $Path -Force | ForEach-Object {
                    &$Bin @BinArgs $_.FullName
                }
            } elseif ($PSCmdlet.ParameterSetName -eq "LiteralPath") {
                Get-Item -LiteralPath $LiteralPath -Force | ForEach-Object {
                    &$Bin @BinArgs $_.FullName
                }
            } else {
                Write-Error "No input provided."
            }
        }.GetNewClosure() -Options AllScope -Force:$Force
        if ($PassThru) {
            $result
        }
    }
}