#!/usr/bin/perl -w
# ApplySnapshot
# Vadim Mikheev, (c) 2000, PostgreSQL Inc.

eval '(exit $?0)' && eval 'exec perl -S $0 ${1+"$@"}'
    & eval 'exec perl -S $0 $argv:q'
    if 0;

BEGIN {
	my $basedir = $0; $basedir =~ s#/[^/]+$##;
	unshift(@INC, "$basedir/../share");
}

use strict;
use IO::File;
use RServ;
use Getopt::Long;

$| = 1;

my ($debug,$verbose) = (0,0);
my ($help,$slavehost,$slaveport,$slaveuser,$slavepassword);
my $snapshot;

my $result = GetOptions(
	"debug!" => \$debug, "verbose!" => \$verbose, "help" => \$help,
	"slavehost=s" => \$slavehost, "slaveport=i" => \$slaveport,
	"slaveuser=s" => \$slaveuser, "slavepassword=s" => \$slavepassword,
	"snapshot=s" => \$snapshot,
	);

if (defined($help) || (scalar(@ARGV) < 1)) {
    print "Usage: $0 [options] slavedb
Options:
	--slavehost=hostname --slaveport=port
	--slaveuser=username --slavepassword=string
	--snapshot=snapshot
";
    exit ((scalar(@ARGV) < 1)? 1:0);
}

my $slave = $ARGV[0] || "slave";

my $sinfo = "dbname=$slave";
$sinfo = "$sinfo host=$slavehost" if (defined($slavehost));
$sinfo = "$sinfo port=$slaveport" if (defined($slaveport));
$sinfo = "$sinfo user=$slaveuser" if (defined($slaveuser));
$sinfo = "$sinfo password=$slavepassword" if (defined($slavepassword));

my $conn = Pg::connectdb($sinfo);

my $inpf;
if (defined($snapshot)) {
    $inpf = new IO::File;
    $inpf->open("< $snapshot"); 
} else {
    $inpf = *STDIN;
}

my $res = ApplySnapshot ($conn, $inpf);

if ($res > 0)
{
	printf STDERR "Snapshot applied\n";
}
elsif ($res != 0)
{
	printf STDERR "ERROR\n";
	exit(1);
}

exit(0);
