#!/bin/bash -
# @configure_input@
# libguestfs virt-sysprep tool
# Copyright (C) 2011 Red Hat Inc.
#
# 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.
#
# 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., 675 Mass Ave, Cambridge, MA 02139, USA.

unset CDPATH
program="virt-sysprep"
version="@PACKAGE_VERSION@"

TEMP=`getopt \
        -o a:c:d:vVx \
        --long help,add:,connect:,domain:,enable:,format::,hostname:,list-operations,verbose,version \
        -n $program -- "$@"`
if [ $? != 0 ]; then
    echo "$program: problem parsing the command line arguments"
    exit 1
fi
eval set -- "$TEMP"

# This array accumulates the arguments we pass through to guestfish.
declare -a guestfish
guestfish[0]="guestfish"
guestfish[1]="--rw"
guestfish[2]="--listen"
guestfish[3]="-i"
i=4

verbose=
add_params=0
enable=
hostname_param=localhost.localdomain

usage ()
{
    echo "Usage:"
    echo "  $program [--options] -d domname"
    echo "  $program [--options] -a disk.img [-a disk.img ...]"
    echo
    echo "Read $program(1) man page for more information."
    echo
    echo "NOTE: $program modifies the guest or disk image *in place*."
    exit $1
}

while true; do
    case "$1" in
        -a|--add)
            guestfish[i++]="-a"
            guestfish[i++]="$2"
            ((add_params++))
            shift 2;;
        -c|--connect)
            guestfish[i++]="-c"
            guestfish[i++]="$2"
            shift 2;;
        -d|--domain)
            guestfish[i++]="-d"
            guestfish[i++]="$2"
            ((add_params++))
            shift 2;;
        --enable)
            if [ -n "$enable" ]; then
                echo "error: --enable option can only be given once"
                exit 1
            fi
            enable="$2"
            shift 2;;
        --format)
            if [ -n "$2" ]; then
                guestfish[i++]="--format=$2"
            else
                guestfish[i++]="--format"
            fi
            shift 2;;
        --help)
            usage 0;;
        --hostname)
            hostname_param="$2"
            shift 2;;
        --list-operations)
            enable=list
            shift;;
        -v|--verbose)
            guestfish[i++]="-v"
            verbose=yes
            shift;;
        -V|--version)
            echo "$program $version"
            exit 0;;
        -x)
            guestfish[i++]="-x"
            shift;;
        --)
            shift
            break;;
        *)
            echo "Internal error!"
            exit 1;;
    esac
done

# Different sysprep operations that can be enabled.  Default is to
# enable all of these, although some of them are only done on certain
# guest types (see details below).
if [ -z "$enable" ]; then
    hostname=yes
    net_hwaddr=yes
    ssh_hostkeys=yes
    udev_persistent_net=yes
elif [ "$enable" = "list" ]; then
    echo "hostname"
    echo "net-hwaddr"
    echo "ssh-hostkeys"
    echo "udev-persistent-net"
    exit 0
else
    for opt in $(echo "$enable" | sed 's/,/ /g'); do
        case "$opt" in
            hostname)              hostname=yes ;;
            net-hwaddr)            net_hwaddr=yes ;;
            ssh-hostkeys)          ssh_hostkeys=yes ;;
            udev-persistent-net)   udev_persistent_net=yes ;;
            *)
                echo "error: unknown --enable feature: $opt"
                exit 1
        esac
    done
fi

# Make sure there were no extra parameters on the command line.
if [ $# -gt 0 ]; then
    echo "error: $program: extra parameters on the command line"
    echo
    usage 1
fi

# Did the user specify at least one -a or -d option?
if [ $add_params -eq 0 ]; then
    echo "error: $program: you need at least one -a or -d option"
    echo
    usage 1
fi

# end of command line parsing
#----------------------------------------------------------------------

set -e

if [ "$verbose" = "yes" ]; then
    echo command: "${guestfish[@]}"
fi

# Create a temporary directory for general purpose use during operations.
tmpdir="$(mktemp -d)"

# Increase the amount of memory allocated to the appliance because
# we're using augeas.  The user can override this by setting
# $LIBGUESTFS_MEMSIZE before running the script.
export LIBGUESTFS_MEMSIZE=${LIBGUESTFS_MEMSIZE:-2048}

# Call guestfish.
GUESTFISH_PID=
eval $("${guestfish[@]}")
if [ -z "$GUESTFISH_PID" ]; then
    echo "$program: guestfish didn't start up, see error messages above"
    exit 1
fi

# Helper.
gf="guestfish --remote --"

cleanup ()
{
    $gf exit >/dev/null 2>&1 ||:
    rm -rf "$tmpdir" ||:
}
trap cleanup EXIT ERR

# Launch back-end, inspect for operating systems, and get the guest
# root disk.
root=$($gf inspect-get-roots)

if [ "$root" = "" ]; then
    echo "$program: no operating system was found on this disk"
    exit 1
fi

if [ "$verbose" = "yes" ]; then
    echo root: "$root"
fi

# Get the guest type.
type="$($gf -inspect-get-type $root)"

if [ "$type" = "linux" ]; then
    distro="$($gf -inspect-get-distro $root)"
fi

if [ "$type" = "windows" ]; then
    systemroot="$($gf -inspect-get-windows-systemroot $root)"
fi

# Start Augeas if it's a Linux guest.
if [ "$type" = "linux" ]; then
    $gf aug-init / 0
    using_augeas=yes
fi

#----------------------------------------------------------------------
# Useful functions.

# erase_line filename regex
#
# Erase line(s) in a file that match the given regex.
erase_line ()
{
    $gf download "$1" "$tmpdir/file"
    sed "/$2/d" < "$tmpdir/file" > "$tmpdir/file.1"
    $gf upload "$tmpdir/file.1" "$1"
}

# prepend_line filename line
#
# Prepend a line to a file (this is better than appending, because it
# works even when the original file isn't terminated with a newline).
prepend_line ()
{
    $gf download "$1" "$tmpdir/file"
    echo "$2" > "$tmpdir/file.1"
    cat "$tmpdir/file.1" "$tmpdir/file" >> "$tmpdir/file.2"
    $gf upload "$tmpdir/file.2" "$1"
}

# rm_files wildcard
#
# Remove files.  Doesn't fail if no files exist.  Note the wildcard
# parameter cannot contain spaces or characters that need special
# quoting.
rm_files ()
{
    files=$($gf glob-expand "$1")
    for f in $files; do
        $gf rm "$f"
    done
}

# rm_file filename
#
# Remove a single file.  No error if the file doesn't exist or is not
# a file.
rm_file ()
{
    t=$($gf is-file "$1")
    if [ "$t" = "true" ]; then
        $gf rm "$1"
    fi
}

#----------------------------------------------------------------------
# The sysprep operations.

if [ "$hostname" = "yes" ]; then
    case "$type/$distro" in
        linux/fedora)
            $gf aug-set /files/etc/sysconfig/network/HOSTNAME "$hostname_param"
            augeas_save_needed=yes
            ;;
        linux/debian|linux/ubuntu)
            $gf write /etc/hostname "$hostname_param"
            ;;
    esac
fi

if [ "$net_hwaddr" = "yes" ]; then
    case "$type/$distro" in
        linux/fedora)
            # XXX these filenames can have spaces and untrusted chars in them!
            nodes=$( $gf aug-ls /files/etc/sysconfig/network-scripts |
                     grep /files/etc/sysconfig/network-scripts/ifcfg- )
            for node in $nodes; do
                $gf -aug-rm "$node/HWADDR" >/dev/null
                augeas_save_needed=yes
            done
            ;;
    esac
fi

if [ "$ssh_hostkeys" = "yes" -a "$type" != "windows" ]; then
    rm_files "/etc/ssh/*_host_*"
fi

if [ "$udev_persistent_net" = "yes" -a "$type" = "linux" ]; then
    rm_file /etc/udev/rules.d/70-persistent-net.rules
fi

#----------------------------------------------------------------------
# Clean up and close down.

if [ "$using_augeas" = "yes" -a "$augeas_save_needed" = "yes" ]; then
    $gf aug-save
    $gf aug-close
fi

$gf umount-all
$gf sync
$gf exit

exit 0