#!/bin/bash

# requires:
# https://aur.archlinux.org/packages/clean-chroot-manager
# https://aur.archlinux.org/packages/aurutils

basedir="$(pwd)"
src_dir="${basedir}/src"
repo_name=aur
pkg_dir="${basedir}/$repo_name"
ccm='sudo ccm64'
ccm_root='/scratch/chroot64'
aurweb='https://aur.archlinux.org'

if [ `whoami` == 'root' ]; then
   echo "Must not be run as root."
   exit 1;
fi

[ -e ppa.cfg ] && source ppa.cfg

clone_update_aur() {
    declare -r startdir=$PWD

    while IFS= read -r; do
        if [[ -d $REPLY ]]
        then

            if [[ ! -d $REPLY/.git-nosub ]]
            then
                rm -rf tmp
                git clone "$aurweb/$REPLY".git tmp
                mv tmp/.git "$REPLY"/.git-nosub
                rm -rf tmp
            fi

            cd "$REPLY"
            mv .git-nosub .git

            git fetch -v # returns 0 even if nothing was fetched
            git reset --hard HEAD
            git merge

            cd "$startdir"
        else
            git clone "$aurweb/$REPLY".git
        fi
        # .git folders make it a git submodule, we don't want that, need to gitignore .git-nosub
        mv "$REPLY"/.git "$REPLY"/.git-nosub
        # we don't ever want to track these
        rm -f "$REPLY"/.SRCINFO
    done
}

add() {
    cd "$src_dir"
    aurchain "$@" | clone_update_aur
}

update() {
    cd "$src_dir"
    if [ "$#" -gt 0 ]; then
        aurchain "$@" | clone_update_aur
    else
        aurchain * | clone_update_aur
    fi
}

sign_packages() {
    cd "$pkg_dir"
    for pkg in "$@"; do
        gpg -v "${pkg}.sig" 2>/dev/null || (rm -f "${pkg}.sig"; gpg --detach-sign --use-agent "$pkg")
    done
}

build() {
    cd "$src_dir"
    if [ "$#" -gt 0 ]; then
        $ccm d
        failed=''
        # create .SRCINFO, so aurqueue can use it next
        for dir in "$@"
        do
            cd $dir
            makepkg --printsrcinfo > .SRCINFO
            cd "$src_dir"
        done
        for dir in $(aurqueue "$@")
        do
            cd $dir
            $ccm s || failed="$failed $dir"
            cd "$src_dir"
        done

        # save packages
        cd "$pkg_dir"
        for pkg in "$ccm_root"/root/repo/*pkg.tar*
        do
            mv "$pkg" .
            sign_packages "$(basename "$pkg")"
        done
        repose -f -z $repo_name
        sign_packages $repo_name.db $repo_name.files

        if [ "$failed" == "" ]
        then
            # success!, nuke the chroot
            echo $ccm n
        else
            # some failures, just delete the packages
            $ccm d
            echo "failed packages:$failed"
        fi

    else
        build *
    fi
}

rebuild_db() {
    cd "$pkg_dir"
    repose -f -z $repo_name
    sign_packages $repo_name.db $repo_name.files
}

echo_deps() {
    cd "$src_dir"
    aurchain "$@"
}

build_deps() {
    cd "$src_dir"
    if [ "$#" -gt 0 ]; then
        build $(aurchain "$@")
    else
        # regular build? if we are building all no use looking up deps...
        build
    fi
}

build_changed() {
    commit="$1"
    [ "$commit" == "" ] && commit=HEAD
    build $(git diff-tree --no-commit-id --name-status -r "$commit" | grep -v ^D | grep PKGBUILD$ | sed -e 's@/PKGBUILD$@@' -e 's@.*/@@' | tr '\n' ' ')
}

build_changed_deps() {
    commit="$1"
    [ "$commit" == "" ] && commit=HEAD
    build_deps $(git diff-tree --no-commit-id --name-status -r "$commit" | grep -v ^D | grep PKGBUILD$ | sed -e 's@/PKGBUILD$@@' -e 's@.*/@@' | tr '\n' ' ')
}

echo_changed() {
    commit="$1"
    [ "$commit" == "" ] && commit=HEAD
    git diff-tree --no-commit-id --name-status -r "$commit" | grep -v ^D | grep PKGBUILD$ | sed -e 's@/PKGBUILD$@@' -e 's@.*/@@' | tr '\n' ' '
}

list_official_packages() {
    cd "$src_dir"
    ls | xargs -n1 -I {} pacman -Ss '^{}$' | egrep -v "^($repo_name| )"
}

reinstall() {
    paclist $repo_name | cut -d' ' -f1 | xargs sudo pacman -S --noconfirm
    comm -12 <(pacman -Qq | sort) <(pacman -Slq $repo_name | sort) | xargs sudo pacman -S --noconfirm
}

clean() {
    # git clean -e 'aur/*' -dxn | grep -v .git
    git clean -e '*.git*' -e 'aur/*' -dxf
}

undo() {
    patch -p1 -R < undo.diff
}

grep_deps() {
    grep "$1" src/*/PKGBUILD
}

if [ "$#" -gt 0 ]; then
    [ -d "$src_dir" ] || mkdir -p "$src_dir"
    [ -d "$pkg_dir" ] || mkdir -p "$pkg_dir"
    $*
else
    echo "Must specify a command, eg: add, build, list, clean"
    exit 1
fi