#!/bin/bash

# set -o xtrace       ## To debug scripts
# set -o errexit      ## To exit on error
# set -o errunset     ## To exit if a variable is referenced but not set


. CONFIG.sh

function main() {
    clear;
    if [[ $(sanity_check) -eq 1 ]]; then echo "\nExiting..."; return; fi

    start_menu_mesage;
    read -p "--> : " ANSR
    while [[ $ANSR != "0" ]] && [[ $ANSR != "1" ]] && \
          [[ $ANSR != "2" ]] && [[ $ANSR != "3" ]] && \
          [[ $ANSR != "4" ]] && [[ $ANSR != "5" ]] && \
                                [[ $ANSR != "6" ]]; do
        read -p "--> : " ANSR
    done
    case $ANSR in
        "1" ) do_all_run; break;;
        # First setup the debootstrap env...
        "2" ) ./step_1_debootstrap.sh; break;;
        # Then setup and run chroot...
        "3" ) ./step_2_chroot.sh; break;;
        # Create the boot structure data...
        "4" ) ./step_3_create_boot_structure.sh; break;;
        # Create the CD...
        "5" ) ./step_4_create_CD.sh; break;;
        # Purge everythin and start fresh...
        "6" ) ./cleanup.sh; break;;
        "0" ) exit; break;;
        * ) echo "Don't know how you got here but that's a bad sign..."; break;;
    esac
}

function do_all_run() {
    # Starting with a clean slate
    ./cleanup.sh;
    # First setup the debootstrap env...
    ./step_1_debootstrap.sh;
    # Then setup and run chroot...
    ./step_2_chroot.sh;
    # Create the boot structure data...
    ./step_3_create_boot_structure.sh;
    # Create the CD...
    ./step_4_create_CD.sh;
}

function sanity_check() {
    # Check for debootstrap command and then install from downloaded deb if not present.
    # We could install from current apt but I want the user and myself to be fully aware
    # of what they are chosing. IE, we could just run the install command itself...
    debootstrap_comm=$(which debootstrap)
    if [[ "${debootstrap_comm}" == "" ]]; then
        echo "No deboostrap command found so will try installing from local directory...\n"
        if [[ ! -f "debootstrap.deb" ]]; then
            echo "No debootstrap deb file found in current directory...";
            echo "Please download one from:\n\thttps://packages.ubuntu.com/search?keywords=debootstrap&searchon=names&suite=all&section=all";
            echo "After download rename it to:  debootstrap.deb\n";
            echo "OR\n"
            echo "Run:  sudo apt install debootstrap"
            return 1;
        else
            sudo dpkg -i ./deboostrap.deb
        fi
    fi

    xyphr=$(which Xephyr)        ## Chroot view window || alt desktop window
    sysLnx=$(which syslinux)     ## Casper stuff
    sysTools=$(which ppmtolss16) ## Part of syslinux-utils
    squash=$(which unsquashfs)   ## Squashfs-tools
    genIso=$(which genisoimage)  ## Iso maker
    netpbm=$(which bmptoppm)     ## Boot splash maker part
    if [[ $xyphr == "" ]] || [[ $sysLnx == "" ]] || \
       [[ $squash == "" ]] || [[ $genIso == "" ]]; then
        echo "# ----  Missing Some Packages  ---- #\n" \
                "Xephyr :" "${xyphr}\n" \
                "Syslinux :" "${sysLnx}\n" \
                "Squashfs-tools :" "${squash}\n" \
                "Genisoimage :" "${genIso}\n" \
                "Netpbm :" "${netpbm}\n" \
                "Going to run :\n" \
                "\tapt-get install xserver-xephyr syslinux squashfs-tools genisoimage netpbm syslinux-utils -y"
        sleep 2
        sudo apt-get install xserver-xephyr syslinux squashfs-tools genisoimage netpbm syslinux-utils -y
    fi

    if [[ "${ARCH}" == "" ]] || [[ "${RELEASE}" == "" ]]; then
        echo "Please check CONFIG.sh and set the ARCH and RELEASE data...";
        return 1;
    fi

    if [[ "${OS_NAME}" == "" ]] || [[ "${LIVE_USER}" == "" ]]; then
        echo "Please check CONFIG.sh and set the OS_NAME and LIVE_USER data...";
        return 1;
    fi

    return 0;
}


main $@;