#!/usr/bin/env bash

#
#  Darkcloud Vim Config Update Tool
#  https://github.com/prurigro/darkcloud-vimconfig
#
#  Written by Kevin MacMartin (prurigro@gmail.com)
#  Released under the MIT license
#

SCRIPT_NAME="${0//*\/}"
SCRIPT_HOME="${0%/*}"
VERSION="$(printf "%s.r%s" "$(git show -s --format=%ci master | sed 's/\ .*//g;s/-//g')" "$(git rev-list --count HEAD)")"
ERRORLOG='update-errors.log'

[[ -t 1 ]] && {
    C_BGBLUE='\e[44m'
    C_BGRED_BOLD='\e[1;41m'
    C_BGBLACK='\e[40m'
    C_BGYELLOW='\e[43m'
    C_FGGREEN_BOLD='\e[1;32m'
    C_FGRED_BOLD='\e[1;31m'
    C_FGWHITE_BOLD='\e[1;37m'
    C_RESET='\e[0m'
} || \
    C_BGBLUE='#'

# error: output and log error
function error() {
    echo -e "$C_BGBLUE $C_RESET$C_BGRED_BOLD ! ERROR: $C_RESET$C_FGRED_BOLD $2 "
    echo -e "$C_BGBLUE $C_RESET$C_BGRED_BOLD ! COMMAND: $C_RESET ${C_FGWHITE_BOLD}=> $1$C_RESET"
    [[ -n "$3" ]] \
        && echo -e "$C_BGBLUE $C_RESET$C_BGRED_BOLD ! OUTPUT: $C_RESET$C_FGWHITE_BOLD $3"

    echo "DATE: @ $(date)" >> "$SCRIPT_HOME/$ERRORLOG"
    echo "ERROR: $2" >> "$SCRIPT_HOME/$ERRORLOG"
    echo "COMMAND: $1" >> "$SCRIPT_HOME/$ERRORLOG"
    [[ -n "$3" ]] \
        && echo "OUTPUT: $3" >> "$SCRIPT_HOME/$ERRORLOG"

    echo >> "$SCRIPT_HOME/$ERRORLOG"
}

# show_version: displays version information
function show_version() {
    echo -e "$SCRIPT_NAME: darkcloud-vimconfig update tool (version: $VERSION)"
}

# show_help: this function displays help output
function show_help() {
    echo -e "\nUSAGE"
    echo -e "  ./$SCRIPT_NAME [OPTION]\n"
    echo "OPTIONS"
    echo -e "  -v, --version\t| output version information and exit"
    echo -e "  -h, --help\t| display this help and exit\n"
    echo -e "Run with no arguments to update darkcloud-vimconfig"
}

### SETUP
cd "$SCRIPT_HOME"

# delete old error log if it exists
[[ -f "$ERRORLOG" ]] \
    && rm "$SCRIPT_HOME/$ERRORLOG"

# parse for arguments (then handle them below)
[[ -n "$1" ]] && {
    case "$1" in
        -h|--help)
            show_version
            show_help
            exit 0
            ;;
        -v|--version)
            show_version
            exit 0
            ;;
        *)
            error "$ERROR" "$1 is not a valid option"
            show_help
            exit 1
            ;;
    esac
}

# display script title
echo -e "\n$C_BGBLACK ~~~ DarkCloud Vimconfig Update Tool ~~~ $C_RESET"

# create vim/bundle.user if it doesn't exist
[[ ! -d 'vim/bundle.user' ]] && {
    echo -ne "\n$C_BGBLUE >> Creating User Plugin Directory:$C_RESET"
    PROCESS_STATUS="$(install -d 'vim/bundle.user' 2>&1)"
    [[ $? = 0 ]] && {
        echo -e "$C_FGGREEN_BOLD SUCCESS! $C_RESET"
    } || {
        echo -e "$C_FGRED_BOLD FAIL! $C_RESET"
        error "install -d vim/bundle.user" "User plugin directory couldn't be created" "$PROCESS_STATUS"
    }
}

# create vim/vimrc.user if it doesn't exist
[[ ! -e 'vim/vimrc.user' ]] && {
    echo -ne "\n$C_BGBLUE >> Creating User Config File:$C_RESET"
    PROCESS_STATUS="$(touch 'vim/vimrc.user' 2>&1)"
    [[ $? = 0 ]] && {
        echo -e '"Autostart Filer in empty buffers: (*1:filer loads in new empty buffers | 0:filer must be triggered)\n"let g:autostartfiler=1\n' >> vim/vimrc.user
        echo -e '"Autocheck syntax checking: (1:start toggled on | *0:start toggled off)\n"let g:autostartchecker=0\n' >> vim/vimrc.user
        echo -e '"Autoload Tagbar: (1:start open | *0:start closed)\n"let g:autostarttagbar=0\n' >> vim/vimrc.user
        echo -e '"Disable automatic tag generation and highlighting: (1:force disabled tag generation and highlighting | *0:enable automatic tag generation and highlighting)\n"let g:disableautotags=0\n' >> vim/vimrc.user
        echo -e '"Disable automatic linebreaking: (1:force disabled globally | *0:let the filetype decide)\n"let g:disablelinebreaks=0\n' >> vim/vimrc.user
        echo -e '"Enable Powerline fonts: (1:expect powerline font | *0:expect regular font)\n"let g:powerlinefonts=0 "(set powerline font for gvim and terminal when enabled)\n' >> vim/vimrc.user
        echo -e '"GVim font selection: (Escaping spaces and use powerline if appropriate)\nset guifont=Monospace\ 12' >> vim/vimrc.user
        echo -e "$C_FGGREEN_BOLD SUCCESS! $C_RESET"
    } || {
        echo -e "$C_FGRED_BOLD FAIL! $C_RESET"
        error "touch vim/vimrc.user" "User config couldn't be created" "$PROCESS_STATUS"
    }
}

### REPO UPDATE
echo -ne "\n$C_BGBLUE >> Updating Repository:$C_RESET"
PROCESS_STATUS="$(git pull origin master 2>&1)"
[[ $? = 0 ]] && {
    echo -e "$C_FGGREEN_BOLD SUCCESS! $C_RESET"
} || {
    echo -e "$C_FGRED_BOLD FAIL! $C_RESET"
    error "git pull origin master" "Git failed to sync the repo" "$PROCESS_STATUS"
    exit 1
}

### SUBMODULE UPDATE {
echo -e "\n$C_BGBLUE >> Updating Plugin Submodules >> $C_RESET"
echo -ne "$C_BGBLUE $C_RESET$C_BGYELLOW + Updating Plugin URLs:$C_RESET"
PROCESS_STATUS="$(git submodule sync 2>&1)"
[[ $? = 0 ]] && {
    echo -e "$C_FGGREEN_BOLD SUCCESS! $C_RESET"
} || {
    echo -e "$C_FGRED_BOLD FAIL! $C_RESET"
    error "git submodule sync" "Git failed to sync the submodules" "$PROCESS_STATUS"
}

# update each submodule to the new head and run 'git fetch --all'
echo -ne "$C_BGBLUE $C_RESET$C_BGYELLOW + Fetching Updates:$C_RESET"
PROCESS_STATUS="$(git submodule foreach git fetch --all 2>&1)"
[[ $? = 0 ]] && {
    PROCESS_STATUS=$(git submodule update --init --recursive 2>&1)
    [[ $? = 0 ]] && {
        echo -e "$C_FGGREEN_BOLD SUCCESS! $C_RESET"
    } || {
        echo -e "$C_FGRED_BOLD FAIL! $C_RESET"
        error "git submodule update --init --recursive" "Git failed to update the submodules" "$PROCESS_STATUS"
    }
} || {
    echo -e "$C_FGRED_BOLD FAIL! $C_RESET"
    error "git submodule foreach git fetch --all" "Git failed to fetch the submodules from their respective remotes" "$PROCESS_STATUS"
}

# run 'git checkout origin/master' on each submodule
echo -ne "$C_BGBLUE $C_RESET$C_BGYELLOW + Checking Out Updates:$C_RESET"
PROCESS_STATUS="$(git submodule foreach git checkout -f origin/master 2>&1)"
[[ $? = 0 ]] && {
    echo -e "$C_FGGREEN_BOLD SUCCESS! $C_RESET"
} || {
    echo -e "$C_FGRED_BOLD FAIL! $C_RESET"
    error "git submodule foreach git checkout -f origin/master" "Git failed to checkout the submodules into origin/master" "$PROCESS_STATUS"
}

# clean plugin directories and remove plugins no longer in the repo
echo -e "\n$C_BGBLUE >> Cleaning Plugin Directories >> $C_RESET"
echo -ne "$C_BGBLUE $C_RESET$C_BGYELLOW + Removing Untracked Files:$C_RESET"

git submodule foreach git clean -dxf >/dev/null \
    && echo -e "$C_FGGREEN_BOLD SUCCESS! $C_RESET" \
    || echo -e "$C_FGRED_BOLD FAIL! $C_RESET"

[[ -f '.gitmodules' ]] && {
    for plugin in vim/bundle/*; do
        [[ -f "$plugin/.git" ]] && {
            FILE=$(echo $plugin | grep -o -e "[^\/]*$")
            [[ $(cat .gitmodules | grep "path = " |  grep -o -e "[^\/]*$" | grep -c -e "${FILE}$") = 0 ]] && {
                [[ -z "$FIRST_OLD" ]] && export FIRST_OLD=1 && echo -ne "$C_BGBLUE $C_RESET$C_BGYELLOW + Removing Old Plugins: $C_RESET\n"
                PROCESS_STATUS=$(rm -rf "$plugin")
                [[ $? = 0 ]] && {
                    echo -e "$C_BGBLUE $C_RESET$C_BGYELLOW = $C_RESET$(echo $plugin\  | sed -re 's|^(.*)/([^/]*)$|\\e\[1;37m\1/\\e\[1;31m\2\\e\[0m|')"
                } || {
                    error "rm -rf $plugin" "Folder couldn't be deleted" "$PROCESS_STATUS"
                    exit 1
                }
            }
        }
    done
    echo
}

### USER PLUGIN UPDATE
[[ -d vim/bundle.user ]] && [[ $(find vim/bundle.user -mindepth 3 -maxdepth 3 -name config | grep '.git/config') ]] && {
    echo -e "$C_BGBLUE >> Updating User Plugins >> $C_RESET"
    pushd vim/bundle.user >/dev/null
    for plugin in *; do
        [[ -d "$plugin/.git" ]] && {
            pushd "$plugin" >/dev/null
            echo -ne "$C_BGBLUE $C_RESET$C_BGYELLOW + Updating 'vim/bundle.user/$plugin':$C_RESET"
            PROCESS_STATUS=$(git pull origin master 2>&1)
            [[ $? = 0 ]] && {
                [[ $(echo $PROCESS_STATUS | grep -c "Already up-to-date") = 0 ]] \
                    && echo -e "$C_FGGREEN_BOLD SUCCESS! $C_RESET" \
                    || echo -e "$C_FGWHITE_BOLD UP TO DATE $C_RESET"
            } || error "git pull origin master" "Failed pulling changes for $plugin" "$PROCESS_STATUS"
            popd >/dev/null
        }
    done
    popd >/dev/null
    echo
}

### GENERATE PLUGIN HELP
[[ $(type -P vim) ]] && {
    echo -ne "$C_BGBLUE >> Generating Plugin Help:$C_RESET"
    timeout --foreground 2m vim -u "$SCRIPT_HOME/vimrc" -c "Helptags|qa!" &> /dev/null
    [[ $? = 0 ]] && {
        echo -e "$C_FGGREEN_BOLD SUCCESS! $C_RESET"
    } || {
        reset -I
        echo -e "$C_FGRED_BOLD FAIL! $C_RESET"
        error "vim -u \"$SCRIPT_HOME/vimrc\" -c \"Helptags|qa!\"" "Generating helpdocs for the submodules failed"
    }
}

echo -e "\n$C_BGBLACK ~~~ Update Complete ~~~ $C_RESET\n"