PSKoans

0.65.4

Public/Set-PSKoanLocation.ps1

using namespace System.Management.Automation

class FolderTransformAttribute : ArgumentTransformationAttribute {
    [object] Transform([EngineIntrinsics]$engineIntrinsics, [object] $inputData) {
        switch ($inputData) {

            { $_ -is [string] } {
                if (-not (Test-Path $_ -IsValid -PathType Container)) {
                    throw [ArgumentTransformationMetadataException]::new(
                        "Could not resolve path: $_",
                        $_.Exception
                    )
                }

                return $engineIntrinsics.SessionState.Path.GetUnresolvedProviderPathFromPSPath($_)
            }

            { $_ -is [System.IO.FileSystemInfo] } {

                if (-not (Test-Path -Path $_.FullName -PathType Container)) {
                    throw [ArgumentTransformationMetadataException]::new(
                        'Path could not be resolved to a valid container.'
                    )
                }
                else {
                    return $inputData.Fullname
                }

            }
        }

        throw [System.IO.FileNotFoundException]::new()
    }
}

function Set-PSKoanLocation {
    [CmdletBinding(SupportsShouldProcess, ConfirmImpact = 'Medium',
        HelpUri = 'https://github.com/vexx32/PSKoans/tree/master/docs/Set-PSKoanLocation.md')]
    [OutputType([void])]
    param(
        [Parameter(Mandatory, Position = 0)]
        [Alias('PSPath', 'Folder')]
        [FolderTransformAttribute()]
        [string]
        $Path,

        [Parameter()]
        [switch]
        $PassThru
    )
    process {
        if ($PSCmdlet.ShouldProcess("Set PSKoans folder location to '$Path'")) {
            Set-PSKoanSetting -Name KoanLocation -Value $Path
        }
        else {
            Write-Warning "PSKoans folder location has not been changed."
        }

        if ($PassThru) {
            $Path
        }
    }
}