#!/bin/bash
#
# MondoRescue test suite
#
# $Id$
#
# Copyright B. Cornec 2012
# Provided under the GPL v2

# First create a dummy account for testing
# In order for this script to run well, you should have sudo access

# You may have multiple distros here
TESTVMDIST="rhel-6-x86_64"
#TESTVMDIST="ubuntu-11.04-i386"
#TESTVMDIST="debian-6.0-x86_64 ubuntu-11.04-i386 fedora-16-x86_64 opensuse-12.1-x86_64 mageia-2-i386 mandriva-2010.1-i386 rhel-6-x86_64 rhel-5-i386 fedora-17-i386 ubuntu-12.10-x86_64"
VMCOMMADIST=`echo $TESTVMDIST | sed 's/ /,/g'`
MRUSER=mrtest
MRTEST="/users/$MRUSER"
export PBPROJ=mondorescue

# Sub function to handle execution as mrtest 
# doesn't change to the home dir of the user as su - can't be used to get env var correct
sudomrtest() {

cmd=$1
msg=$2

echo " "
if [ _"$cmd" == _"/tmp/mrtestcmd" ]; then
	echo "INFO (as $MRUSER): Now running `head -1 $cmd` [$msg]"
else
	echo "INFO (as $MRUSER): Now running $cmd [$msg]"
fi
sudo -n -E su - $MRUSER -c "$cmd"
ret=$?
if [ $ret -ne 0 ] && [ _"$msg" == _"" ]; then
	echo "================================"
	echo "ERROR: Command $cmd failed: $ret"
	echo "================================"
fi
return $ret
}

sudocmd() {
cmd=$1

echo " "
echo "INFO (as root): Now running $cmd"
sudo -n $cmd
ret=$?
if [ $ret -ne 0 ]; then
	echo "================================"
	echo "ERROR: Command $cmd failed: $ret"
	echo "================================"
fi
return $ret
}

# First start building for all these distro we want
pb -p mondorescue -g -r branches/3.0 -m $VMCOMMADIST sbx2vm mondo mindi mindi-busybox

sudocmd "rm -rf $MRTEST/* $MRTEST/.??*"
sudocmd "rmdir $MRTEST"
sudocmd "killall -u $MRUSER"
sudocmd "userdel $MRUSER"
sudocmd "useradd $MRUSER -d $MRTEST"
sudocmd "grep -q $MRUSER /etc/sudoers"
if [ $? -ne 0 ]; then
	cat > /tmp/sudoers << EOF
# This is for the $MRUSER account for the VM only
$MRUSER ALL=NOPASSWD:/usr/sbin/mondoarchive
$MRUSER ALL=NOPASSWD:/usr/bin/qemu-kvm
EOF
	sudocmd "cat /etc/sudoers /tmp/sudoers" | grep -v "INFO " > /tmp/sudoers2
	sudocmd "chmod 440 /tmp/sudoers2"
	sudocmd "chown root:root /tmp/sudoers2"
	sudocmd "mv /tmp/sudoers2 /etc/sudoers"
	rm -f /tmp/sudoers /tmp/sudoers2
fi
# Force $MRUSER (for now mruser hardcoded!) account passwd is pbtest
sudocmd "perl -pi -e 's|^mrtest:.*|mrtest:$2a$08$2Q.tEty4v/fU0RyPeDelk.AjLAtXUXU3DpSs/l/8YrFmau1LJRCqC:15483:0:99999:7:::|' /etc/shadow"
sudomrtest "echo export PBPROJ=$PBPROJ >> $MRTEST/.bash_profile"
sudomrtest "env | grep PBPROJ"

sudomrtest "mkdir -p $MRTEST/{ftp,vm,.ssh,mr/images}"
sudocmd "grep -q $MRUSER /etc/exports"
if [ $? -ne 0 ]; then
	cat > /tmp/exports << EOF
$MRTEST/mr	*(rw,no_root_squash,async,insecure)
EOF
	sudocmd "cat /etc/exports /tmp/exports" | grep -v "INFO " > /tmp/exports2
	sudocmd "chmod 640 /tmp/exports2"
	sudocmd "chown root:root /tmp/exports2"
	sudocmd "mv /tmp/exports2 /etc/exports"
	sudocmd "exportfs -a"
	rm -f /tmp/exports2 /tmp/exports
fi
sudomrtest "chmod 700 $MRTEST/.ssh"

sudomrtest "pb clean 2>&1 > /dev/null" 1

cat > /tmp/.pbrc << EOF
# Projects
pbdefdir default = \$ENV{'HOME'}/mr
pburl mondorescue = svn://svn.mondorescue.org/mondorescue/branches
pbconfurl mondorescue = svn://svn.project-builder.org/pb/projects/mondorescue/pbconf
pbprojdir mondorescue = $ENV{'HOME'}/mr/svn
pbconfdir mondorescue = $ENV{'HOME'}/mr/pbconf
sshhost mondorescue = localhost
sshlogin mondorescue = $MRUSER
sshdir mondorescue = \$ENV{'HOME'}/ftp
#
pbpackager default = Big Chief <bigchief@hp.com>
vmport mondorescue = 2300
pbparallel mondorescue = 6
vmpath default = $MRTEST/vm
EOF
sudocmd "mv /tmp/.pbrc $MRTEST/.pbrc"

cat > /tmp/.pbrc << EOF
vmtype default = kvm
vmcmd default = qemu-kvm
vmsize fedora = 10G
vmmonport default = 4000
vmlogin default = $MRUSER
vmmem default = 1024
EOF
echo "vmlist default = $VMCOMMADIST" >> /tmp/.pbrc
sudocmd "mv /tmp/.pbrc $MRTEST/vm/.pbrc"

sudocmd "chown ${MRUSER}:$MRUSER $MRTEST/.pbrc $MRTEST/vm/.pbrc"

sudocmd "cp -a /users/bruno/.ssh/pb* $MRTEST/.ssh"
sudocmd "chown $MRUSER $MRTEST/.ssh/pb_dsa.pub $MRTEST/.ssh/pb_dsa"
sudomrtest "chmod 600 $MRTEST/.ssh/pb*"
sudomrtest "ssh-keygen -f $MRTEST/.ssh/id_rsa -t rsa -b 768 -v -N ''"
sudomrtest "cp $MRTEST/.ssh/id_rsa.pub $MRTEST/.ssh/authorized_keys"
sudomrtest "chmod 600 $MRTEST/.ssh/authorized_keys $MRTEST/.ssh/id_rsa.pub"
sudomrtest "ssh localhost touch $MRTEST/ftp/README"

for d in $TESTVMDIST; do
	sudomrtest "qemu-img create -f qcow2 -b /users/qemu/$d.qemu $MRTEST/vm/$d.qemu"
	#sudocmd "cp /users/qemu/$d.qemu $MRTEST/vm"
	#sudocmd "chown $MRUSER:$MRUSER $MRTEST/vm/$d.qemu"
	# pb isue workaround
	sudomrtest "mkdir -p $MRTEST/mr/$PBPROJ/delivery"
	sudomrtest "touch $MRTEST/mr/$PBPROJ/delivery/pbrc"
	updcmd=`pbdistrocheck -s $d | cut -d, -f9`
	dd=`echo $d | cut -d'-' -f1`
	vv=`echo $d | cut -d'-' -f2`
	aa=`echo $d | cut -d'-' -f3`
	date=`date +%Y%m%d%H%M`
	cat > /tmp/script << EOF
#!/bin/bash
#
set -x
echo "$updcmd" | grep -w yum 
if [ \$? -eq 0 ]; then
	$updcmd nfs-utils
	cd /etc/yum.repos.d
	wget ftp://ftp.mondorescue.org/test/$dd/$vv/$aa/mondorescue-test.repo
	wget ftp://ftp.mondorescue.org/$dd/$vv/$aa/mondorescue.repo
	rpm --import ftp://ftp.mondorescue.org/$dd/$vv/$aa/mondorescue.pubkey
	if [ -f /etc/pki/rpm-gpg/RPM-GPG-KEY-redhat-release ]; then
		# Import Red Hat keys for deps
		rpm --import /etc/pki/rpm-gpg/RPM-GPG-KEY-redhat-release
	fi
fi
echo "$updcmd" | grep -w apt-get 
if [ \$? -eq 0 ]; then
	$updcmd nfs-common
	cd /etc/apt/sources.list.d
	wget ftp://ftp.mondorescue.org/test/$dd/$vv/mondorescue-test.sources.list
	wget ftp://ftp.mondorescue.org/$dd/$vv/mondorescue.sources.list
	cat >> /etc/apt/preferences << EOT
Package: mindi
Pin: version 2.1.*
Pin-Priority: 1001

Package: mondo
Pin: version 3.0.*
Pin-Priority: 1001
EOT
fi

$updcmd mondo

TEMP_PATH=$MRTEST/mr/tmp
SCRATCH_PATH=$MRTEST/mr/scratch

NFS_HOST=10.0.2.2
NFS_MOUNT=$MRTEST/mr
MOUNT_PATH=/mnt

mount | grep \$MOUNT_PATH | wc -l
if [ \$? -ne 0 ]; then 
	umount \$MOUNT_PATH;
fi

# Relative path that will be used locally and remotely on the NFS server
BCKUP_PATH=images

if [ ! -d \$MOUNT_PATH ]; then 
	mkdir -p \$MOUNT_PATH;
	echo "Mount directory \$MOUNT_PATH created"
fi

echo "Try to mount NFS point \$NFS_HOST:\$NFS_MOUNT on \$MOUNT_PATH..."

mount \$NFS_HOST:\$NFS_MOUNT \$MOUNT_PATH
if [ \$? -eq 0 ]; then
	echo "NFS point \$NFS_HOST:\$NFS_MOUNT mounted on \$MOUNT_PATH"
else
	echo "Unable to mount \$NFS_HOST:\$NFS_MOUNT on \$MOUNT_PATH"
	exit -1
fi

mkdir -p \$TEMP_PATH;
echo "Temp directory \$TEMP_PATH created"

mkdir -p \$SCRATCH_PATH;
echo "Scratch directory \$SCRATCH_PATH created"

if [ ! -d \$MOUNT_PATH/\$BCKUP_PATH ]; then
	mkdir -p \$MOUNT_PATH/\$BCKUP_PATH;
	echo "Backup directory \$MOUNT_PATH/\$BCKUP_PATH created"
fi

BCKUP_NAME=$d-$date
echo "Backup name: \$BCKUP_NAME"

CMD_LINE="mondoarchive -O -n \$NFS_HOST:\$NFS_MOUNT -p \$BCKUP_NAME -d \$BCKUP_PATH -s 4380m -S \$SCRATCH_PATH -T \$TEMP_PATH -G -N -E '/var/log|/usr/share/doc|/var/cache|/usr/src|/usr/share/texfm|/share/games|/usr/share/webmin|/var/tmp|/tmp' -H -z"
echo "Will run command: \$CMD_LINE"

\$CMD_LINE
echo "Mondorescue terminated"

umount \$MOUNT_PATH
echo "Backup directory \$MOUNT_PATH unmounted"

rm -rf \$TEMP_PATH
echo "Temp directory \$TEMP_PATH removed"

rm -rf \$SCRATCH_PATH
echo "Scratch directory \$SCRATCH_PATH removed"

echo "Backup over"

echo "Halting system"
halt -p
EOF
	chmod 755 /tmp/script
	sudocmd "chmod 666 /dev/kvm"
	#sudocmd "chmod 622 /dev/tty0"
	port=2300
	sudomrtest "qemu-kvm --nographic -m 2048 -redir tcp:${port}:10.0.2.15:22 $MRTEST/vm/$d.qemu"&
	sleep 240
	sudomrtest "scp -i $MRTEST/.ssh/pb_dsa -P $port /tmp/script root@localhost:"
	sudomrtest "ssh -i $MRTEST/.ssh/pb_dsa -p $port root@localhost ./script"
	sudomrtest "qemu-img create -f qcow2 $MRTEST/vm/$d.rest 8G"
	sudomrtest "chmod 666 $MRTEST/vm/$d.rest"
	#sudomrtest "ssh -i $MRTEST/.ssh/pb_dsa -p $port root@localhost /sbin/halt -p" &
	/usr/bin/qemu-kvm -m 1024 -drive file=$MRTEST/vm/$d.rest,media=disk -redir tcp:2400:10.0.2.15:22 -boot d -cdrom $MRTEST/mr/images/$d-$date-1.iso
	/usr/bin/qemu-kvm -m 1024 -drive file=$MRTEST/vm/$d.rest,media=disk -redir tcp:2400:10.0.2.15:22
done

# cleanup
sudomrtest "pb clean"
exit
