use JSON;
use File::Path qw(make_path);
+use PVE::APIClient::JSONSchema;
use PVE::APIClient::Exception qw(raise);
+use PVE::APIClient::CLIFormatter;
+use PVE::APIClient::CLIHandler;
+use PVE::APIClient::PTY;
use Encode::Locale;
use Encode;
use HTTP::Status qw(:constants);
my $pve_api_definition;
-my $pve_api_path_hash;
my $pve_api_definition_fn = "/usr/share/pve-client/pve-api-definition.dat";
-my $method_map = {
+our $method_map = {
create => 'POST',
set => 'PUT',
get => 'GET',
delete => 'DELETE',
};
-my $build_pve_api_path_hash;
-$build_pve_api_path_hash = sub {
+my $__real_remove_formats; $__real_remove_formats = sub {
+ my ($properties) = @_;
+
+ foreach my $pname (keys %$properties) {
+ if (my $d = $properties->{$pname}) {
+ if (defined(my $format = $d->{format})) {
+ if (ref($format)) {
+ $__real_remove_formats->($format);
+ } elsif (!PVE::APIClient::JSONSchema::get_format($format)) {
+ # simply remove unknown format definitions
+ delete $d->{format};
+ }
+ }
+ }
+ }
+};
+
+my $remove_unknown_formats; $remove_unknown_formats = sub {
my ($tree) = @_;
my $class = ref($tree);
- return $tree if !$class;
+ return if !$class;
if ($class eq 'ARRAY') {
- foreach my $el (@$tree) {
- $build_pve_api_path_hash->($el);
- }
+ foreach my $el (@$tree) {
+ $remove_unknown_formats->($el);
+ }
} elsif ($class eq 'HASH') {
- if (defined($tree->{leaf}) && defined(my $path = $tree->{path})) {
- $pve_api_path_hash->{$path} = $tree;
+ if (my $info = $tree->{info}) {
+ for my $method (qw(GET PUT PUSH DELETE)) {
+ next if !$info->{$method};
+ my $properties = $info->{$method}->{parameters}->{properties};
+ $__real_remove_formats->($properties) if $properties;
+ }
}
- foreach my $k (keys %$tree) {
- $build_pve_api_path_hash->($tree->{$k});
+ if ($tree->{children}) {
+ $remove_unknown_formats->($tree->{children});
}
}
+ return;
};
-my $default_output_format = 'text';
-my $client_output_format = $default_output_format;
+sub get_api_definition {
-sub set_output_format {
- my ($format) = @_;
+ if (!defined($pve_api_definition)) {
+ open(my $fh, '<', $pve_api_definition_fn) ||
+ die "unable to open '$pve_api_definition_fn' - $!\n";
+ $pve_api_definition = Storable::fd_retrieve($fh);
- if (!defined($format)) {
- $client_output_format = $default_output_format;
- } else {
- $client_output_format = $format;
+ $remove_unknown_formats->($pve_api_definition);
}
-}
-sub get_output_format {
- return $client_output_format;
+ return $pve_api_definition;
}
-sub print_result {
- my ($data, $result_schema) = @_;
-
- my $format = get_output_format();
+my $map_path_to_info = sub {
+ my ($child_list, $stack, $uri_param) = @_;
- return if $result_schema->{type} eq 'null';
+ while (defined(my $comp = shift @$stack)) {
+ foreach my $child (@$child_list) {
+ my $text = $child->{text};
- # TODO: implement different output formats ($format)
-
- if ($format eq 'json') {
- print to_json($data, {utf8 => 1, allow_nonref => 1, canonical => 1, pretty => 1 });
- } elsif ($format eq 'text') {
- my $type = $result_schema->{type};
- if ($type eq 'object') {
- die "implement me";
- } elsif ($type eq 'array') {
- my $item_type = $result_schema->{items}->{type};
- if ($item_type eq 'object') {
- die "implement me";
- } elsif ($item_type eq 'array') {
- die "implement me";
+ if ($text eq $comp) {
+ # found
+ } elsif ($text =~ m/^\{(\S+)\}$/) {
+ # found
+ $uri_param->{$1} = $comp;
} else {
- foreach my $el (@$data) {
- print "$el\n"
- }
+ next; # text next child
+ }
+ if ($child->{leaf} || !scalar(@$stack)) {
+ return $child;
+ } else {
+ $child_list = $child->{children};
+ last; # test next path component
}
- } else {
- print "$data\n";
}
- } else {
- die "internal error: unknown output format"; # should not happen
}
-}
-
-sub get_api_definition {
-
- if (!defined($pve_api_definition)) {
- open(my $fh, '<', $pve_api_definition_fn) ||
- die "unable to open '$pve_api_definition_fn' - $!\n";
- $pve_api_definition = Storable::fd_retrieve($fh);
- $build_pve_api_path_hash->($pve_api_definition);
- }
-
- return $pve_api_definition;
-}
+ return undef;
+};
-sub lookup_api_method {
- my ($path, $method, $noerr) = @_;
+sub find_method_info {
+ my ($path, $method, $uri_param, $noerr) = @_;
- get_api_definition(); # make sure API data is loaded
+ $uri_param //= {};
- my $info = $pve_api_path_hash->{$path};
+ my $stack = [ grep { length($_) > 0 } split('\/+' , $path)]; # skip empty fragments
- if (!$info) {
- return undef if $noerr;
- die "unable to find API info for path '$path'\n";
- }
+ my $api = get_api_definition();
- my $data = $info->{info}->{$method};
+ my $child = scalar(@$stack) ? $map_path_to_info->($api->{children}, $stack, $uri_param) : $api;
- if (!$data) {
+ if (!($child && $child->{info} && $child->{info}->{$method})) {
return undef if $noerr;
die "unable to find API method '$method' for path '$path'\n";
}
- return $data;
+ return $child->{info}->{$method};
}
sub complete_api_call_options {
&$print_result(@option_list);
}
+sub merge_api_definition_properties {
+ my ($path, $method, $properties) = @_;
+
+ my $info = PVE::APIClient::Helpers::find_method_info($path, $method);
+
+ return { %{$info->{parameters}->{properties}}, %$properties };
+}
+
sub complete_api_path {
my ($text) = @_;
- get_api_definition(); # make sure API data is loaded
+ my $api = get_api_definition(); # make sure API data is loaded
$text =~ s!^/!!;
my $info;
if (!defined($dir)) {
$dir = '';
- $info = { children => $pve_api_definition };
+ $info = $api;
} else {
- $info = $pve_api_path_hash->{"/$dir"};
+ my $stack = [ grep { length($_) > 0 } split('\/+' , $dir)]; # skip empty fragments
+ $info = $map_path_to_info->($api->{children}, $stack, {});
}
my $res = [];
if ($info) {
+ my $prefix = length($dir) ? "/$dir/" : '/';
if (my $children = $info->{children}) {
foreach my $c (@$children) {
- if ($c->{path} =~ m!\Q$dir/$rest!) {
- push @$res, $c->{path};
- push @$res, "$c->{path}/" if $c->{children};
+ my $ctext = $c->{text};
+ push @$res, "${prefix}$ctext" if $ctext =~ m/^\Q$rest/;
+ if ($ctext =~ m/^\{(\S+)\}$/) {
+ if (length($rest) && $rest ne $ctext) {
+ push @$res, "$prefix$rest";
+ }
}
}
}
}
+
+ if (scalar(@$res) == 1) {
+ $res = [$res->[0], "$res->[0]/"];
+ }
+
return $res;
}
# test for command lines with api calls (or similar bash completion calls):
# example1: pveclient api get remote1 /cluster
sub extract_path_info {
+ my ($uri_param) = @_;
my $info;
my $path = $args->[4];
if (my $method = $method_map->{$args->[2]}) {
- $info = lookup_api_method($path, $method, 1);
+ $info = find_method_info($path, $method, $uri_param, 1);
}
};
}
sub poll_task {
- my ($conn, $node, $upid) = @_;
+ my ($conn, $node, $upid, $quiet) = @_;
my $path = "api2/json/nodes/$node/tasks/$upid/status";
my $task_status;
+ my $last_line = 0;
while(1) {
+ if (!$quiet) {
+ my $path = "api2/json/nodes/$node/tasks/$upid/log";
+ my $task_log = $conn->get($path, {start => $last_line});
+
+ my $printme = '';
+ for my $li (@$task_log) {
+ if ($li->{t} eq 'no content') {
+ next;
+ }
+ $printme .= $li->{t} . "\n";
+ $last_line = $li->{n};
+ }
+
+ if ($printme ne '') {
+ print $printme;
+ }
+ }
+
$task_status = $conn->get($path, {});
if ($task_status->{status} eq "stopped") {
last;
}
- sleep(10);
+ sleep(2);
+ }
+
+ if ($task_status->{exitstatus} ne "OK") {
+ die $task_status->{exitstatus};
}
return $task_status->{exitstatus};
die $@ if $@;
}
+sub extract_even_elements {
+ my ($list) = @_;
+
+ my $ind = 0;
+ return [ grep { ($ind++ % 2) == 0 } @$list ];
+}
+
+sub print_ordered_result {
+ my ($property_list, $data, $result_schema, $options) = @_;
+
+ my $param_order = extract_even_elements($property_list);
+
+ PVE::APIClient::CLIFormatter::print_api_result($data, $result_schema, $param_order, $options);
+}
1;