#!/bin/bash
#functions
function echo() {
    command echo -e "$@"
}

function header_success() {
    echo ${GREEN}
    echo "====$(for i in $(seq ${#1}); do echo "=\c"; done)===="
    echo "==  ${1}  =="
    echo "====$(for i in $(seq ${#1}); do echo "=\c"; done)===="
    echo ${RST}
}

function header_error() {
    echo ${RED}
    echo "====$(for i in $(seq ${#1}); do echo "=\c"; done)===="
    echo "==  ${1}  =="
    echo "====$(for i in $(seq ${#1}); do echo "=\c"; done)===="
    echo ${RST}
}

#colors
RED="\033[0;31m"
GREEN="\033[0;32m"
RST="\033[0m"
BROWN="\033[0;33m"
CYAN="\033[0;36m"
NC="\033[0m"
BOLD="\033[1m"

#directories
KERNEL_DIR=$PWD
KERN_IMG=$KERNEL_DIR/arch/arm64/boot/Image.gz-dtb
CONFIG_DIR=$KERNEL_DIR/arch/arm64/configs

#export
export CROSS_COMPILE="$HOME/aarch64-linux-android-4.9/bin/aarch64-linux-android-"
export ARCH=arm64
export SUBARCH=arm64
export KBUILD_BUILD_USER=AnggaSyncVeganaShooter

#misc
CONFIG=lineageos_kenzo_defconfig
THREAD="-j$(grep -c ^processor /proc/cpuinfo)"

#main script
while true; do
echo -e "\n${NC}[1]Build kernel"
echo -e "[2]Regenerate defconfig"
echo -e "[3]Source cleanup"
echo -e "[4]Quit${NC}"
echo -ne "\n${CYAN}Please enter a choice[1-4]:${NC} "

read choice

if [ "$choice" == "1" ]; then
  echo " "
  BUILD_START=$(date +"%s")
  DATE=`date`
  echo "${NC}Build started at $DATE${NC}"
  make lineageos_kenzo_defconfig
  make menuconfig
  make $THREAD

  if ! [ -a $KERN_IMG ]; then
    header_error "KERNEL COMPILATION FAILED"
    exit 1
  fi
  BUILD_END=$(date +"%s")
  DIFF=$(($BUILD_END - $BUILD_START))
  header_success "BUILD SUCCESSFUL"
  echo "${BOLD}Script duration: $(($DIFF / 60)) MINUTES AND $(($DIFF % 60)) SECONDS.${NC}"
fi


if [ "$choice" == "2" ]; then
  echo " "
  make $CONFIG
  cp .config $CONFIG_DIR/$CONFIG
  header_success "DEFCONFIG GENERATED"
fi


if [ "$choice" == "3" ]; then
  make clean &>/dev/null
  make mrproper &>/dev/null
  header_success "KERNEL SOURCES CLEANED"
fi


if [ "$choice" == "4" ]; then
 exit 1
fi
done