[pmg-devel] [PATCH pmg-api 09/11] pbs-integration: add CLI calls to pmgbackup

[Stoiko Ivanov]

add two new categories for commands to pmgbackup:
* pmgbackup remote - for managing PBS instances' configuration, cluster-wide
* pmgbackup pbsjob - for managing backups, restores, pruning

Signed-off-by: Stoiko Ivanov <s.ivanov at proxmox.com>
---
 src/PMG/CLI/pmgbackup.pm | 96 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 96 insertions(+)

diff --git a/src/PMG/CLI/pmgbackup.pm b/src/PMG/CLI/pmgbackup.pm
index 69224e5..2a85b76 100644
--- a/src/PMG/CLI/pmgbackup.pm
+++ b/src/PMG/CLI/pmgbackup.pm
@@ -3,14 +3,20 @@ package PMG::CLI::pmgbackup;
 use strict;
 use warnings;
 use Data::Dumper;
+use POSIX qw(strftime);
 
 use PVE::Tools;
 use PVE::SafeSyslog;
 use PVE::INotify;
 use PVE::CLIHandler;
+use PVE::CLIFormatter;
+use PVE::JSONSchema qw(get_standard_option);
+use PVE::PBSTools;
 
 use PMG::RESTEnvironment;
 use PMG::API2::Backup;
+use PMG::API2::PBS::Remote;
+use PMG::API2::PBS::Job;
 
 use base qw(PVE::CLIHandler);
 
@@ -28,10 +34,100 @@ my $format_backup_list = sub {
     }
 };
 
+__PACKAGE__->register_method ({
+    name => 'prune',
+    path => 'prune',
+    method => 'GET',
+    description => "Prune all remotes",
+    protected => 1,
+    permissions => { check => [ 'admin', 'audit' ] },
+    parameters => {
+	additionalProperties => 0,
+	properties => {
+	    node => get_standard_option('pve-node'),
+	    remote => {
+		description => "only prune the provided Proxmox Backup Server.",
+		type => 'string', format => 'pve-configid',
+		optional => 1,
+	    },
+	    'dry-run' => { type => 'boolean', optional => 1, default => 0},
+	    'verbose' => { type => 'boolean', optional => 1, default => 0},
+	},
+    },
+    returns => { type => "string" },
+    code => sub {
+	my ($param) = @_;
+
+	my $remote = $param->{remote};
+	my $node = $param->{node};
+
+	my $verbose = defined($param->{verbose}) && $param->{verbose};
+
+	my $group = "host/$node";
+
+	my $conf = PMG::PBSConfig->new();
+
+	my $pbs_opts = {};
+	if (defined($param->{'dry-run'}) && $param->{'dry-run'}) {
+	    $pbs_opts->{'dry-run'} = 1;
+	}
+
+	my $run_prune = sub {
+	    my ($remote_config, $remote) = @_;
+
+	    print "running prune for $remote\n" if $verbose;
+
+	    my $prune_opts = $conf->prune_options($remote);
+	    my $res = PVE::PBSTools::prune_group($remote_config, $remote, $pbs_opts, $prune_opts, $group);
+
+	    if ($verbose) {
+		foreach my $pruned (@$res){
+		    my $time = strftime("%FT%TZ", gmtime($pruned->{'backup-time'}));
+		    my $snap = $pruned->{'backup-type'} . '/' . $pruned->{'backup-id'} . '/' .  $time;
+		    print "pruned snapshot: $snap\n";
+		}
+
+	    }
+	    print "finished pruning $remote\n" if $verbose;
+	};
+
+	if (defined($remote)) {
+	    my $remote_config = $conf->{ids}->{$remote};
+	    die "PBS remote '$remote' does not exist\n" if !$remote_config;
+	    die "PBS remote '$remote' is disabled\n" if $remote_config->{disable};
+	    $run_prune->($remote_config, $remote);
+	} else {
+	    foreach my $remote (sort keys %{$conf->{ids}}){
+		my $cfg = $conf->{ids}->{$remote};
+		next if $cfg->{disable};
+		$run_prune->($cfg, $remote);
+	    }
+	}
+    }});
+
 our $cmddef = {
     backup => [ 'PMG::API2::Backup', 'backup', undef, { node => $nodename } ],
     restore => [ 'PMG::API2::Backup', 'restore', undef, { node => $nodename } ],
     list => [ 'PMG::API2::Backup', 'list', undef, { node => $nodename }, $format_backup_list ],
+    remote => {
+	list => ['PMG::API2::PBS::Remote', 'list', undef, undef,  sub {
+	    my ($data, $schema, $options) = @_;
+	    PVE::CLIFormatter::print_api_result($data, $schema, ['remote', 'server', 'datastore', 'username' ], $options);
+	}, $PVE::RESTHandler::standard_output_options ],
+	add => ['PMG::API2::PBS::Remote', 'create', ['remote'] ],
+	remove => ['PMG::API2::PBS::Remote', 'delete', ['remote'] ],
+	set => ['PMG::API2::PBS::Remote', 'update_config', ['remote'] ],
+    },
+    pbsjob => {
+	list_backups => ['PMG::API2::PBS::Job', 'get_snapshots', ['remote'] , { node => $nodename },  sub {
+	    my ($data, $schema, $options) = @_;
+	    PVE::CLIFormatter::print_api_result($data, $schema, ['time', 'size'], $options);
+	}, $PVE::RESTHandler::standard_output_options ],
+	forget => ['PMG::API2::PBS::Job', 'forget_snapshot', ['remote', 'time'], { node => $nodename} ],
+	run => ['PMG::API2::PBS::Job', 'run_backup', ['remote'], { node => $nodename} ],
+	restore => ['PMG::API2::PBS::Job', 'restore', ['remote'], { node => $nodename} ],
+	prune => [__PACKAGE__, 'prune', undef , { node => $nodename } ],
+    },
 };
 
 1;
-- 
2.20.1