#!/bin/bash

### Initialize 
set -u
set -e

TESTING=

DO_SHOWSPEC=y
DO_VIRTINSTALL=

scripts=$(dirname $0)/scripts
. $scripts/lib.sh

### Command-line input
## Command-line options
function _help () {
    cat <<END
Usage: $1 [options] <name> [param=value [...]]

Options:
    -h      Show this help text

    -D      Debug
    -v      Verbose
    -q      Quiet

    -N      Mock command execution
    -P      Skip command prompts

    -I      Do a virt-install

    -T      Just testing..

Parameters are given as:

    name=value-{FOO}

END
}

while getopts "hDvqNPTI" opt; do
    case "$opt" in
        h)
            _help $0

            exit 0
            ;;

        D)
            LOG_DEBUG=y
            LOG_CMD=y

            ;;

        v)
            LOG_CMD=y

            log_debug "log: Commands"
            ;;

        q)
            LOG_CMD=
            LOG_INFO=
            DO_SHOWSPEC=
            
            log_debug "log: Quiet"
            ;;

        N)
            log_info "Mock-executing commands (this will break functionality..)"

            CMD_MOCK=y
            ;;

        P)
            log_info "Skipping command confirmations"

            CMD_PROMPT=
            ;;

        T)
            log_info "Just testing.."

            TESTING=y
            ;;

        I)
            log_info "Do virt-install"

            DO_VIRTINSTALL=y
            ;;
        ?)
            _help $0

            exit 1
            ;;
    esac
done


# forget them
[ $OPTIND -gt 1 ] && shift $(( $OPTIND - 1 ))

## Command-line arguments
# Name
[ -z $1 ] && die "Machine name must be given as first argument"

opt_name="$1"; shift

# Defaults
function define_opt () {
    local name=$1
    local value=${2:-}

    log_debugf "%-20s = %s" $name "$value"
    eval "opt_${name}=${value}"
}

function resolve_name () {
    local name=$1
    local out=$(dig +short $name)

    [ -z "$out" ] && die "Hostname lookup failed: $name"

    echo $out
}

define_opt  ram         1G
define_opt  cpus        2
define_opt  os          debiansqueeze
define_opt  disk_size   10G
define_opt  disk_vg     pvl
define_opt  disk_lv     $opt_name
define_opt  disk_bus    virtio
define_opt  guest_disk  /dev/vda
define_opt  hostname    $opt_name
define_opt  bridge      br-lan
define_opt  dns_domain  paivola.fi
define_opt  ip          
define_opt  puppet      
define_opt  puppet_master   puppet

log_info "Processing ${#@} parameters:"
for param in "$@"; do
    name=${param%=*}
    value=${param##*=}

    # evaluate
    value=$(expand_line "$value")

    define_opt ${name} "${value}"
done

# resolve defaults
if [ -z $opt_ip ]; then
    define_opt  ip          $(resolve_name ${opt_name}.${opt_dns_domain})
fi

### Virtual machine config
## libvirt guest info
# Name
GUEST_NAME=$opt_name

# Basic params
GUEST_RAM=$opt_ram
GUEST_VCPUS=$opt_cpus

# OS variant (for virt-install)
GUEST_OS_VARIANT=$opt_os

## Disk
# Size of LV to create
DISK_SIZE=$opt_disk_size

# LVM vg to use
DISK_VG=$opt_disk_vg

# LVM lv to use
DISK_NAME=$opt_disk_lv

# Path to disk block device
DISK_PATH=/dev/mapper/${DISK_VG}-${DISK_NAME}
GUEST_DISK_BUS=$opt_disk_bus

### Preseed content
GUEST_DISK=$opt_guest_disk

## Network
# Network configuration, for /etc/network/interfaces
NET_DOMAIN=paivola.fi
NET_HOSTNAME=$opt_hostname
NET_BRIDGE=$opt_bridge
NET_IPADDR=$opt_ip
NET_NETMASK=255.255.255.0
NET_GATEWAY=194.197.235.1
NET_NAMESERVERS=( 194.197.235.210 194.197.235.252 )

## Clock/time
TIME_ZONE='Europe/Helsinki'

# only used during install, not stored in target
TIME_NTP_SERVER=ntp.paivola.fi                      # XXX: harcoded

## User account
function getent_user_attr () {
    local user=$1
    local db=$2
    local attr=$3

    line=$(getent $db $user) || die "Unable to read $db database for $user"
    
    echo "$line" | cut -d ':' -f $attr
}
function user_fullname () {
    local user=$1

    getent_user_attr $user passwd 5
}
function user_shadow () {
    local user=$1

    if [ $UID -eq 0 ]; then
        log_debug "Get user password from shadow: $user"
        getent_user_attr $user shadow 2
    else
        echo -n "Install target login ($USER_NAME) " >&2
        mkpasswd -m sha-512
    fi
}

# XXX: hardcoded
USER_CREATE='true'
USER_NAME=$USER
USER_FULLNAME=$(user_fullname $USER)
USER_SHADOW=$(user_shadow $USER)
USER_GROUPS=( cdrom sudo adm )

## Misc
PACKAGE_INCLUDES=( sudo screen vim )
PRESEED_LATE_COMMANDS=( )
PRESEED_INCLUDES=( 'passwords.cfg' 'host.cfg' )


### Extra
## Puppet
if [ $opt_puppet ]; then
    log_info "Puppetizing preseed"

    PUPPET_PACKAGES=( puppet )
    PUPPET_COMMANDS=( \
#        "in-target sed -i 's/START=no/START=yes/' /etc/default/puppet"  \
#        "echo '[agent]\nserver = ${opt_puppet_master}\n' >> /etc/puppet/puppet.conf" \
    )

    # XXX: we use files in preseed/files/..., should modularize those

    PUPPET_MASTER="${opt_puppet_master}"

    PACKAGE_INCLUDES=( ${PACKAGE_INCLUDES[@]} ${PUPPET_PACKAGES[@]} )
    PRESEED_LATE_COMMANDS=( "${PRESEED_LATE_COMMANDS[@]:-}" "${PUPPET_COMMANDS[@]:-}" )
fi

### Installer setup
## Installation image
# Original Debian Installer image (iso)
INSTALLER_NAME="debian-6.0.3-amd64"
INSTALLER_ISO="iso-in/${INSTALLER_NAME}-netinst.iso"
INSTALLER_TREE="iso-in/$INSTALLER_NAME"
INSTALLER_FLAG="${INSTALLER_TREE}.unpacked"

# Customized preseed image name
INSTALL_NAME="debian-6.0.3-amd64_${GUEST_NAME}"

# Customized image content
INSTALL_TREE="images/${INSTALL_NAME}"
INSTALL_ISO="iso-out/${INSTALL_NAME}.iso"

## Preseed files
# Preseed output file in install tree
PRESEED_NAME="preseed.cfg"

# Directory containing our source templates
PRESEED_SOURCE_DIR="preseed"

# Mount path of preseed target in installer
PRESEED_MOUNT="/cdrom"

# Prefix for target files in install tree
# XXX: not implemented
#PRESEED_TARGET_PREFIX=""

# Main preseed source template
PRESEED_TEMPLATE="${PRESEED_SOURCE_DIR}/${PRESEED_NAME}"

# Target path for preseed in install tree
PRESEED_FILE="${INSTALL_TREE}/${PRESEED_NAME}"

# Checksum of target preseed.cfg
PRESEED_CHECKSUM= # set later

## Configs
CONF_FILES_SOURCE='preseed/files'
CONF_FILES_TARGET_NAME='preseed-files'
CONF_FILES_TARGET="${INSTALL_TREE}/${CONF_FILES_TARGET_NAME}"
CONF_FILES_CP_SRC="${PRESEED_MOUNT}/${CONF_FILES_TARGET_NAME}"
CONF_FILES_CP_DST='/target'

PRESEED_LATE_COMMANDS=( "${PRESEED_LATE_COMMANDS[@]:-}" \
    "cp -rd -- ${CONF_FILES_CP_SRC}/* ${CONF_FILES_CP_DST}"  \
)


# Additional files to copy
PRESEED_INCLUDE_FILES=( $(for preseed in ${PRESEED_INCLUDES[@]}; do echo "${PRESEED_SOURCE_DIR}/${preseed}"; done) )
#PRESEED_INCLUDE_FILES=("preseed/passwords.cfg" "preseed/host.cfg")

# preseed.cfg 'includes' line
PRESEED_INCLUDE=${PRESEED_INCLUDES[@]}

# preseed command execution
PRESEED_LATE_COMMAND=$(for cmd in "${PRESEED_LATE_COMMANDS[@]}"; do if [ "$cmd" ]; then echo -n "$cmd;" $'\\\n    '; fi; done; echo true)


## Isolinux
# Source for install tree isolinux file
PRESEED_ISOLINUX="${PRESEED_SOURCE_DIR}/isolinux.cfg"

# XXX: automagics? :)
#      virt-install assumes we have an X $DISPLAY if given --vnc
#      but we also want to enable the serial console...
#      d-i automagically configures the serial boot/login console, but only if the installer is run under serial...
#      we want to have both :(
if [ $DISPLAY ]; then
    INSTALL_BOOT_CONSOLE=""
else
    INSTALL_BOOT_CONSOLE="console=ttyS0"
fi

# installer boot args
INSTALL_BOOT_ARGS="auto=true priority=critical preseed/file=${PRESEED_MOUNT}/${PRESEED_NAME} preseed/file/checksum=${PRESEED_CHECKSUM} -- quiet ${INSTALL_BOOT_CONSOLE}"



### External progs
## Bootable .iso for Debian isolinux-based installer CDs
GENISOIMAGE=/usr/bin/genisoimage
GENISOIMAGE_OPTS="-r -J -no-emul-boot -boot-load-size 4 -boot-info-table -b isolinux/isolinux.bin -c isolinux/boot.cat"

## LVM
LVM=/sbin/lvm

## Libvirt 
# --connect URL
LIBVIRT=qemu:///system
VIRSH=/usr/bin/virsh

function virsh () {
    $VIRSH --connect $LIBVIRT "$@"
}

# type of guest to create
LIBVIRT_TYPE=kvm

## virt-install
VIRT_INSTALL="/usr/bin/virt-install"

## SELinux?
#SEMANAGE=/usr/sbin/semanage
#RESTORECON=/sbin/restorecon



### Prepare
if [ $DO_SHOWSPEC ]; then
    cat <<END
Guest:
    Name:       $GUEST_NAME
    CPUs:       $GUEST_VCPUS
    RAM:        $GUEST_RAM
    OS:         $GUEST_OS_VARIANT

    Disk:       $GUEST_DISK

Disk:
    Method:     LVM
    Size:       $DISK_SIZE
    LVM:
        VG:     $DISK_VG
        LV:     $DISK_NAME

    Path:       $DISK_PATH
    Bus:        $GUEST_DISK_BUS

Net:
    Hostname:   $NET_HOSTNAME
    Method:     Bridge
    IP:         $NET_IPADDR

    Bridge:
        Name:   $NET_BRIDGE

User:
    username:   $USER_NAME
    fullname:   $USER_FULLNAME
    shadow:     $USER_SHADOW

Installer:
    Name:       $INSTALLER_NAME
    ISO:        $INSTALLER_ISO
    Tree:       $INSTALLER_TREE

Install:
    Name:       $INSTALL_NAME
    Tree:       $INSTALL_TREE
    ISO:        $INSTALL_ISO

Preseed:
    Name:       $PRESEED_NAME
    Template:   $PRESEED_TEMPLATE
    Target:     $PRESEED_FILE
    Isolinux:   $PRESEED_ISOLINUX
    Includes:   $PRESEED_INCLUDES

END
fi

[ $TESTING ] && exit 0

### Check
# Parameters given?
[ -z $NET_IPADDR ] && die "net: No IP-address given: ip"

# Installer exists?
if cmd test ! -f ${INSTALLER_ISO}; then
    die "Installer not found: ${INSTALLER_ISO}"
fi

### Prepare install
## Extract .iso
if [ -f ${INSTALLER_FLAG} ]; then
    log_info "Installer already unpacked: ${INSTALLER_TREE}"

else
    log_info "Unpacking installer: ${INSTALLER_ISO}"
    cmd extract_iso ${INSTALLER_ISO} ${INSTALLER_TREE}
    cmd touch ${INSTALLER_FLAG}
fi

# Copy to customized tree
[ -d ${INSTALL_TREE} ] && cmd rm -r ${INSTALL_TREE}
cmd cp -rd ${INSTALLER_TREE} ${INSTALL_TREE}
cmd chmod -R u=rwX,og=rX ${INSTALL_TREE}

log_info "Installer extracted: $INSTALL_TREE"

## Customize preseed
cmd expand_template ${PRESEED_TEMPLATE} ${PRESEED_FILE}

# md5sum
PRESEED_CHECKSUM=$(my_md5sum $PRESEED_FILE)

# Isolinux .cfg
cmd expand_template ${PRESEED_ISOLINUX} ${INSTALL_TREE}/isolinux/isolinux.cfg

# Others
for file in ${PRESEED_INCLUDE_FILES[@]}; do
    name=$(basename $file)

    cmd expand_template $file ${INSTALL_TREE}/${name}
done

log_info "Preseed generated: $PRESEED_FILE"

# Files
log_info "Copy preseed-files"...
cmd expand_tree ${CONF_FILES_SOURCE} ${CONF_FILES_TARGET}

## Create .iso
[ -f ${INSTALL_ISO} ] && cmd rm -f ${INSTALL_ISO}

# generates a lot of output
cmd ${GENISOIMAGE} -o ${INSTALL_ISO} -quiet ${GENISOIMAGE_OPTS} ${INSTALL_TREE} 

log_info "Install ISO generated: $INSTALL_ISO"

### Create virtual machine
if [ $DO_VIRTINSTALL ]; then
    ## Check
    # VM exists?
    if cmd virsh domid ${GUEST_NAME} > /dev/null 2> /dev/null; then
        die "Virtual machine already exists: ${GUEST_NAME}"
    fi

    ## Disk
    # Create LV (unless it already exists)
    [ -e $DISK_PATH ] || cmd_confirm sudo $LVM lvcreate -L $DISK_SIZE -n $DISK_NAME $DISK_VG

    ## SELinux?
    #$SEMANAGE fcontext -a -t virt_image_t $DISK
    #$RESTORECON -v $DISK

    ## virt-install
    cmd_confirm $VIRT_INSTALL \
            --connect $LIBVIRT \
            --name $GUEST_NAME \
            --ram $(expand_MB $GUEST_RAM) --vcpus $GUEST_VCPUS \
            --cdrom "$INSTALL_ISO" \
            --os-variant $GUEST_OS_VARIANT \
            --disk path=$DISK_PATH,bus=$GUEST_DISK_BUS \
            --network bridge:$NET_BRIDGE \
            --vnc \
            --virt-type $LIBVIRT_TYPE \
            --accelerate --hvm \
            --serial pty
fi