splattyps

0.1

SplattyPS.psm1

Set-StrictMode -Version 'Latest'

function Get-SplattedCommand {
    param (
        [Parameter(Mandatory = $true, Position = 0, ValueFromPipeline = $true, ValueFromPipelineByPropertyName = $true)]
        [ValidateNotNullOrEmpty()]        
        [string]$Name,

        [ValidateNotNullOrEmpty()]
        [Alias('htn')]
        [string]$HashTableName = 'Params',

        [ValidateNotNullOrEmpty()]
        [Alias('ps')]
        [string]$ParameterSet,

        [Alias('All')]
        [switch]$AllParameters,
    
        [Alias('icp')]
        [switch]$IncludeCommonParameters,
        
        [Alias('esb')]
        [switch]$EmitScriptBlock,
        
        [Alias('sth')]
        [switch]$ShowTypeHint,

        [Alias('c')]
        [switch]$CompactHashTable,

        [Alias('ILoveSemis')]
        [switch]$ILoveSemiColons,

        [Alias('is')]
        [int]$IndentSize = 4,

        [Alias('il')]
        [int]$IndentLevel = 0,

        [Alias('ut', 'nooooo')]
        [Switch]$UseTabs,

        [Alias('ra')]
        [Switch]$ResolveAlias,

        [Alias('avn')]
        [ValidateNotNullOrEmpty()]
        [string]$AssignmentVariableName
    )

    begin {
        $CommonParamNames = 'Debug', 'Verbose', 'ErrorAction', 'WarningAction', 'InformationAction', 'ErrorVariable', 'WarningVariable', 'InformationVariable', 'OutVariable', 'OutBuffer', 'PipelineVariable', 'WhatIf', 'Confirm'

        $IndentChar = ' '

        if ($UseTabs.IsPresent) {
            $IndentChar = "`t"

            if (-not $PSBoundParameters.ContainsKey('IndentSize')) {
                $IndentSize = 1
            }
        }
        
        $Indent = $IndentChar * $IndentSize

        if ($ILoveSemiColons.IsPresent) {
            $LineEnding = ';'
        } 
        else {
            $LineEnding = ''
        }

        $Assignment = ''

        if ($AssignmentVariableName) {
            $Assignment = "`$$AssignmentVariableName = "
        }
    }   

    process {
        $Command = (Get-Command $Name)

        if ($Command.CommandType -eq 'Alias') {
            $Command = (Get-Command $Command.Definition)
            if ($ResolveAlias.IsPresent) {
                $Name = $Command.Name
            }
        }

        $DefaultParameterSet = $PSBoundParameters["ParameterSet"]
        
        if (!$DefaultParameterSet) {
            $DefaultParameterSet = $Command.DefaultParameterSet
        }
    
        if (!$DefaultParameterSet) {
            $DefaultParameterSet = "__AllParameterSets"
        }

        Write-Verbose "DefaultParameterSet = $DefaultParameterSet"
          
        $Output = @()
        
        if ($EmitScriptBlock.IsPresent) {
            $Output += "$($Indent * $IndentLevel)Invoke-Command {"
            $IndentLevel++
        }        
        $Output += "$($Indent * $IndentLevel)`$$HashTableName = @{"
        $IndentLevel++       

        $MaxLength = 0

        ($Command.ParameterSets | Where-Object Name -eq $DefaultParameterSet).Parameters | ForEach-Object {        
            if ($AllParameters.IsPresent -or $_.IsMandatory) {
                $MaxLength = [math]::Max($_.Name.Length, $MaxLength)
            }
        }                    

        $MaxLength = $MaxLength + ($MaxLength % 4)

        ($Command.ParameterSets | Where-Object Name -eq $DefaultParameterSet).Parameters | ForEach-Object {
            if ($AllParameters.IsPresent -or $_.IsMandatory) {
                if ($_.IsMandatory) {
                    $MandatoryComment = '#mandatory'
                }
                else {
                    $MandatoryComment = '#optional'
                }

                if ($ShowTypeHint.IsPresent) {
                    $TypeHint = $_.ParameterType.ToString()
                } 
                else {
                    $TypeHint = ''
                }

                if ($IncludeCommonParameters -or $_.Name -inotin $CommonParamNames) {
                    $x = ' ' * ($MaxLength - $_.Name.Length)
                    $Output += "$($Indent * $IndentLevel)$($_.Name) = $LineEnding$($x) $mandatoryComment $TypeHint"
                }
            }    
        }
    
        $IndentLevel--
        $Output += "$($Indent * $IndentLevel)}`r`n`r`n"
        $Output += "$($Indent * $IndentLevel)$Assignment$($name) @$($HashTableName)$($LineEnding)"

        if ($EmitScriptBlock.IsPresent) {
            $IndentLevel--
            $Output += "$($Indent * $IndentLevel)}"
        }

        Write-Output $Output    
    }    
}