#!/bin/bash
# set -x # uncomment for debug
# Builds packages from ABS recursively. It tries to find dependencies that
# aren't built or need update and then makepkg them in order.

# TODO move __build to chroot

source /etc/makepkg.conf
source /etc/abs.conf
source /etc/libretools.conf

if [ -z $XDG_CONFIG_HOME ]; then # Avoid /libretools dir doesn't exist errors

    error "There's no XDG_CONFIG_HOME var set"; exit 1

elif [ -e $XDG_CONFIG_HOME/libretools/libretools.conf ]; then

    source $XDG_CONFIG_HOME/libretools/libretools.conf

fi


usage() {

    echo "cd to a dir containing a PKGBUILD and run:"
    echo "$0 [options]"
    printf "This script will check dependencies, build them if possible "
    printf "and stage the packages on it's repo."
    echo
    echo "OPTIONS:"
    echo " -h           : this message."
    echo " -a absdir    : set absdir as ABSROOT."
    echo " -b build_dir : use a fullpkg build_dir and only build."
    echo " -c           : check deps only, do not build."
    echo " -d build_dir : use this dir to build. Defaults to mktemp."
    echo " -n           : don't update pacman db."
    echo " -m max_level : check deps until this level"
    echo " -r \"command\" : use this instead of \"$FULLBUILDCMD\""
    echo
    exit 1

}

# Removes a package from the buildorder
# $1 package name
# $2 buildorder file
remove_buildorder() {
  grep -Evw "${1}" ${2} > ${2}2
  mv -f ${2}2 ${2}

  return $?
}

# Get repo name. Asumes ${ABSROOT}/repo/package/PKGBUILD
guess_repo() {
    basename $(dirname $(pwd))
}

# return : full version spec, including epoch (if necessary), pkgver, pkgrel
#  usage : get_fullver( ${epoch:-0}, $pkgver, $pkgrel )
get_fullver() {
    if [[ $1 -eq 0 ]]; then
# zero epoch case, don't include it in version
        echo $2-$3
    else
        echo $1:$2-$3
    fi

}

# Cleans the build_dir.
cleanup() {
# Do nothing or already cleaned.
    [[ "${do_cleanup}" = false || ! -d ${build_dir} ]] && return 0

# Only do cleanup on level 0.
    msg "Cleaning up..."
    [ $level -eq 0 ] && rm -rf $build_dir/

}

# Checks ABSROOT and look for target pkg deps. Adds them if not built or outdated.
find_deps() {
# Check this level
    source PKGBUILD

    local repo=${repo:-$(guess_repo)}
    local pkgbase=${pkgbase:-${pkgname[0]}}
    local epoch=${epoch:-0}
    local fullver=$(get_fullver ${epoch} ${pkgver} ${pkgrel})

    if is_built "${pkgbase}>=${fullver}"; then # pkg is built and updated
        exit 0
    fi

    echo "${level}:${pkgbase}" >> "${build_dir}/BUILDORDER" # greater levels are built first

    if [ -d "${build_dir}/${pkgbase}" ]; then # PKGBUILD is already there
        exit 0
    else # Copy dir to build_dir
        cp -r ../${pkgbase}/ ${build_dir}/
        echo "repo=$repo" > "${build_dir}/${pkgbase}/.INFO" # to identify repo later
    fi

    msg2 "%${level}s${pkgbase}-${fullver}" # current package plus a space for every level

    declare -i next_level=$level+1 ## Check next levels

    deps=$(echo "${depends[@]} ${makedepends[@]}" | \
           sed "s/[=<>]\+[^ ]\+//g" | \
           tr ' ' "\n" | \
           sort -u) # All deps in separate line, only once, without version.

    for _dep in ${deps[@]}; do

        local found=false

        for _repo in ${REPOS[@]}; do # TODO ask toru where the pkgbuild is

            if [ -e "${ABSROOT}/${_repo}/${_dep}/PKGBUILD" ]; then # ABSROOT/repo/package

                pushd "${ABSROOT}/${_repo}/${_dep}" > /dev/null
                $0 -c -d ${build_dir} -l ${next_level} # run this cmd on dep's PKGBUILD dir
                [ $? -eq 20 ] && return 20 # probable circular deps
                popd > /dev/null
                local found=true
                break 1 # found, end cycle
            fi

        done

        if ${found}; then # go to next dep
            continue 1
        else
            echo "dep_not_found:$_dep" >> $build_dir/log
        fi

    done

## End variable block

    unset next_level dir
# unset PKGBUILD variables
    unset pkgname pkgver pkgrel epoch pkgdesc arch url license groups depends \
          makedepens checkdepends optdepends provides conflicts replaces backup \
          options install changelog source noextract md5sums build check package
}

__build() {
    pushd ${build_dir}  > /dev/null

    build_packages=($(sort -gr $buildorder | cut -d: -f2)) # greater levels must be built first

    while [ ${#build_packages[@]} -ge 1 ]; do
        pushd $build_dir/${build_packages[0]} > /dev/null
        source PKGBUILD

        msg2 "${pkgbase:-${pkgname[0]}} $pkgver-$pkgrel"

        msg2 "Checking for non free deps"
        pkgbuild-check-nonfree || {

            if [ $? -eq 15 ]; then # this error means nonfree others means fail.

              echo "nonfree:$(basename $PWD)" >> $build_dir/log

              remove_buildorder "$(basename $PWD)" $buildorder # take out package from $buildorder

              continue # build next package
            fi
        }

        msg2 "Building $(basename $PWD)"

        $FULLBUILDCMD; r=$? # this buildcmd is on libretools.conf
        case $r in

            0) ## Succesfull build
                plain "The build was succesful."
                if source .INFO && [ -n $repo ]; then

                    if [ ! -z $HOOKLOCALRELEASE ]; then # Calls a local release script
                        find -name "*.pkg.tar.?z" -print0 | xargs -0 $HOOKLOCALRELEASE $repo
                    fi

                    librestage $repo || echo "unstaged:$(basename $PWD)" >> $build_dir/log

                    msg "Updating pacman db and packages"
                    sudo pacman -Sy || true

                fi

                echo "built:$(basename $PWD)" >> $build_dir/log
            ;;

            *) ## Build failed
                error "There were errors while trying to build the package."
                echo "failed:$(basename $PWD)" >> $build_dir/log
            ;;
        esac

        remove_buildorder "${build_packages[0]}" $buildorder || true

        build_packages=($(sort -gr $buildorder | cut -d: -f2)) # find out next package
        popd > /dev/null
    done

    pkgs=($(grep "nonfree:" $build_dir/log)) && {
        error "Those packages contain nonfree deps:"
        echo ${pkgs[@]} | tr " " "\n" | cut -d: -f2
    }

    pkgs=($(grep "built:" $build_dir/log)) && {
        msg "Those packages were built and staged:"
        echo ${pkgs[@]} | tr " " "\n" | cut -d: -f2
    }

    pkgs=($(grep "failed:" $build_dir/log)) && {
        error "Those packages failed to build:"
        echo ${pkgs[@]} | tr " " "\n" | cut -d: -f2
    }

    pkgs=($(grep "unstaged:" $build_dir/log)) && {
        error "Those packages couldn't be staged because of missing reponame:"
        echo ${pkgs[@]} | tr " " "\n" | cut -d: -f2
    }

    popd > /dev/null
}

# End inmediately but print a useful message
trap_exit() {

  error "$@"
  warning "Leftover files left on $build_dir"

  exit 1
}

# Trap signals from makepkg
set -E
trap 'cleanup' 0
trap 'trap_exit "(prfullpkg:${level}) TERM signal caught. Exiting..."' TERM HUP QUIT
trap 'trap_exit "(prfullpkg:${level}) Aborted by user! Exiting..."' INT
trap 'trap_exit "(prfullpkg:${level}) An unknown error has occurred. Exiting..."' ERR

ban_file=$XDG_CONFIG_HOME/libretools/ban
force_build=""
level=0
noupdate=false
build_only=false
check_deps_only=false
do_cleanup=false
max_level=21

while getopts 'ha:b:cCd:l:nm:r:' arg; do
    case $arg in
        h) usage ;;
        a) ABSROOT="$OPTARG" ;;
        b) build_only=true
           build_dir="$OPTARG"
           if [ -z ${build_dir} ]; then
               usage
           fi
           if [ ! -r ${build_dir}/BUILDORDER ] ; then
               error "${build_dir}/BUILDORDER doesn't exist."
               exit 1
           fi;;
        c) check_deps_only=true ;;
        C) do_cleanup=true;;
        d) build_dir="$OPTARG" ;;
# hidden option to know dep level.
        l) level=$OPTARG ;;
        n) noupdate=true;;
        m) max_level=$OPTARG ;;
        r) FULLBUILDCMD="$OPTARG" ;;
    esac
done

if ! (( ${build_only} )); then

    if [ ! -r PKGBUILD ]; then # Check if we are actually on a build directory. Do this early.
        error "This isn't a build directory"
        usage
    fi

    if [ ! -z "$HOOKPKGBUILDMOD" ]; then
        "$HOOKPKGBUILDMOD"
    fi

fi

if [ $level -eq 0 ]; then

    build_dir=${build_dir:-$(mktemp -d /tmp/fullpkg.XXXXXX)} # use -d option or else mktemp

    if [ ! -d ${build_dir} ]; then # in case of custom -d option
        mkdir -p ${build_dir}
    else
        cleanup # files already there can screw find_deps
    fi

    touch ${build_dir}/{log,BUILDORDER} ${ban_file} # make files for log and buildorder
    buildorder=${build_dir}/BUILDORDER

    if ! (( noupdate )); then

        msg "Updating pacman db and packages"
        sudo pacman -Syu --noconfirm || true

    fi

    if (( build_only )); then

        msg "Build Packages"

        __build

        exit 0

    fi

    msg "Checking dependencies"
fi

# Probable circular deps
[ $level -ge $max_level ] && exit 20

find_deps || {

# Probable circular deps
    if [ $? -eq 20 ]; then

# Show error only on level 0
       if [ $level -eq 0 ]; then
           error "Check for circular deps on $build_dir/BUILDORDER";
       fi

    fi
# Pass message 20
    exit 20
}

# only build on level 0
$( (( check_deps_only )) || [ $level -gt 0 ] ) && exit 0

msg "Building packages:"

# Build the packages
__build

echo
msg2 "Check if your system works fine and librerelease if it does"

exit 0