#!/bin/bash
#-----------------------------------------------------------------------------
# /var/install/bin/xen-create-domain - create XEN domain from template
#
# Last Update:  $Id$
#
# Copyright (c) 2006-2012 the eisfair team, team(at)eisfair(dot)org
#
# 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 2 of the License, or
# (at your option) any later version.
#-----------------------------------------------------------------------------
# Load desired libs
. /var/install/include/xenlib
. /etc/config.d/base
#exec 2> /tmp/xen-create-domain-trace$$.log
#set -x

read_dhcp_leases () {
    if [ "$IP_NET_1_STATIC_IP" = "no" ]
    then
        leasefile=/var/lib/dhcp3/dhclient.eth0.leases
        if [ -f $leasefile ]
        then
            IP_NET_1_NETMASK=`grep 'subnet-mask' $leasefile | awk 'BEGIN { RS=""; FS="\n" } { print $NF }' | sed -e 's#[^0-9.]##g'`
            IP_NET_1_GATEWAY=`grep 'router' $leasefile | awk 'BEGIN { RS=""; FS="\n" } { print $NF }' | sed -e 's#[^0-9.]##g'`
            DOMAIN_NAME=`grep 'domain-name' $leasefile | awk 'BEGIN { RS=""; FS="\n" } { print $NF }' | sed  -e 's#^[^"]*##g' -e 's#[^"]*$##' -e 's#["]##g'`
            DNS_SERVER=`grep 'domain-name-servers' $leasefile | awk 'BEGIN { RS=""; FS="\n" } { print $NF }' | sed -e 's#[^0-9.]##g'`
        fi
    fi
}

cleanUpOnError () {
    cd # important for umount
    mecho --error "$1"
    ${swapImageCreated:-false} && {
        echo "Deleting swap image..."
        #remove $swapImage
        mecho --warn " -`lvremove -f $swapImage`"
    }
    ${diskImageCreated:-false} && {
        echo "Deleting disk image..."
        unMount "$mountpoint"
        mecho --warn " -`lvremove -f $diskImage`"
    }
    [ -d "$mountpoint" ] && remove "$mountpoint"
    remove ""
    exit 1
}

cleanUpNormal() {
    cd # important for umount
    [ -d "$mountpoint" ] && {
        unMount "$mountpoint"
        remove "$mountpoint"
        # register created domain:
        registrate_domain $domainName
    }
    echo "Done."
    anykey
}

cleanUpOnERR () {
    cleanUpOnError "An error occoured! Aborting..."
}

cleanUpOnSIGINT () {
    cleanUpOnError "Ctrl-C pressed! Aborting..."
}

### TODO: Validate input data in a nice(!) way
networkTemplateScriptInput() {
    mecho --info "Collect network settings:"
    hostname=$domainName
    address=`askIP "question=an IP-Address (e.g. 192.168.0.100)" "type=ip-address"`
    netmask=`askIP "question=the Subnetmask" "proposal=$IP_NET_1_NETMASK" "type=subnet mask"`
    gateway=`askIP "question=the default gateway" "proposal=$IP_NET_1_GATEWAY" "type=default gateway"`
    searchdomain=`/var/install/bin/ask --tty --info "Please enter the Domain" "${DOMAIN_NAME}" "+"`
    nameserver=`askIP "question=the DNS" "proposal=$DNS_SERVER" "type=DNS"`
}

# Here it begins
# main

clrhome

template="${*}"
if [ -z "$template" ]
then
    /var/install/bin/list-files.cui -t "Select a template" \
               -p "$xenTemplatePath" \
               -f '*.tar.gz' -f '*.tar.bz2' \
               -o 1 \
               -q 'Use template >>%ls<<?' \
               --helpfile=/var/install/help/xen \
               --helpname=XEN_CREATE_DOMU \
               --helpview \
               -s "$0"
    exit 0
fi
#trapdebug on
trap cleanUpOnSIGINT 2
trap cleanUpNormal EXIT
defaultName=`findDefaultname ${template}`
proposedName=`customizeHostname $defaultName index`
read_dhcp_leases

#-----------------------------------------------------------------------------
# user dialog
#-----------------------------------------------------------------------------
domainName="`determineName $proposedName index | awk 'BEGIN { RS=""; FS="\n"} {print $NF}'`"


case "$template" in
    fli4l*)
	mecho --info "Preparing the fli4l domU ..."
	cp "$defaultDomUConfigfile" "$domUConfigfile$delimiter$domainName"
	setVariable $domUConfigfile$delimiter$domainName IS_FLI4L yes
	fli4l_name=${template%%".tar"*}
	fli4l_version=${fli4l_name##*-}
	prepare_fli4l ${fli4l_version}
    ;;

    eisfair2.tar.bz2|eisfair1.tar.bz2)
	disk_n=1

	mecho --info "Template: $template"
	echo

	existingDisk=`/var/install/bin/ask --tty --info "Do you want to use an existing partition as disk device?\n" "no"`
	if [ "$existingDisk" = "no" ]
	then
    	    diskSizeM=`/var/install/bin/ask --tty --info "Disk size in MB\n" "4096" "32-1000000"`
	    diskSizeK=`expr $diskSizeM \* 1024`
	    diskImage=/dev/$volumeGroup/$domainName
	else
    	    diskImage=`/var/install/bin/ask --tty --info "Please enter the device to be used as image (e.g. /dev/vg/foo)?\n" "/dev/vg/foo" "+"`
        fi
	memM=`/var/install/bin/ask --tty --info "Memory size in MB\n" "64" "32-4096"`
	existingSwap=`/var/install/bin/ask --tty --info "Do you want to use an existing partition as swap device?\n" "no"`
	if [ "$existingSwap" = "no" ]
	then
    	    recommendedSwapSize=`expr $memM \* 2`
    	    swapSizeM=`/var/install/bin/ask --tty --info "Swap space in MB\n" "$recommendedSwapSize" "0-4096"`
    	    swapImage=/dev/$volumeGroup/${domainName}-swap
	else
    	    swapImage=`/var/install/bin/ask --tty --info "Please enter the device to be used as image (e.g. /dev/vg/foo)?\m" "/dev/vg/foo" "+"`
    	    swapSizeM=1
	fi
	bootPriority=20
	autostart=`/var/install/bin/ask --tty --info "Power on when Domain-0 boots?\n" "yes"`
	echo

	networkTemplateScriptInput
	passwordTemplateScriptInput
	trap cleanUpOnERR ERR

	# no nead to check if images dir exists, mkdir does that allready
	mkdir -p $xenImagePath

	# creating and populating images
	if [ $swapSizeM -gt 0 ]; then
    	    if [ "$existingSwap" = "no" ]; then
        	swapSizeK=`expr $swapSizeM \* 1024`
        	mecho --info "Creating swap image..."
        	swapImageCreated='true'
        	if vgs | grep -qs $volumeGroup; then
            	    createLvmImage swap $domainName $volumeGroup $swapSizeK swap
        	else
            	    createDiskImage swap $domainName $swapSizeK swap
            	    swapImage=$xenImagePath/$domainName.swap
        	fi
    	    else
        	useExistingDiskImage $swapImage
    	    fi
    	    disk_n=`expr $disk_n + 1`
	fi

	if [ "$existingDisk" = "no" ]
	then
    	    mecho --info "Creating disk image..."
    	    diskImageCreated='true'
    	    if vgs | grep -qs $volumeGroup
    	    then
        	mountOpts='-t ext3'
        	createLvmImage disk $domainName $volumeGroup $diskSizeK ext3
    	    else
        	mountOpts='-o loop'
        	createDiskImage disk $domainName $diskSizeK ext3
        	diskImage=$xenImagePath/$domainName.disk
    	    fi
	else
    	    useExistingDiskImage $diskImage
	fi
	mountpoint=`mktemp -d`
	mount $mountOpts $diskImage $mountpoint
	cd $mountpoint
	unpack verbose=$debugLvl "tarbal=$xenTemplatePath/$template" "info=Extracting template..."
	# copy kernel modules into domU
	# TODO this has to be dependant of the setting USE_CUSTOM_KERNEL
	copy_modules
	echo
	generatedMacAddress=`generateMac`
	# applying template specific settings
	mecho --info "Running postcreate script..."
	runasJob $debugLvl /data/xen/templates/scripts/$defaultName-postcreate.sh\
                                                    "VE_ROOT=$mountpoint"\
                                     "useSwap=${swapImageCreated:-false}"\
                                                    "passwd=$newPassword"\
                                                     "hostname=$hostname"\
                                                       "address=$address"\
                                                       "netmask=$netmask"\
                                                       "gateway=$gateway"\
                                                 "nameserver=$nameserver"\
                                                   "domain=$searchdomain"\
                               "generatedMacAddress=$generatedMacAddress"
	# saving settings
	mecho --info "Adding domain to configuration..."
	runasJob $debugLvl cp "$defaultDomUConfigfile" "$domUConfigfile$delimiter$domainName"
	setVariable "$domUConfigfile$delimiter$domainName" AUTO_START "$autostart"
	setVariable "$domUConfigfile$delimiter$domainName" MEMORY "$memM"
	setVariable "$domUConfigfile$delimiter$domainName" DISK_N "$disk_n"
	setVariable "$domUConfigfile$delimiter$domainName" DISK_1_UNAME "$diskImage"
	setVariable "$domUConfigfile$delimiter$domainName" VIF_1_MAC "$generatedMacAddress"
	${swapImageCreated:-false} && setVariable "$domUConfigfile$delimiter$domainName" DISK_2_UNAME "$swapImage"
	ln_relative -s "$domUConfigfile$delimiter$domainName" "$domUConfigfile_${bootPriority:-20}_$domainName"
    ;;

    hvm.tar.bz2)
        disk_n=1

        mecho --info "Template: $template"
        echo

        existingDisk=`/var/install/bin/ask --tty --info "Do you want to use an existing partition as disk device?\n" "no"`
        if [ "$existingDisk" = "no" ]
        then
            diskSizeM=`/var/install/bin/ask --tty --info "Disk size in MB\n" "4096" "32-1000000"`
            diskSizeK=`expr $diskSizeM \* 1024`
            diskImage=/dev/$volumeGroup/$domainName
        else
            diskImage=`/var/install/bin/ask --tty --info "Please enter the device to be used as image (e.g. /dev/vg/foo)?\n" "/dev/vg/foo" "+"`
        fi
        memM=`/var/install/bin/ask --tty --info "Memory size in MB\n" "64" "32-4096"`

        bootPriority=20
        autostart=`/var/install/bin/ask --tty --info "Power on when Domain-0 boots?\n" "yes"`
        echo

        trap cleanUpOnERR ERR

        # no nead to check if images dir exists, mkdir does that allready
        mkdir -p $xenImagePath

        # creating and populating images
        if [ "$existingDisk" = "no" ]
        then
            mecho --info "Creating disk image..."
            diskImageCreated='true'
            if vgs | grep -qs $volumeGroup
            then
		mountOpts='-t ext3'
		createLvmImage disk $domainName $volumeGroup $diskSizeK none
	    else
		mountOpts='-o loop'
		createDiskImage disk $domainName $diskSizeK none
		diskImage=$xenImagePath/$domainName.disk
	    fi
	else
            useExistingDiskImage $diskImage
	fi

        generatedMacAddress=`generateMac`

        # saving settings
        mecho --info "Adding domain to configuration..."
        runasJob verbose cp "$defaultDomUConfigfile" "$domUConfigfile$delimiter$domainName"
        setVariable "$domUConfigfile$delimiter$domainName" AUTO_START "$autostart"
        setVariable "$domUConfigfile$delimiter$domainName" MEMORY "$memM"
        setVariable "$domUConfigfile$delimiter$domainName" DISK_N 2
        setVariable "$domUConfigfile$delimiter$domainName" DISK_1_UNAME "$diskImage"
        setVariable "$domUConfigfile$delimiter$domainName" DISK_1_MAP_TO hda
	setVariable "$domUConfigfile$delimiter$domainName" DISK_2_SUBTYPE hdd
        setVariable "$domUConfigfile$delimiter$domainName" DISK_2_TYPE file
	setVariable "$domUConfigfile$delimiter$domainName" DISK_2_SUBTYPE cd
        setVariable "$domUConfigfile$delimiter$domainName" DISK_2_UNAME /data/xen/images/hvm.iso
        setVariable "$domUConfigfile$delimiter$domainName" DISK_2_MAP_TO hdc
        setVariable "$domUConfigfile$delimiter$domainName" DISK_2_MODE r
        setVariable "$domUConfigfile$delimiter$domainName" VIF_1_MAC "$generatedMacAddress"
        setVariable "$domUConfigfile$delimiter$domainName" HVM yes
        setVariable "$domUConfigfile$delimiter$domainName" BOOT d
        setVariable "$domUConfigfile$delimiter$domainName" LOCALTIME yes

	mecho --warn "Please copy your installation image to '/data/xen/images/hvm.iso' or modify domU configuration!!!"
	anykey
    ;;

    *)
        mecho --error "Unknown template: ${template}!"
        exit 1
    ;;

esac

# to create all desired symlinks
preSetup
runasJob $debugLvl sh /var/install/config.d/$domUConfigPrefix$delimiter$domainName.sh
/etc/init.d/xen start
exit 0