Goto.psm1
|
$script:GotoDataPath = Join-Path -Path $env:APPDATA -ChildPath 'PowerShell\Goto' $script:AliasFilePath = Join-Path -Path $script:GotoDataPath -ChildPath "GotoAliases.xml" $script:UpdateInfoFile = Join-Path -Path $script:GotoDataPath -ChildPath "UpdateInfo.xml" $script:BackupFilePath = Join-Path -Path $script:GotoDataPath -ChildPath "GotoAliases.bak" $Global:DirectoryAliases = @{} function Backup-Aliases { if (Test-Path $script:AliasFilePath) { Copy-Item -Path $script:AliasFilePath -Destination $script:BackupFilePath -Force Write-Host "Aliases backed up to $script:BackupFilePath" -ForegroundColor Green } } function Restore-AliasesFromBackup { if (Test-Path $script:BackupFilePath) { Copy-Item -Path $script:BackupFilePath -Destination $script:AliasFilePath -Force Write-Host "Aliases restored from backup." -ForegroundColor Green Import-Aliases } else { Write-Warning "No backup file found. Unable to restore aliases." } } function Update-GotoModule { try { Backup-Aliases $currentAliases = $Global:DirectoryAliases.Clone() $currentModule = Get-InstalledModule -Name Goto -ErrorAction Stop $updateInfo = Test-ForUpdates if ($updateInfo) { Write-Host "Current Goto version: $($updateInfo.CurrentVersion)" -ForegroundColor Cyan Write-Host "New version available: $($updateInfo.NewVersion)" -ForegroundColor Green $confirmation = Read-Host "Do you want to update now? (Y/N)" if ($confirmation -eq 'Y') { # Let's try to use Update-Module try { Update-Module -Name Goto -Force -ErrorAction Stop $updatedModule = Get-InstalledModule -Name Goto if ($updatedModule.Version -gt $currentModule.Version) { Write-Host "Goto has been successfully updated to version $($updatedModule.Version)!" -ForegroundColor Green $Global:DirectoryAliases = $currentAliases Save-Aliases Write-Host "Aliases have been restored." -ForegroundColor Green Initialize-GotoEnvironment Write-Host "Please restart your PowerShell session to use the new version." -ForegroundColor Yellow } else { throw "Update failed: Version did not change" } } catch { Write-Host "Error during Update-Module. Trying alternative update method..." -ForegroundColor Yellow # Alternative update method try { Uninstall-Module -Name Goto -AllVersions -Force -ErrorAction Stop Install-Module -Name Goto -Force -AllowClobber -ErrorAction Stop $updatedModule = Get-InstalledModule -Name Goto if ($updatedModule.Version -gt $currentModule.Version) { Write-Host "Goto has been successfully updated to version $($updatedModule.Version)!" -ForegroundColor Green $Global:DirectoryAliases = $currentAliases Save-Aliases Write-Host "Aliases have been restored." -ForegroundColor Green Initialize-GotoEnvironment Write-Host "Please restart your PowerShell session to use the new version." -ForegroundColor Yellow } else { throw "Alternative update failed: Version did not change" } } catch { throw "Both update methods failed. Error: $_" } } } else { Write-Host "Update cancelled." -ForegroundColor Yellow } } else { Write-Host "Goto is already at the latest version ($($currentModule.Version))." -ForegroundColor Green } } catch { Write-Host "An error occurred while updating Goto: $_" -ForegroundColor Red Write-Host "Error details: $($_.Exception.Message)" -ForegroundColor Red Write-Host "Stack trace: $($_.ScriptStackTrace)" -ForegroundColor Red Restore-AliasesFromBackup } } function Test-ForUpdates { $currentModule = Get-InstalledModule -Name Goto -ErrorAction SilentlyContinue if (-not $currentModule) { return $null } try { $onlineModule = Find-Module -Name Goto -Repository PSGallery -ErrorAction Stop $newVersion = $onlineModule.Version } catch { Write-Warning "Unable to check for updates. Please check your internet connection and try again later." return $null } if ($newVersion -gt $currentModule.Version) { return @{ CurrentVersion = $currentModule.Version NewVersion = $newVersion } } return $null } function Initialize-GotoEnvironment { if (-not (Test-Path -Path $script:GotoDataPath)) { New-Item -Path $script:GotoDataPath -ItemType Directory -Force | Out-Null } Import-Aliases if (-not (Test-Path -Path $script:AliasFilePath)) { Save-Aliases } $profileContent = @" # Import Goto module Import-Module Goto -DisableNameChecking # Error handling for WinGet `$ErrorActionPreference = 'Continue' try { `$null = Get-Command winget -ErrorAction Stop Import-Module Microsoft.WinGet.Client -ErrorAction SilentlyContinue } catch { Write-Verbose "WinGet not detected. Some features may be unavailable." } "@ if (-not (Test-Path -Path $PROFILE)) { New-Item -Path $PROFILE -ItemType File -Force | Out-Null } $currentProfileContent = Get-Content -Path $PROFILE -Raw -ErrorAction SilentlyContinue if ($currentProfileContent -notmatch [regex]::Escape($profileContent)) { Add-Content -Path $PROFILE -Value $profileContent Write-Host "Goto module import and WinGet error handling have been added to your PowerShell profile." -ForegroundColor Green Write-Host "Please restart your PowerShell session or run '. `$PROFILE' to apply changes." -ForegroundColor Yellow } # Asynchronous verification of updates every 30 days $lastCheckFile = Join-Path $script:GotoDataPath "LastUpdateCheck.txt" $lastCheck = if (Test-Path $lastCheckFile) { Get-Content $lastCheckFile } else { "1970-01-01" } if ((Get-Date) -gt ([DateTime]::Parse($lastCheck).AddDays(30))) { Start-Job -ScriptBlock { Import-Module Goto $updateInfo = Test-ForUpdates if ($updateInfo) { Write-Host "New version $($updateInfo.NewVersion) of Goto is available. Your version: $($updateInfo.CurrentVersion). Run 'goto -Update' to update." -ForegroundColor Yellow } Get-Date -Format "yyyy-MM-dd" | Set-Content $using:lastCheckFile } | Out-Null } } function Import-Aliases { if (Test-Path $script:AliasFilePath) { try { $Global:DirectoryAliases = Import-Clixml -Path $script:AliasFilePath $Global:DirectoryAliases.GetEnumerator() | ForEach-Object { $alias = $_.Key $path = $_.Value Set-Alias -Name $alias -Value (Get-Command -Name Set-Location) -Force -Scope Global Set-Item -Path "Alias:$alias" -Value $path -Force } } catch { Write-Warning "Failed to load aliases. Recreating alias file." $Global:DirectoryAliases = @{} Save-Aliases } } else { $Global:DirectoryAliases = @{} Save-Aliases } } function Save-Aliases { try { $Global:DirectoryAliases | Export-Clixml -Path $script:AliasFilePath -ErrorAction Stop Import-Aliases } catch { Write-Warning "Failed to save aliases. Error: $_" } } function _goto_print_similar { param( [string]$aliasInput, [string]$action = "navigate to" ) $normalizedInput = $aliasInput.ToLower() $matchingAliases = $Global:DirectoryAliases.Keys | Where-Object { $alias = $_.ToLower() $inputChars = $normalizedInput.ToCharArray() $aliasChars = $alias.ToCharArray() $matchingCharsCount = ($inputChars | Where-Object { $aliasChars -contains $_ }).Count $matchingCharsCount -eq $inputChars.Count } $matchedAliases = $matchingAliases | ForEach-Object { $alias = $_ $aliasChars = $alias.ToLower().ToCharArray() $matchingCharsCount = ($normalizedInput.ToCharArray() | Where-Object { $aliasChars -contains $_ }).Count [PSCustomObject]@{ Alias = $alias Path = $Global:DirectoryAliases[$alias] Score = $matchingCharsCount } } | Sort-Object -Property Score -Descending if ($matchedAliases.Count -eq 1) { $selectedAlias = $matchedAliases[0].Alias $path = $matchedAliases[0].Path Write-Host "Only one matching alias found: '$selectedAlias' -> '$path'. Navigating..." -ForegroundColor Green Set-Location $path return $selectedAlias } elseif ($matchedAliases.Count -gt 1) { Write-Host "Did you mean one of these? Type the number to $action, or press ENTER to cancel:" -ForegroundColor Yellow Write-Host $maxAliasLength = ($matchedAliases | Measure-Object -Property { $_.Alias.Length } -Maximum).Maximum $numberWidth = $matchedAliases.Count.ToString().Length for ($i = 0; $i -lt $matchedAliases.Count; $i++) { $alias = $matchedAliases[$i].Alias $path = $matchedAliases[$i].Path $paddedAlias = $alias.PadRight($maxAliasLength) $paddedNumber = ($i + 1).ToString().PadLeft($numberWidth) Write-Host "[" -NoNewline -ForegroundColor DarkGray Write-Host $paddedNumber -NoNewline -ForegroundColor Cyan Write-Host "]:" -NoNewline -ForegroundColor DarkGray Write-Host " $paddedAlias" -NoNewline -ForegroundColor Green Write-Host " -> " -NoNewline -ForegroundColor DarkGray Write-Host $path -ForegroundColor Yellow } Write-Host "" # Additional indentation before selection $choice = Read-Host "Enter your choice (1-$($matchedAliases.Count))" if ([string]::IsNullOrWhiteSpace($choice)) { Write-Host "No selection made. No action taken." -ForegroundColor Red } elseif ($choice -match '^\d+$' -and [int]$choice -ge 1 -and [int]$choice -le $matchedAliases.Count) { $selectedAlias = $matchedAliases[[int]$choice - 1].Alias $path = $matchedAliases[[int]$choice - 1].Path Write-Host "Navigating to '$selectedAlias' -> '$path'." -ForegroundColor Green Set-Location $path return $selectedAlias } else { Write-Host "Invalid selection. No action taken." -ForegroundColor Red } } else { Write-Host "No similar aliases found for '$aliasInput'." -ForegroundColor Red } return $null } function goto { [CmdletBinding(DefaultParameterSetName = 'Navigate')] param( [Parameter(ParameterSetName = 'Navigate', Position = 0)] [Parameter(ParameterSetName = 'Unregister', Position = 1)] [Parameter(ParameterSetName = 'Expand', Position = 1)] [Parameter(ParameterSetName = 'Push', Position = 1)] [string]$Alias, [Parameter(ParameterSetName = 'Register', Mandatory = $true)] [Alias('r')] [switch]$Register, [Parameter(ParameterSetName = 'Unregister', Mandatory = $true)] [Alias('u')] [switch]$Unregister, [Parameter(ParameterSetName = 'Push', Mandatory = $true)] [Alias('p')] [switch]$Push, [Parameter(ParameterSetName = 'Pop', Mandatory = $true)] [Alias('o')] [switch]$Pop, [Parameter(ParameterSetName = 'List', Mandatory = $true)] [Alias('l')] [switch]$List, [Parameter(ParameterSetName = 'Expand', Mandatory = $true)] [Alias('x')] [switch]$Expand, [Parameter(ParameterSetName = 'Cleanup', Mandatory = $true)] [Alias('c')] [switch]$Cleanup, [Parameter(ParameterSetName = 'Help', Mandatory = $true)] [Alias('h')] [switch]$Help, [Parameter(ParameterSetName = 'Update', Mandatory = $true)] [switch]$Update, [Parameter(ParameterSetName = 'Version', Mandatory = $true)] [Alias('v')] [switch]$Version, [Parameter(ParameterSetName = 'Register', Position = 1)] [Parameter(ParameterSetName = 'Navigate', Position = 1)] [string]$Path ) switch ($PSCmdlet.ParameterSetName) { 'Register' { if (-not [string]::IsNullOrWhiteSpace($Alias) -and -not [string]::IsNullOrWhiteSpace($Path)) { if (Test-Path $Path) { $resolvedPath = Resolve-Path -Path $Path | Select-Object -ExpandProperty Path $Global:DirectoryAliases[$Alias] = $resolvedPath New-Item -Path Function:\ -Name "Global:$Alias" -Value { Set-Location $resolvedPath } | Out-Null Write-Host "Alias '$Alias' registered for path '$resolvedPath'." -ForegroundColor Green Save-Aliases } else { Write-Host "The specified path does not exist." -ForegroundColor Red } } else { Write-Host "Alias name or path not specified. Usage: goto r <alias_name> <path>" -ForegroundColor Yellow } } 'Unregister' { if (-not $Alias) { Write-Host "Usage: goto u <alias>" -ForegroundColor Yellow return } $selectedAlias = _goto_print_similar -aliasInput $Alias -action "unregister" if ($selectedAlias) { Write-Host "Are you sure you want to unregister the alias '$selectedAlias' which points to '$($Global:DirectoryAliases[$selectedAlias])'? [Y/N]: " -ForegroundColor Yellow -NoNewline $confirmation = Read-Host if ($confirmation -eq 'Y') { $Global:DirectoryAliases.Remove($selectedAlias) Remove-Item -Path "Function:$selectedAlias" -ErrorAction SilentlyContinue Write-Host "Alias '$selectedAlias' unregistered." -ForegroundColor Green Save-Aliases } else { Write-Host "Unregistration cancelled." -ForegroundColor Yellow } } else { Write-Host "No matching alias found for '$Alias'." -ForegroundColor Red } } 'List' { if ($Global:DirectoryAliases.Count -eq 0) { Write-Host "No aliases registered." -ForegroundColor Yellow } else { $maxAliasLength = ($Global:DirectoryAliases.Keys | Measure-Object -Maximum -Property Length).Maximum $maxPathLength = ($Global:DirectoryAliases.Values | Measure-Object -Maximum -Property Length).Maximum Write-Host "`nRegistered Aliases:" -ForegroundColor Cyan Write-Host ("-" * (4 + $maxAliasLength + 4 + $maxPathLength)) -ForegroundColor DarkGray $Global:DirectoryAliases.GetEnumerator() | Sort-Object Name | ForEach-Object { $aliasDisplay = $_.Key.PadRight($maxAliasLength) $pathDisplay = $_.Value.PadRight($maxPathLength) Write-Host " " -NoNewline Write-Host $aliasDisplay -ForegroundColor Green -NoNewline Write-Host " -> " -ForegroundColor DarkGray -NoNewline Write-Host $pathDisplay -ForegroundColor Yellow } Write-Host ("-" * (4 + $maxAliasLength + 4 + $maxPathLength)) -ForegroundColor DarkGray } } 'Expand' { if (-not $Alias) { Write-Host "Usage: goto x <alias>" -ForegroundColor Yellow return } $selectedAlias = _goto_print_similar -aliasInput $Alias -action "expand" if ($selectedAlias) { $path = $Global:DirectoryAliases[$selectedAlias] Write-Host "Alias: " -ForegroundColor Cyan -NoNewline Write-Host "$selectedAlias" -ForegroundColor Green Write-Host "Path: " -ForegroundColor Cyan -NoNewline Write-Host "$path" -ForegroundColor Yellow if (Test-Path $path) { Write-Host "Status:" -ForegroundColor Cyan -NoNewline Write-Host " Path exists" -ForegroundColor Green } else { Write-Host "Status:" -ForegroundColor Cyan -NoNewline Write-Host " Path does not exist" -ForegroundColor Red } } else { Write-Host "No matching alias found for '$Alias'." -ForegroundColor Red } } 'Cleanup' { $invalidAliases = $Global:DirectoryAliases.GetEnumerator() | Where-Object { -not (Test-Path $_.Value) } if ($invalidAliases.Count -eq 0) { Write-Host "All aliases point to valid paths. No cleanup needed." -ForegroundColor Green return } Write-Host "The following aliases point to non-existent paths:" -ForegroundColor Yellow foreach ($invalidAlias in $invalidAliases) { Write-Host " $($invalidAlias.Key) -> $($invalidAlias.Value)" -ForegroundColor Red } $confirmation = Read-Host "Do you want to remove these aliases? [Y/N]" if ($confirmation -eq 'Y') { foreach ($invalidAlias in $invalidAliases) { $Global:DirectoryAliases.Remove($invalidAlias.Key) Remove-Item -Path "Function:$($invalidAlias.Key)" -ErrorAction SilentlyContinue Write-Host "Removed alias '$($invalidAlias.Key)'." -ForegroundColor Green } Save-Aliases Write-Host "Cleanup complete. Removed $(($invalidAliases | Measure-Object).Count) invalid aliases." -ForegroundColor Green } else { Write-Host "Cleanup cancelled. No aliases were removed." -ForegroundColor Yellow } } 'Help' { function Show-Help { $helpText = @" usage: goto [<option>] <alias> [<directory>] default usage: goto <alias> - changes to the directory registered for the given alias OPTIONS: -r, -Register: registers an alias goto -r|-Register <alias> <directory> -u, -Unregister: unregisters an alias goto -u|-Unregister <alias> -p, -Push: pushes the current directory onto the stack, then performs goto goto -p|-Push <alias> -o, -Pop: pops the top directory from the stack, then changes to that directory goto -o|-Pop -l, -List: lists aliases goto -l|-List -x, -Expand: expands an alias goto -x|-Expand <alias> -c, -Cleanup: cleans up non existent directory aliases goto -c|-Cleanup -h, -Help: prints this help goto -h|-Help -v, -Version: displays the version of the goto module goto -v|-Version -Update: updates the goto module goto -Update "@ Write-Host $helpText } Show-Help } 'Push' { $currentPath = (Get-Location).Path Push-Location if ($Global:DirectoryAliases.ContainsKey($Alias)) { $path = $Global:DirectoryAliases[$Alias] Write-Host "Pushed current directory '$currentPath' to stack and moved to alias '$Alias' at path '$path'." -ForegroundColor Green Set-Location $path } else { if (-not [string]::IsNullOrWhiteSpace($Alias)) { $selectedAlias = _goto_print_similar -aliasInput $Alias if ($selectedAlias) { $path = $Global:DirectoryAliases[$selectedAlias] Write-Host "Pushed current directory '$currentPath' to stack and moved to alias '$selectedAlias' at path '$path'." -ForegroundColor Green Set-Location $path } else { Pop-Location Write-Host "No matching alias found. Current directory was not changed." -ForegroundColor Yellow } } else { Write-Host "Pushed current directory '$currentPath' to stack." -ForegroundColor Yellow } } } 'Pop' { if ((Get-Location -Stack).Count -eq 0) { Write-Host "The location stack is empty." -ForegroundColor Yellow return } $previousPath = (Get-Location).Path Pop-Location $newPath = (Get-Location).Path if ($previousPath -ne $newPath) { Write-Host "Popped and moved from '$previousPath' to '$newPath'." -ForegroundColor Green } else { Write-Host "Already at the bottom of the location stack." -ForegroundColor Yellow } } 'Version' { $moduleInfo = Get-Module Goto $moduleVersion = $moduleInfo.Version.ToString() Write-Host "goto version $moduleVersion" } 'Update' { Update-GotoModule } 'Navigate' { if (-not $Alias) { Show-Help return } if ($Global:DirectoryAliases.ContainsKey($Alias)) { $path = $Global:DirectoryAliases[$Alias] if (Test-Path $path) { Write-Host "Exact match found. Navigating to '$Alias' -> '$path'." -ForegroundColor Green Set-Location $path } else { Write-Host "The path '$path' for alias '$Alias' does not exist." -ForegroundColor Red } } else { $selectedAlias = _goto_print_similar -aliasInput $Alias if (-not $selectedAlias) { Write-Host "No matching alias found for '$Alias'." -ForegroundColor Red Write-Host "Usage: goto [ <alias> | -r <alias> <path> | -u <alias> | -l | -x <alias> | -c | -p <alias> | -o | -v | -h | -Update ]" -ForegroundColor Yellow } } } } } # Initialization when loading module Initialize-GotoEnvironment # Asynchronous verification of updates $updateCheckJob = Start-Job -ScriptBlock { param($GotoDataPath, $UpdateInfoFile) # I import the module inside the Job Import-Module Goto if (Test-Path $UpdateInfoFile) { $updateInfo = Import-Clixml -Path $UpdateInfoFile if (($null -eq $updateInfo.LastCheckTime) -or ((Get-Date) - $updateInfo.LastCheckTime).Days -ge 30) { Test-ForUpdates } } else { Test-ForUpdates } } -ArgumentList $script:GotoDataPath, $script:UpdateInfoFile # Register the ending event of the Job Register-ObjectEvent -InputObject $updateCheckJob -EventName StateChanged -Action { if ($Event.SourceEventArgs.JobStateInfo.State -eq "Completed") { $result = Receive-Job -Job $Event.SourceJob if ($result) { Write-Host $result -ForegroundColor Yellow } Unregister-Event -SourceIdentifier $Event.SourceIdentifier Remove-Job -Job $Event.SourceJob } } | Out-Null # Export of functions Export-ModuleMember -Function goto, Import-Aliases, Initialize-GotoEnvironment, Update-GotoModule, Save-Aliases Register-ArgumentCompleter -CommandName 'goto' -ScriptBlock { param($commandName, $wordToComplete, $commandAst, $fakeBoundParameters) $commands = @{ 'r' = 'Register' 'u' = 'Unregister' 'l' = 'List' 'x' = 'Expand' 'c' = 'Cleanup' 'p' = 'Push' 'o' = 'Pop' 'v' = 'Version' 'h' = 'Help' 'Update' = 'Update' } $aliases = $Global:DirectoryAliases.Keys if ($wordToComplete.StartsWith('-')) { $commands.GetEnumerator() | Where-Object { $_.Key -like "$($wordToComplete.TrimStart('-'))*" -or $_.Value -like "$($wordToComplete.TrimStart('-'))*" } | ForEach-Object { [System.Management.Automation.CompletionResult]::new("-$($_.Value)", "-$($_.Value)", 'ParameterName', "Perform $($_.Value) operation") } } elseif ($fakeBoundParameters.ContainsKey('Unregister') -or $fakeBoundParameters.ContainsKey('u') -or $fakeBoundParameters.ContainsKey('Expand') -or $fakeBoundParameters.ContainsKey('x') -or $fakeBoundParameters.ContainsKey('Push') -or $fakeBoundParameters.ContainsKey('p')) { $aliases | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object { [System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_) } } else { $aliases | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object { [System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_) } } } #Export of CD functions for each Alias $Global:DirectoryAliases.Keys | ForEach-Object { $aliasName = $_ $aliasPath = $Global:DirectoryAliases[$aliasName] New-Item -Path Function:\ -Name "Global:cd$aliasName" -Value { Set-Location $aliasPath }.GetNewClosure() | Out-Null Export-ModuleMember -Function "cd$aliasName" } |