Get-Splat.ps1
|
function Get-Splat { <# .Synopsis Gets a splat .Description Gets a splat for a command .Link Find-Splat .Link Use-Splat .Example @{id=$pid} | Get-Splat .Example @{id=$Pid} | ?@ # ?@ is an alias for Get-Splat .Example @{id=$pid} | & ${?@} # Get-Splat as a script block #> param( # The command that is being splatted. [Parameter(Mandatory=$true,Position=0)] [PSObject[]] $Command, # The input object [Parameter(Mandatory=$true,ValueFromPipeline=$true,Position=1)] [Alias('InputObject')] [PSObject] $Splat, # If set, will return regardless of if parameters map, are valid, and have enough mandatory parameters [switch] $Force ) begin { # Declare some caches: if (-not ${script:_@p}) { ${script:_@p} = @{} } # * All Parameters if (-not ${script:_@c}) { ${script:_@c} = @{} } # * All commands if (-not ${script:_@mp}) { ${script:_@mp} = @{} } # * All Mandatory Parameters if (-not ${script:_@pp}) { ${script:_@pp} = @{} } # * All Pipelined Parameters $ValidateAttributes = { param( [Parameter(Mandatory)]$value, [Parameter(Mandatory)]$attributes ) foreach ($attr in $attributes) { $_ = $this = $value if ($attr -is [Management.Automation.ValidateScriptAttribute]) { $result = . $attr.ScriptBlock if ($result -ne $true) { $attr } } elseif ($attr -is [Management.Automation.ValidatePatternAttribute] -and (-not [Regex]::new($attr.RegexPattern, $attr.Options, '00:00:05').IsMatch($value)) ) { $attr } elseif ($attr -is [Management.Automation.ValidateSetAttribute] -and $attr.ValidValues -notcontains $value) { $attr } elseif ($attr -is [Management.Automation.ValidateRangeAttribute] -and ( ($value -gt $attr.MaxRange) -or ($value -lt $attr.MinRange) )) {$attr} } } } process { $ap,$ac,$amp = ${script:_@p},${script:_@c}, ${script:_@mp} #region Turn dictionaries into PSObjects if ($Splat -is [Collections.IDictionary]) { if ($splat.GetType().Name -ne 'PSBoundParametersDictionary') { $Splat = [PSCustomObject]$Splat } else { $splat = [PSCustomObject]([Ordered]@{} + $Splat) } } #endregion Turn dictionaries into PSObjects $in = $Splat foreach ($cmd in $Command) { # Walk over each command $rc = if ($ac.$cmd) { # use cache if available, otherwise: $ac.$cmd } elseif ($cmd -is [string]) { # *find it if it's a [string] $fc = $ExecutionContext.SessionState.InvokeCommand.GetCommand($cmd,'Function,Cmdlet,ExternalScript,Alias') $fc = if ($fc -is [Management.Automation.AliasInfo]) { $fc.ResolvedCommand } else { $fc } $ac.$cmd = $fc $fc } elseif ($cmd -is [ScriptBlock]) { # * Make a temporary command if it's a [ScriptBlock] $hc = $cmd.GetHashCode() $ExecutionContext.SessionState.PSVariable.Set("function:f$hc", $cmd) $c = $ExecutionContext.SessionState.InvokeCommand.GetCommand("f$hc",'Function') $ac.$cmd = $c $c } elseif ($cmd -is [Management.Automation.CommandInfo]) { # * Otherwise, use the command info $ac.$cmd = $cmd $cmd } if (-not $rc) {continue} $cmd = $rc $outSplat,$Invalid,$Unmapped,$paramMap,$Pipe,$NoPipe = foreach ($_ in 1..6){[ordered]@{}} $params = [Collections.ArrayList]::new() $props = @($in.psobject.properties) $pc = $props.Count if (-not ${script:_@pp}.$cmd) { ${script:_@pp}.$cmd = @( foreach ($param in $cmd.Parameters.Values) { foreach ($attr in $param.Attributes) { if ($attr.ValueFromPipeline) { $param } } }) } $cmdMd = $cmd -as [Management.Automation.CommandMetaData] $problems = @( foreach ($vfp in ${script:_@pp}.$cmd) { if ($in -is $vfp.ParameterType -or ($vfp.ParameterType.IsArray -and $in -as $vfp.ParameterType) ) { $v = $in $badAttributes = & $ValidateAttributes $v $vfp.Attributes if ($badAttributes) { @{$vfp.Name=$badAttributes} } if (-not $badAttributes -or $Force) { $null = $params.Add($vfp.Name) $pipe[$vfp.Name] = $v $outSplat[$vfp.Name] = $v $paramMap[$vfp.Name] = $vfp.Name $pipelineParameterSets = @(foreach ($attr in $vfp.Attributes) { if ($attr.ParameterSetName) { $attr.ParameterSetName} }) } } } :NextProperty foreach ($prop in $props) { $cp=$cmd.Parameters $pn = $prop.Name $pv = $prop.Value if (-not $cp) { continue } $param = $cp.$pn if (-not $param) { $k = "${cmd}:$pn" $param = if ($ap[$k]) { $ap[$k] } else { foreach ($p in $cp.Values) { foreach ($a in $p.Aliases) { $ap["${cmd}:$a"] = $p } if ($ap[$k]) { $ap[$k]; break } } } } if (-not $param) { $pn continue } $paramMap[$param.Name] = $pn if ($params -contains $param) { continue } $pt=$param.ParameterType $paramSets = @(foreach ($attr in $param.Attributes) { if ($attr.ParameterSetName) { $attr.ParameterSetName } }) if ($pipelineParameterSets) { $ok = $false foreach ($cmdPs in $paramSets) { $ok = $ok -bor ($pipelineParameterSets -contains $cmdPs) } if (-not $ok -and -not $Force) { continue } } $v = $pv -as $pt if (-not $v -and ($pt -eq [ScriptBlock] -or $pt -eq [ScriptBlock[]])) { $sb = try { foreach ($_ in $pv) { [ScriptBlock]::Create($_) }} catch {$null} if ($sb) { $v = $sb } } if ($null -ne $v) { $nv = try { [PSVariable]::new("$pn", $v, 'Private',$param.Attributes) } catch { @{$pn=$_} } if ($nv -is [PSVariable] -or $Force) { $null = $params.Add($param) :CanItPipe do { foreach ($attr in $param.Attributes) { if ($attr.ValueFromPipeline -or $attr.ValueFromPipelineByPropertyName -and ((-not $pipelineParameterSets) -or ($pipelineParameterSets -contains $attr.ParameterSetName)) ) { $pipe[$prop.Name] = $v break CanItPipe } } $NoPipe[$prop.Name] = $v } while ($false) $outSplat[$prop.Name] = $v } if ($nv -isnot [PSVariable]) { $nv } } else { @{$pn = $param} } }) $Mandatory = @{} foreach ($param in $cmdMd.Parameters.Values) { foreach ($a in $param.Attributes) { if (-not $a.Mandatory) { continue } if ($a -isnot [Management.Automation.ParameterAttribute]) { continue } if (-not $Mandatory[$a.ParameterSetName]) { $Mandatory[$a.ParameterSetName] = [Ordered]@{} } $mp = ($paramMap.($param.Name)) $Mandatory[$a.ParameterSetName].($param.Name) = if ($mp) { if ($pipelineParameterName -contains $param.Name) { $in } else { $outSplat.$mp } } } } $amp.$cmd = $Mandatory $mandatory = $amp.$cmd $missingMandatory = @{} foreach ($m in $Mandatory.GetEnumerator()) { $missingMandatory[$m.Key] = @(foreach ($_ in $m.value.GetEnumerator()) { if ($null -eq $_.Value) { $_.Key } }) } $couldRun = if (-not $Mandatory.Count) { $true } elseif ($missingMandatory.'__AllParameterSets') { $false } else { foreach ($_ in $missingMandatory.GetEnumerator()) { if (-not $_.Value) { $true;break } } } if (-not $couldRun -and -not $Force) { continue } foreach ($p in $problems) { if ($p -is [Hashtable]) { $Invalid += $p } else { $Unmapped[$p] = $in.$p } } if ($Invalid.Count -eq 0) { $Invalid = $null } if ($Unmapped.Count -eq 0) { $Unmapped = $null } $realCmd = if ($cmd -is [Management.Automation.FunctionInfo] -and $cmd.Name.Contains($cmd.ScriptBlock.GetHashCode().ToString())) { $cmd.ScriptBlock } else { $cmd } foreach($_ in ([Ordered]@{ Command = $realCmd CouldRun = $couldRun Invalid = $Invalid Missing = $missingMandatory PercentFit = $(if ($pc) {$outSplat.Count / $pc } else { 0}) Unmapped = $Unmapped PipelineParameter = $Pipe NonPipelineParameter = $NoPipe }).GetEnumerator()) { $outSplat.psobject.properties.Add([Management.Automation.PSNoteProperty]::new($_.Key,$_.Value)) } $outSplat } } } |