#!/usr/bin/perl

# Copyright (c) 2008, 2011 Oracle and/or its affiliates. All rights reserved.
# Copyright (c) 2014, SkySQL Ab.
# Use is subject to license terms.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; version 2 of the License.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301
# USA

use strict;
use lib 'lib';
use lib "$ENV{RQG_HOME}/lib";
#use List::Util 'shuffle';
use Cwd;
use File::Path 'remove_tree';
use GenTest;
use GenTest::Constants;
use Getopt::Long;
use File::Basename;

Getopt::Long::Configure("pass_through");

if (defined $ENV{RQG_HOME}) {
	if (osWindows()) {
		$ENV{RQG_HOME} = $ENV{RQG_HOME}.'\\';
	} else {
		$ENV{RQG_HOME} = $ENV{RQG_HOME}.'/';
	}
} else {
	$ENV{RQG_HOME} = dirname(Cwd::abs_path($0));
}

if ( osWindows() )
{
	require Win32::API;
	my $errfunc = Win32::API->new('kernel32', 'SetErrorMode', 'I', 'I');
	my $initial_mode = $errfunc->Call(2);
	$errfunc->Call($initial_mode | 2);
};

$| = 1;
my $ctrl_c = 0;
	
$SIG{INT} = sub { $ctrl_c = 1 };
$SIG{TERM} = sub { exit(0) };
$SIG{CHLD} = "IGNORE" if osWindows();

my ($vardir, $vardir1, $vardir2, $trials, $force, $old, $exit_status, @exit_status);

my $max_result = 0;

my $opt_result = GetOptions(
	'vardir=s' => \$vardir,
	'vardir1=s' => \$vardir1,
	'vardir2=s' => \$vardir2,
	'trials=i' => \$trials,
	'force' => \$force,
	'old' => \$old,
	'exit_status=s' => \$exit_status,
	'exit-status=s' => \$exit_status,
);

$trials = 1 unless defined $trials;

push @ARGV, "--vardir=$vardir" if defined $vardir;
push @ARGV, "--vardir1=$vardir1" if defined $vardir1;
push @ARGV, "--vardir2=$vardir2" if defined $vardir2;

my $comb_str = join(' ', @ARGV);		

if ($exit_status) {
	@exit_status = split(',', $exit_status);
}

foreach my $trial_id (1..$trials) {

	say("##########################################################");
	say("Running trial ".$trial_id."/".$trials);
	my $runall = $old?"runall.pl":"runall-new.pl";

	my $command = "perl ".
		(defined $ENV{RQG_HOME} ? $ENV{RQG_HOME}."/" : "" ).
		"$runall $comb_str ";

	$command =~ s{[\t\r\n]}{ }sgio;
	$command =~ s{"}{\\"}sgio;

	unless (osWindows())
	{
		$command = 'bash -c "set -o pipefail; '.$command.'"';
	}

	say("$command");

	my $result = system($command) >> 8;
	my $result_name = status2text($result);
	say("Trial $trial_id ended with exit status $result_name ($result)");
	unless (check_for_desired_status($result_name)) {
		say("Exit status $result_name is not on the list of desired status codes ($exit_status), it will be ignored");
		next;
	}
	exit($result) if not defined $force;

	$max_result = $result if $result > $max_result;

	if ($result > 0) {
		# Storing vardirs for the failure
		foreach my $v ($vardir,$vardir1,$vardir2) {
			next unless defined $v;
			# Remove trailing slashes
			$v =~ s/[\/\\]+$//;
			my $from = $v;
			my $to = $v.'_trial'.$trial_id;
			say("Copying $from to $to");
			remove_tree($to) if (-e $to);
			remove_tree($to.'_slave') if (-e $to.'_slave');
			if (osWindows()) {
				system("xcopy \"$from\" \"$to\" /E /I /Q");
				system("xcopy \"$from"."_slave\" \"$to"."_slave\" /E /I /Q") if -e $from.'_slave';
				open(OUT, ">$to/command");
				print OUT $command;
				close(OUT);
			} elsif ($command =~ m{--mem}) {
				system("cp -r /dev/shm/var $to");
				open(OUT, ">$to/command");
				print OUT $command;
				close(OUT);
			} else {
				system("cp -r $from $to");
				system("cp -r $from"."_slave $to"."_slave") if -e $from.'_slave';
				open(OUT, ">$to/command");
				print OUT $command;
				close(OUT);
			}
		}
	}
}

say("$0 will exit with exit status ".status2text($max_result)."($max_result)");
exit($max_result);

sub check_for_desired_status {
	my $resname = shift;
	if (!$exit_status) {
		# No desired codes, anything except for STATUS_OIK will do
		return $resname ne 'STATUS_OK';
	}
	else {
		foreach (@exit_status) {
			return 1 if $resname eq $_;
		}
		return 0;
	}
}