fully implement api get/set/create/delete

[pve-client.git] / PVE / APIClient / Helpers.pm

package PVE::APIClient::Helpers;

use strict;
use warnings;

use Storable;
use JSON;
use PVE::APIClient::Exception qw(raise);
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 = {
    create => 'POST',
    set => 'PUT',
    get => 'GET',
    delete => 'DELETE',
};

my $build_pve_api_path_hash;
$build_pve_api_path_hash = sub {
    my ($tree) = @_;

    my $class = ref($tree);
    return $tree if !$class;

    if ($class eq 'ARRAY') {
        foreach my $el (@$tree) {
            $build_pve_api_path_hash->($el);
        }
    } elsif ($class eq 'HASH') {
        if (defined($tree->{leaf}) && defined(my $path = $tree->{path})) {
            $pve_api_path_hash->{$path} = $tree;
        }
        foreach my $k (keys %$tree) {
            $build_pve_api_path_hash->($tree->{$k});
        }
    }
};

my $default_output_format = 'table';
my $client_output_format =  $default_output_format;

sub set_output_format {
    my ($format) = @_;

    if (!defined($format)) {
        $client_output_format =  $default_output_format;
    } else {
        $client_output_format =  $format;
    }
}

sub get_output_format {
    return $client_output_format;
}

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;
}

sub lookup_api_method {
    my ($path, $method, $noerr) = @_;

    get_api_definition(); # make sure API data is loaded

    my $info = $pve_api_path_hash->{$path};

    if (!$info) {
        return undef if $noerr;
        die "unable to find API info for path '$path'\n";
    }

    my $data = $info->{info}->{$method};

    if (!$data) {
        return undef if $noerr;
        die "unable to find API method '$method' for path '$path'\n";
    }

    return $data;
}

sub complete_api_call_options {
    my ($cmd, $prop, $prev, $cur, $args) = @_;

    my $print_result = sub {
        foreach my $p (@_) {
            print "$p\n" if $p =~ m/^$cur/;
        }
    };

    my $print_parameter_completion = sub {
        my ($pname) = @_;
        my $d = $prop->{$pname};
        if ($d->{completion}) {
            my $vt = ref($d->{completion});
            if ($vt eq 'CODE') {
                my $res = $d->{completion}->($cmd, $pname, $cur, $args);
                &$print_result(@$res);
            }
        } elsif ($d->{type} eq 'boolean') {
            &$print_result('0', '1');
        } elsif ($d->{enum}) {
            &$print_result(@{$d->{enum}});
        }
    };

    my @option_list = ();
    foreach my $key (keys %$prop) {
        push @option_list, "--$key";
    }

    if ($cur =~ m/^-/) {
        &$print_result(@option_list);
        return;
    }

    if ($prev =~ m/^--?(.+)$/ && $prop->{$1}) {
        my $pname = $1;
        &$print_parameter_completion($pname);
        return;
    }

    &$print_result(@option_list);
}

sub complete_api_path {
    my ($text) = @_;

    get_api_definition(); # make sure API data is loaded

    $text =~ s!^/!!;

    my ($dir, $rest) = $text =~ m|^(?:(.*)/)?(?:([^/]*))?$|;

    my $info;
    if (!defined($dir)) {
        $dir = '';
        $info = { children => $pve_api_definition };
    } else {
        $info = $pve_api_path_hash->{"/$dir"};
    }

    my $res = [];
    if ($info) {
        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};
                }
            }
        }
    }
    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 $info;

    my $test_path_properties = sub {
        my ($args) = @_;

        return if scalar(@$args) < 5;
        return if $args->[1] ne 'api';

        my $path = $args->[4];
        if (my $method = $method_map->{$args->[2]}) {
            $info = lookup_api_method($path, $method, 1);
        }
    };

    if (defined(my $cmd = $ARGV[0])) {
        if ($cmd eq 'api') {
            $test_path_properties->([$0, @ARGV]);
        } elsif ($cmd eq 'bashcomplete') {
            my $cmdline = substr($ENV{COMP_LINE}, 0, $ENV{COMP_POINT});
            my $args = PVE::APIClient::Tools::split_args($cmdline);
            $test_path_properties->($args);
        }
    }

    return $info;
}

sub get_vmid_resource {
    my ($conn, $vmid) = @_;

    my $resources = $conn->get('api2/json/cluster/resources', {type => 'vm'});

    my $resource;
    for my $tmp (@$resources) {
        if ($tmp->{vmid} eq $vmid) {
            $resource = $tmp;
            last;
        }
    }

    if (!defined($resource)) {
        die "\"$vmid\" not found";
    }

    return $resource;
}

sub poll_task {
    my ($conn, $node, $upid) = @_;

    my $path = "api2/json/nodes/$node/tasks/$upid/status";

    my $task_status;
    while(1) {
        $task_status = $conn->get($path, {});

        if ($task_status->{status} eq "stopped") {
            last;
        }

        sleep(10);
    }

    return $task_status->{exitstatus};
}

1;