#!/bin/bash

echo '              _____   _       ___             '
echo '             |__ __| | |___  | _ |            '
echo '               | |   |  _  | | __|            '
echo '               |_|   |_| |_| |___|            '
echo '          ____                                '
echo '         |  __|   ____  ___  ___  ____        '
echo '         |  | __ | ___|| _ || _ || __ |       '
echo '         |  |_| || |   | __|| __|| || |       '
echo '         |______||_|   |___||___||_||_|       '
echo '  ________                                    '
echo ' |        | ______  ____  _      _  ____  ___ '
echo ' |  |  |  ||  _   ||  __|| |___ | || __ || __|'
echo ' |  |  |  || |_|  || |__ |  _  || || || || __|'
echo ' |__|__|__||____|_||____||_| |_||_||_||_||___|'
echo
echo '        Compile Script by YoshiShaPow         '

echo "What device is this? "
read device
echo "You're building for $input_variable"

export ARCH=arm
export SUBARCH=arm

build="/home/luca/android/tgm_builds"
kernel="The-Green-Machine-Kernel"
rom="lollipop"
version="v2.0.0"
config="tgm_"$device"_defconfig"
kerneltype="zImage"
if uname -a | grep Darwin; then
    jobcount=$(sysctl hw.ncpu | awk '{print $2}')
    t1=`gdate`
else
    jobcount="-j$(grep -c ^processor /proc/cpuinfo)"
    t1=`date`
fi
zipname=""$kernel"-"$rom"-"$device"-"$version""

#toolchains
sabernaro_4=/home/yoshishapow/kernels/toolchains/SaberNaro/bin/arm-eabi-
linaromod_4=/home/yoshishapow/kernels/toolchains/LinaroMod/bin/arm-eabi-
linaromod_5=/home/yoshishapow/kernels/toolchains/LinaroMod_5/bin/arm-eabi-
sabermod_5=/home/yoshishapow/kernels/toolchains/SaberMod_5/bin/arm-eabi-
uber_5=/home/yoshishapow/kernels/toolchains/UBER_5/bin/arm-eabi-

echo
echo "Which toolchain would you like to use?"

while read -p "sabernaro_4, linaromod_4, linaromod_5, sabermod_5, or uber_5? " cchoice
do
case "$cchoice" in
    sabernaro_4 )
	toolchain="sabernaro_4.9"
	export CROSS_COMPILE=$sabernaro_4
	break
	;;
    linaromod_4 )
	toolchain="linaromod_4.9"
	export CROSS_COMPILE=$linaromod_4
	break
	;;
    linaromod_5 )
	toolchain="linaromod_5.0"
	export CROSS_COMPILE=$linaromod_5
	break
	;;
    sabermod_5 )
	toolchain="sabermod_5.0"
	export CROSS_COMPILE=$sabermod_5
	break
	;;
    uber )
	toolchain="uber_5.0"
	export CROSS_COMPILE=$uber_5
	break
	;;
esac
done

while read -p "Do you want to make clean (y/n)? " cchoice
do
echo
case "$cchoice" in
    y|Y )
	if [ -f arch/arm/boot/"$kerneltype" ]; then
		echo "  CLEAN   ozip"
	fi
	rm -rf ozip/kernel/dt.img
	rm -rf ozip/kernel/zImage
	rm -rf arch/arm/boot/"$kerneltype"
	make clean && make mrproper
	echo "Working directory cleaned..."
	break
	;;
    n|N)
	break
	;;
    * )
	echo
	echo "Invalid try again!"
	echo
	;;
esac
done

while read -p "Do you want to build (y/n)? " cchoice
do
echo
case "$cchoice" in
	y|Y )
		make "$config"
		make "$jobcount"
		echo "Done"
		break
		;;
	n|N )
		break
		;;
	* )
		echo
		echo "Invalid try again!"
		echo
		;;
esac
done

echo "Extracting files..."
if [ -f arch/arm/boot/"$kerneltype" ]; then
	cp arch/arm/boot/"$kerneltype" ozip/kernel/.
fi

echo "Making dt.img..."
if [ -f arch/arm/boot/"$kerneltype" ]; then
	./dtbToolCM -2 -o ozip/kernel/dt.img -s 2048 -p scripts/dtc/ arch/arm/boot/
	echo "dt.img created"
else
	echo "No build found..."
	exit 0;
fi

echo "Zipping..."
if [ -f arch/arm/boot/"$kerneltype" ]; then
	cd ozip
	zip -r ../$zipname.zip .
	mkdir -p $build
	mv ../$zipname.zip $build/.
	cd ..
	echo "Done..."
	echo "Output zip: $zipname.zip"
	exit 0;
else
	echo "No $kerneltype found..."
	exit 0;
fi

if uname -a | grep Darwin; then
    t2=`gdate +"%s`
else
    t2=`date +"%s`
fi

time=$(($t2 - $t1))
echo "Build took $(($time / 60)) minute(s) and $(($time % 60)) seconds."
echo