#! /bin/sh

function help() {
    echo "${0##*/} [drbs] [push|pull] [remote]"
    echo -e "-d --git-dir       set GIT_DIR"
    echo -e "-r --git-remote    set GIT_REMOTE"
    echo -e "-b --git-branch    set GIT_BRANCH"
    echo -e "-s --use-stash     set USE_STASH"
    echo -e "-c --clean         set USE_GC"
    exit 0
}

for arg in $@; do
    option=`echo "'$arg'" | cut -d'=' -f1 | tr -d "'"`
    value=`echo "'$arg'" | cut -d'=' -f2- | tr -d "'"`
    if [ "$value" == "$option" ]; then value=""; fi
    case "$option" in
        pull) action="pull";;
        push) action="push";;
        -d|--git-dir) GIT_DIR=$value;;
        -r|--git-remote) GIT_REMOTE=$value;;
        -b|--git-branch) GIT_BRANCH=$value;;
        -s|--use-stash) USE_STASH=1;;
        -h|--help) help;;
        *) GIT_REMOTE=$arg;;
    esac
done

GIT_DIR=${GIT_DIR:-$HOME/usr/git}
GIT_REMOTE=${GIT_REMOTE:-origin}
GIT_BRANCH=${GIT_BRANCH:-master}
USE_STASH=${USE_STASH:-0}
STASH=0
USE_GC=${USE_GC:-0}

RESET="\033[0m"
RED="\033[0;31m"
GREEN="\033[0;32m"
BROWN="\033[0;33m"
PURPLE="\033[0;35m"
CYAN="\033[0;36m"
YELLOW="\033[1;33m"

function success() {
    echo -e "$1${GREEN}OK${RESET}"
}

function fail() {
    echo -e "${RED}$1 ${RESET}${CYAN}leave${RESET}\n" && cd ..
}

function search_remote() {
    echo -ne "    ${CYAN}search remote${RESET}: ${YELLOW}${GIT_REMOTE}${RESET} ... " && \
        git remote show ${GIT_REMOTE} >/dev/null 2>&1 && success && return 0
    [ $? -ne 0 ] && fail "remote not found" && return 1
}

function checkout_branch() {
    echo -ne "    ${CYAN}checkout branch ${YELLOW}${GIT_BRANCH}${RESET} ... " && \
        git branch | grep -q -e " ${GIT_BRANCH}\$" && git checkout -q ${GIT_BRANCH} 2>/dev/null && success && return 0
    [ $? -ne 0 ] && fail "KO" && return 1
}

function gc() {
    [ $USE_GC -eq 0 ] && return 0
    echo -ne "    ${CYAN}clean$RESET ... " && git gc --prune --quiet && git repack >/dev/null && git diff --quiet && success
    [ $? -eq 0 ] && return 0
    if [ "$action" = "push" ]; then
        success "dirty but "
    elif [ $USE_STASH -eq 1 ]; then
        STASH=1
        echo -e "${RED}head is not clean, ${CYAN}git stash save${RESET}" && git stash save -q
    else
        fail "head is not clean, use -s flag to use stash ... " && return 1
    fi
    return 0
}

function pull() {
    [ "$action" = "push" ] && return 0
    echo -ne "    ${CYAN}pull${RESET}: ${YELLOW}${GIT_REMOTE}${RESET}:${PURPLE}${GIT_BRANCH}${RESET} ... " && \
        git pull -q ${GIT_REMOTE} ${GIT_BRANCH} && success && return 0
    [ $? -ne 0 ] && fail "KO" && return 1
}

function push() {
    [ "${GIT_REMOTE}" = "origin" -o "$action" = "pull" ] && return 0
    echo -ne "    ${CYAN}push${RESET}: ${YELLOW}${GIT_REMOTE}${RESET}:${PURPLE}${GIT_BRANCH}${RESET} ... " && \
        git push ${GIT_REMOTE} ${GIT_BRANCH} 2>/dev/null && success
    [ $? -ne 0 ] && fail "KO" && return 1
    [ $STASH -eq 1 ] && echo -e "    ${RED}was not clean, ${CYAN}git stash pop${RESET}" && git stash pop -q
    # TODO check success
}

echo -e "remote:${YELLOW}${GIT_REMOTE}${RESET}:${PURPLE}${GIT_BRANCH}${RESET} USE_STASH=${RED}${USE_STASH}${RESET}"
for gitdir in ${GIT_DIR}; do
    echo -e "enter ${BROWN}${gitdir}${RESET}" && cd ${gitdir} || return 1
    for dir in $(ls -1); do
        STASH=0
        [ -d ${dir} ] && [ -e ${dir}/.git/config ] || continue
        echo -e "  enter ${BROWN}${dir}${RESET}" && cd $dir || exit 1
        search_remote || continue
        checkout_branch || continue
        gc || continue
        pull || continue
        push || continue
        echo -e "  leave ${BROWN}${dir}${RESET}\n" && cd ..
    done
    echo -e "leave ${BROWN}${gitdir}${RESET}"
done