#!/bin/bash
#
# Copyright (c) 2015 Igor Pecovnik, igor.pecovnik@gma**.com
#
# This file is licensed under the terms of the GNU General Public
# License version 2. This program is licensed "as is" without any
# warranty of any kind, whether express or implied.
#
# This file is a part of tool chain https://github.com/igorpecovnik/lib
#
#
# Main program
#
#

# Include here to make "display_alert" and "prepare_host" available
source $SRC/lib/general.sh					# General functions

# compress and remove old logs
(cd $DEST/debug && tar -czf logs-$(date +"%d_%m_%Y-%H_%M_%S").tgz *.log) > /dev/null 2>&1
rm -f $DEST/debug/*.log > /dev/null 2>&1

# Script parameters handling
for i in "$@"; do
	if [[ "$i" == *"="* ]]; then
		parameter=${i%%=*}
		value=${i##*=}
		display_alert "Command line: setting $parameter to" "${value:-(empty)}" "info"
		eval $parameter=$value
	fi
done

if [ "$PROGRESS_DISPLAY" = "none" ]; then
	OUTPUT_VERYSILENT=yes;
elif [ "$PROGRESS_DISPLAY" != "plain" ]; then
	OUTPUT_DIALOG=yes;
fi
if [ "$PROGRESS_LOG_TO_FILE" != "yes" ]; then unset PROGRESS_LOG_TO_FILE; fi

# compile.sh version checking
ver1=$(grep '^# VERSION' "$SRC/compile.sh" | cut -d'=' -f2)
ver2=$(grep '^# VERSION' "$SRC/lib/compile.sh" | cut -d'=' -f2)
if [ -z "$ver1" ] || [ "$ver1" -lt "$ver2" ]; then
	display_alert "File $0 is outdated. Please overwrite is with updated version from" "$SRC/lib" "wrn"
	read -p "Press <Ctrl-C> to abort compilation, <Enter> to ignore and continue"
fi

# We'll use this title on all menus
backtitle="Armbian building script, http://www.armbian.com | Author: Igor Pecovnik"

# if language not set, set to english
[ "$LANGUAGE" == "" ] && export LANGUAGE="en_US:en"

# default console if not set
[ "$CONSOLE_CHAR" == "" ] && export CONSOLE_CHAR="UTF-8"

# Check and fix dependencies, directory structure and settings
prepare_host

# Choose destination - creating board list from file configuration.sh
if [ "$BOARD" == "" ]; then
	IFS=";"
	MYARRAY=($(cat $SRC/lib/configuration.sh | awk '/\)#enabled/ || /#des/' | sed -e 's/\t\t//' | sed 's/)#enabled//g' \
	| sed 's/#description //g' | sed -e 's/\t//' | sed ':a;N;$!ba;s/\n/;/g'))
	MYPARAMS=( --title "Choose a board" --backtitle $backtitle --menu "\n Supported:" 34 67 24 )
	i=0; j=1
	while [[ $i -lt ${#MYARRAY[@]} ]];	do
		MYPARAMS+=( "${MYARRAY[$i]}" "         ${MYARRAY[$j]}" )
		i=$[$i+2]; j=$[$j+2]
	done
	whiptail "${MYPARAMS[@]}" 2>results  
	BOARD=$(<results)
	rm results
	unset MYARRAY
fi
	
if [ "$BOARD" == "" ]; then echo "ERROR: You have to choose one board"; exit; fi

# This section is left out if we only compile kernel
if [ "$KERNEL_ONLY" != "yes" ]; then
	
	# Choose for which distribution you want to compile
	if [ "$RELEASE" == "" ]; then
		IFS=";"
		declare -a MYARRAY=('wheezy' 'Debian 7 Wheezy | oldstable' 'jessie' 'Debian 8 Jessie | stable' \
		'trusty' 'Ubuntu Trusty Tahr 14.04.x LTS');
		MYPARAMS=( --title "Choose a distribution" --backtitle $backtitle --menu "\n Root file system:" 10 60 3 )
		i=0; j=1	
		while [[ $i -lt ${#MYARRAY[@]} ]]; do
			MYPARAMS+=( "${MYARRAY[$i]}" "         ${MYARRAY[$j]}" )
			i=$[$i+2]; j=$[$j+2]
		done
		whiptail "${MYPARAMS[@]}" 2>results  
		RELEASE=$(<results)
		rm results
		unset MYARRAY
	fi

	if [ "$RELEASE" == "" ]; then echo "ERROR: You have to choose one distribution"; exit; fi

	# Choose to build a desktop
	if [ "$BUILD_DESKTOP" == "" ]; then
		IFS=";"
		declare -a MYARRAY=('No' 'Command line interface' 'Yes' 'XFCE graphical interface');
		MYPARAMS=( --title "Install desktop?" --backtitle $backtitle --menu "" 10 60 3 )
		i=0; j=1
		while [[ $i -lt ${#MYARRAY[@]} ]]; do
			MYPARAMS+=( "${MYARRAY[$i]}" "         ${MYARRAY[$j]}" )
			i=$[$i+2]; j=$[$j+2]
		done
		whiptail "${MYPARAMS[@]}" 2>results  
		BUILD_DESKTOP=$(<results)
		BUILD_DESKTOP=${BUILD_DESKTOP,,}
		rm results
		unset MYARRAY
	fi

	if [ "$BUILD_DESKTOP" == "" ]; then echo "ERROR: You need to choose"; exit; fi

fi

# Choose for which branch you want to compile
if [ "$BRANCH" == "" ]; then	
	# get info crom configuration which kernel can be build for certain board
	line_number=$(grep -n "$BOARD)" $SRC/lib/configuration.sh | grep -Eo '^[^:]+' | head -1)
	display_para=$(tail -n +$line_number $SRC/lib/configuration.sh | grep -in "#build" | head -1 | awk '{print $NF}')

	if [[ "$display_para" == *wip ]]; then 
		display_para=${display_para//[!0-9]/}
		whiptail --title "Warning Warning Warning" --msgbox "This is a work in progress. \
		Building might not succeed. \n\nYou must hit OK to continue." 9 63
	fi

	IFS=";"
	
	# define all possible combinations
	if [[ $display_para == "1" ]]; then 
		declare -a MYARRAY=('default' '3.4.x - 3.14.x legacy'); 
	fi
	if [[ $display_para == "2" ]]; then 
		declare -a MYARRAY=('next' 'Latest stable @kernel.org'); 
		fi
	if [[ $display_para == "3" ]]; then 
		declare -a MYARRAY=('default' '3.4.x - 3.14.x legacy' 'next' 'Latest stable @kernel.org'); 
		fi
	if [[ $display_para == "4" ]]; then 
		declare -a MYARRAY=('dev' 'Latest dev @kernel.org'); 
		fi
	if [[ $display_para == "5" ]]; then 
		declare -a MYARRAY=('next' 'Latest stable @kernel.org' 'dev' 'Latest dev @kernel.org'); 
		fi
	if [[ $display_para == "6" || $display_para == "0" ]]; then 
		declare -a MYARRAY=('default' '3.4.x - 3.14.x legacy' 'next' 'Latest stable @kernel.org' 'dev' 'Latest dev @kernel.org'); 
		fi
	
	MYPARAMS=( --title "Choose a branch" --backtitle "$backtitle" --menu "\n Kernel:" 10 60 3 )
	i=0; j=1
	while [[ $i -lt ${#MYARRAY[@]} ]]; do
		MYPARAMS+=( "${MYARRAY[$i]}" "         ${MYARRAY[$j]}" )
		i=$[$i+2]; j=$[$j+2]
	done
	whiptail "${MYPARAMS[@]}" 2>results  
	BRANCH=$(<results)
	rm results
	unset MYARRAY
fi

if [ "$BRANCH" == "" ]; then echo "ERROR: You have to choose one branch"; exit; fi

# don't compile external modules on mainline
if [ "$BRANCH" != "default" ]; then EXTERNAL="no"; fi

# back to normal
unset IFS

# naming to distro 
if [[ "$RELEASE" == "precise" || "$RELEASE" == "trusty" ]]; then DISTRIBUTION="Ubuntu"; else DISTRIBUTION="Debian"; fi

# set hostname to the board
HOST="$BOARD"

# Load libraries
source $SRC/lib/configuration.sh			# Board configuration
source $SRC/lib/debootstrap.sh 				# System specific install
source $SRC/lib/distributions.sh 			# System specific install
source $SRC/lib/patching.sh 				# Source patching
source $SRC/lib/boards.sh 				# Board specific install
source $SRC/lib/desktop.sh 				# Desktop specific install
source $SRC/lib/common.sh 				# Functions
source $SRC/lib/makeboarddeb.sh 			# Create board support package

# The name of the job
VERSION="Armbian $REVISION ${BOARD^} $DISTRIBUTION $RELEASE $BRANCH"

echo `date +"%d.%m.%Y %H:%M:%S"` $VERSION >> $DEST/debug/install.log

# needed if process failed in the middle
umount_image

# let's start with fresh screen
clear
display_alert "Starting Armbian build script" "@host" "info"

# optimize build time with 100% CPU usage
CPUS=$(grep -c 'processor' /proc/cpuinfo)
if [ "$USEALLCORES" = "yes" ]; then
	CTHREADS="-j$(($CPUS + $CPUS/2))";
else
	CTHREADS="-j${CPUS}";
fi

# Use C compiler cache
if [ "$USE_CCACHE" = "yes" ]; then 
	CCACHE="ccache"; 
else 
	CCACHE=""; 
fi

# display what we do	
if [ "$KERNEL_ONLY" == "yes" ]; then
	display_alert "Compiling kernel" "$BOARD" "info"
else
	display_alert "Building" "$VERSION" "info"
fi

# sync clock
if [ "$SYNC_CLOCK" != "no" ]; then
	display_alert "Syncing clock" "host" "info"
	ntpdate -s time.ijs.si
fi
start=`date +%s`

# fetch_from_github [repository, sub directory]

if [ "$FORCE_CHECKOUT" = "yes" ]; then FORCE="-f"; else FORCE=""; fi

# Some old branches are tagged
#if [ "$BRANCH" == "default" ]; then	KERNELTAG="$LINUXBRANCH"; fi

[[ "$CLEAN_LEVEL" == *sources* ]] && cleaning "sources"

display_alert "source downloading" "@host" "info"
fetch_from_github "$BOOTLOADER" "$BOOTSOURCE" "$BOOTBRANCH" "yes"
BOOTSOURCEDIR=$BOOTSOURCE/$GITHUBSUBDIR
fetch_from_github "$LINUXKERNEL" "$LINUXSOURCE" "$KERNELBRANCH" "yes"	
LINUXSOURCEDIR=$LINUXSOURCE/$GITHUBSUBDIR

if [[ -n "$MISC1" ]]; then fetch_from_github "$MISC1" "$MISC1_DIR"; fi
if [[ -n "$MISC2" ]]; then fetch_from_github "$MISC2" "$MISC2_DIR"; fi
if [[ -n "$MISC3" ]]; then fetch_from_github "$MISC3" "$MISC3_DIR"; fi
if [[ -n "$MISC4" ]]; then fetch_from_github "$MISC4" "$MISC4_DIR"; fi
if [[ -n "$MISC5" ]]; then fetch_from_github "$MISC5" "$MISC5_DIR"; fi

# compile sunxi tools
if [[ $LINUXFAMILY == sun*i ]]; then 
	compile_sunxi_tools
	[[ $BRANCH != "default" ]] && LINUXFAMILY="sunxi"
fi

# define some packages
branch="${BRANCH//default/}"
[[ -n "$branch" ]] && branch="-"$branch

CHOOSEN_UBOOT=linux-u-boot"$branch"-"$BOARD"_"$REVISION"_armhf.deb
CHOOSEN_KERNEL=linux-image"$branch"-"$CONFIG_LOCALVERSION$LINUXFAMILY"_"$REVISION"_armhf.deb
CHOOSEN_ROOTFS=linux-"$RELEASE"-root"$branch"-"$BOARD"_"$REVISION"_armhf
HEADERS_CACHE="${CHOOSEN_KERNEL/image/cache}"

for option in $(tr ',' ' ' <<< "$CLEAN_LEVEL"); do
	[ "$option" != "sources" ] && cleaning "$option"
done

[ ! -f "$DEST/debs/$CHOOSEN_UBOOT" ] && needs_uboot=yes
[ ! -f "$DEST/debs/$CHOOSEN_KERNEL" ] && needs_kernel=yes

# patching sources if we need to compile u-boot or kernel
[[ $needs_uboot == yes || $needs_kernel == yes ]] && patching_sources

# Compile source if packed not exists
[ "$needs_uboot" = "yes" ] && compile_uboot
[ "$needs_kernel" = "yes" ] && compile_kernel

[[ -n "$RELEASE" ]] && create_board_package

if [ "$KERNEL_ONLY" == "yes" ]; then
	display_alert "Kernel building done" "@host" "info"
	display_alert "Target directory" "$DEST/debs/" "info"
	display_alert "File name" "$CHOOSEN_KERNEL" "info"
else
	
	# create or use prepared root file-system
	custom_debootstrap

	# add kernel to the image
	install_kernel

	# install board specific applications
	install_distribution_specific
	install_board_specific

	# install desktop
	if [ "$BUILD_DESKTOP" = "yes" ]; then
		install_desktop
	fi

	# install external applications
	if [ "$EXTERNAL" = "yes" ]; then
		install_external_applications
	fi

	# closing image
	closing_image
fi

end=`date +%s`
runtime=$(((end-start)/60))	
display_alert "Runtime" "$runtime min" "info"