#!/bin/bash

###############################################################################
#Variables
export PROJECT_LIST=$(find project | grep "GCC/Makefile$")
export FEATURE_LIST=$(find project | grep "GCC/feature.*.mk$")
export BOARD_LIST="project/*"
export OUT="${PWD}/out"
export FLASHGENERATOR="tools/flashgen/flashgen.pl"
feature_mk=""
feature_mk_o=""
bl_feature_mk=""
bl_feature_mk_o=""
bl_feature_mk_o_flag=0
compatibleTest="off"
cobuild="off"

platform=$(uname)
if [[ "${platform}" =~ "MINGW" || "${platform}" =~ "MSYS" ]]; then
    max_jobs=$(( $(WMIC CPU Get NumberOfLogicalProcessors|tail -2|awk '{print $1}') - 1))
    if [ "${max_jobs}" == "0" ]; then
        max_jobs="1"
    fi
elif [[ "${platform}" =~ "Darwin" ]]; then
    export PATH="/usr/local/bin:${PATH}"
    max_jobs=`sysctl -n hw.ncpu`
else
    max_jobs=`cat /proc/cpuinfo |grep ^processor|wc -l`
fi

# check /bin/bash
if [ -f /bin/bash ]; then
    export SHELL="/bin/bash"
fi

bl_extra_opt=-j${max_jobs}

source tools/scripts/build/usage_function

use_new_out_folder="false"
target_check () {
    for p in ${PROJECT_LIST}
    do
        q=$(echo ${p} | grep "project/${1}/")
        if [ ! -z "${q}" ]; then
            r=$(echo ${q} | sed 's/GCC\/Makefile//')
            s=`basename ${r}`
            if [ "${s}" == "${2}" ]; then
                echo "UE BUILD BOARD: ${1}"
                echo "UE BUILD PROJECT: ${2}"
                if [ ${use_new_out_folder} == "false" ]; then
                    OUT=${OUT}/${1}/${2}
                fi
                BUILD=project/${1}/${2}
                export TARGET_PATH=$(dirname ${q})
                return 0
            fi
        fi
    done
    return 1
}

# support MinGW
mingw_check () {
    echo "platform=${platform}"
		if [[ "${platform}" =~ "MINGW" || "${platform}" =~ "MSYS" ]]; then
        pwdpath=$(pwd)
        echo ${pwdpath}
        if [[ "${pwdpath}" =~ "\[|\]| " ]]; then
            echo "Build.sh Exception: The codebase folder name should not have spacing, [ or ]."
            exit 1
        fi
    fi
}

clean_out () {
    rm -rf "${1}"
    echo "rm -rf ${1}"
}

check_folder_length () {
    #echo -e "\nout_folder_length_limit=$out_folder_length_limit"
    out_folder_length=$(echo ${2}| wc -m)
    if [[ "${platform}" =~ "MINGW" || "${platform}" =~ "MSYS" ]]; then
        if [ "${out_folder_length}" -gt "$(( out_folder_length_limit + pwd_folder_length_limit ))" ]; then

            echo -e "\n Error: over maximum path length"
            if [ "${pwdLength}" -gt "${pwd_folder_length_limit}" ]; then
                echo -e "    1. Please install SDK in a path less than ${pwd_folder_length_limit} characters"
            fi

            # here's out_folder_length using absolute path
            if [ "$(( out_folder_length - pwdLength ))" -gt "${out_folder_length_limit}" ]; then
                echo -e "    2. Please shorter the name of project or feature."
            fi
            exit 1
        else
            return 0
        fi
    fi
    return 0
}

###############################################################################
#Begin here
if [ "$#" -eq "0" ]; then
    show_usage
    exit 1
fi

# parsing arguments
declare -a argv=($0)
ori_argv=$@
do_make_clean="none"
for i in $@
do
    case ${i} in
        -o=*|--option=*)
            opt=" ${i#*=}"
            echo "UE BUILD OPTION:${opt}"
            echo "${opt}" | grep -q -E " OUT="
            if [[ $? -eq 0 ]]; then
                OUT=`echo ${opt} | grep -o "OUT=[^ |^	]*" | cut -d '=' -f2 | tr -d ' '`
                if [ -z "${OUT}" ]; then
                    echo "Error: -o=OUT= cannot be empty!"
                    show_usage
                    exit 1
                fi
                OUT=${PWD}/${OUT}
                use_new_out_folder="true"
                echo "output folder change to: ${OUT}"
            fi
            if [[ "${opt}" =~ " FEATURE=" ]]; then
                feature_mk_o="${opt#*=}"
                shift
                continue
            fi
            if [[ "${opt}" =~ " BL_FEATURE=" ]]; then
                bl_feature_mk_o="${opt#*=}"
                bl_feature_mk_o_flag=1
            fi
            if [[ "${opt}" == " IOT_SDK_GCC_VER=V9" ]]; then
                    echo "IOT_SDK_GCC_VERSION : 9.2.1"
                    bl_extra_opt+=${opt}
            fi
            extra_opt+=${opt}
            shift
            ;;
        -blo=*|--bloption=*)
            opt=" ${i#*=}"
            echo "${opt}" | grep -q -E " OUT="
            if [[ $? -eq 0 ]]; then
                echo "Error: Unsupported -o=OUT= in [-blo|-bloption]."
                exit 1
            fi
            bl_extra_opt+=${opt}
            do_make_clean="true"
            shift
            ;;
        -f=*|--feature=*)
            feature_mk="${i#*=}"
            shift
            ;;
        -j=*|--jobs=*)
            max_jobs="${i#*=}"
            shift
            ;;
        list)
            if [ "${#argv[@]}" == "3" ]; then
                target_check ${argv[1]} ${argv[2]}
                if [ "$?" -eq "0" ]; then
                    show_available_proj ${argv[1]} ${argv[2]}
                    exit 0
                else
                    echo "Error: ${argv[1]} ${argv[2]} is not available board & project"
                    show_usage
                    exit 1
                fi
            elif [ "${#argv[@]}" == "1" ]; then
                show_available_proj
                exit 0
            else
                #echo "args : ${#argv[@]}"
                echo "Error: wrong usage!!"
                exit 1
            fi
            ;;
        -compTest)
            compatibleTest="on"
            ;;
        -cobuild)
            cobuild="on"
            ;;
        -*)
            echo "Error: unknown parameter \"${i}\""
            show_usage
            exit 1
            ;;
        *)
            argv+=($i)
            ;;
    esac
done

if [ ! -z ${feature_mk_o} ]; then
    if [ ! -z ${feature_mk} ]; then
        echo "Override -f=${feature_mk} by -o=FEATURE=${feature_mk_o}"
        echo "FEATURE=${feature_mk_o}"
    fi
    feature_mk=${feature_mk_o}
fi

export PROJ_NAME=${argv[2]}
export EXTRA_VAR=-j${max_jobs}
###############################################################################
if [ "${argv[3]}" == "bl" ]; then
    if [ "${#argv[@]}" != "4" ]; then
        show_usage
        exit 1
    fi
    target_check ${argv[1]} ${argv[2]}
    if [ "$?" -ne "0" ]; then
        echo "Error: ${argv[1]} ${argv[2]} is not available board & project"
        show_usage
        exit 1
    fi

    mingw_check
    where_to_find_feature_mk=${TARGET_PATH}
    if [ ! -z ${feature_mk} ]; then
        if [ ! -e "${TARGET_PATH}/${feature_mk}" ]; then
            echo "Error: cannot find ${feature_mk} under ${TARGET_PATH}."
            exit 1
        fi
        EXTRA_VAR+=" FEATURE=${feature_mk}"
    else
        where_to_find_feature_mk=`grep "^TARGET_PATH\ *[?:]\{0,1\}=\ *" ${TARGET_PATH}/Makefile | cut -d '=' -f2 | tr -d ' ' | tr -d '\r' | tail -1`
        if [ -z ${where_to_find_feature_mk} ]; then
            where_to_find_feature_mk=${TARGET_PATH}
        fi
        feature_mk=`grep "^FEATURE\ *[?:]\{0,1\}=\ *" ${TARGET_PATH}/Makefile | cut -d '=' -f2 | tr -d ' ' | tr -d '\r'| tail -1`
    fi
    echo "FEATURE = ${feature_mk}"

    if [ -e "${OUT}/obj/${TARGET_PATH}/tmp.mk" ]; then
        diff -q ${where_to_find_feature_mk}/${feature_mk} ${OUT}/obj/${TARGET_PATH}/tmp.mk
        if [ $? -ne 0 ]; then
            do_make_clean="true"
        fi
    fi
    if [ -e "${OUT}/obj/${TARGET_PATH}/extra_opts.lis" ]; then
        echo ${extra_opt} | grep -Eo  "[_[:alnum:]]+=[[:graph:]]*" | sort | uniq > ${OUT}/obj/${TARGET_PATH}/extra_opts.current
        diff -q ${OUT}/obj/${TARGET_PATH}/extra_opts.current ${OUT}/obj/${TARGET_PATH}/extra_opts.lis
        if [ $? -ne 0 ]; then
            do_make_clean="true"
        else
            if [ ${do_make_clean} != "true" ]; then
                do_make_clean="false"
                rm -f "${OUT}/obj/${TARGET_PATH}/extra_opts.current"
            fi
        fi
    fi
    if [ ${do_make_clean} == "true" ]; then
        clean_out ${OUT}
    fi
    mkdir -p ${OUT}/obj/${TARGET_PATH}
    echo ${extra_opt} | grep -Eo  "[_[:alnum:]]+=[[:graph:]]*" | sort | uniq > ${OUT}/obj/${TARGET_PATH}/extra_opts.lis
    cp -f "${where_to_find_feature_mk}/${feature_mk}" "${OUT}/obj/${TARGET_PATH}/tmp.mk"

    CM4_TARGET_PATH_BAK=${TARGET_PATH}
    TARGET_PATH="project/${argv[1]}/apps/bootloader/GCC"

    # Check bootloader feature makefile
    if [ "${bl_feature_mk_o_flag}" -eq "0" ]; then
        bl_feature_mk=`grep "^BL_FEATURE\ *[?:]\{0,1\}=\ *" ${CM4_TARGET_PATH_BAK}/${feature_mk} | cut -d '=' -f2 | tr -d ' ' | tr -d '\r' | tail -1`
    else
        bl_feature_mk=${bl_feature_mk_o}
    fi
    if [ ! -z ${bl_feature_mk} ]; then
        if [ ! -e "${TARGET_PATH}/${bl_feature_mk}" ]; then
            echo "Error: cannot find ${bl_feature_mk} under ${TARGET_PATH}."
            exit 1
        fi
        bl_extra_opt+=" FEATURE=${bl_feature_mk}"
    fi
    echo "BL_FEATURE = ${bl_feature_mk}"

    mkdir -p ${OUT}/log
    echo "$0 ${ori_argv}" > ${OUT}/log/build_time.log
    echo "Start Build: "`date` >> ${OUT}/log/build_time.log
    echo ""
    echo "Build bootloader..."
    # Check if the source dir is existed
    if [ ! -d "project/${argv[1]}/apps/bootloader" ]; then
        echo "Error: no bootloader source in project/${argv[1]}/apps/bootloader"
        exit 1
    fi

    mkdir -p ${OUT}
    echo "make -C ${TARGET_PATH} BUILD_DIR=${OUT}/obj/bootloader OUTPATH=${OUT} BL_MAIN_PROJECT=${argv[2]} ${bl_extra_opt}"
    make -C ${TARGET_PATH} BUILD_DIR=${OUT}/obj/bootloader OUTPATH=${OUT} BL_MAIN_PROJECT=${argv[2]} ${bl_extra_opt} 2>> ${OUT}/err.log
    BUILD_RESULT=$?
    mkdir -p ${OUT}/lib
    mkdir -p ${OUT}/log
    #echo "End Build: "`date` >> ${OUT}/log/build_time.log
    if [ $BUILD_RESULT -ne 0 ]; then
        echo "Error: bootloader build failed!!"
        echo "BOOTLOADER BUILD : FAIL" >> ${OUT}/log/build_time.log
        if [ "$compatibleTest" == "off" ]; then
            mv -f ${OUT}/*.a "${OUT}/lib/" 2> /dev/null
            mv -f ${OUT}/*.log "${OUT}/log/" 2> /dev/null
            exit 2;
        fi
    else
        mv -f "${OUT}/copy_firmware_opts.log" "${OUT}/log/copy_firmware_opts_bootloader.log" 2> /dev/null
        echo "BOOTLOADER BUILD : PASS" >> ${OUT}/log/build_time.log
    fi
    echo "Build bootloader...Done"
    echo ""

    # build cm4 firmware
    echo "Build CM4 Firmware..."
    TARGET_PATH=${CM4_TARGET_PATH_BAK}
    mkdir -p ${OUT}/autogen
    EXTRA_VAR+="${extra_opt}"
    echo "make -C ${TARGET_PATH} BUILD_DIR=${OUT}/obj OUTPATH=${OUT} $EXTRA_VAR"
    make -C ${TARGET_PATH} BUILD_DIR=${OUT}/obj OUTPATH=${OUT} $EXTRA_VAR 2>> ${OUT}/err.log
    BUILD_RESULT=$?
    mkdir -p ${OUT}/lib
    mkdir -p ${OUT}/log
    mv -f ${OUT}/*.a "${OUT}/lib/" 2> /dev/null
    mv -f ${OUT}/*.log "${OUT}/log/" 2> /dev/null
    echo "Build CM4 Firmware...Done"
    echo "End Build: "`date` >> ${OUT}/log/build_time.log
    cat ${OUT}/log/build.log | grep "MODULE BUILD" >> ${OUT}/log/build_time.log
    if [ "$BUILD_RESULT" -eq "0" ]; then
        echo "MCU warnings: "`grep warning: ${OUT}/log/err.log |grep -v -f ${PWD}/tools/scripts/build/skip_warning_check.txt -c` >> ${OUT}/log/build_time.log
        echo "MCU build log: ${OUT}/log/err.log " >> ${OUT}/log/build_time.log
        echo "TOTAL BUILD: PASS" >> ${OUT}/log/build_time.log
    else
        echo "TOTAL BUILD: FAIL" >> ${OUT}/log/build_time.log
    fi
    echo "=============================================================="
    cat ${OUT}/log/build_time.log
    exit $BUILD_RESULT
elif [ "${argv[3]}" == "clean" ]; then
    if [ "${#argv[@]}" != "4" ]; then
        show_usage
        exit 1
    fi
    if [ "$use_new_out_folder" == "true" ]; then
        rm -rf "${OUT}"
    else
        rm -rf "${OUT}/${argv[1]}/${argv[2]}"
    fi
elif [ "${argv[2]}" == "clean" ]; then
    if [ "${#argv[@]}" != "3" ]; then
        show_usage
        exit 1
    fi
    if [ "$use_new_out_folder" == "true" ]; then
        rm -rf "${OUT}"
    else
        rm -rf "${OUT}/${argv[1]}"
    fi
elif [ "${argv[1]}" == "clean" ]; then
    if [ "${#argv[@]}" != "2" ]; then
        show_usage
        exit 1
    fi
    rm -rf "${OUT}"
else
    if [ "${#argv[@]}" != "3" ]; then
        show_usage
        exit 1
    fi
    target_check ${argv[1]} ${argv[2]}
    if [ "$?" -ne "0" ]; then
        echo "Error: ${argv[1]} ${argv[2]} is not available board & project or module"
        show_usage
        exit 1
    fi

    mingw_check
    where_to_find_feature_mk=${TARGET_PATH}
    if [ ! -z ${feature_mk} ]; then
        if [ ! -e "${TARGET_PATH}/${feature_mk}" ]; then
            echo "Error: cannot find ${feature_mk} under ${TARGET_PATH}."
            exit 1
        fi
        EXTRA_VAR+=" FEATURE=${feature_mk}"
    else
        where_to_find_feature_mk=`grep "^TARGET_PATH\ *[?:]\{0,1\}=\ *" ${TARGET_PATH}/Makefile | cut -d '=' -f2 | tr -d ' ' | tr -d '\r' | tail -1`
        if [ -z ${where_to_find_feature_mk} ]; then
            where_to_find_feature_mk=${TARGET_PATH}
        fi
        feature_mk=`grep "^FEATURE\ *[?:]\{0,1\}=\ *" ${TARGET_PATH}/Makefile | cut -d '=' -f2 | tr -d ' ' | tr -d '\r'| tail -1`
    fi
    echo "FEATURE = ${feature_mk}"

    if [ -e "${OUT}/obj/${TARGET_PATH}/tmp.mk" ]; then
        diff -q ${where_to_find_feature_mk}/${feature_mk} ${OUT}/obj/${TARGET_PATH}/tmp.mk
        if [ $? -ne 0 ]; then
            do_make_clean="true"
        fi
    fi
    if [ -e "${OUT}/obj/${TARGET_PATH}/extra_opts.lis" ]; then
        echo ${extra_opt} | grep -Eo  "[_[:alnum:]]+=[[:graph:]]*" | sort | uniq > ${OUT}/obj/${TARGET_PATH}/extra_opts.current
        diff -q ${OUT}/obj/${TARGET_PATH}/extra_opts.current ${OUT}/obj/${TARGET_PATH}/extra_opts.lis
        if [ $? -ne 0 ]; then
            do_make_clean="true"
        else
            if [ ${do_make_clean} != "true" ]; then
                do_make_clean="false"
                rm -f "${OUT}/obj/${TARGET_PATH}/extra_opts.current"
            fi
        fi
    fi
    if [ ${do_make_clean} == "true" ]; then
        clean_out ${OUT}
    fi
    mkdir -p ${OUT}/obj/${TARGET_PATH}
    echo ${extra_opt} | grep -Eo  "[_[:alnum:]]+=[[:graph:]]*" | sort | uniq > ${OUT}/obj/${TARGET_PATH}/extra_opts.lis
    cp -f "${where_to_find_feature_mk}/${feature_mk}" "${OUT}/obj/${TARGET_PATH}/tmp.mk"

    CM4_TARGET_PATH_BAK=${TARGET_PATH}
    TARGET_PATH="project/${argv[1]}/apps/bootloader/GCC"

    # Check bootloader feature makefile
    if [ "${bl_feature_mk_o_flag}" -eq "0" ]; then
        bl_feature_mk=`grep "^BL_FEATURE\ *[?:]\{0,1\}=\ *" ${CM4_TARGET_PATH_BAK}/${feature_mk} | cut -d '=' -f2 | tr -d ' ' | tr -d '\r' | tail -1`
    else
        bl_feature_mk=${bl_feature_mk_o}
    fi
    if [ ! -z ${bl_feature_mk} ]; then
        if [ ! -e "${TARGET_PATH}/${bl_feature_mk}" ]; then
            echo "Error: cannot find ${bl_feature_mk} under ${TARGET_PATH}."
            exit 1
        fi
        bl_extra_opt+=" FEATURE=${bl_feature_mk}"
    fi
    echo "BL_FEATURE = ${bl_feature_mk}"

    mkdir -p ${OUT}/autogen
    mkdir -p ${OUT}/log
    mkdir -p ${OUT}/lib
    echo "$0 ${ori_argv}" > ${OUT}/log/build_time.log
    # build bootloader by BL_FEATURE in project feature mk
    if [ ! -z ${bl_feature_mk} ]; then
        #echo "$0 ${argv[1]} bootloader -f=${bl_feature_mk}" > ${OUT}/log/build_time.log
        echo "Start Build: "`date` >> ${OUT}/log/build_time.log
        echo ""
        echo "Build bootloader..."
        # Check if the source dir is existed
        if [ ! -d "project/${argv[1]}/apps/bootloader" ]; then
            echo "Error: no bootloader source in project/${argv[1]}/apps/bootloader"
            exit 1
        fi

        echo "make -C ${TARGET_PATH} BUILD_DIR=${OUT}/obj/bootloader OUTPATH=${OUT} BL_MAIN_PROJECT=${argv[2]} ${bl_extra_opt}"
        make -C ${TARGET_PATH} BUILD_DIR=${OUT}/obj/bootloader OUTPATH=${OUT} BL_MAIN_PROJECT=${argv[2]} ${bl_extra_opt} 2>> ${OUT}/err.log
        BUILD_RESULT=$?
        #echo "End Build: "`date` >> ${OUT}/log/build_time.log
        if [ $BUILD_RESULT -ne 0 ]; then
            echo "Error: bootloader build failed!!"
            echo "BOOTLOADER BUILD : FAIL" >> ${OUT}/log/build_time.log
            if [ "$compatibleTest" == "off" ]; then
                mv -f ${OUT}/*.a "${OUT}/lib/" 2> /dev/null
                mv -f ${OUT}/*.log "${OUT}/log/" 2> /dev/null
                exit 2;
            fi
        else
            mv -f "${OUT}/copy_firmware_opts.log" "${OUT}/log/copy_firmware_opts_bootloader.log" 2> /dev/null
            echo "BOOTLOADER BUILD : PASS" >> ${OUT}/log/build_time.log
        fi
        echo "Build bootloader...Done"
        echo " "
    fi

    # build Build CM4 Firmware...
    echo "Build CM4 Firmware..."
    TARGET_PATH=${CM4_TARGET_PATH_BAK}
    echo "Start Build: "`date` >> ${OUT}/log/build_time.log
    if [ ! -z ${feature_mk} ]; then
        EXTRA_VAR+=" FEATURE=${feature_mk}"
    fi
    EXTRA_VAR+="${extra_opt}"
    #echo "make -C ${TARGET_PATH} BUILD_DIR=${OUT}/obj OUTPATH=${OUT} $EXTRA_VAR"
    make -C ${TARGET_PATH} BUILD_DIR=${OUT}/obj OUTPATH=${OUT} $EXTRA_VAR 2>> ${OUT}/err.log
    BUILD_RESULT=$?
    mv -f ${OUT}/*.a "${OUT}/lib/" 2> /dev/null
    mv -f ${OUT}/*.log "${OUT}/log/" 2> /dev/null
    echo "End Build: "`date` >> ${OUT}/log/build_time.log
    cat ${OUT}/log/build.log | grep "MODULE BUILD" >> ${OUT}/log/build_time.log
    if [ "${BUILD_RESULT}" -eq "0" ]; then
        echo "MCU warnings: "`grep warning: ${OUT}/log/err.log |grep -v -f ${PWD}/tools/scripts/build/skip_warning_check.txt -c` >> ${OUT}/log/build_time.log
        echo "MCU build log: ${OUT}/log/err.log " >> ${OUT}/log/build_time.log
        echo "TOTAL BUILD: PASS" >> ${OUT}/log/build_time.log
    else
        echo "TOTAL BUILD: FAIL" >> ${OUT}/log/build_time.log
    fi
    echo "=============================================================="
    cat ${OUT}/log/build_time.log
    exit ${BUILD_RESULT}
fi