WieldingLocation

0.0.3-Alpha

WieldingLocation.psm1

class QLocationData {
    $Locations = @{}
    [string]$LastLocation
}

$QuickLocation = New-Object -TypeName QLocationData

function Test-IsDirectory {
    param (
        $item
    )

    foreach ($attribute in $item.Attributes) {
        $isReparsePoint = ($attribute -band [System.IO.FileAttributes]::ReparsePoint) -eq [System.IO.FileAttributes]::ReparsePoint
        $isDirectory = ($attribute -band [System.IO.FileAttributes]::Directory) -eq [System.IO.FileAttributes]::Directory

        if ($isReparsePoint -or $isDirectory) {
            return $True
        }
    }

    return $False
}

function Set-QuickLocation {
    <#
 .SYNOPSIS
    Sets or changes to a stored directory location.
 
 .DESCRIPTION
    Sets or changes to a stored directory location.
 
 .PARAMETER Alias
    The short name to use to reference the directory
 
 .PARAMETER Location
    The folder that will be referenced by the Alias value
 
  .EXAMPLE
    Set-QuickLocation -Alias doc -Location "$($env:USERPROFILE)\Documents"
    Sets a quick location named "doc" to the users Documents folder
 
 .EXAMPLE
    Set-QuickLocation -Alias doc
    Changes to the directory that has an alias of "doc"
 
 .EXAMPLE
    Set-QuickLocation
    Switches to the last directory that was changed to using Set-QuickLocation.
 
 .NOTES
    Author: Andrew Kunkel
     
 
 .LINK
    https://github.com/Wielding/WieldingLocation
     
#>        
    [CmdletBinding()]
    param (
        [string]$Alias = "",
        [string]$Location = ""
    )

    $hasAlias = ($Alias -ne "")
    $hasLocation = ($Location -ne "")


    if (!$hasAlias -and !$hasLocation) {
        $newLocation = $QuickLocation.LastLocation
        $QuickLocation.LastLocation = $PWD
        Set-Location -Path $newLocation
        return
    }

    if ($hasAlias) {
        if (!$hasLocation) {
            if ($QuickLocation.Locations.Contains($Alias)) {
                $item = Get-ItemProperty $QuickLocation.Locations[$Alias]
                if (Test-IsDirectory $item) {
                    $QuickLocation.LastLocation = $PWD
                    Set-Location -Path $item.FullName
                }
                else {
                    Invoke-Item $item.FullName
                }
                return
            }
            else {
                Write-Error "Location name [$Alias] does not exist"
                return
            }
        }
        else {
            if (Test-Path $Location) {
                try {
                    $item = Get-ItemProperty $Location
                }
                catch {
                    Write-Error "Error accessing location [$item]"
                    return
                }
            }
            else {
                Write-Error "Unknown location [$Location]"
                return
            }
            $QuickLocation.Locations[$Alias] = $item.FullName
        }
    }
}

function Show-QuickLocation {
    $QuickLocation.Locations
}

$locationCompleter = {
    param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters)

    $QuickLocation.Locations.Keys | Sort-Object | ForEach-Object -Process { if ($_.StartsWith($wordToComplete)) { $_ } }
}

Set-Alias -Name "ql" -Value Set-QuickLocation
Set-Alias -Name "qll" -Value Show-QuickLocation

Register-ArgumentCompleter -CommandName Set-QuickLocation -ParameterName Alias -ScriptBlock $locationCompleter

Export-ModuleMember -Function Out-Default, 'Set-QuickLocation'
Export-ModuleMember -Function Out-Default, 'Show-QuickLocation'
Export-ModuleMember -Variable 'QuickLocation'
Export-ModuleMember -Alias 'ql'
Export-ModuleMember -Alias 'qll'