#!/bin/bash
set -e
PROGNAME=$(basename $0)
#  $HeadURL: http://sos16-vip/svn/kickstart/trunk/bin/defaults-dumper $ $LastChangedRevision: 669 $

#  Load libraries
. $(dirname $0)/../include/mkdi.sh

main()
{
    local TARGET_HOSTNAME TARGET_IPADDR TARGET_MACADDR TARGET_RELEASE

    #  Defaults for options
    VERBOSELEVEL=2

    #  Process options
    while [ "X$1" != X ]; do
        case $1 in 
            #  Application specific options
            --target-hostname=*) TARGET_HOSTNAME=${1#*=} ;;
            --target-ipaddr=*)   TARGET_IPADDR=${1#*=} ;;
            --target-macaddr=*)  TARGET_MACADDR=${1#*=} ;;
            --target-release=*)  TARGET_RELEASE=${1#*=} ;;
            #  General options
            --help|-h)           usage 0 ;;
            --verbose|-v)        VERBOSELEVEL=3 ;;
            -d)                  VERBOSELEVEL=$2; shift ;;
            --debug=*)           VERBOSELEVEL=${1#*=} ;;
            --)                  shift; break ;;
            -*)                  usage ;;
            *)                   break ;; 
        esac
        shift
    done

    #  Process arguments
    [ $# -ge 1 ] || usage
    MODE=$1; shift
    debug 10 "main: MODE=$MODE"

    #  Sanity checks and derivations

    #  Guts
    case $MODE in
        create) create --target-hostname=$TARGET_HOSTNAME --target-ipaddr=$TARGET_IPADDR --target-macaddr=$TARGET_MACADDR --target-release=$TARGET_RELEASE "$@" || return $? ;;
        delete) delete --target-hostname=$TARGET_HOSTNAME --target-ipaddr=$TARGET_IPADDR --target-macaddr=$TARGET_MACADDR --target-release=$TARGET_RELEASE "$@" || return $? ;;
        list)   list "$@" || return $? ;;
        purge)  purge "$@" || return $? ;;
        *)      usage ;;
    esac

    return 0
}

create()
{
    local TARGET_HOSTNAME TARGET_IPADDR TARGET_MACADDR TARGET_RELEASE

    #  Process options
    while [ "X$1" != X ]; do
        case $1 in
            #  Application specific options
            --target-hostname=*) TARGET_HOSTNAME=${1#*=} ;;
            --target-ipaddr=*)   TARGET_IPADDR=${1#*=} ;;
            --target-macaddr=*)  TARGET_MACADDR=${1#*=} ;;
            --target-release=*)       TARGET_RELEASE=${1#*=} ;;
            #  General options
            --)                  shift; break ;;
            -*)                  internal "create: $1: invalid option" ;;
            *)                   break ;; 
        esac
        shift
    done

    #  Process arguments
    [ $# = 0 ] || internal "create: $#: wrong arg count"
    
    #  Create entry
    info "creating PXE file ..."
    create_or_delete --target-hostname=$TARGET_HOSTNAME --target-ipaddr=$TARGET_IPADDR --target-macaddr=$TARGET_MACADDR --target-release=$TARGET_RELEASE create || return $?

    return 0
}

delete()
{
    local TARGET_HOSTNAME TARGET_IPADDR TARGET_MACADDR TARGET_RELEASE

    #  Process options
    while [ "X$1" != X ]; do
        case $1 in
            #  Application specific options
            --target-hostname=*) TARGET_HOSTNAME=${1#*=} ;;
            --target-ipaddr=*)   TARGET_IPADDR=${1#*=} ;;
            --target-macaddr=*)  TARGET_MACADDR=${1#*=} ;;
            --target-release=*)  TARGET_RELEASE=${1#*=} ;;
            #  General options
            --)                  shift; break ;;
            -*)                  internal "delete: $1: invalid option" ;;
            *)                   break ;;
        esac
        shift
    done

    #  Process arguments
    [ $# = 0 ] || internal "delete: $#: wrong arg count"
   
    #  Delete entry
    info "deleting PXE file ..."
    create_or_delete --target-hostname=$TARGET_HOSTNAME --target-ipaddr=$TARGET_IPADDR --target-macaddr=$TARGET_MACADDR --target-release=$TARGET_RELEASE delete || return $?

    return 0
}

create_or_delete()
{
    local TARGET_HOSTNAME TARGET_IPADDR TARGET_MACADDR TARGET_RELEASE
    local MODE LINE DHCPD_OUTPUT

    #  Process options
    while [ "X$1" != X ]; do
        case $1 in
            #  Application specific options
            --target-hostname=*) TARGET_HOSTNAME=${1#*=} ;;
            --target-ipaddr=*)   TARGET_IPADDR=${1#*=} ;;
            --target-macaddr=*)  TARGET_MACADDR=${1#*=} ;;
            --target-release=*)  TARGET_RELEASE=${1#*=} ;;
            #  General options
            --)                  shift; break ;;
            -*)                  internal "create_or_delete: $1: invalid option" ;;
            *)                   break ;;
        esac
        shift
    done

    #  Process arguments
    [ $# = 1 ] || internal "create_or_delete: $#: wrong arg count"
    MODE=$1
    
    #  Sanity checks and derivations
    validate_vars MODE TARGET_HOSTNAME TARGET_IPADDR TARGET_MACADDR TARGET_RELEASE || return $?
    debug 10 "main: MODE=$MODE, TARGET_HOSTNAME=$TARGET_HOSTNAME, TARGET_IPADDR=$TARGET_IPADDR, TARGET_MACADDR=$TARGET_MACADDR, TARGET_RELEASE=$TARGET_RELEASE"

    #  Guts
    if [ $MODE = create ]; then
        debug 10 "create: creating file ..."
        {
            echo "default linux"
            echo "prompt 0"
            echo "timeout 30"
            echo
            echo "label linux"
            echo "    kernel linux_$TARGET_RELEASE"
            echo "    append initrd=initrd.gz_$TARGET_RELEASE debian-installer/language=en debian-installer/locale=en_US.UTF-8 console-keymaps-at/keymap=us netcfg/get_hostname=debian netcfg/get_domain=pasta.net preseed/url=http://$PRESEEDSERVER_IPADDR${PRESEEDSERVER_DIR#$PRESEEDSERVER_DOCROOT}/$TARGET_HOSTNAME.cfg"
        } > /var/lib/tftpboot/pxelinux.cfg/01-${TARGET_MACADDR//:/-}
 
    elif [ $MODE = delete ]; then
        debug 10 "create_or_delete: deleting file ..."
        RM_OUTPUT=$(rm /var/lib/tftpboot/pxelinux.cfg/01-${TARGET_MACADDR//:/-} 2>&1) || warning "rm: failed (output was: $RM_OUTPUT)"
    fi

    return 0
}

purge()
{
    #  Guts
    rm /var/lib/tftpboot/pxelinux.cfg/01-??-??-??-??-??-?? 2>/dev/null

    return 0
}

list()
{
   #  Process options
    while [ "X$1" != X ]; do
        case $1 in
            #  Application specific options
            #  General options
            --)                  shift; break ;;
            -*)                  internal "list: $1: invalid option" ;;
            *)                   break ;;
        esac
        shift
    done

    #  Process arguments
    [ $# = 0 ] || internal "list: $#: wrong arg count"

    #  Guts
    ls -l /var/lib/tftpboot/pxelinux.cfg/01-* 2>/dev/null

    return 0
}

usage()
{
    local RC

    RC=${1:-1}

    {
        echo "Usage: $PROGNAME [ <options> ] { create | delete | list }"
        echo
    } | if [ $RC = 0 ]; then
        cat
    else
        cat >&2
    fi

    exit $RC
}

main "$@"