#!/bin/bash
# Creation:     2013-05-27 jv <jens@eisfair.org>
# Copyright (c) 2000-2013 The eisfair Team <team@eisfair.org>
#-------------------------------------------------------------------------------
# ToDo: LVM option for /data or/and /
#       correct timezone select
#       optional view logfile

hw_backtitle() {
    echo "Alpine Linux with eisfair-ng - Installation   $PDRIVE"
    return 0
}


isValidIp() {
    if echo $1 | egrep -qs '\b(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\b' ; then
        return 0
    else
        return 1
    fi
}


countDisks() {
    if [ $# -gt 1 ] ; then 
        return 0
    else
        return 1
    fi
}


calulate_swap_size() {
    local memtotal_kb=$(awk '$1 == "MemTotal:" {print $2}' /proc/meminfo)
    local size=$(( $memtotal_kb * 2 / 1024 ))
    local disk= disksize=
    for disk in $@; do
        local sysfsdev=$(echo ${disk#/dev/} | sed 's:/:!:g')
        local sysfspath=/sys/block/$sysfsdev/size
        maxsize=$(awk '{ printf "%i", $0 / 8192 }' $sysfspath )
        if [ $size -gt $maxsize ]; then
            size=$maxsize
        fi
    done
    if [ $size -gt 4096 ]; then
        # dont ever use more than 4G
        size=4096
    elif [ $size -lt 64 ]; then
        # dont bother create swap smaller than 64MB
        size=0
    fi
    echo $size
}


getNextMenuItem () {
    : $(( n_item++ ))
}


PDRIVE=""
PRAIDLEVEL="0"
PSWAPSIZE="512"
PKEYBLAYOUT="de"
PKEYBVARIANT="de-latin1"
PNETIPSTATIC="1"
PIPADDRESS="192.168.1.2"
PNETMASK="255.255.255.0"
PGATEWAY="192.168.1.1"
PHOSTNAME="eis"
PDOMAIN="eisfair.home"
PDNSSERVER=""
PTIMEZONE="Europe/Berlin"
PPASSWORD="eis"
POPTIONS=""

n_item="1"

# add packages for install setup and load default keymap
apk add -q bkeymaps
[ -f "/usr/share/bkeymaps/$PKEYBLAYOUT/$PKEYBVARIANT.bmap" ] && cat "/usr/share/bkeymaps/$PKEYBLAYOUT/$PKEYBVARIANT.bmap" | loadkmap

while true ; do
    if [ "$PNETIPSTATIC" = "1" ] ; then
        n_item=$(dialog --stdout --no-shadow --no-cancel --item-help \
            --backtitle "$(hw_backtitle)" \
            --default-item "$n_item" \
            --title " Server configuration "  --clear \
            --menu "Base setup" 21 50 14 \
             0 "Shell login" "Return to login" \
             1 "Select disc" "Select disc for installation." \
             2 "Adjust partition size" "Adjust the size of swap/root partition." \
             3 "Keyboard layout" "Setup the keyboard layout." \
             4 "Use DHCP for network" "Automatic IP-address of first network interface." \
             5 "IP-address" "IP-address of first network interface." \
             6 "Netmask" "Netmask of IP interface."\
             7 "Gateway" "Default Gateway for interface." \
             8 "DNS Server" "DNS Server." \
             9 "Hostname" "System Hostname."\
            10 "Domain" "DNS Domain name." \
            11 "Root password" "Set password for user root" \
            12 "Start installation" "Start installation" \
            13 "Reboot server" "Reboot server after installation"  )
    else
        n_item=$(dialog --stdout --no-shadow --no-cancel --item-help \
            --backtitle "$(hw_backtitle)" \
            --default-item "$n_item" \
            --title "Server configuration"  --clear \
            --menu "Base setup" 21 50 14 \
             0 "Shell login" "Return to login" \
             1 "Select disc" "Select disc for installation." \
             2 "Adjust partition size" "Adjust the size of swap/root partition." \
             3 "Keyboard layout" "Setup the keyboard layout." \
             4 "Use DHCP for network" "Automatic IP-address of first network interface." \
             9 "Hostname" "System Hostname."\
            10 "Domain" "DNS Domain name." \
            11 "Root password" "Set password for user root" \
            12 "Start installation" "Start installation" \
            13 "Reboot server" "Reboot server after installation"  )
    fi

    case ${n_item} in
        1)
            ### Select drive ######################################################
            if [ `fdisk -l | grep "^Disk /.*:" | wc -l` = '1' ] ; then
                drivelist=$(fdisk -l | sed -n 's/^Disk \(\/dev\/[^:]*\): \([^, ]*\) \([MGTB]*\).*$/\1 \2_\3 on/p')
            else
                drivelist=$(fdisk -l | sed -n 's/^Disk \(\/dev\/[^:]*\): \([^, ]*\) \([MGTB]*\).*$/\1 \2_\3 off/p')
            fi
            if [ -z "$drivelist" ] ; then
                dialog --backtitle "$(hw_backtitle)" --title "" \
                    --msgbox " No drive found!\n Please try again." 6 30
            else
                new=$(dialog --stdout --no-shadow \
                    --backtitle "$(hw_backtitle)" \
                    --title "Select drive" \
                    --clear \
                    --checklist "Select drive(s) to partition:" 12 40 6 \
                    $drivelist )
                if [ -n "$new" ] ; then
                    if countDisks $new ; then
                        dialog --stdout --no-shadow \
                            --backtitle "$(hw_backtitle)" \
                            --title "Software RAID installation"  --clear \
                            --yesno "Use drives for RAID:\n${new}" 7 40
                        if [ "$?" = "0" ] ; then
                            PRAIDLEVEL="1"
                            PDRIVE="$new"
                            getNextMenuItem
                        fi
                    else
                        PRAIDLEVEL="0"
                        PDRIVE="$new"
                        getNextMenuItem
                    fi
                fi
            fi
            ;;
        2)
            if [ -z "$PDRIVE" ] ; then
                n_item="1"
            else
                PSWAPSIZE=$(calulate_swap_size ${PDRIVE})
                new=$(dialog --stdout --no-shadow \
                    --backtitle "$(hw_backtitle)" \
                    --title "Adjust size of swap Partition"  --clear \
                    --inputbox "Size in MB:" 10 45 "$PSWAPSIZE")
                if [ "$?" -eq 0 ] ; then
                    PPSWAPSIZE="$new"
                    getNextMenuItem
                fi
            fi
            ;;
        3)
            ### Keyboard configuration #########################################
            sellist=""
            cd /usr/share/bkeymaps
            for I in * ; do
                sellist="${sellist}${I} . "
            done
            PKEYBLAYOUT=$(dialog --stdout --no-shadow --no-cancel \
                --backtitle "$(hw_backtitle)" \
                --default-item "$PKEYBLAYOUT" \
                --title "Keyboard configuration"  --clear \
                --menu "Select keyboard layout:" 18 40 11 \
                $sellist )
            sellist=""
            cd /usr/share/bkeymaps/$PKEYBLAYOUT
            for I in * ; do
                stemp="$(basename $I .bmap)"
                sellist="${sellist}$stemp . "
            done
            new=$(dialog --stdout --no-shadow --no-cancel \
                --backtitle "$(hw_backtitle)" \
                --default-item "$PKEYBVARIANT" \
                --title "Keyboard configuration"  --clear \
                --menu "Available variants:" 18 40 11 \
                $sellist )
            if [ "$?" -eq 0 ] ; then
                PKEYBVARIANT="$new"
                cat /usr/share/bkeymaps/$PKEYBLAYOUT/$PKEYBVARIANT.bmap | loadkmap
                getNextMenuItem
            fi
            ;;
            ### DHCP ###########################################################
        4)
            dialog --stdout --no-shadow \
                --backtitle "$(hw_backtitle)" \
                --title "Network interface IP-address"  --clear \
                --yesno "Get automatic IP-address with DHCP?" 5 40
            PNETIPSTATIC="$?"
            if [ "$PNETIPSTATIC" = "1" ] ; then
                getNextMenuItem
            else
                n_item="9"
            fi
            ;;
        5)
            ### IP-address #####################################################
            while true ; do
                new=$(dialog --stdout --no-shadow \
                    --backtitle "$(hw_backtitle)" \
                    --title "Network interface IP-address"  --clear \
                    --inputbox "Enter the IP-Address (192.168.1.2)" 10 45 "$PIPADDRESS")
                if [ "$?" -eq 0 ] ; then
                    if isValidIp $new ; then
                        PIPADDRESS="$new"
                        getNextMenuItem
                        break
                    else
                        dialog --backtitle "$(hw_backtitle)" --title "" \
                            --msgbox " Wrong IP-address!\n Please try again." 6 30
                    fi
                else
                    break
                fi
            done
            ;;
        6)
            ### Netmask ########################################################
            while true ; do
                new=$(dialog --stdout --no-shadow \
                    --backtitle "$(hw_backtitle)" \
                    --title "Network mask"  --clear \
                    --inputbox "Enter the network mask (255.255.255.0)" 10 45 "$PNETMASK")
                if [ "$?" -eq 0 ] ; then
                    if isValidIp $new ; then
                        PNETMASK="$new"
                        getNextMenuItem
                        break
                    else
                        dialog --backtitle "$(hw_backtitle)" --title "" \
                            --msgbox " Wrong network mask!\n Please try again." 6 30
                    fi
                else
                    break
                fi
            done
            ;;
        7)
            ### Gateway ########################################################
            while true ; do
                new=$(dialog --stdout --no-shadow \
                    --backtitle "$(hw_backtitle)" \
                    --title "Default gateway"  --clear \
                    --inputbox "Enter the default gateway (192.168.1.1)" 10 45 "$PGATEWAY")
                if [ "$?" -eq 0 ] ; then
                    if isValidIp $new ; then
                        PGATEWAY="$new"
                        [ -z "$PDNSSERVER" ] && PDNSSERVER="$new"
                        getNextMenuItem
                        break
                    else
                        dialog --backtitle "$(hw_backtitle)" --title "" \
                            --msgbox " Wrong default gateway!\n Please try again." 6 30
                    fi
                else
                    break
                fi
            done
            ;;
        8)
            ### DNS Server #####################################################
            new=$(dialog --stdout --no-shadow \
                --backtitle "$(hw_backtitle)" \
                --title "Management DNS Servers"  --clear \
                --inputbox " Enter a space delimited list of DNS Servers to be used by the Managment Interface" 10 45 "${PDNSSERVER}")
            if [ "$?" -eq 0 ] ; then
                PDNSSERVER="${new}"
                getNextMenuItem
            fi
            ;;
        9)
            ### Hostname #######################################################
            while true ; do
                new=$(dialog --stdout --no-shadow --backtitle "$(hw_backtitle)" \
                    --title "Management Hostname"  --clear \
                    --inputbox "Enter the system Hostname (Host only part of FQDN)\n For example, the Hostname of the FQDN \"myhost.some.domain\" is \"myhost\"." 10 55 "${PHOSTNAME}")
                if [ $? -eq 0 ] ; then
                    if [ "x${new}" == "x" ] ; then
                        dialog --stdout --no-shadow --cancel-label "Return to main menu"\
                            --backtitle "$(hw_backtitle)" \
                            --title "Management Hostname"  --clear \
                            --msgbox "\n   Hostname cannot be null." 10 55
                        [ $? -eq 1 ] && break
                    elif [ "$(echo "${new}" | egrep -c "[\.|_]")" -gt 0 ] ; then
                        dialog --stdout --no-shadow --cancel-label "Return to main menu"\
                            --backtitle "$(hw_backtitle)" \
                            --title "Management Hostname"  --clear \
                            --msgbox "\n   Hostname cannot contian \".\" or \"_\" ." 10 55
                        [ $? -eq 1 ] && break
                    else
                        PHOSTNAME="${new}"
                        getNextMenuItem
                        break
                    fi
                elif [ $? -eq 1 ] ; then
                    break
                fi
            done
            ;;
       10)
            ### Domain #########################################################
            while true ; do
                new=$(dialog --stdout --no-shadow \
                    --backtitle "$(hw_backtitle)" \
                    --title "Domain Name"  --clear \
                    --inputbox "Enter the DNS domain if you have one (localhost if not).  The DNS domain will be appended to the Hostname to form the fully qualified domain name (FQDN)." 12 55 "${PDOMAIN}")

                if [ $? -eq 0 ] ; then
                    if [ "x${new}" == "x" ] ; then
                        dialog --stdout --no-shadow --cancel-label "Return to main menu"\
                            --backtitle "$(hw_backtitle)" \
                            --title "Management Domain Name"  --clear \
                            --msgbox "\n   Domain cannot be null." 10 55
                        [ $? -eq 1 ] && break;
                    else
                        PDOMAIN="${new}"
                        getNextMenuItem
                        break
                    fi
                elif [ $? -eq 1 ] ; then
                    break
                fi
            done
            ;;
        11)
            ### root password ##################################################
            new=$(dialog --stdout --no-shadow \
                --backtitle "$(hw_backtitle)" \
                --title "Set root password" \
                --clear --no-cancel --insecure \
                --passwordbox "Enter a password for user root" 10 45 "${free}")
            new2=$(dialog --stdout --no-shadow \
                --backtitle "$(hw_backtitle)" \
                --title "Confirm root password" \
                --clear --no-cancel --insecure \
                --passwordbox "Re enter password for user root" 10 45 "${free}")
            if [ "$?" -eq 0 ] ; then
                if [ "$new" = "$new2" ] ; then
                    PPASSWORD="${new}"
                    getNextMenuItem
                else
                    dialog --backtitle "$(hw_backtitle)" --title "" \
                      --msgbox " Password not match!\n Please try again." 6 30
                fi
            fi
            ;;
        12)
            ### Start installation #############################################
            if [ -n "$PDRIVE" ] ; then
                dialog --stdout --no-shadow \
                  --backtitle "$(hw_backtitle)" \
                  --title "Start installation"  --clear \
                  --yesno "Delete all partitions on drive(s):\n${PDRIVE}\nand start installation?" 8 40
                if [ "$?" = "0" ] ; then
                    [ "$PNETIPSTATIC" = "0" ] && POPTIONS="$POPTIONS -d"
                    [ "$PRAIDLEVEL" = "1" ] && POPTIONS="$POPTIONS -r"
                    PRINTK=$(cat /proc/sys/kernel/printk)
                    echo "0" >/proc/sys/kernel/printk
                    tempfile=/tmp/install.$$
                    trap "rm -f $tempfile" 0 1 2 5 15
                    date > /tmp/fdisk.log 
                    dialog --no-shadow \
                      --title "Start installation"  \
                      --backtitle "$(hw_backtitle)" \
                      --no-kill \
                      --tailboxbg /tmp/fdisk.log 21 75 2>$tempfile
                    /bin/eis-install.setup-disk -e "$PKEYBVARIANT" -E "$PKEYBLAYOUT" \
                        -H "$PHOSTNAME" -D "$PDOMAIN" -I "$PIPADDRESS" -N "$PNETMASK" \
                        -G "$PGATEWAY" -F "$PDNSSERVER" -P "$PPASSWORD" -s "$PSWAPSIZE" \
                        ${POPTIONS} $PDRIVE >>/tmp/fdisk.log 2>&1
                    sleep 3; kill -3 `cat $tempfile` 2>&1 >/dev/null 2>/dev/null
                    echo "$PRINTK" > /proc/sys/kernel/printk
                    getNextMenuItem
                fi
            else
                n_item="1"
            fi
            ;;
        13)
            echo "1" > /reboot
            clear 
            break
            ;;
        0)
            break
            ;;
    esac
done

exit 0