X-Git-Url: https://git.proxmox.com/?p=pve-common.git;a=blobdiff_plain;f=src%2FPVE%2FCLIHandler.pm;h=222051dcdb3e02fa186d17babe319c0476994a19;hp=5e2d3b8d193acb664b0c62555e7d0b17bd1ea151;hb=f9326469095aa3ae90a7439f19b3536dff8a66dc;hpb=4e0214faae9a5fe3eb62c710c44f447a23883430 diff --git a/src/PVE/CLIHandler.pm b/src/PVE/CLIHandler.pm index 5e2d3b8..222051d 100644 --- a/src/PVE/CLIHandler.pm +++ b/src/PVE/CLIHandler.pm @@ -75,6 +75,7 @@ sub resolve_cmd { my ($argv, $is_alias) = @_; my ($def, $cmd) = ($cmddef, $argv); + my $cmdstr = $exename; if (ref($argv) eq 'ARRAY') { my $expanded = {}; @@ -82,6 +83,7 @@ sub resolve_cmd { for my $i (0..$last_arg_id) { $cmd = $expand_command_name->($def, $argv->[$i]); + $cmdstr .= " $cmd"; $expanded->{$argv->[$i]} = $cmd if $cmd ne $argv->[$i]; last if !defined($def->{$cmd}); $def = $def->{$cmd}; @@ -89,7 +91,7 @@ sub resolve_cmd { if (ref($def) eq 'ARRAY') { # could expand to a real command, rest of $argv are its arguments my $cmd_args = [ @$argv[$i+1..$last_arg_id] ]; - return ($cmd, $def, $cmd_args, $expanded); + return ($cmd, $def, $cmd_args, $expanded, $cmdstr); } if (defined($def->{alias})) { @@ -102,9 +104,9 @@ sub resolve_cmd { # got either a special command (bashcomplete, verifyapi) or an unknown # cmd, just return first entry as cmd and the rest of $argv as cmd_arg my $cmd_args = [ @$argv[1..$last_arg_id] ]; - return ($argv->[0], $def, $cmd_args, $expanded); + return ($argv->[0], $def, $cmd_args, $expanded, $cmdstr); } - return ($cmd, $def); + return ($cmd, $def, undef, undef, $cmdstr); } sub generate_usage_str { @@ -118,9 +120,10 @@ sub generate_usage_str { $indent //= ''; my $read_password_func = $cli_handler_class->can('read_password'); - my $param_mapping_func = $cli_handler_class->can('string_param_file_mapping'); + my $param_mapping_func = $cli_handler_class->can('param_mapping') || + $cli_handler_class->can('string_param_file_mapping'); - my ($subcmd, $def) = resolve_cmd($cmd); + my ($subcmd, $def, undef, undef, $cmdstr) = resolve_cmd($cmd); my $generate; $generate = sub { @@ -167,9 +170,6 @@ sub generate_usage_str { return $str; }; - my $cmdstr = $exename; - $cmdstr .= ' ' . join(' ', @$cmd) if defined($cmd); - return $generate->($indent, $separator, $def, $cmdstr); } @@ -461,8 +461,7 @@ my $handle_cmd = sub { $cmddef->{help} = [ __PACKAGE__, 'help', ['extra-args'] ]; - my $cmd_str = join(' ', @$args); - my ($cmd, $def, $cmd_args) = resolve_cmd($args); + my ($cmd, $def, $cmd_args, undef, $cmd_str) = resolve_cmd($args); $abort->("no command specified") if !$cmd; @@ -481,15 +480,14 @@ my $handle_cmd = sub { } # checked special commands, if def is still a hash we got an incomplete sub command - $abort->("incomplete command '$exename $cmd_str'") if ref($def) eq 'HASH'; + $abort->("incomplete command '$cmd_str'") if ref($def) eq 'HASH'; &$preparefunc() if $preparefunc; my ($class, $name, $arg_param, $uri_param, $outsub) = @{$def || []}; $abort->("unknown command '$cmd_str'") if !$class; - my $prefix = "$exename $cmd_str"; - my $res = $class->cli_handler($prefix, $name, $cmd_args, $arg_param, $uri_param, $read_password_func, $param_mapping_func); + my $res = $class->cli_handler($cmd_str, $name, $cmd_args, $arg_param, $uri_param, $read_password_func, $param_mapping_func); &$outsub($res) if $outsub; }; @@ -546,7 +544,8 @@ sub run_cli_handler { my $preparefunc = $params{prepare}; my $read_password_func = $class->can('read_password'); - my $param_mapping_func = $class->can('string_param_file_mapping'); + my $param_mapping_func = $cli_handler_class->can('param_mapping') || + $class->can('string_param_file_mapping'); $exename = &$get_exe_name($class);