# This file contains some usefull common functions

# Copyright 2007 Yann E. MORIN

# Licensed under the GPL v2. See COPYING in the root of this package

CT_OnError() {

    ret=$?

    CT_DoLog ERROR "Build failed in step \"${CT_STEP_MESSAGE[${CT_STEP_COUNT}]}\""

    for((step=(CT_STEP_COUNT-1); step>1; step--)); do

        CT_DoLog ERROR "      called in step \"${CT_STEP_MESSAGE[${step}]}\""

    done

    CT_DoLog ERROR "Error happened in \"${BASH_SOURCE[1]}\" in function \"${FUNCNAME[1]}\" (line unknown, sorry)"

    for((depth=2; ${BASH_LINENO[$((${depth}-1))]}>0; depth++)); do

        CT_DoLog ERROR "      called from \"${BASH_SOURCE[${depth}]}\" at line # ${BASH_LINENO[${depth}-1]} in function \"${FUNCNAME[${depth}]}\""

    done

    CT_DoLog ERROR "Look at \"${CT_ACTUAL_LOG_FILE}\" for more info on this error."

    exit $ret

}

trap CT_OnError ERR

set -E

set -o pipefail

# The different log levels:

CT_LOG_LEVEL_ERROR=0

CT_LOG_LEVEL_WARN=1

CT_LOG_LEVEL_INFO=2

CT_LOG_LEVEL_EXTRA=3

CT_LOG_LEVEL_DEBUG=4

CT_LOG_LEVEL_ALL=5

# Attributes

_A_NOR="\\033[0m"

_A_BRI="\\033[1m"

_A_DIM="\\033[2m"

_A_UND="\\033[4m"

_A_BRB="\\033[5m"

_A_REV="\\033[7m"

_A_HID="\\033[8m"

# Fore colors

_F_BLK="\\033[30m"

_F_RED="\\033[31m"

_F_GRN="\\033[32m"

_F_YEL="\\033[33m"

_F_BLU="\\033[34m"

_F_MAG="\\033[35m"

_F_CYA="\\033[36m"

_F_WHI="\\033[37m"

# A function to log what is happening

# Different log level are available:

#   - ERROR:   A serious, fatal error occurred

#   - WARN:    A non fatal, non serious error occurred, take your responsbility with the generated build

#   - INFO:    Informational messages

#   - EXTRA:   Extra informational messages

#   - DEBUG:   Debug messages

#   - ALL:     Component's build messages

# Usage: CT_DoLog <level> [message]

# If message is empty, then stdin will be logged.

CT_DoLog() {

    local max_level LEVEL level cur_l cur_L

    local l

    eval max_level="\${CT_LOG_LEVEL_${CT_LOG_LEVEL_MAX}}"

    # Set the maximum log level to DEBUG if we have none

    [ -z "${max_level}" ] && max_level=${CT_LOG_LEVEL_DEBUG}

    LEVEL="$1"; shift

    eval level="\${CT_LOG_LEVEL_${LEVEL}}"

    if [ $# -eq 0 ]; then

        cat -

    else

        echo "${1}"

    fi |( IFS="\n" # We want the full lines, even leading spaces

          CT_PROG_BAR_CPT=0

          indent=$((2*CT_STEP_COUNT))

          while read line; do

              case "${CT_LOG_SEE_TOOLS_WARN},${line}" in

                y,*"warning:"*)         cur_L=WARN; cur_l=${CT_LOG_LEVEL_WARN};;

                *"error:"*)             cur_L=ERROR; cur_l=${CT_LOG_LEVEL_ERROR};;

                *"make["?*"]:"*"Stop.") cur_L=ERROR; cur_l=${CT_LOG_LEVEL_ERROR};;

                *)                      cur_L="${LEVEL}"; cur_l="${level}";;

              esac

              l="`printf \"[%-5s]%*s%s%s\" \"${cur_L}\" \"${indent}\" \" \" \"${line}\"`"

              # There will always be a log file, be it /dev/null

              echo -e "${l}" >>"${CT_ACTUAL_LOG_FILE}"

              color="CT_${cur_L}_COLOR"

              normal="CT_NORMAL_COLOR"

              if [ ${cur_l} -le ${max_level} ]; then

                  echo -e "\r${!color}${l}${!normal}"

              fi

              if [ "${CT_LOG_PROGRESS_BAR}" = "y" ]; then

                  str=`CT_DoDate +%s`

                  elapsed=$((str-(CT_STAR_DATE/(1000*1000*1000))))

                  [ ${CT_PROG_BAR_CPT} -eq 0  ] && bar="/"

                  [ ${CT_PROG_BAR_CPT} -eq 10 ] && bar="-"

                  [ ${CT_PROG_BAR_CPT} -eq 20 ] && bar="\\"

                  [ ${CT_PROG_BAR_CPT} -eq 30 ] && bar="|"

                  printf "\r[%02d:%02d] %s " $((elapsed/60)) $((elapsed%60)) "${bar}"

                  CT_PROG_BAR_CPT=$(((CT_PROG_BAR_CPT+1)%40))

              fi

          done

        )

    return 0

}

# Abort the execution with a error message

# Usage: CT_Abort <message>

CT_Abort() {

    CT_DoLog ERROR "$1" >&2

    exit 1

}

# Test a condition, and print a message if satisfied

# Usage: CT_Test <message> <tests>

CT_Test() {

    local ret

    local m="$1"

    shift

    test "$@" && CT_DoLog WARN "$m"

    return 0

}

# Test a condition, and abort with an error message if satisfied

# Usage: CT_TestAndAbort <message> <tests>

CT_TestAndAbort() {

    local m="$1"

    shift

    test "$@" && CT_Abort "$m"

    return 0

}

# Test a condition, and abort with an error message if not satisfied

# Usage: CT_TestAndAbort <message> <tests>

CT_TestOrAbort() {

    local m="$1"

    shift

    test "$@" || CT_Abort "$m"

    return 0

}

# Test the presence of a tool, or abort if not found

# Usage: CT_HasOrAbort <tool>

CT_HasOrAbort() {

    CT_TestAndAbort "\"${1}\" not found and needed for successfull toolchain build." -z "`which \"${1}\"`"

    return 0

}

# Get current date with nanosecond precision

# On those system not supporting nanosecond precision, faked with rounding down

# to the highest entire second

# Usage: CT_DoDate <fmt>

CT_DoDate() {

    date "$1" |sed -r -e 's/%N$/000000000/;'

}

CT_STEP_COUNT=1

CT_STEP_MESSAGE[${CT_STEP_COUNT}]="<none>"

# Memorise a step being done so that any error is caught

# Usage: CT_DoStep <loglevel> <message>

CT_DoStep() {

    local start=`CT_DoDate +%s%N`

    CT_DoLog "$1" "================================================================="

    CT_DoLog "$1" "$2"

    CT_STEP_COUNT=$((CT_STEP_COUNT+1))

    CT_STEP_LEVEL[${CT_STEP_COUNT}]="$1"; shift

    CT_STEP_START[${CT_STEP_COUNT}]="${start}"

    CT_STEP_MESSAGE[${CT_STEP_COUNT}]="$1"

    return 0

}

# End the step just being done

# Usage: CT_EndStep

CT_EndStep() {

    local stop=`CT_DoDate +%s%N`

    local duration=`printf "%032d" $((stop-${CT_STEP_START[${CT_STEP_COUNT}]})) |sed -r -e 's/([[:digit:]]{2})[[:digit:]]{7}$/\.\1/; s/^0+//; s/^\./0\./;'`

    local level="${CT_STEP_LEVEL[${CT_STEP_COUNT}]}"

    local message="${CT_STEP_MESSAGE[${CT_STEP_COUNT}]}"

    CT_STEP_COUNT=$((CT_STEP_COUNT-1))

    CT_DoLog "${level}" "${message}: done in ${duration}s"

    return 0

}

# Pushes into a directory, and pops back

CT_Pushd() {

    pushd "$1" >/dev/null 2>&1

}

CT_Popd() {

    popd >/dev/null 2>&1

}

# Makes a path absolute

# Usage: CT_MakeAbsolutePath path

CT_MakeAbsolutePath() {

    # Try to cd in that directory

    if [ -d "$1" ]; then

        CT_Pushd "$1"

        pwd

        CT_Popd

    else

        # No such directory, fail back to guessing

        case "$1" in

            /*)  echo "$1";;

            *)   echo "`pwd`/$1";;

        esac

    fi

    return 0

}

# Creates a temporary directory

# $1: variable to assign to

# Usage: CT_MktempDir foo

CT_MktempDir() {

    # Some mktemp do not allow more than 6 Xs

    eval "$1"="`mktemp -q -d \"${CT_BUILD_DIR}/.XXXXXX\"`"

    CT_TestOrAbort "Could not make temporary directory" -n "${!1}" -a -d "${!1}"

}

# Echoes the specified string on stdout until the pipe breaks.

# Doesn't fail

# $1: string to echo

# Usage: CT_DoYes "" |make oldconfig

CT_DoYes() {

    yes "$1" || true

}

# Get the file name extension of a component

# Usage: CT_GetFileExtension <component_name-component_version>

# If found, echoes the extension to stdout

# If not found, echoes nothing on stdout.

CT_GetFileExtension() {

    local ext

    local file="$1"

    local got_it=1

    CT_Pushd "${CT_TARBALLS_DIR}"

    for ext in .tar.gz .tar.bz2 .tgz .tar; do

        if [ -f "${file}${ext}" ]; then

            echo "${ext}"

            got_it=0

            break

        fi

    done

    CT_Popd

    return 0

}

# Download an URL using wget

# Usage: CT_DoGetFileWget <URL>

CT_DoGetFileWget() {

    # Need to return true because it is legitimate to not find the tarball at

    # some of the provided URLs (think about snapshots, different layouts for

    # different gcc versions, etc...)

    # Some (very old!) FTP server might not support the passive mode, thus

    # retry without

    # With automated download as we are doing, it can be very dangerous to use

    # -c to continue the downloads. It's far better to simply overwrite the

    # destination file

    wget -nc --progress=dot:binary --tries=3 --passive-ftp "$1" || wget -nc --progress=dot:binary --tries=3 "$1" || true

}

# Download an URL using curl

# Usage: CT_DoGetFileCurl <URL>

CT_DoGetFileCurl() {

	# Note: comments about wget method are also valid here

	# Plus: no good progreess indicator is available with curl,

	#       so output is consigned to oblivion

	curl --ftp-pasv -O --retry 3 "$1" >/dev/null || curl -O --retry 3 "$1" >/dev/null || true

}

# Wrapper function to call one of curl or wget

# Usage: CT_DoGetFile <URL>

CT_DoGetFile() {

    local _wget=`which wget`

    local _curl=`which curl`

    case "${_wget},${_curl}" in

        ,)  CT_DoError "Could find neither wget nor curl";;

        ,*) CT_DoGetFileCurl "$1" 2>&1 |CT_DoLog DEBUG;;

        *)  CT_DoGetFileWget "$1" 2>&1 |CT_DoLog DEBUG;;

    esac

}

# Download the file from one of the URLs passed as argument

# Usage: CT_GetFile <filename> <url> [<url> ...]

CT_GetFile() {

    local got_it

    local ext

    local url

    local file="$1"

    shift

    # Do we already have it?

    ext=`CT_GetFileExtension "${file}"`

    if [ -n "${ext}" ]; then

        CT_DoLog DEBUG "Already have \"${file}\""

        return 0

    fi

    CT_DoLog EXTRA "Retrieving \"${file}\""

    CT_Pushd "${CT_TARBALLS_DIR}"

    # File not yet downloaded, try to get it

    got_it=0

    # We'd rather have a bzip2'ed tarball, then gzipped, and finally plain tar.

    for ext in .tar.bz2 .tar.gz .tgz .tar; do

        if [ ${got_it} -ne 1 ]; then

            # Try local copy first, if it exists

            if [ -r "${CT_LOCAL_TARBALLS_DIR}/${file}${ext}" -a \

                 "${CT_FORCE_DOWNLOAD}" != "y" ]; then

                cp -v "${CT_LOCAL_TARBALLS_DIR}/${file}${ext}" "${file}${ext}" |CT_DoLog DEBUG

                got_it=1

                break 1

            else

                # Try all urls in turn

                for url in "$@"; do

                    case "${url}" in

                        *)  CT_DoLog DEBUG "Trying \"${url}/${file}${ext}\""

                            CT_DoGetFile "${url}/${file}${ext}"

                            ;;

                    esac

                    [ -f "${file}${ext}" ] && got_it=1 && break 2 || true

                done

            fi

        fi

    done

    CT_Popd

    CT_TestAndAbort "Could not download \"${file}\", and not present in \"${CT_LOCAL_TARBALLS_DIR}\"" ${got_it} -eq 0

}

# Extract a tarball and patch the resulting sources if necessary.

# Some tarballs need to be extracted in specific places. Eg.: glibc addons

# must be extracted in the glibc directory; uCLibc locales must be extracted

# in the extra/locale sub-directory of uClibc.

CT_ExtractAndPatch() {

    local file="$1"

    local base_file=`echo "${file}" |cut -d - -f 1`

    local ver_file=`echo "${file}" |cut -d - -f 2-`

    local official_patch_dir

    local custom_patch_dir

    local libc_addon

    local ext=`CT_GetFileExtension "${file}"`

    CT_TestAndAbort "\"${file}\" not found in \"${CT_TARBALLS_DIR}\"" -z "${ext}"

    local full_file="${CT_TARBALLS_DIR}/${file}${ext}"

    CT_Pushd "${CT_SRC_DIR}"

    # Add-ons need a little love, really.

    case "${file}" in

        glibc-[a-z]*-*)

            CT_TestAndAbort "Trying to extract the C-library addon/locales \"${file}\" when C-library not yet extracted" ! -d "${CT_LIBC_FILE}"

            cd "${CT_LIBC_FILE}"

            libc_addon=y

            [ -f ".${file}.extracted" ] && return 0

            touch ".${file}.extracted"

            ;;

        uClibc-locale-*)

            CT_TestAndAbort "Trying to extract the C-library addon/locales \"${file}\" when C-library not yet extracted" ! -d "${CT_LIBC_FILE}"

            cd "${CT_LIBC_FILE}/extra/locale"

            libc_addon=y

            [ -f ".${file}.extracted" ] && return 0

            touch ".${file}.extracted"

            ;;

    esac

    # If the directory exists, then consider extraction and patching done

    if [ -d "${file}" ]; then

        CT_DoLog DEBUG "Already extracted \"${file}\""

        return 0

    fi

    CT_DoLog EXTRA "Extracting \"${file}\""

    case "${ext}" in

        .tar.bz2)     tar xvjf "${full_file}" |CT_DoLog ALL;;

        .tar.gz|.tgz) tar xvzf "${full_file}" |CT_DoLog ALL;;

        .tar)         tar xvf  "${full_file}" |CT_DoLog ALL;;

        *)            CT_Abort "Don't know how to handle \"${file}\": unknown extension" ;;

    esac

    # Snapshots might not have the version number in the extracted directory

    # name. This is also the case for some (old) packages, such as libfloat.

    # Overcome this issue by symlink'ing the directory.

    if [ ! -d "${file}" -a "${libc_addon}" != "y" ]; then

        case "${ext}" in

            .tar.bz2)     base=`tar tjf "${full_file}" |head -n 1 |cut -d / -f 1 || true`;;

            .tar.gz|.tgz) base=`tar tzf "${full_file}" |head -n 1 |cut -d / -f 1 || true`;;

            .tar)         base=`tar tf  "${full_file}" |head -n 1 |cut -d / -f 1 || true`;;

        esac

        CT_TestOrAbort "There was a problem when extracting \"${file}\"" -d "${base}" -o "${base}" != "${file}"

        ln -s "${base}" "${file}"

    fi

    # Kludge: outside this function, we wouldn't know if we had just extracted

    # a libc addon, or a plain package. Apply patches now.

    CT_DoLog EXTRA "Patching \"${file}\""

    # If libc addon, we're already in the correct place.

    [ -z "${libc_addon}" ] && cd "${file}"

    [ "${CUSTOM_PATCH_ONLY}" = "y" ] || official_patch_dir="${CT_TOP_DIR}/patches/${base_file}/${ver_file}"

    [ "${CT_CUSTOM_PATCH}" = "y" ] && custom_patch_dir="${CT_CUSTOM_PATCH_DIR}/${base_file}/${ver_file}"

    for patch_dir in "${official_patch_dir}" "${custom_patch_dir}"; do

        if [ -n "${patch_dir}" -a -d "${patch_dir}" ]; then

            for p in "${patch_dir}"/*.patch; do

                if [ -f "${p}" ]; then

                    CT_DoLog DEBUG "Applying patch \"${p}\""

                    patch -g0 -F1 -p1 -f <"${p}" |CT_DoLog ALL

                    CT_TestAndAbort "Failed while applying patch file \"${p}\"" ${PIPESTATUS[0]} -ne 0

                fi

            done

        fi

    done

    CT_Popd

}

# Compute the target triplet from what is provided by the user

# Usage: CT_DoBuildTargetTriplet

# In fact this function takes the environment variables to build the target

# triplet. It is needed both by the normal build sequence, as well as the

# sample saving sequence.

CT_DoBuildTargetTriplet() {

    case "${CT_ARCH_BE},${CT_ARCH_LE}" in

        y,) target_endian_eb=eb; target_endian_el=;;

        ,y) target_endian_eb=; target_endian_el=el;;

    esac

    case "${CT_ARCH}" in

        arm)  CT_TARGET="${CT_ARCH}${target_endian_eb}";;

        mips) CT_TARGET="${CT_ARCH}${target_endian_el}";;

        x86*) # Much love for this one :-(

              # Ultimately, we should use config.sub to output the correct

              # procesor name. Work for later...

              arch="${CT_ARCH_ARCH}"

              [ -z "${arch}" ] && arch="${CT_ARCH_TUNE}"

              case "${CT_ARCH}" in

                  x86_64)      CT_TARGET=x86_64;;

              	  *)  case "${arch}" in

                          "")                                       CT_TARGET=i386;;

                          i386|i486|i586|i686)                      CT_TARGET="${arch}";;

                          winchip*)                                 CT_TARGET=i486;;

                          pentium|pentium-mmx|c3*)                  CT_TARGET=i586;;

                          nocona|athlon*64|k8|athlon-fx|opteron)    CT_TARGET=x86_64;;

                          pentiumpro|pentium*|athlon*)              CT_TARGET=i686;;

                          *)                                        CT_TARGET=i586;;

                      esac;;

              esac;;

    esac

    case "${CT_TARGET_VENDOR}" in

        "") CT_TARGET="${CT_TARGET}-unknown";;

        *)  CT_TARGET="${CT_TARGET}-${CT_TARGET_VENDOR}";;

    esac

    case "${CT_KERNEL}" in

        linux*)  CT_TARGET="${CT_TARGET}-linux";;

        cygwin*) CT_TARGET="${CT_TARGET}-cygwin";;

    esac

    case "${CT_LIBC}" in

        glibc)  CT_TARGET="${CT_TARGET}-gnu";;

        uClibc) CT_TARGET="${CT_TARGET}-uclibc";;

    esac

    case "${CT_ARCH_ABI}" in

        eabi)   CT_TARGET="${CT_TARGET}eabi";;

    esac

    CT_TARGET="`${CT_TOP_DIR}/tools/config.sub ${CT_TARGET}`"

}