#!/bin/bash
# This script installs the Debian image to the microSD card, and should
# be run on your laptop/desktop with the microSD card plugged in.

# License
# =======
#
# Copyright (C) 2014-2015 Bob Mottram <bob@robotics.uk.to>
#
# 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 3 of the License, 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 this program. If not, see <http://www.gnu.org/licenses/>.

# The number of arguments
NO_OF_ARGS=$#

# Version number of this script
VERSION="1.00"

# typically /dev/sdb or /dev/sdc, depending upon how
# many drives there are on your system
MICROSD_DRIVE=

# IP address of the router (gateway)
ROUTER_IP_ADDRESS="192.168.1.254"

# The fixed IP address of the Beaglebone Black on your local network
BBB_FIXED_IP_ADDRESS="192.168.1.55"

MICROSD_MOUNT_POINT="/media/$USER"

DEBIAN_FILE_NAME="debian-jessie-console-armhf-2014-08-13"

# Downloads for the Debian installer
DOWNLOAD_LINK1="https://rcn-ee.net/deb/rootfs/jessie/$DEBIAN_FILE_NAME.tar.xz"
DOWNLOAD_LINK2="http://ynezz.ibawizard.net/beagleboard/jessie/$DEBIAN_FILE_NAME.tar.xz"

ROOTFS='rootfs'
BOOT='BOOT'

function show_help {
    echo ''
    echo 'freedombone-prep -d [microSD device] --ip [BBB LAN IP address] --iprouter [Router LAN IP address] --mount [mount directory]'
    echo ''
    echo 'See the manpage for more details'
    echo ''
}

# if no arguments are given
if [[ $NO_OF_ARGS == 0 ]]; then
  show_help
  exit 0
fi

if [ ! -d $MICROSD_MOUNT_POINT ]; then
    if [ -d /media ]; then
        # different directories for Debian
        if [ -d /media/usb1 ]; then
            MICROSD_MOUNT_POINT=/media
            ROOTFS=usb1
        fi
        if [ -d /media/usb0 ]; then
            MICROSD_MOUNT_POINT=/media
            BOOT=usb0
        fi
    fi
    if [ ! -d $MICROSD_MOUNT_POINT ]; then
        echo "The mount directory $MICROSD_MOUNT_POINT does not exist."
        echo 'Use the --mount option to specify where the microSD gets mounted to.'
        exit 67563
    fi
fi

while [[ $# > 1 ]]
do
key="$1"

case $key in
    -h|--help)
    show_help
    ;;
    # Drive path for the microSD
    -d|--drive)
    shift
    MICROSD_DRIVE="$1"
    ;;
    # BBB static IP address on the LAN
    --ip)
    shift
    BBB_FIXED_IP_ADDRESS="$1"
    ;;
    # Router IP address on the LAN
    --iprouter)
    shift
    ROUTER_IP_ADDRESS="$1"
    ;;
    # mount point
    --mount)
    shift
    MICROSD_MOUNT_POINT="$1"
    ;;
    *)
    # unknown option
    ;;
esac
shift
done


if [ ! $MICROSD_DRIVE ]; then
    echo 'You need to specify a drive for the connected microSD.'
    echo 'This can most easily be found by removing the microSD, then'
    echo 'running:'
    echo ''
    echo '  ls /dev/sd*'
    echo ''
    echo 'Then plugging the microSD back in and entering the same command again'
    exit 1
fi

if [ ! -b ${MICROSD_DRIVE}1 ]; then
    echo "The microSD drive could not be found at ${MICROSD_DRIVE}1"
    exit 2
fi

if [ ! -d ~/freedombone ]; then
    mkdir ~/freedombone
fi
cd ~/freedombone
if [ ! -f ~/freedombone/$DEBIAN_FILE_NAME.tar.xz ]; then
    wget $DOWNLOAD_LINK1
fi
if [ ! -f ~/freedombone/$DEBIAN_FILE_NAME.tar.xz ]; then
    # try another site
    wget $DOWNLOAD_LINK2
    if [ ! -f ~/freedombone/$DEBIAN_FILE_NAME.tar.xz ]; then
        echo 'The Debian installer could not be downloaded'
        exit 3
    fi
fi

echo 'Extracting files...'
tar xJf $DEBIAN_FILE_NAME.tar.xz
if [ ! -d ~/freedombone/$DEBIAN_FILE_NAME ]; then
    echo "Couldn't extract files"
    exit 4
fi
cd $DEBIAN_FILE_NAME

SUDO=
if [ -f /usr/bin/sudo ]; then
    SUDO='sudo'
fi
$SUDO ./setup_sdcard.sh --mmc $MICROSD_DRIVE --dtb beaglebone

echo ''
echo ''
read -p "Eject the microSD card, re-insert it and wait a minute for it to mount, then press any key to continue... " -n1 -s

if [ ! -b ${MICROSD_DRIVE}1 ]; then
    echo ''
    echo "The microSD drive could not be found at ${MICROSD_DRIVE}1"
    read -p "Wait for the drive to mount then press any key... " -n1 -s
    if [ ! -b ${MICROSD_DRIVE}1 ]; then
        echo "microSD drive not found at ${MICROSD_DRIVE}1"
        exit 5
    fi
fi

if [ ! -d $MICROSD_MOUNT_POINT/$BOOT ]; then
    echo "The boot partition $MICROSD_MOUNT_POINT/$BOOT was not found."
    ls $MICROSD_MOUNT_POINT
    exit 67857
fi

if [ ! -d $MICROSD_MOUNT_POINT/$ROOTFS ]; then
    echo "The rootfs partition $MICROSD_MOUNT_POINT/$ROOTFS was not found."
    ls $MICROSD_MOUNT_POINT
    exit 65688
fi

if [ ! -d $MICROSD_MOUNT_POINT/$ROOTFS/home ]; then
    echo "The rootfs partition was not written correctly."
    ls $MICROSD_MOUNT_POINT/$ROOTFS
    exit 65688
fi

$SUDO cp $MICROSD_MOUNT_POINT/$BOOT/bbb-uEnv.txt $MICROSD_MOUNT_POINT/$BOOT/uEnv.txt

$SUDO sed -i 's/iface eth0 inet dhcp/iface eth0 inet static/g' $MICROSD_MOUNT_POINT/$ROOTFS/etc/network/interfaces
$SUDO sed -i '/iface eth0 inet static/a\    dns-nameservers 213.73.91.35 85.214.20.141' $MICROSD_MOUNT_POINT/$ROOTFS/etc/network/interfaces
$SUDO sed -i "/iface eth0 inet static/a\    gateway $ROUTER_IP_ADDRESS" $MICROSD_MOUNT_POINT/$ROOTFS/etc/network/interfaces
$SUDO sed -i '/iface eth0 inet static/a\    netmask 255.255.255.0' $MICROSD_MOUNT_POINT/$ROOTFS/etc/network/interfaces
$SUDO sed -i "/iface eth0 inet static/a\    address $BBB_FIXED_IP_ADDRESS" $MICROSD_MOUNT_POINT/$ROOTFS/etc/network/interfaces
$SUDO sed -i '/iface usb0 inet static/,/    gateway 192.168.7.1/ s/^/#/' $MICROSD_MOUNT_POINT/$ROOTFS/etc/network/interfaces

$SUDO sed -i 's/nameserver.*/nameserver 213.73.91.35/g' $MICROSD_MOUNT_POINT/$ROOTFS/etc/resolv.conf
$SUDO sed -i '/nameserver 213.73.91.35/a\nameserver 85.214.20.141' $MICROSD_MOUNT_POINT/$ROOTFS/etc/resolv.conf

# change the motd to show further install instructions
echo 'Become the root user by typing:' > $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '    su' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo 'Using the password "root". Change the root user password by typing:' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '    passwd' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo 'Then create a user for the system with:' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '    adduser [username]' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo 'Enter the command "exit" a couple of times to get back to your main system' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo 'then log back in as the user you just created with:' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '    ssh [username]@$BBB_FIXED_IP_ADDRESS' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo 'and use the "su" command to become the root user again. You can then load' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo 'the freedombone main installation script with:' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '    apt-get update' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '    apt-get -y install git dialog build-essential' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '    git clone https://github.com/bashrc/freedombone.git' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '    cd freedombone' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '    make install' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo 'Finally you can use the freedombone command to install a server configuration:' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
echo '    freedombone menuconfig' >> $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd

clear
echo '*** Initial microSD card setup is complete ***'
echo ''
echo 'The microSD card can now be removed and inserted into the Beaglebone Black.'
echo 'Once the Beaglebone has booted then you can log in with:'
echo ''
echo "    ssh debian@$BBB_FIXED_IP_ADDRESS"
echo ''
echo 'The password is "temppwd".'
cat $MICROSD_MOUNT_POINT/$ROOTFS/etc/motd
exit 0