#!/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"

# DNS
NAMESERVER1='213.73.91.35'
NAMESERVER2='85.214.20.141'

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 /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

echo "MICROSD_MOUNT_POINT=$MICROSD_MOUNT_POINT"
echo "BOOT=$BOOT"

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"
    ;;
    # nameserver 1
    --ns1)
    shift
    NAMESERVER1="$1"
    ;;
    # nameserver 2
    --ns2)
    shift
    NAMESERVER2="$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 apt-get install u-boot-tools dosfstools git-core kpartx wget parted
$SUDO ./setup_sdcard.sh --mmc $MICROSD_DRIVE --dtb beaglebone

echo ''
echo ''
$SUDO umount $MICROSD_MOUNT_POINT/$ROOTFS
$SUDO umount $MICROSD_MOUNT_POINT/$BOOT
sync
sleep 5
$SUDO mkdir -p $MICROSD_MOUNT_POINT/$BOOT
$SUDO mkdir -p $MICROSD_MOUNT_POINT/$ROOTFS
$SUDO mount ${MICROSD_DRIVE}1 $MICROSD_MOUNT_POINT/$BOOT
$SUDO mount ${MICROSD_DRIVE}2 $MICROSD_MOUNT_POINT/$ROOTFS
sync

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 ''
    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 ''
    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 ''
    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 $NAMESERVER1 $NAMESERVER2" $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 $NAMESERVER1/g" $MICROSD_MOUNT_POINT/$ROOTFS/etc/resolv.conf
$SUDO sed -i "/nameserver $NAMESERVER1/a\nameserver $NAMESERVER2" $MICROSD_MOUNT_POINT/$ROOTFS/etc/resolv.conf

# copy the commands to the card
$SUDO cp -f /usr/local/bin/freedombone* $MICROSD_MOUNT_POINT/$ROOTFS/usr/local/bin/
if [ ! -f $MICROSD_MOUNT_POINT/$ROOTFS/usr/local/bin/freedombone ]; then
    echo 'There was a problem with writing freedombone commands to the SD card'
    exit 8736
fi

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

$SUDO cp -f /tmp/freedombone_motd $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 /tmp/freedombone_motd
rm /tmp/freedombone_motd
$SUDO umount $MICROSD_MOUNT_POINT/$ROOTFS
$SUDO umount $MICROSD_MOUNT_POINT/$BOOT
sync
if [ -d $MICROSD_MOUNT_POINT/$ROOTFS ]; then
    $SUDO rm -rf $MICROSD_MOUNT_POINT/$ROOTFS
fi
if [ -d $MICROSD_MOUNT_POINT/$BOOT ]; then
    $SUDO rm -rf $MICROSD_MOUNT_POINT/$BOOT
fi
exit 0
