#!/bin/sh

################################################################################
#      This file is part of OpenELEC - http://www.openelec.tv
#      Copyright (C) 2009-2012 Stephan Raue (stephan@openelec.tv)
#
#  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, 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 OpenELEC; see the file COPYING.  If not, write to
#  the Free Software Foundation, 51 Franklin Street, Suite 500, Boston, MA 02110, USA.
#  http://www.gnu.org/copyleft/gpl.html
################################################################################

# some DOCs:

# list devices:
# cat /proc/partitions | sed -n "s/\ *[0-9][0-9]*\ *[0-9][0-9]*\ *[0-9][0-9]*\ \([a-z]*\)$/\1/p"

# list all partitionnumbers from /dev/sda:
# parted -m /dev/sda print |grep -v ^/dev |grep -v ^BYT | cut -f1 -d ":"

# list device data from /dev/sda:
# parted -m /dev/sda print |grep /dev/sda

# list mounted partitions:
# mount |grep ^/dev

# list modelnumber:
# parted -m /dev/sda print |grep /dev/sda | cut -f7 -d ":" | sed "s/;//"
# list size:
# parted -m /dev/sda print |grep /dev/sda | cut -f2 -d ":"

# exclude mounted partitions
# for i in `cat /proc/mounts | grep ^/dev/ | cut -f1 -d " " | sed "s/[0-9]//"`; do TEST="$TEST `echo "| grep -v $i"`"; done

# Interpret embedded "\Z" sequences in the dialog text by the following
# character, which tells dialog to set colors or video attributes: 0 through 7
# are the ANSI used in curses: black, red, green, yellow, blue, magenta, cyan
# and white respectively. Bold is set by 'b', reset by 'B'. Reverse is set
# by 'r', reset by 'R'. Underline is set by 'u', reset by 'U'. The settings are
# cumulative, e.g., "\Zb\Z1" makes the following text bold (perhaps bright)
# red. Restore normal settings with "\Zn".

[ -f /etc/installer.conf ] && . /etc/installer.conf || exit 0

dbglg() {
  # Acts just like echo cmd, with automatic redirection

    echo "" >> $LOGFILE
    echo "###################################################################" >> $LOGFILE
    echo "# $@" >> $LOGFILE
    echo "###################################################################" >> $LOGFILE
    echo "" >> $LOGFILE
}

get_device_unmount() {
  # get all unmounted devices
  # usage: get_devices_unmount
  # uses:     -
  # provides: DEVICES

    DEVICES=""
    DEVICES=$(parted -s -m -l | grep ^/dev/sd | cut -f1 -d ":")

    for i in $(cat /proc/mounts | grep ^/dev/sd | cut -f1 -d " " | sed "s/[0-9]//"); do
      DEVICES=$(echo $DEVICES |sed -e "s|$i||")
    done
}

get_partition() {
  # get all partitions of a specifed device
  # usage:    get_partitions /dev/sda
  # uses:     -
  # provides: PARTITIONS

    PARTITIONS=$(parted -s -m $1 print |grep -v ^/dev |grep -v BYT | cut -f1 -d ":")
}

create_device_list() {
  # creates device list to use in menus
  # usage:    create_devices_list
  # uses:     get_device_unmount
  # provides: DEVICE_MODEL, DEVICE_SIZE, DEVICE_LIST, DEVICE_NAME,
  #           DEVICES (get_device_unmount)

    DEVICE_MODEL=""
    DEVICE_SIZE=""
    DEVICE_LIST=""
    DEVICE_NAME=""

    get_device_unmount

    if [ "$DEVICES" = "" ]; then
      msg_no_device
    fi

    for i in $DEVICES; do
      DEVICE_MODEL=$(parted -s $i -m print | grep ^$i | cut -f7 -d ":" | sed "s/;//")
      DEVICE_SIZE=$(parted -s $i -m print | grep ^$i | cut -f2 -d ":")
      DEVICE_NAME=$(echo $DEVICE_MODEL ${DEVICE_SIZE} | sed 's/ /_/g')
      DEVICE_LIST="$DEVICE_LIST $i $DEVICE_NAME"
    done
}

create_partition_list() {
  # get an overview of all partitions of a specified device
  # usage:    create_partition_list /dev/sda
  # uses:     get_partition
  # provides: PARTITION_NUMBER, PARTITION_SIZE, PARTITION_FORMAT,
  #           PARTITION_LIST, PARTITIONS (get_partition)

    PARTITION_NUMBER=""
    PARTITION_SIZE=""
    PARTITION_FORMAT=""
    PARTITION_LIST=""

    get_partition $1

    for partition in $PARTITIONS; do
      PARTITION_NUMBER=$(parted -s -m $1 print |grep -v ^/dev |grep -v BYT | cut -f1 -d ":")
      PARTITION_SIZE=$(parted -s -m $1 print |grep -v ^/dev |grep -v BYT | cut -f4 -d ":")
      PARTITION_FORMAT=$(parted -s -m $1 print |grep -v ^/dev |grep -v BYT | cut -f5 -d ":")
      PARTITION_LIST=" $PARTITION_LIST \n Partition $1$PARTITION_NUMBER Size: $PARTITION_SIZE Format: $PARTITION_FORMAT"
    done
}

do_install_mbr() {

  # show menu
    MSG_TITLE="\Z4[ (RE)INSTALL MBR ]\Zn"
    MSG_MENU="\n Please select where to install MBR.\n\n Please select a device:"
    MSG_CANCEL="Back"

    create_device_list

    dialog --colors --backtitle "$BACKTITLE" --cancel-label "$MSG_CANCEL" \
      --title "$MSG_TITLE" --menu "$MSG_MENU" 20 50 5 \
      $DEVICE_LIST 2> $TMPDIR/device_for_install

  # now we must do everything
    case $? in
      0)
        INSTALL_DEVICE=$(cat "$TMPDIR/device_for_install")

        # installing mbr
          prompt_gpt
          if [ "$GPT" = "1" ]; then
            cat /usr/share/syslinux/gptmbr.bin > $INSTALL_DEVICE
          else
            cat /usr/share/syslinux/mbr.bin > $INSTALL_DEVICE
          fi

          msg_install_ready "Master Boot Record installed on $INSTALL_DEVICE"
        ;;
      1)
        menu_main
        ;;
      255)
        do_poweroff
        ;;
    esac
}

do_install_quick() {

  # show menu
    MSG_TITLE="\Z4[ QUICK INSTALL MENU ]\Zn"
    MSG_MENU="\nUse the up/down arrows to select the correct device.\n\n Please select a device:"
    MSG_CANCEL="Back"

    create_device_list

    dialog --colors --backtitle "$BACKTITLE" --cancel-label "$MSG_CANCEL" \
      --title "$MSG_TITLE" --menu "$MSG_MENU" 20 50 5 \
      $DEVICE_LIST 2> $TMPDIR/device_for_install

  # now we must do everything
    case $? in
      0)
        INSTALL_DEVICE=$(cat "$TMPDIR/device_for_install")

        prompt_gpt
        prompt_ssh
        prompt_backup_unpack

        EXTLINUX_SSH=""
        if [ "$SSH" = "1" ]; then
          EXTLINUX_SSH="ssh"
        fi

        # remove all partitions
          msg_progress_install "1" "get all partitions $INSTALL_DEVICE"
            get_partition $INSTALL_DEVICE

          msg_progress_install "5" "wiping disk $INSTALL_DEVICE"
            dd if=/dev/zero of=$INSTALL_DEVICE bs=4096 count=1024

        # create 2 new partitions (first $PARTSIZE_SYSTEM, second rest)

          msg_progress_install "7" "creating label on $INSTALL_DEVICE"
            if [ "$GPT" = "1" ]; then
              parted -s $INSTALL_DEVICE mklabel gpt >> $LOGFILE 2>&1
            else
              parted -s $INSTALL_DEVICE mklabel msdos >> $LOGFILE 2>&1
            fi

          msg_progress_install "9" "writing Master Boot Record on $INSTALL_DEVICE"
            if [ "$GPT" = "1" ]; then
              cat /usr/share/syslinux/gptmbr.bin > $INSTALL_DEVICE
            else
              cat /usr/share/syslinux/mbr.bin > $INSTALL_DEVICE
            fi

          msg_progress_install "10" "creating partition on $INSTALL_DEVICE"
            parted -s $INSTALL_DEVICE unit cyl mkpart primary ext2 -- 0 $PARTSIZE_SYSTEM >> $LOGFILE 2>&1

          msg_progress_install "13" "creating partition on $INSTALL_DEVICE"
            parted -s $INSTALL_DEVICE unit cyl mkpart primary ext2 -- $PARTSIZE_SYSTEM -2 >> $LOGFILE 2>&1

          msg_progress_install "16" "setup bootflag on partition 1 of $INSTALL_DEVICE"
            parted -s $INSTALL_DEVICE set 1 boot on >> $LOGFILE 2>&1
            if [ "$GPT" = "1" ]; then
              parted -s $INSTALL_DEVICE set 1 legacy_boot on >> $LOGFILE 2>&1
            fi

          msg_progress_install "20" "tell the kernel we have a new partitiontable on $INSTALL_DEVICE"
            partprobe $INSTALL_DEVICE >> $LOGFILE 2>&1

        # create filesystem
          msg_progress_install "23" "creating filesystem on ${INSTALL_DEVICE}1"
            mke2fs -t ext4 -m 0 ${INSTALL_DEVICE}1 >> $LOGFILE 2>&1

          msg_progress_install "25" "set uuid and disklabel $DISKLABEL_SYSTEM on ${INSTALL_DEVICE}1"
            tune2fs -U random -L $DISKLABEL_SYSTEM  ${INSTALL_DEVICE}1 >> $LOGFILE 2>&1

          msg_progress_install "28" "creating filesystem on ${INSTALL_DEVICE}2"
            mke2fs -t ext4 -m 0 ${INSTALL_DEVICE}2 >> $LOGFILE 2>&1

          msg_progress_install "30" "set uuid and disklabel $DISKLABEL_STORAGE on ${INSTALL_DEVICE}2"
            tune2fs -U random -L $DISKLABEL_STORAGE  ${INSTALL_DEVICE}2 >> $LOGFILE 2>&1

        # mount system partition
          msg_progress_install "35" "creating $TMPDIR/part1"
            mkdir -p $TMPDIR/part1 >> $LOGFILE 2>&1

          msg_progress_install "40" "mounting ${INSTALL_DEVICE}1 to $TMPDIR/part1"
            mount -t ext4 ${INSTALL_DEVICE}1 $TMPDIR/part1 >> $LOGFILE 2>&1

        # installing extlinux
          msg_progress_install "50" "installing extlinux to $TMPDIR/part1"
            extlinux -i $TMPDIR/part1 >> $LOGFILE 2>&1

        # install system files
          msg_progress_install "60" "installing Kernel"
            cp /flash/KERNEL $TMPDIR/part1 >> $LOGFILE 2>&1

          msg_progress_install "65" "installing System"
            cp /flash/SYSTEM $TMPDIR/part1 >> $LOGFILE 2>&1
            sync

        # configuring bootloader
          msg_progress_install "80" "setup bootloader with boot label = $DISKLABEL_SYSTEM and disk label = $DISKLABEL_STORAGE"
            echo "DEFAULT linux" > $TMPDIR/part1/extlinux.conf
            echo "PROMPT 0" >> $TMPDIR/part1/extlinux.conf
            echo " " >> $TMPDIR/part1/extlinux.conf
            echo "LABEL linux" >> $TMPDIR/part1/extlinux.conf
            echo " KERNEL /KERNEL" >> $TMPDIR/part1/extlinux.conf
            echo " APPEND boot=LABEL=$DISKLABEL_SYSTEM disk=LABEL=$DISKLABEL_STORAGE $EXTLINUX_PARAMETERS $EXTLINUX_SSH quiet" >> $TMPDIR/part1/extlinux.conf
            sync

        # umount system partition, remove mountpoint
          msg_progress_install "85" "unmount $TMPDIR/part1"
            umount $TMPDIR/part1 >> $LOGFILE 2>&1

          msg_progress_install "87" "remove $TMPDIR/part1"
            rmdir $TMPDIR/part1 >> $LOGFILE 2>&1

          if [ "$BACKUP_UNPACK" = "1" ]; then
            # mount storage partition
              msg_progress_install "89" "creating $TMPDIR/part2"
                mkdir -p $TMPDIR/part2 >> $LOGFILE 2>&1

              msg_progress_install "90" "mounting ${INSTALL_DEVICE}2 to $TMPDIR/part2"
                mount -t ext4 ${INSTALL_DEVICE}2 $TMPDIR/part2 >> $LOGFILE 2>&1

              msg_progress_install "92" "restoring backup"
                [ -f /flash/backup.tar.bz2 ] && tar -xjf /flash/backup.tar.bz2 -C $TMPDIR/part2 >> $LOGFILE 2>&1
                [ -f /flash/backup.zip ] && unzip -qq /flash/backup.zip -d $TMPDIR/part2 >> $LOGFILE 2>&1
                sync

            # umount system partition, remove mountpoint
              msg_progress_install "97" "unmount $TMPDIR/part2"
                umount $TMPDIR/part2 >> $LOGFILE 2>&1

              msg_progress_install "100" "remove $TMPDIR/part2"
                rmdir $TMPDIR/part2 >> $LOGFILE 2>&1
          fi

        menu_main
        ;;
      1)
        menu_main
        ;;
      255)
        do_poweroff
        ;;
    esac
}

msg_not_implemented() {
  # show a dialog that this function is not yet implemented
    MSG_TITLE="\Z2[ WORK IN PROGRESS ]\Zn"
    MSG_INFOBOX=" This function is not implemented yet."

    dialog --colors --backtitle "$BACKTITLE" --title "$MSG_TITLE" --msgbox "$MSG_INFOBOX" 7 70
}

msg_oem_only() {
  # show a dialog that this function is only implemented on special builds
    MSG_TITLE="\Z2[ FOR OEM ONLY ]\Zn"
    MSG_INFOBOX=" OEM only feature, this function is not implemented in this build. \n if you have questions about this feature \n visit http://www.openelec.tv"

    dialog --colors --backtitle "$BACKTITLE" --title "$MSG_TITLE" --msgbox "$MSG_INFOBOX" 7 70
}

msg_warning_beta() {
  # show a warning dialog if we use beta software
    MSG_TITLE="\Z1[ BETA WARNING ]\Zn"
    MSG_INFOBOX=" This installer is  for beta versions. \n This means this sofware have not been tested yet. \n Please make sure you have a backup of your files."

    dialog --colors --backtitle "$BACKTITLE" --title "$MSG_TITLE" --msgbox "$MSG_INFOBOX" 7 70
}

msg_no_device() {
  # show a warning dialog if we dont find not mounted devices for install and return to main menu
    MSG_TITLE="\Z1[ WARNING ]\Zn"
    MSG_INFOBOX=" No devices were found. \n If you are trying to install on a brand new harddisk you must \n create atleast one partition. \n Otherwise it won't be found. \n If you dont know how, ask in the forum or on IRC."

    dialog --colors --backtitle "$BACKTITLE" --title "$MSG_TITLE" --msgbox "$MSG_INFOBOX" 9 70

    menu_main
}

msg_install_ready() {
  # show a dialog that we have installed
    MSG_TITLE="\Z1[ INFORMATION ]\Zn"

    dialog --colors --backtitle "$BACKTITLE" --title "$MSG_TITLE" --msgbox " $1" 7 70

    menu_main
}

msg_progress_install() {
  # show the progress dialog
    MSG_TITLE="\Z1[ INSTALLING ]\Zn"

    dbglg "$2"
    dialog --colors --backtitle "$BACKTITLE" --title "$MSG_TITLE" --gauge "$2 ..." 6 70 $1 &
}

prompt_gpt() {
  # Prompt for GPT use
  # usage:    prompt_gpt
  # uses:     INSTALL_DEVICE
  # provides: GPT
    MSG_TITLE="\Z1[ Partition Table Type ]\Zn"
    # Get "msdos" or "gpt"
    INSTALL_DEVICE_PARTITION_TYPE=$(parted -s -m $INSTALL_DEVICE print | grep $INSTALL_DEVICE | cut -f6 -d ":")
    # Get size in GB
    INSTALL_DEVICE_SIZE=$(($(cat /sys/block/${INSTALL_DEVICE#/dev/}/size)*512/1000/1000/1000))
    if [ "$INSTALL_DEVICE_PARTITION_TYPE" = "gpt" ]; then
        MSG_DETAIL="GUID Partition Table detected on the destination disk. It is recommended that you keep it."
        DIALOG_OPTIONS=""
    # 2^41 bytes is the DOS limit = 2199023255552 (2.2TB)
    elif [ "$INSTALL_DEVICE_SIZE" -ge 2200 ] 2>/dev/null; then
        MSG_DETAIL="Destination disk is too large to use a DOS partition table. You will need to use a GUID Partition Table."
        DIALOG_OPTIONS=""
    else
        MSG_DETAIL="You should only use a GUID Partition Table if you know what you are doing."
        DIALOG_OPTIONS="--defaultno"
    fi
    if dialog --colors --backtitle "$BACKTITLE" --title "$MSG_TITLE" $DIALOG_OPTIONS --yesno "Use GPT partitions?\n$MSG_DETAIL" 0 0; then
        GPT="1"
    else
        GPT="0"
    fi
}

prompt_ssh() {
  # Prompt for SSH enabling
  # usage:    prompt_ssh
  # uses:
  # provides: SSH
    MSG_TITLE="\Z1[ Enable SSH Server on start ]\Zn"
    MSG_DETAIL="Enable SSH server per default.\nYou should only enable SSH server if you know what you are doing."
    DIALOG_OPTIONS="--defaultno"
    if dialog --colors --backtitle "$BACKTITLE" --title "$MSG_TITLE" $DIALOG_OPTIONS --yesno "$MSG_DETAIL" 0 0; then
        SSH="1"
    else
        SSH="0"
    fi
}

prompt_backup_unpack() {
  # Prompt for unpacking backup files to /storage
  # usage:    prompt_backup_unpack
  # uses:
  # provides: BACKUP_UNPACK
  BACKUP_UNPACK="0"
  if [ -f /flash/backup.tar.bz2 -o -f /flash/backup.zip ]; then
    MSG_TITLE="\Z1[ Restore backup files ]\Zn"
    MSG_DETAIL="Restore backup files to storage partition.\nFile backup.tar.bz2 or/and backup.zip exist on\ninstallation USB stick."
    DIALOG_OPTIONS="--defaultno"
    if dialog --colors --backtitle "$BACKTITLE" --title "$MSG_TITLE" $DIALOG_OPTIONS --yesno "$MSG_DETAIL" 0 0; then
        BACKUP_UNPACK="1"
    fi
  fi
}

menu_main() {
  # show the mainmenu
    MSG_TITLE="\Z4[ MAIN MENU ]\Zn"
    MSG_MENU="\n\ZbQuick Install:\Zn do a default installation on a specific device \
              \Z1\Zb(this will delete ALL data on this device!)\Zn \
              \n\ZbCustom Install:\Zn do a custom installation \
              \n\ZbSetup:\Zn change some settings to run OpenELEC \
              \n\ZbBIOS Update:\Zn backup and update your BIOS (only for OEMs) \
              \n\ZbShow logfile:\Zn show and save the logfile \
              \n \
              \nPlease select:"
    MSG_CANCEL="Reboot"

    dialog --colors --backtitle "$BACKTITLE" --cancel-label "$MSG_CANCEL" \
      --title "$MSG_TITLE" --menu "$MSG_MENU" 20 70 5 \
        1 "Quick Install of OpenELEC" \
        2 "Custom Install of OpenELEC" \
        3 "Setup OpenELEC" \
        4 "BIOS update (only for OEM's)" \
        5 "Show logfile" 2> $TMPDIR/mainmenu

    case $? in
      0)
        ITEM_MAINMENU=$(cat "$TMPDIR/mainmenu")
        case $ITEM_MAINMENU in
          1) do_install_quick; break;;
          2) menu_custom; break;;
          3) menu_setup; break;;
          4) menu_bios; break;;
          5) logfile_show; break;;
        esac
        ;;
      1)
        do_reboot
        ;;
      255)
        do_poweroff
        ;;
    esac
}

menu_setup() {
  # TODO: show the setupmenu
    msg_not_implemented
    menu_main
}

menu_bios() {
  # show the biosmenu
    MSG_TITLE="\Z4[ BIOS MENU ]\Zn"
    MSG_MENU="\n You can use the UP/DOWN arrow keys,\n or the number of the choice as a hotkey,\n to choose an option.\n\n Please choose an option:"
    MSG_CANCEL="Back"

    dialog --colors --backtitle "$BACKTITLE" --cancel-label "$MSG_CANCEL" \
      --title "$MSG_TITLE" --menu "$MSG_MENU" 20 70 5 \
        1 "Backup installed BIOS" \
        2 "Update BIOS" \
        3 "Erase BIOS (not recommended)" 2> $TMPDIR/biosmenu

    case $? in
      0)
        ITEM_BIOSMENU=$(cat "$TMPDIR/biosmenu")
        case $ITEM_BIOSMENU in
          1) bios_backup; break;;
          2) bios_update; break;;
          3) bios_erase; break;;
        esac
        ;;
      1)
        menu_main
        ;;
      255)
        do_poweroff
        ;;
    esac
}

menu_custom() {
  # TODO: show the installmenu
    msg_not_implemented
    menu_main
}

bios_backup() {
  # create a backup of the installed bios
    if [ "$BIOS_UPDATE" = "yes" -a -f "$BIOS_FILE" ]; then

      clear
      echo "##### backup old BIOS to $BIOS_BACKUP #####"
        usleep 1000000
        mount -o remount,rw /flash
        flashrom --read /flash/$BIOS_BACKUP
        mount -o remount,ro /flash

      echo "##### Please control the output and press any key to continue #####"
      read -sn1

    else
      msg_oem_only
    fi
    menu_bios
}

bios_update() {
  # update the bios
    if [ "$BIOS_UPDATE" = "yes" -a -f "$BIOS_FILE" ]; then

      clear
      echo "##### erasing BIOS #####"
        usleep 1000000
        flashrom --erase

      echo "##### writing new BIOS from $BIOS_FILE #####"
        usleep 1000000
        flashrom --write "$BIOS_FILE"

      echo "##### Please control the output and press any key to continue #####"
      read -sn1

    else
      msg_oem_only
    fi
    menu_bios
}

bios_erase() {
  # erase the bios
    if [ "$BIOS_UPDATE" = "yes" -a -f "$BIOS_FILE" ]; then

      clear
      echo "##### erasing BIOS #####"
        usleep 1000000
        flashrom --erase

      echo "##### Please control the output and press any key to continue #####"
      read -sn1

    else
      msg_oem_only
    fi
    menu_bios
}

logfile_show() {
  # TODO: show the logfile
    dialog --textbox "$LOGFILE" 20 70
    clear
    menu_main
}

do_reboot() {
  # reboot on request
    clear
    sync
    reboot
}

do_poweroff() {
  # powerdown on request
    clear
    sync
    poweroff
}

# setup needed variables
  BETA="yes"
  INSTALLER_VERSION="0.2.6"
  OS_VERSION=$(lsb_release)
  BACKTITLE="OpenELEC Installer $INSTALLER_VERSION - $OS_VERSION"

  TMPDIR="/tmp/installer"
  LOGFILE="$TMPDIR/install.log"

# prepare temporary directory
  rm -rf $TMPDIR
  mkdir -p $TMPDIR

# main
  [ "$BETA" = "yes" ] && msg_warning_beta

  while true; do
    menu_main
  done

# exit cleanly
  exit 0
