posh-cargo.psm1
|
if (Get-Module 'posh-cargo') { return } function Get-AliasPattern($exe) { $aliases = @($exe) + @(Get-Alias | Where-Object { $_.Definition -eq $exe } | Select-Object -Exp Name) "($($aliases -join '|'))" } function get_manifest() { $manifest = (cargo metadata --format-version 1 --no-deps --quiet 2> $null) if (!$manifest -and $Global:Error.Count -ne 0) { $Global:Error.RemoveAt(0) } return $manifest } function get_manifest_path() { $proj = (cargo locate-project 2> $null) | Select-Object -First 1 | ForEach-Object { $_.SubString(9, $_.Length - 11) -replace '\\\\', '\' } if ($proj) { return $proj } elseif ($Global:Error.Count -ne 0) { $Global:Error.RemoveAt(0) } return $null } function get_project_directory() { $manifest_path = get_manifest_path if ($manifest_path) { Split-Path -Path $manifest_path } } function get_target_from_manifest_by_kind() { param($kind) return get_manifest | ConvertFrom-Json | Select-Object -ExpandProperty 'packages' | Select-Object -ExpandProperty 'targets' | Where-Object { $_.kind -eq $kind } | Select-Object -ExpandProperty 'name' } function get_package_from_manifest() { $packages = get_manifest | ConvertFrom-Json | Select-Object -ExpandProperty 'packages' $packages | Select-Object -ExpandProperty 'name' $packages | Select-Object -ExpandProperty 'dependencies' | Select-Object -ExpandProperty 'name' } function get_feature_from_manifest() { $manifest = get_manifest | ConvertFrom-Json $packages = ($manifest | Select-Object -ExpandProperty packages) $packages | Select-Object -ExpandProperty features | ForEach-Object { $_.PSObject.Properties.Name $_.PSObject.Properties.Value } $packages | Select-Object -ExpandProperty 'dependencies' | Select-Object -ExpandProperty 'name' } function get_cargo_commands() { return (cargo --list 2> $null) | Select-Object -Skip 1 | ForEach-Object { $_.Trim() -match '^\S+' > $null $Matches[0] } } function get_available_target_triples($toolchain) { $pattern = '(\s\(default\)|\s\(installed\))$' $targets = @() if ($toolchain) { $targets = $(rustup target list --toolchain $toolchain) } else { $targets = $(rustup target list) } return $targets | Where-Object { $_ -match $pattern } | ForEach-Object { $_ -replace $pattern, '' } } function get_toolchain_path($toolchain) { $rustc_path = '' if ($toolchain) { $rustc_path = $(rustup run $toolchain rustup which rustc) } else { $rustc_path = $(rustup which rustc) } return Resolve-Path ((Split-Path $rustc_path) + '\..') } function get_rustc_errorcode_list($toolchain) { $toolchain_path = (get_toolchain_path $toolchain) $doc_path = Join-Path $toolchain_path 'share\doc\rust\html\error-index.html' Get-Content $doc_path | Where-Object { $_ -match 'error-described error-used\"><h2 id=\"(\S+)\"' } | ForEach-Object { $Matches[1] } } function get_available_toolchains() { $toolchain_list = (rustup toolchain list | ForEach-Object{ $i = $_.IndexOf(' ') if ($i -ge 0) { $_.SubString(0, $i) } else { $_ } }) $toolchain_list $toolchain_list | ForEach-Object { $t = $_ -split '-' $t[0] $t[0] + '-' + $t[4] } } function get_installed_packages() { return ( cargo install --list | Where-Object { $_ -match '^(\S+)\s+\S+:$' } | ForEach-Object{ $Matches[1] } ) } $vcs = @('git', 'hg', 'pijul', 'none') $color = @('auto', 'always', 'never') $msg_format = @('human', 'json') $opt_help = @('-h', '--help') $opt_verbose = @('-v', '--verbose') $opt_very_verbose = @($opt_verbose + @('-vv')) $opt_quiet = @('-q', '--quiet') $opt_color = @('--color') $opt_common = @($opt_help + $opt_very_verbose + $opt_quiet + $opt_color) $opt_pkg = @('-p', '--package') $opt_feat = @('--features', '--all-features', '--no-default-features') $opt_mani = @('--manifest-path') $opt_jobs = @('-j', '--jobs') $opt_force = @('-f', '--force') $opt_lock = @('--frozen', '--locked') $opt_bin = @('--bin', '--bins') $opt_example = @('--example', '--examples') $opt_test = @('--test', '--tests', '--bench', '--benches') $opt_bin_types = @($opt_bin + $opt_example + $opt_test) $opt_exclude = @('--all', '--exclude') $opt_version = @('-V', '--version') $opt__fetch = @($opt_common + $opt_mani + $opt_lock) $opt___nocmd = $opt_common + $opt_version + @('--list', '--explain') $opt___commands = @( @('bench', @($opt_common + $opt_pkg + $opt_feat + $opt_mani + $opt_lock + $opt_jobs + $opt_bin_types + $opt_exclude + @('--message-format', '--target', '--lib', '--no-run', '--no-fail-fast'))), @('benchcmp', @($opt_help + $opt_color + @('--version', '--include-missing', '--threshold', '--varience', '--improvements', '--regressions'))), @('build', @($opt_common + $opt_pkg + $opt_feat + $opt_mani + $opt_lock + $opt_jobs + $opt_bin_types + $opt_exclude + @('--message-format', '--target', '--lib', '--release'))), @('check', @($opt_common + $opt_pkg + $opt_feat + $opt_mani + $opt_lock + $opt_jobs + $opt_bin_types + $opt_exclude + @('--message-format', '--target', '--lib', '--release'))), @('clean', @($opt_common + $opt_pkg + $opt_mani + $opt_lock + @('--target', '--release'))), @('doc', @($opt_common + $opt_pkg + $opt_feat + $opt_mani + $opt_lock + $opt_jobs + $opt_bin + @('--all', '--message-format', '--lib', '--target', '--open', '--no-deps', '--release'))), @('expand', @($opt_common + $opt_pkg + $opt_feat + $opt_mani + $opt_lock + $opt_jobs + $opt_bin_types + @('--message-format', '--profile', '--target', '--lib', '--release'))), @('fetch', $opt__fetch), @('fmt', @($opt_help + $opt_verbose + $opt_quiet + $opt_pkg + @('--all'))), @('generate-lockfile', $opt__fetch), @('git-checkout', @($opt_common + $opt_lock + @('--reference', '--url'))), @('graph', @($opt_help + $opt_version + @('-I', '--include-versions', '--build-color', '--build-deps', '--build-line-color', '--build-line-style', '--build-shape', '--dev-color', '--dev-deps', '--dev-line-color', '--dev-line-style', '--dev-shape', '--dot-file', '--lock-file', '--manifest-file', '--optional-color', '--optional-deps', '--optional-line-color', '--optional-line-style', '--optional-shape'))), @('help', $opt_help), @('init', @($opt_common + $opt_lock + @('--bin', '--lib', '--name', '--vcs'))), @('install', @($opt_common + $opt_feat + $opt_jobs + $opt_lock + $opt_force + $opt_bin + $opt_example + @('--branch', '--debug', '--git', '--list', '--path', '--rev', '--root', '--tag', '--vers'))), @('install-update', @($opt_help + $opt_force + $opt_version + @('-a', '--all', '-i', '--allow-no-update', '-l', '--list', '-c', '--cargo-dir'))), @('install-update-config', @($opt_help + $opt_version + @('-c', '--cargo-dir', '-d', '--default-features', '-f', '--feature', '-n', '--no-feature', '-t', '--toolchain'))), @('locate-project', @($opt_mani + $opt_help)), @('login', @($opt_common + $opt_lock + @('--host'))), @('metadata', @($opt_common + $opt_feat + $opt_mani + $opt_lock + @('--format-version', '--no-deps'))), @('modules', @($opt_help + $opt_version + @('-b', '--bin', '-l', '--lib', '-o', '--orphans', '-p', '--plain'))), @('new', @($opt_common + $opt_lock + @('--vcs', '--bin', '--lib', '--name'))), @('outdated', @($opt_help + $opt_verbose + $opt_mani + $opt_pkg + $opt_version + @('-d', '--depth', '--exit-code', '-l', '--lockfile-path', '-m', '-R', '--root-deps-only'))), @('owner', @($opt_common + $opt_lock + @('-a', '--add', '-r', '--remove', '-l', '--list', '--index', '--token'))), @('package', @($opt_common + $opt_mani + $opt_lock + $opt_jobs + @('--allow-dirty', '-l', '--list', '--no-verify', '--no-metadata'))), @('pkgid', @($opt__fetch + $opt_pkg)), @('publish', @($opt_common + $opt_mani + $opt_lock + $opt_jobs + @('--allow-dirty', '--dry-run', '--host', '--token', '--no-verify'))), @('read-manifest', @($opt_help + $opt_very_verbose + $opt_mani + $opt_color + @('--no-deps'))), @('run', @($opt_common + $opt_feat + $opt_mani + $opt_lock + $opt_jobs + @('--bin', '--example', '--message-format', '--target', '--release'))), @('rustc', @($opt_common + $opt_pkg + $opt_feat + $opt_mani + $opt_lock + $opt_jobs + $opt_bin_types + @('--message-format', '--profile', '--target', '--lib', '--release'))), @('rustdoc', @($opt_common + $opt_pkg + $opt_feat + $opt_mani + $opt_lock + $opt_jobs + $opt_bin_types + @('--message-format', '--target', '--lib', '--release', '--open'))), @('search', @($opt_common + $opt_lock + @('--host', '--limit'))), @('test', @($opt_common + $opt_pkg + $opt_feat + $opt_mani + $opt_lock + $opt_jobs + $opt_bin_types + $opt_exclude + @('--message-format', '--doc', '--target', '--lib', '--no-run', '--release', '--no-fail-fast'))), @('tree', @($opt_help + $opt_verbose + $opt_quiet + $opt_color + $opt_pkg + $opt_feat + $opt_lock + $opt_mani + $opt_version + @('-a', '--all', '--charset', '-d', '--duplicates', '-f', '--format', '-i', '--invert', '-k', '--kind', '--no-indent', '--target'))), @('uninstall', @($opt_common + $opt_lock + @('--bin', '--root'))), @('update', @($opt_common + $opt_pkg + $opt_mani + $opt_lock + @('--aggressive', '--precise'))), @('verify-project', @($opt__fetch)), @('version', @($opt_help + $opt_very_verbose + $opt_color)), @('yank', @($opt_common + $opt_lock + @('--vers', '--undo', '--index', '--token'))) ) function parse_commands ($parameters) { if ($parameters -match "^(?<firstToken>\S+)\s+(\S+\s+)*(?<lastToken>\S*)\s*$") { $opt___commands | Where-Object { $_[0].Equals($Matches['firstToken']) } | Select-Object -First 1 | ForEach-Object { $command = $_[0] $completionList = $_[1] $wordToComplete = $Matches['lastToken'] return @($command, $completionList, $wordToComplete) } } } function CargoTabExpansion($lastBlock) { $toolchain = '' $parameters = ($lastBlock -replace "^$(Get-AliasPattern cargo)(\.exe)?\s+","") if ($parameters -and $parameters[0] -eq '+') { $i = $parameters.IndexOf(' ') if ($i -gt 0) { $toolchain = $parameters.SubString(1, $i - 1) $parameters = $parameters.SubString($i).TrimStart() } } $parsed = (parse_commands $parameters) if ($parsed.Length -eq 3) { $command = $parsed[0] $completionList = $parsed[1] $wordToComplete = $parsed[2] } if ($command) { $other_command_specific = $false if ($command -eq 'help') { $completionList = $completionList + (get_cargo_commands) } elseif ($parameters -match '\s+(?<lastOption>\-\S+)\s+(?<lastToken>\S*)$') { $lastOption = $Matches['lastOption'] $lastToken = $Matches['lastToken'] switch -regex ($lastOption) { '^\-\-color$' { $completionList = $color } '^\-\-message-format$' { $completionList = $msg_format } '^\-\-vcs$' { $completionList = $vcs } '^\-\-target$' { $completionList = (get_available_target_triples $toolchain) } '^(\-\-bin|\-\-test|\-\-example|\-\-bench)$' { $kind = $lastOption.SubString(2) if (!(($kind -eq 'bin') -and ($command -eq 'init' -or $command -eq 'new'))) { $possibleList = get_target_from_manifest_by_kind $kind if (!$possibleList) { $possibleList = @('<NAME>') } $wordToComplete = $lastToken $completionList = $possibleList } } '^(\-p|\-\-package|\-\-exclude)$' { $possibleList = (get_package_from_manifest) if (!$possibleList) { $possibleList = @('<SPEC>') } $wordToComplete = $lastToken $completionList = $possibleList } '^\-\-features$' { $possibleList = (get_feature_from_manifest) if (!$possibleList) { $possibleList = @('<FEATURE>') } $wordToComplete = $lastToken $completionList = $possibleList } '^(\-t|\-\-toolchain)$' { $completionList = (get_available_toolchains) } '^(\-t|\-\-manifest-path)$' { $completionList = @() } default { $other_command_specific = $true } } } else { $other_command_specific = $true } if ($other_command_specific) { if ($command -eq 'pkgid') { $completionList = $completionList + (get_package_from_manifest) } elseif ($command -eq 'uninstall' -or $command -eq 'install-update' -or $command -eq 'install-update-config') { $installed_packages = (get_installed_packages) $completionList = $completionList + $installed_packages } elseif ($command -eq 'graph') { if ($parameters -match '\s+(?<lastOption>\-\S+)\s+(?<lastToken>\S*)$') { $lastOption = $Matches['lastOption'] switch -regex ($lastOption) { '^\-\-\S+\-color$' { $completionList = @('blue', 'black', 'yellow', 'purple', 'green', 'red', 'white', 'orange') } '^\-\-\S+\-style$' { $completionList = @('solid', 'dotted', 'dashed') } '^\-\-\S+\-shape$' { $completionList = @('box', 'round', 'diamond', 'triangle') } '^\-\-\S+\-deps$' { $completionList = @('true', 'false') } '^\-\-\S+\-file$' { $completionList = @() } } } } elseif ($command -eq 'modules') { if ($parameters -match '\s+(?<lastOption>\-\S+)\s+(?<lastToken>\S*)$') { $lastOption = $Matches['lastOption'] switch -regex ($lastOption) { '^\-b$' { $possibleList = get_target_from_manifest_by_kind 'bin' if (!$possibleList) { $possibleList = @('<NAME>') } $wordToComplete = $lastToken $completionList = $possibleList } } } } elseif ($command -eq 'outdated') { if ($parameters -match '\s+(?<lastOption>\-\S+)\s+(?<lastToken>\S*)$') { $lastOption = $Matches['lastOption'] switch -regex ($lastOption) { '^\-(d|\-\depth|\-exit-code)$' { $completionList = @(0.. 9) + @('<NUM>') } '^\-\-\S+\-path$' { $completionList = @() } } } } elseif ($command -eq 'tree') { if ($parameters -match '\s+(?<lastOption>\-\S+)\s+(?<lastToken>\S*)$') { $lastOption = $Matches['lastOption'] switch -regex ($lastOption) { '^(\-k|\-\-kind)$' { $completionList = @('normal', 'dev', 'build') } '^\-\-charset$' { $completionList = @('utf8', 'ascii') } '^(\-f|\-\-format)$' { $completionList = @('<FORMAT>', '""') } } } } } } else { $wordToComplete = $parameters $completionList = $opt___nocmd + (get_cargo_commands) if (!$toolchain) { $completionList += get_available_toolchains | ForEach-Object { '+' + $_ } } if ($parameters -match '(^|\s+)(?<lastOption>\-\S+)\s+(?<lastToken>\S*)$') { $lastToken = $Matches['lastToken'] $lastOption = $Matches['lastOption'] switch -regex ($lastOption) { '^\-\-color$' { $wordToComplete = $lastToken $completionList = $color } '^\-\-explain$' { $wordToComplete = $lastToken $completionList = (get_rustc_errorcode_list $toolchain) if (!$completionList) { $completionList = @('<CODE>') } } } } } $finalCandidates = $completionList | Where-Object { $_ -like "${wordToComplete}*" } | Sort-Object -CaseSensitive -Unique $finalCandidates | Where-Object { $_[0] -ne '-' -and $_[0] -ne '+' } $finalCandidates | Where-Object { $_[0] -eq '-' -and $_.Length -gt 1 -and $_[1] -ne '-' } $finalCandidates | Where-Object { $_[0] -eq '-' -and $_.Length -gt 1 -and $_[1] -eq '-' } $finalCandidates | Where-Object { $_[0] -eq '+' } } $PowerTab_RegisterTabExpansion = if (Get-Module -Name powertab) { Get-Command Register-TabExpansion -Module powertab -ErrorAction SilentlyContinue } if ($PowerTab_RegisterTabExpansion) { & $PowerTab_RegisterTabExpansion "cargo" -Type Command { param($Context, [ref]$TabExpansionHasOutput, [ref]$QuoteSpaces) # 1: $line = $Context.Line $lastBlock = [regex]::Split($line, '[|;]')[-1].TrimStart() $TabExpansionHasOutput.Value = $true CargoTabExpansion $lastBlock } return } if (Test-Path Function:\TabExpansion) { Rename-Item Function:\TabExpansion TabExpansionBackup } function TabExpansion($line, $lastWord) { $lastBlock = [regex]::Split($line, '[|;]')[-1].TrimStart() switch -regex ($lastBlock) { # Execute cargo tab completion for all cargo-related commands "^$(Get-AliasPattern cargo)(\.exe)? (.*)" { CargoTabExpansion $lastBlock } # Fall back on existing tab expansion default { if (Test-Path Function:\TabExpansionBackup) { TabExpansionBackup $line $lastWord } } } } Export-ModuleMember -Function 'TabExpansion' |