#!/usr/bin/env bash

# init
# Craig Jennings <c@cjennings.net>
# The initial archsetup script. Downloads and calls other scripts.


# welcome user, present options, and setup ssh
# server from the live cd. Functions to installing base arch and setup
# the workstation occur in other scripts, called by this script.

# Run as root after booting the live cd and issuing the following command:
#   curl -s https://cjennings.net/archsetup/init | bash -i 

welcome () {
    clear
    echo ""
    echo "ArchSetup"
    echo "---------"
    echo ""
    echo ""
}

check_root_or_exit () {
    if [ "$EUID" -ne 0 ]
    then echo "Please run as root"
         exit
    fi
}

setup_ssh () {
    # retrieve keys from repo and decrypt
    echo "retrieving keys for decryption..."
    {
        mkdir -p "$HOME/.ssh/"
        # retrieve keys from dotfiles repo and decrypt the keys
        (git clone --depth 1 https://git.cjennings.net/dotfiles.git \
             "$HOME/dotfiles/") >> /dev/null
        cp "$HOME/dotfiles/system/.ssh/*" "$HOME/.ssh/"
        rm -rf "$HOME/dotfiles/"
    } > /dev/null
    cd "$HOME/.ssh/"
    gpg --decrypt "$HOME/ssh.tar.gz.gpg" | tar -zxf -

    # set proper permissions on .ssh directory
    cd "$HOME" && chmod 600 $HOME/.ssh/* && chmod 700 $HOME/.ssh \
        && chmod 644 $HOME/.ssh/*.pub

    # setup ssh
    echo "setting up ssh daemon..."
    systemctl start sshd
    echo "root:welcome" | chpasswd
    hostname=$(</etc/hostname)
    ipaddress=$(ip addr show wlan0 | grep 'inet ' | awk '{print $2}' | cut -d/ -f1)

    # notify user
    echo ""; echo ""
    echo "you may ssh root@$hostname or ssh root@$ipaddress using the password 'welcome'"
}

############ SCRIPT

check_root_or_exit

# present menu
echo "Choose:"
echo " (1) create an arch linux base install"
echo " (2) turn a base install into a full workstation"
echo " (3) generate a workstation in an arch distrobox container"
echo " (4) allow ssh to this live iso"
echo " (Enter 1, 2, 3, or 4) :"

IFS= read -r setup
# read setup

echo "You chose '$SETUP'."

case $setup in
    1)
        echo "this will call the archinstall script"
        ;;
    2)
        echo "this will call the archsetup script"
        ;;
    3)
        echo "this will call the archsetup-distrobox script"
        ;;
    4)
        setup_ssh
        ;;
    *)
        echo "Error: Unknown setup option. Please enter 1,2,3,or 4."
        exit 1
        ;;
esac