#!/usr/bin/env bash

#=============================================================================
# install.sh --- bootstrap script for SpaceVim
# Copyright (c) 2016-2022 Shidong Wang & Contributors
# Author: Shidong Wang < wsdjeg@outlook.com >
# URL: https://spacevim.org
# License: GPLv3
#=============================================================================

# Init option {{{
Color_off='\033[0m'       # Text Reset

# terminal color template {{{
# Regular Colors
Black='\033[0;30m'        # Black
Red='\033[0;31m'          # Red
Green='\033[0;32m'        # Green
Yellow='\033[0;33m'       # Yellow
Blue='\033[0;34m'         # Blue
Purple='\033[0;35m'       # Purple
Cyan='\033[0;36m'         # Cyan
White='\033[0;37m'        # White

# Bold
BBlack='\033[1;30m'       # Black
BRed='\033[1;31m'         # Red
BGreen='\033[1;32m'       # Green
BYellow='\033[1;33m'      # Yellow
BBlue='\033[1;34m'        # Blue
BPurple='\033[1;35m'      # Purple
BCyan='\033[1;36m'        # Cyan
BWhite='\033[1;37m'       # White

# Underline
UBlack='\033[4;30m'       # Black
URed='\033[4;31m'         # Red
UGreen='\033[4;32m'       # Green
UYellow='\033[4;33m'      # Yellow
UBlue='\033[4;34m'        # Blue
UPurple='\033[4;35m'      # Purple
UCyan='\033[4;36m'        # Cyan
UWhite='\033[4;37m'       # White

# Background
On_Black='\033[40m'       # Black
On_Red='\033[41m'         # Red
On_Green='\033[42m'       # Green
On_Yellow='\033[43m'      # Yellow
On_Blue='\033[44m'        # Blue
On_Purple='\033[45m'      # Purple
On_Cyan='\033[46m'        # Cyan
On_White='\033[47m'       # White

# High Intensity
IBlack='\033[0;90m'       # Black
IRed='\033[0;91m'         # Red
IGreen='\033[0;92m'       # Green
IYellow='\033[0;93m'      # Yellow
IBlue='\033[0;94m'        # Blue
IPurple='\033[0;95m'      # Purple
ICyan='\033[0;96m'        # Cyan
IWhite='\033[0;97m'       # White

# Bold High Intensity
BIBlack='\033[1;90m'      # Black
BIRed='\033[1;91m'        # Red
BIGreen='\033[1;92m'      # Green
BIYellow='\033[1;93m'     # Yellow
BIBlue='\033[1;94m'       # Blue
BIPurple='\033[1;95m'     # Purple
BICyan='\033[1;96m'       # Cyan
BIWhite='\033[1;97m'      # White

# High Intensity backgrounds
On_IBlack='\033[0;100m'   # Black
On_IRed='\033[0;101m'     # Red
On_IGreen='\033[0;102m'   # Green
On_IYellow='\033[0;103m'  # Yellow
On_IBlue='\033[0;104m'    # Blue
On_IPurple='\033[0;105m'  # Purple
On_ICyan='\033[0;106m'    # Cyan
On_IWhite='\033[0;107m'   # White
# }}}

# version
Version='2.5.0-dev'
#System name
System="$(uname -s)"
# }}}

XDGSpaceDir="${XDG_CONFIG_HOME:-${HOME}/.}${XDG_CONFIG_HOME:+/}SpaceVim"
XDGvimDir="${XDG_CONFIG_HOME:-${HOME}/.}${XDG_CONFIG_HOME:+/}vim"
XDGnvimDir="${XDG_CONFIG_HOME:-${HOME}/.config/}${XDG_CONFIG_HOME:+/}nvim"

# need_cmd {{{
need_cmd () {
    if ! hash "$1" &>/dev/null; then
        error "Need '$1' (command not found)"
        exit 1
    fi
}
# }}}

# success/info/error/warn {{{
msg() {
    printf '%b\n' "$1" >&2
}

success() {
    msg "${Green}[✔]${Color_off} ${1}${2}"
}

info() {
    msg "${Blue}[➭]${Color_off} ${1}${2}"
}

error() {
    msg "${Red}[✘]${Color_off} ${1}${2}"
    exit 1
}

warn () {
    msg "${Yellow}[⚠]${Color_off} ${1}${2}"
}
# }}}

# echo_with_color {{{
echo_with_color () {
    printf '%b\n' "$1$2$Color_off" >&2
}
# }}}

# fetch_repo {{{
fetch_repo () {
    need_cmd 'git'
    if [[ -d "${XDGSpaceDir:-}" ]]; then
        info "Trying to update SpaceVim"
        (
            cd "${XDGSpaceDir:?}"
            git pull
        )
        success "Successfully update SpaceVim"
    else
        info "Trying to clone SpaceVim"
        git clone --depth 1 https://github.com/SpaceVim/SpaceVim.git "${XDGSpaceDir:-}"
        if [ $? -eq 0 ]; then
            success "Successfully clone SpaceVim"
        else
            error "Failed to clone SpaceVim"
            exit 0
        fi
    fi
}
# }}}

# install_vim {{{
install_vim () {
    if [[ -f "$HOME/.vimrc" ]]; then
        mv "$HOME/.vimrc" "$HOME/.vimrc_back"
        success "Backup $HOME/.vimrc to $HOME/.vimrc_back"
    fi

    if [[ -d "${XDGvimDir:-}" ]]; then
        if [[ "$(readlink "${XDGvimDir:-}")" =~ \.?SpaceVim$ ]]; then
            success "SpaceVim already installed in '${XDGvimDir:-}'"
            return
        fi

        mv "${XDGvimDir:?}" "${XDGvimDir:-}_back"
        success "BackUp '${XDGvimDir}' to '${XDGvimDir}_back'"
    fi

    ln -s "${XDGSpaceDir:?}" "${XDGvimDir:?}"
    success "Installed SpaceVim for vim in '${XDGvimDir}'"
}
# }}}

# install_neovim {{{
install_neovim () {
    if [[ -d "${XDGnvimDir:-}" ]]; then
        if [[ "$(readlink "${XDGnvimDir:-}")" =~ \.?SpaceVim$ ]]; then
            success "SpaceVim already installed in '${XDGnvimDir:-}'"
            return
        fi

        mv "${XDGnvimDir:?}" "${XDGnvimDir:-}_back"
        success "BackUp '${XDGnvimDir}' to '${XDGnvimDir}_back'"
    fi

    if [[ ! -d "$(dirname "${XDGnvimDir}")" ]]; then
        mkdir "$(dirname "${XDGnvimDir}")"
    fi

    ln -s "${XDGSpaceDir:?}" "${XDGnvimDir:?}"
    success "Installed SpaceVim for nvim in '${XDGnvimDir}'"
}
# }}}

# uninstall_vim {{{
uninstall_vim () {
    if [[ -d "${XDGvimDir:-}" ]] &&
       [[ "$(readlink "${XDGvimDir:?}")" =~ \.?SpaceVim$ ]]; then
        rm "${XDGvimDir:?}"
        success "Uninstall SpaceVim for vim"
        if [[ -d "${XDGvimDir}_back" ]]; then
            mv "${XDGvimDir}_back" "${XDGvimDir}"
            success "Recover from ${XDGvimDir}_back"
        fi
    fi
    if [[ -f "$HOME/.vimrc_back" ]]; then
        mv "$HOME/.vimrc_back" "$HOME/.vimrc"
        success "Recover from $HOME/.vimrc_back"
    fi
}
# }}}

# uninstall_neovim {{{
uninstall_neovim () {
    if [[ -d "${XDGnvimDir:-}" ]] &&
       [[ "$(readlink "${XDGnvimDir:?}")" =~ \.?SpaceVim$ ]]; then
        rm "${XDGnvimDir:?}"
        success "Uninstall SpaceVim for neovim"
        if [[ -d "${XDGnvimDir}_back" ]]; then
            mv "${XDGnvimDir}_back" "${XDGnvimDir}"
            success "Recover from ${XDGnvimDir}_back"
        fi
    fi
}
# }}}

# check_requirements {{{
check_requirements () {
    info "Checking Requirements for SpaceVim"
    if hash "git" &>/dev/null; then
        git_version=$(git --version)
        success "Check Requirements: ${git_version}"
    else
        warn "Check Requirements : git"
    fi
    if hash "mkfontscale" &>/dev/null; then
        mkfontscale_version=$(mkfontscale -v)
        success "Check Requirements: ${mkfontscale_version}"
    else
        warn "Check Requirements : mkfontscale"
    fi
    if hash "vim" &>/dev/null; then
        is_vim8=$(vim --version | grep "Vi IMproved 8")
        is_vim74=$(vim --version | grep "Vi IMproved 7.4")
        if [ -n "$is_vim8" ]; then
            success "Check Requirements: vim 8.0"
        elif [ -n "$is_vim74" ]; then
            success "Check Requirements: vim 7.4"
        else
            if hash "nvim" &>/dev/null; then
                success "Check Requirements: nvim"
            else
                warn "SpaceVim need vim 7.4 or above"
            fi
        fi
        if hash "nvim" &>/dev/null; then
            success "Check Requirements: nvim"
        fi
    else
        if hash "nvim" &>/dev/null; then
            success "Check Requirements: nvim"
        else
            warn "Check Requirements : vim or nvim"
        fi
    fi
    info "Checking true colors support in terminal:"
    bash -c "$(curl -fsSL https://raw.githubusercontent.com/JohnMorales/dotfiles/master/colors/24-bit-color.sh)"
}
# }}}

# usage {{{
usage () {
    echo "SpaceVim install script : V ${Version}"
    echo ""
    echo "Usage : curl -sLf https://spacevim.org/install.sh | bash -s -- [option] [target]"
    echo ""
    echo "  This is bootstrap script for SpaceVim."
    echo ""
    echo "OPTIONS"
    echo ""
    echo " -i, --install            install spacevim for vim or neovim"
    echo " -v, --version            Show version information and exit"
    echo " -u, --uninstall          Uninstall SpaceVim"
    echo " -c, --checkRequirements  checkRequirements for SpaceVim"
    echo " --no-fonts               skip downloading fonts"
    echo ""
    echo "EXAMPLE"
    echo ""
    echo "    Install SpaceVim for vim and neovim"
    echo ""
    echo "        curl -sLf https://spacevim.org/install.sh | bash"
    echo ""
    echo "    Install SpaceVim for vim only or neovim only"
    echo ""
    echo "        curl -sLf https://spacevim.org/install.sh | bash -s -- --install vim"
    echo "        curl -sLf https://spacevim.org/install.sh | bash -s -- --install neovim"
    echo ""
    echo "    Uninstall SpaceVim"
    echo ""
    echo "        curl -sLf https://spacevim.org/install.sh | bash -s -- --uninstall"
}
# }}}

# install_done {{{
install_done () {
    echo_with_color ${Yellow} ""
    echo_with_color ${Yellow} "Almost done!"
    echo_with_color ${Yellow} "=============================================================================="
    echo_with_color ${Yellow} "==    Open Vim or Neovim and it will install the plugins automatically      =="
    echo_with_color ${Yellow} "=============================================================================="
    echo_with_color ${Yellow} ""
    echo_with_color ${Yellow} "That's it. Thanks for installing SpaceVim. Enjoy!"
    echo_with_color ${Yellow} ""
}
# }}}

# welcome {{{
welcome () {
    echo_with_color ${Yellow} "        /######                                     /##    /##/##             "
    echo_with_color ${Yellow} "       /##__  ##                                   | ##   | #|__/             "
    echo_with_color ${Yellow} "      | ##  \__/ /######  /######  /####### /######| ##   | ##/##/######/#### "
    echo_with_color ${Yellow} "      |  ###### /##__  ##|____  ##/##_____//##__  #|  ## / ##| #| ##_  ##_  ##"
    echo_with_color ${Yellow} "       \____  #| ##  \ ## /######| ##     | ########\  ## ##/| #| ## \ ## \ ##"
    echo_with_color ${Yellow} "       /##  \ #| ##  | ##/##__  #| ##     | ##_____/ \  ###/ | #| ## | ## | ##"
    echo_with_color ${Yellow} "      |  ######| #######|  ######|  ######|  #######  \  #/  | #| ## | ## | ##"
    echo_with_color ${Yellow} "       \______/| ##____/ \_______/\_______/\_______/   \_/   |__|__/ |__/ |__/"
    echo_with_color ${Yellow} "               | ##                                                           "
    echo_with_color ${Yellow} "               | ##                                                           "
    echo_with_color ${Yellow} "               |__/                                                           "
    echo_with_color ${Yellow} "                      version : ${Version}      by : spacevim.org             "
}
# }}}

# download_font {{{
download_font () {
    url="https://raw.githubusercontent.com/ryanoasis/nerd-fonts/master/patched-fonts/SourceCodePro/Regular/complete/${1// /%20}"
    path="$HOME/.local/share/fonts/$1"
    if [[ -f "$path" && ! -s "$path" ]]
    then
        rm "$path"
    fi
    if [[ -f "$path" ]]
    then
        success "Downloaded $1"
    else
        info "Downloading $1"
        curl -s -o "$path" "$url"
        success "Downloaded $1"
    fi
}
# }}}

# install_fonts {{{
install_fonts () {
    if [[ ! -d "$HOME/.local/share/fonts" ]]; then
        mkdir -p $HOME/.local/share/fonts
    fi
    download_font "Sauce Code Pro Nerd Font Complete.ttf"
    info "Updating font cache, please wait ..."
    if [ $System == "Darwin" ];then
        if [ ! -e "$HOME/Library/Fonts" ];then
            mkdir "$HOME/Library/Fonts"
        fi
        cp $HOME/.local/share/fonts/* $HOME/Library/Fonts/
    else
        # need_cmd 'mkfontdir'
        # need_cmd 'mkfontscale'
        fc-cache -fv > /dev/null
        mkfontdir "$HOME/.local/share/fonts" > /dev/null
        mkfontscale "$HOME/.local/share/fonts" > /dev/null
    fi
    success "font cache done!"
}
# }}}

### main {{{
main () {
    if [ $# -gt 0 ]
    then
        case $1 in
            --uninstall|-u)
                info "Trying to uninstall SpaceVim"
                uninstall_vim
                uninstall_neovim
                echo_with_color ${BWhite} "Thanks!"
                exit 0
                ;;
            --checkRequirements|-c)
                check_requirements
                exit 0
                ;;
            --install|-i)
                welcome
                fetch_repo
                if [ $# -eq 2 ]
                then
                    case $2 in
                        neovim)
                            install_neovim
                            install_fonts
                            install_done
                            exit 0
                            ;;
                        vim)
                            install_vim
                            install_fonts
                            install_done
                            exit 0
                    esac
                fi
                install_vim
                install_neovim
                install_fonts
                install_done
                exit 0
                ;;
            --help|-h)
                usage
                exit 0
                ;;
            --no-fonts)
                welcome
                fetch_repo
                install_vim
                install_neovim
                install_done
                exit 0
                ;;
            --version|-v)
                msg "${Version}"
                exit 0
        esac
    else
        welcome
        fetch_repo
        install_vim
        install_neovim
        install_fonts
        install_done
    fi
}
# }}}

main $@

# vim:set nofoldenable foldmethod=marker: