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 10, 2023
1 parent 28e8bb4 commit ee5e84a
Show file tree
Hide file tree
Showing 12 changed files with 205 additions and 1 deletion.
9 changes: 8 additions & 1 deletion clap_complete/src/shells/zsh.rs
Original file line number Diff line number Diff line change
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
Original file line number Diff line number Diff line change
Expand Up @@ -84,6 +84,18 @@ fn value_hint() {
);
}

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

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

pub fn value_terminator_command(name: &'static str) -> clap::Command {
clap::Command::new(name)
.arg(
clap::Arg::new("arguments")
.help("multi-valued argument with a value terminator")
.num_args(1..)
.value_terminator("--"),
)
}

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
Original file line number Diff line number Diff line change
Expand Up @@ -83,3 +83,15 @@ fn value_hint() {
name,
);
}

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

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

#[test]
fn value_terminator() {
let name = "my-app";
let cmd = common::value_terminator_command(name);
common::assert_matches_path(
"tests/snapshots/value_terminator.ps1",
clap_complete::shells::PowerShell,
cmd,
name,
);
}
38 changes: 38 additions & 0 deletions clap_complete/tests/snapshots/value_terminator.bash
Original file line number Diff line number Diff line change
@@ -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 [arguments]..."
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/value_terminator.elvish
Original file line number Diff line number Diff line change
@@ -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]
}
1 change: 1 addition & 0 deletions clap_complete/tests/snapshots/value_terminator.fish
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
complete -c my-app -s h -l help -d 'Print help'
32 changes: 32 additions & 0 deletions clap_complete/tests/snapshots/value_terminator.ps1
Original file line number Diff line number Diff line change
@@ -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
}
30 changes: 30 additions & 0 deletions clap_complete/tests/snapshots/value_terminator.zsh
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
#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]' \
'*--::arguments -- multi-valued argument with a value terminator:' \
&& 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
Original file line number Diff line number Diff line change
Expand Up @@ -83,3 +83,15 @@ fn value_hint() {
name,
);
}

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

0 comments on commit ee5e84a

Please sign in to comment.