#!/bin/bash
# main autocluster script
#
# Copyright (C) Andrew Tridgell  2008
# Copyright (C) Martin Schwenke  2008
#
# 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; either version 3 of the License, or
# (at your option) any later version.
#   
# 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, see <http://www.gnu.org/licenses/>.

if [ -f "$0" ]; then
    installdir="`dirname \"$0\"`"
else
    autocluster=`which $0`
    installdir="`dirname \"$autocluster\"`"
fi

####################
# show program usage
usage ()
{
    cat <<EOF
Usage: autocluster [OPTION] ... <COMMAND>
  options:
    -c <file>    specify config file (default is "config")
    -x           enable script debugging
    --dump       dump config settings and exit

  configuration options:
EOF

    usage_config_options

    cat <<EOF

  commands:
     create base
           create a base image

     create cluster CLUSTERNAME
           create a full cluster

     create node CLUSTERNAME NODENUMBER
           create one cluster node

     create tsm CLUSTERNAME
           create a TSM server node

     mount DISK
           mount a qemu disk on mnt/

     unmount
           unmount a qemu disk from mnt/

     bootbase
           boot the base image

     testproxy
           test your proxy setup
EOF
    exit 1
}


###############################
# create a single node
create_node() {
    CLUSTER="$1"
    NODENUMBER="$2"
    NAME="$CLUSTER""n$NODENUMBER"
    BASE="$VIRTBASE/$BASENAME.img"
    DISK="$VIRTBASE/$CLUSTER/$NAME.qcow2"
    NUMSHARED=3

    # first node might need more memory for SoFS GUI
    if [ $NODENUMBER = 1 -a $GUIMEM -gt $MEM ]; then
	NODEMEM=$GUIMEM
    else
	NODEMEM=$MEM
    fi
    echo "Creating cluster node $NAME"

    mkdir -p $VIRTBASE/$CLUSTER

    echo "Creating the disk"
    rm -f "$DISK"
    qemu-img create -b "$BASE" -f qcow2 "$DISK"

    IPNUM=`expr $FIRSTIP + $NODENUMBER`

    mount_disk $DISK
    setup_base
    setup_network
    unmount_disk

    mkdir -p tmp

    MAC1=`get_macaddr $CLUSTER $NODENUMBER 1`
    MAC2=`get_macaddr $CLUSTER $NODENUMBER 2`
    MAC3=`get_macaddr $CLUSTER $NODENUMBER 3`
    UUID=`uuidgen`
    
    echo "Creating $NAME.xml"
    substitute_vars $NODE_TEMPLATE tmp/$NAME.xml
    
    # install the XML file
    $VIRSH undefine $NAME > /dev/null 2>&1 || true
    $VIRSH define tmp/$NAME.xml
}


###############################
# create a TSM node
create_tsm() {
    CLUSTER="$1"
    NAME="$CLUSTER""tsm"
    BASE="$VIRTBASE/$BASENAME.img"
    DISK="$VIRTBASE/$CLUSTER/$NAME.qcow2"
    TSMDISK="$VIRTBASE/$CLUSTER/tsmstorage.qcow2"
    
    echo "Creating TSM cluster node $NAME"
    
    mkdir -p $VIRTBASE/$CLUSTER tmp
    
    echo "Creating the disk"
    rm -f "$DISK"
    qemu-img create -b "$BASE" -f qcow2 "$DISK"
    
    echo "Creating tsm disk"
    qemu-img create -f qcow2 "$TSMDISK" $TSMDISKSIZE
    
    # TSM server is first IP in the cluster
    IPNUM=$FIRSTIP

    mount_disk $DISK
    setup_base
    setup_network
    unmount_disk
    
    UUID=`uuidgen`
    MAC1=`get_macaddr $CLUSTER 0 0`
    
    echo "Creating $NAME.xml"
    substitute_vars $TSM_TEMPLATE tmp/$NAME.xml
    
    # install the XML file
    $VIRSH undefine $NAME > /dev/null 2>&1 || true
    $VIRSH define tmp/$NAME.xml
}


###############################
# create a whole cluster
create_cluster() {
    CLUSTER="$1"

    mkdir -p $VIRTBASE/$CLUSTER tmp
    mkdir -p $KVMLOG

    echo "Creating 3 shared disks"
    for i in `seq 1 3`; do
	# setup a nice ID at the start of the disk
	qemu-img create -f raw $VIRTBASE/$CLUSTER/shared$i $SHAREDDISKSIZE
	echo "SOFS-`uuidgen`" > tmp/diskid
	dd if=tmp/diskid of=$VIRTBASE/$CLUSTER/shared$i conv=notrunc bs=1 > /dev/null 2>&1
    done

    echo "Creating $NUMNODES base nodes"
    for i in `seq 1 $NUMNODES`; do
	create_node "$CLUSTER" $i
    done

    if [ $WITH_TSM_NODE -eq 1 ]; then
      echo "Creating TSM server node"
      create_tsm "$CLUSTER"
    fi

    echo "# autocluster $CLUSTER" > hosts.$CLUSTER
    [ $WITH_TSM_NODE -eq 1 ] && {
	echo "$IPBASE.0.$FIRSTIP ${CLUSTER}tsm.$LOWDOMAIN ${CLUSTER}tsm" >> hosts.$CLUSTER
    }
    for i in `seq 1 $NUMNODES`; do
	echo "$IPBASE.0.`expr $FIRSTIP + $i` ${CLUSTER}n$i.$LOWDOMAIN ${CLUSTER}n$i" >> hosts.$CLUSTER
    done
    echo >> hosts.$CLUSTER

    echo "Cluster $CLUSTER created"
    echo "You may want to add this to your /etc/hosts file:"
    cat hosts.$CLUSTER

    echo
}

###############################
# test the proxy setup
test_proxy() {
    export http_proxy=$WEBPROXY
    wget -O /dev/null $INSTALL_SERVER || {
	echo "Your WEBPROXY setting \"$WEBPROXY\" is not working"
	exit 1
    }
    echo "Proxy OK"
}


###################
# create base image
create_base() {

    NAME="$BASENAME"
    DISK="$VIRTBASE/$NAME.img"

    mkdir -p $KVMLOG

    echo "Testing WEBPROXY $WEBPROXY"
    test_proxy

    echo "Creating the disk"
    qemu-img create -f $BASE_FORMAT "$DISK" $DISKSIZE

    rm -rf tmp
    mkdir -p mnt tmp tmp/ISO

    setup_timezone
    setup_repos

    echo "Creating kickstart file from template"
    substitute_vars "$KICKSTART" "tmp/ks.cfg"

    if [ $INSTALLKEY = "--skip" ]; then
	cat <<EOF
--------------------------------------------------------------------------------------
WARNING: You have not entered an install key. Some RHEL packages will not be installed.

Please enter a valid RHEL install key in your config file like this:

  INSTALLKEY="1234-5678-0123-4567"

The install will continue without an install key in 5 seconds
--------------------------------------------------------------------------------------
EOF
	sleep 5
    fi
    
    echo "Creating kickstart floppy"
    dd if=/dev/zero of=tmp/floppy.img bs=1024 count=1440
    mkdosfs tmp/floppy.img
    mount -o loop -t msdos tmp/floppy.img mnt
    cp tmp/ks.cfg mnt
    mount -o loop,ro $ISO tmp/ISO
    
    echo "Setting up bootloader"
    cp tmp/ISO/isolinux/isolinux.bin tmp
    cp tmp/ISO/isolinux/vmlinuz tmp
    cp tmp/ISO/isolinux/initrd.img tmp
    umount tmp/ISO
    umount mnt

    UUID=`uuidgen`

    substitute_vars $INSTALL_TEMPLATE tmp/$NAME.xml

    rm -f $KVMLOG/serial.$NAME

    # boot the install CD
    $VIRSH create tmp/$NAME.xml

    echo "Waiting for install to start"
    sleep 2
    
    # wait for the install to finish
    if ! waitfor $KVMLOG/serial.$NAME "you may safely reboot your system" 3600; then
	$VIRSH destroy $NAME
	echo "Failed to create base image $DISK"
	exit 1
    fi
    
    $VIRSH destroy $NAME

    ls -l $DISK
    cat <<EOF

Install finished, base image $DISK created

You may wish to run
   chattr +i $DISK
To ensure that this image does not change

Note that the root password has been set to $ROOTPASSWORD

EOF
}



###############################
# boot the base disk
boot_base() {
    CLUSTER="$1"

    NAME="$BASENAME"
    DISK="$VIRTBASE/$NAME.img"

    rm -rf tmp
    mkdir -p tmp

    IPNUM=$FIRSTIP
    CLUSTER="base"

    mount_disk $DISK
    setup_base
    unmount_disk

    UUID=`uuidgen`
    
    echo "Creating $NAME.xml"
    substitute_vars $BOOT_TEMPLATE tmp/$NAME.xml
    
    # boot the base system
    $VIRSH create tmp/$NAME.xml
}

######################################################################

. "$installdir/functions"
. "$installdir/config.default"

############################
# parse command line options
long_opts=$(getopt_config_options)
getopt_output=$(getopt -n autocluster -o "c:xh" -l help,dump -l "$long_opts" -- "$@")
[ $? != 0 ] && usage

use_default_config=true

# We 2 passes of the options.  The first time we just handle usage and
# check whether -c is being used.
eval set -- "$getopt_output"
while true ; do
    case "$1" in
	-c) shift 2 ; use_default_config=false ;;
	--) shift ; break ;;
	--dump|-x) shift ;;
	-h|--help) usage ;; # Usage should be shown here for real defaults.
	--*) shift 2 ;; # Assume other long opts are valid and take an arg.
	*) usage ;; # shouldn't happen, so this is reasonable.
    esac
done

config="./config"
$use_default_config && [ -r "$config" ] && . "$config"

eval set -- "$getopt_output"

while true ; do
    case "$1" in
	-c) . "`dirname $2`/$2" ; shift 2 ; conf_done=true ;;
	-x) set -x; shift ;;
	--dump) dump_config ;;
	--) shift ; break ;;
	-h|--help) usage ;; # Redundant.
	--*)
	    # Putting --opt1|opt2|... into a variable and having case
	    # match against it as a pattern doesn't work.  The | is
	    # part of shell syntax, so we need to do this.  Look away
	    # now to stop your eyes from bleeding! :-)
	    x=",${long_opts}" # Now each option is surrounded by , and :
	    if [ "$x" != "${x#*,${1#--}:}" ] ; then
		# Our option, $1, surrounded by , and : was in $x, so is legal.
		setconf_longopt "$1" "$2"; shift 2
	    else
		usage
	    fi
	    ;;
	*) usage ;; # shouldn't happen, so this is reasonable.
    esac
done

# catch errors
set -e
set -E
trap 'es=$?; 
      echo ERROR: failed in function \"${FUNCNAME}\" at line ${LINENO} of ${BASH_SOURCE[0]} with code $es; 
      exit $es' ERR

LOWDOMAIN=`echo $DOMAIN | tr A-Z a-z`

# check for needed programs 
check_command nbd-client
check_command expect
check_command $QEMU_NBD

[ $# -lt 1 ] && usage

command="$1"
shift

case $command in
    create)
	type=$1
	shift
	case $type in
	    base)
		[ $# != 0 ] && usage
		create_base;
		;;
	    cluster)
		[ $# != 1 ] && usage
		create_cluster "$1";
		;;
	    node)
		[ $# != 2 ] && usage
		create_node "$1" "$2";
		;;
	    tsm)
		[ $# != 1 ] && usage
		create_tsm "$1";
		;;
	    *)
		usage;
		;;
	esac
	;;
    mount)
	[ $# != 1 ] && usage
	mount_disk "$1"
	;;
    unmount)
	[ $# != 0 ] && usage
	unmount_disk
	;;
    bootbase)
	boot_base;
	;;
    testproxy)
	test_proxy;
	;;
    *)
	usage;
	;;
esac