Skip to content

Commit

Permalink
fix(complete): Support two multi-length arguments
Browse files Browse the repository at this point in the history
Emit the value_terminator into the zsh completion pattern when
completing multi-length arguments on zsh to avoid doubled
rest-arguments definitions:

    $ my-app <TAB>
    _arguments:comparguments:325: doubled rest argument definition:
    *::second -- second set of of multi-length arguments:

Follow the suggestion from
clap-rs#3266 (comment)
by including the value terminator in order to resolve the ambiguity.

Closes clap-rs#3022
Signed-off-by: David Aguilar <davvid@gmail.com>
  • Loading branch information
davvid committed Jan 9, 2023
1 parent 3aea884 commit a686e61
Show file tree
Hide file tree
Showing 12 changed files with 211 additions and 1 deletion.
9 changes: 8 additions & 1 deletion clap_complete/src/shells/zsh.rs
Expand Up @@ -621,13 +621,20 @@ fn write_positionals_of(p: &Command) -> String {
debug!("write_positionals_of;");

let mut ret = vec![];
let mut cardinality_value;

for arg in p.get_positionals() {
debug!("write_positionals_of:iter: arg={}", arg.get_id());

let num_args = arg.get_num_args().expect("built");
let cardinality = if num_args.max_values() > 1 {
"*:"
match arg.get_value_terminator() {
Some(terminator) => {
cardinality_value = format!("*{}:", escape_value(&terminator));
cardinality_value.as_str()
}
None => "*:",
}
} else if !arg.is_required_set() {
":"
} else {
Expand Down
12 changes: 12 additions & 0 deletions clap_complete/tests/bash.rs
Expand Up @@ -84,6 +84,18 @@ fn value_hint() {
);
}

#[test]
fn two_multi_length_arguments() {
let name = "my-app";
let cmd = common::two_multi_length_arguments_command(name);
common::assert_matches_path(
"tests/snapshots/two_multi_length_arguments.bash",
clap_complete::shells::Bash,
cmd,
name,
);
}

#[cfg(feature = "unstable-dynamic")]
#[test]
fn register_minimal() {
Expand Down
15 changes: 15 additions & 0 deletions clap_complete/tests/common.rs
Expand Up @@ -252,6 +252,21 @@ pub fn value_hint_command(name: &'static str) -> clap::Command {
)
}

pub fn two_multi_length_arguments_command(name: &'static str) -> clap::Command {
clap::Command::new(name)
.arg(
clap::Arg::new("first")
.help("first set of multi-length arguments")
.num_args(1..)
.value_terminator("--"),
)
.arg(
clap::Arg::new("second")
.help("second set of of multi-length arguments")
.raw(true),
)
}

pub fn assert_matches_path(
expected_path: impl AsRef<std::path::Path>,
gen: impl clap_complete::Generator,
Expand Down
12 changes: 12 additions & 0 deletions clap_complete/tests/elvish.rs
Expand Up @@ -83,3 +83,15 @@ fn value_hint() {
name,
);
}

#[test]
fn two_multi_length_arguments() {
let name = "my-app";
let cmd = common::two_multi_length_arguments_command(name);
common::assert_matches_path(
"tests/snapshots/two_multi_length_arguments.elvish",
clap_complete::shells::Elvish,
cmd,
name,
);
}
12 changes: 12 additions & 0 deletions clap_complete/tests/fish.rs
Expand Up @@ -83,3 +83,15 @@ fn value_hint() {
name,
);
}

#[test]
fn two_multi_length_arguments() {
let name = "my-app";
let cmd = common::two_multi_length_arguments_command(name);
common::assert_matches_path(
"tests/snapshots/two_multi_length_arguments.fish",
clap_complete::shells::Fish,
cmd,
name,
);
}
12 changes: 12 additions & 0 deletions clap_complete/tests/powershell.rs
Expand Up @@ -83,3 +83,15 @@ fn value_hint() {
name,
);
}

#[test]
fn two_multi_length_arguments_command() {
let name = "my-app";
let cmd = common::two_multi_length_arguments_command(name);
common::assert_matches_path(
"tests/snapshots/two_multi_length_arguments.ps1",
clap_complete::shells::PowerShell,
cmd,
name,
);
}
38 changes: 38 additions & 0 deletions clap_complete/tests/snapshots/two_multi_length_arguments.bash
@@ -0,0 +1,38 @@
_my-app() {
local i cur prev opts cmds
COMPREPLY=()
cur="${COMP_WORDS[COMP_CWORD]}"
prev="${COMP_WORDS[COMP_CWORD-1]}"
cmd=""
opts=""

for i in ${COMP_WORDS[@]}
do
case "${cmd},${i}" in
",$1")
cmd="my__app"
;;
*)
;;
esac
done

case "${cmd}" in
my__app)
opts="-h --help [first]... [second]..."
if [[ ${cur} == -* || ${COMP_CWORD} -eq 1 ]] ; then
COMPREPLY=( $(compgen -W "${opts}" -- "${cur}") )
return 0
fi
case "${prev}" in
*)
COMPREPLY=()
;;
esac
COMPREPLY=( $(compgen -W "${opts}" -- "${cur}") )
return 0
;;
esac
}

complete -F _my-app -o bashdefault -o default my-app
26 changes: 26 additions & 0 deletions clap_complete/tests/snapshots/two_multi_length_arguments.elvish
@@ -0,0 +1,26 @@

use builtin;
use str;

set edit:completion:arg-completer[my-app] = {|@words|
fn spaces {|n|
builtin:repeat $n ' ' | str:join ''
}
fn cand {|text desc|
edit:complex-candidate $text &display=$text' '(spaces (- 14 (wcswidth $text)))$desc
}
var command = 'my-app'
for word $words[1..-1] {
if (str:has-prefix $word '-') {
break
}
set command = $command';'$word
}
var completions = [
&'my-app'= {
cand -h 'Print help'
cand --help 'Print help'
}
]
$completions[$command]
}
@@ -0,0 +1 @@
complete -c my-app -s h -l help -d 'Print help'
32 changes: 32 additions & 0 deletions clap_complete/tests/snapshots/two_multi_length_arguments.ps1
@@ -0,0 +1,32 @@

using namespace System.Management.Automation
using namespace System.Management.Automation.Language

Register-ArgumentCompleter -Native -CommandName 'my-app' -ScriptBlock {
param($wordToComplete, $commandAst, $cursorPosition)

$commandElements = $commandAst.CommandElements
$command = @(
'my-app'
for ($i = 1; $i -lt $commandElements.Count; $i++) {
$element = $commandElements[$i]
if ($element -isnot [StringConstantExpressionAst] -or
$element.StringConstantType -ne [StringConstantType]::BareWord -or
$element.Value.StartsWith('-') -or
$element.Value -eq $wordToComplete) {
break
}
$element.Value
}) -join ';'

$completions = @(switch ($command) {
'my-app' {
[CompletionResult]::new('-h', 'h', [CompletionResultType]::ParameterName, 'Print help')
[CompletionResult]::new('--help', 'help', [CompletionResultType]::ParameterName, 'Print help')
break
}
})

$completions.Where{ $_.CompletionText -like "$wordToComplete*" } |
Sort-Object -Property ListItemText
}
31 changes: 31 additions & 0 deletions clap_complete/tests/snapshots/two_multi_length_arguments.zsh
@@ -0,0 +1,31 @@
#compdef my-app

autoload -U is-at-least

_my-app() {
typeset -A opt_args
typeset -a _arguments_options
local ret=1

if is-at-least 5.2; then
_arguments_options=(-s -S -C)
else
_arguments_options=(-s -C)
fi

local context curcontext="$curcontext" state line
_arguments "${_arguments_options[@]}" \
'-h[Print help]' \
'--help[Print help]' \
'*--::first -- first set of multi-length arguments:' \
'*::second -- second set of of multi-length arguments:' \
&& ret=0
}

(( $+functions[_my-app_commands] )) ||
_my-app_commands() {
local commands; commands=()
_describe -t commands 'my-app commands' commands "$@"
}

_my-app "$@"
12 changes: 12 additions & 0 deletions clap_complete/tests/zsh.rs
Expand Up @@ -83,3 +83,15 @@ fn value_hint() {
name,
);
}

#[test]
fn two_multi_length_arguments_command() {
let name = "my-app";
let cmd = common::two_multi_length_arguments_command(name);
common::assert_matches_path(
"tests/snapshots/two_multi_length_arguments.zsh",
clap_complete::shells::Zsh,
cmd,
name,
);
}

0 comments on commit a686e61

Please sign in to comment.