improve bash completion

[pve-client.git] / pveclient

diff --git a/pveclient b/pveclient
index 30d86681805a2e49357eca32460a06e48bd935df..091af96bb160a2ee7ab79a3e029794984770a9d0 100755 (executable)
--- a/pveclient
+++ b/pveclient
@@ -3,6 +3,7 @@
 use strict;
 use warnings;
 use lib '/usr/share/pve-client';
+use lib '.';
 use Data::Dumper;
 
 use PVE::JSONSchema;
@@ -11,13 +12,22 @@ use PVE::CLIHandler;
 use PVE::APIClient::LWP;
 use PVE::APIClient::Helpers;
 use PVE::APIClient::Commands::remote;
+use PVE::APIClient::Commands::lxc;
 
 use JSON;
 
 sub print_usage {
 
-    die "Usage: implement me";
-    exit(-1);
+    my $text = "pveclient usage:\n\n";
+
+    $text .= "pveclient remote <help|add|remove> {options}\n\n";
+
+    $text .= "pveclient lxc <help|create|destroy|...> {options}\n\n";
+
+    $text .= "pveclient <get/set/create/delete> <path> {options}\n\n";
+
+    print STDERR $text;
+
 }
 
 sub call_method {
@@ -50,7 +60,12 @@ sub call_method {
 #my $res = $conn->get("/", {});
 #print to_json($res, { pretty => 1, canonical => 1});
 
-my $cmd = shift || print_usage();
+my $cli_class_handlers = {
+    lxc => 'PVE::APIClient::Commands::lxc',
+    remote => 'PVE::APIClient::Commands::remote',
+};
+
+my $cmd = shift || (print_usage() && exit(-1));
 
 if ($cmd eq 'get') {
     my $method = 'GET';
@@ -66,8 +81,38 @@ if ($cmd eq 'get') {
     die "implement me";
 } elsif ($cmd eq 'delete') {
     die "implement me";
-} elsif ($cmd eq 'remote') {
-    PVE::APIClient::Commands::remote->run_cli_handler();
+} elsif (my $class = $cli_class_handlers->{$cmd}) {
+    $class->run_cli_handler();
+} elsif ($cmd eq 'bashcomplete') {
+
+    exit(0) if !(defined($ENV{COMP_LINE}) && defined($ENV{COMP_POINT}));
+
+    my $cmdlist = join('|', keys %$cli_class_handlers);
+    if ($ENV{COMP_LINE} =~ m/^(.*pveclient\s+($cmdlist)\s+)(.*)$/) {
+       my $cmd = $2;
+       my $class = $cli_class_handlers->{$cmd} || die "internal error";
+       $ENV{COMP_LINE} = "pveclient $3";
+       $ENV{COMP_POINT} = length($ENV{COMP_LINE});
+       @ARGV = ('bashcomplete', 'pveclient', $ARGV[1], $ARGV[2]);
+
+       $class->run_cli_handler();
+
+    } else {
+
+       my $cmdline = substr($ENV{COMP_LINE}, 0, $ENV{COMP_POINT});
+       my ($bash_command, $cur, $prev) = @ARGV;
+       $cmdline =~ s/$cur$//;
+
+       my $args = PVE::Tools::split_args($cmdline);
+
+       my @cmds = ('get', 'set', 'create', 'delete', keys %$cli_class_handlers);
+       if (scalar(@$args) == 1) {
+           foreach my $p (@cmds) {
+               print "$p\n" if $p =~ m/^$cur/;
+           }
+       }
+    }
+
 } else {
     print_usage();
 }