setup.sh

#!/usr/bin/env bash
usage() { echo "Usage: $0 [-o force overwrite]" 1>&2; exit 1; }
declare BASHRC_D="$HOME/.shellrc/bashrc.d"

setup_vim() {
    destination=$HOME/.vim/autoload/plug.vim
    created=false
    if [ ! -d "$destination" ] ; then
        curl -fLo $destination --create-dirs https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim
        created=true
    fi
    if [ $overwrite = true ] || [ $created = true ] ; then
        #Install all vundle plugins
        ln -s $(pwd)/vimrc $HOME/.vimrc
        vim +PlugInstall +qall
    fi
    if [ $overwrite = true ] || [ ! -d "$HOME/.vim/after" ] ; then
        cp $cp_options -r vim/after $HOME/.vim/after
    fi

    mkdir -p $HOME/.config
    cp $cp_options flake8 $HOME/.config/flake8

    if [ $overwrite = true ] || [ ! -d "$HOME/.config/nvim" ] ; then
        mkdir -p $HOME/.config/nvim
        ln -s $(pwd)/nvim/init.vim $HOME/.config/nvim/init.vim
        ln -s $(pwd)/nvim/lua $HOME/.config/nvim/lua
        ln -s $(pwd)/nvim/plugins $HOME/.config/nvim/plugins
    fi
}

setup_pyenv() {
    destination=$HOME/.pyenv
    if [ ! -d "$destination" ] ; then
        git clone https://github.com/pyenv/pyenv.git $destination
    fi
    if [ ! -d "$destination/plugins/pyenv-virtualenv" ] ; then
        git clone https://github.com/pyenv/pyenv-virtualenv.git $destination/plugins/pyenv-virtualenv
    fi
}

setup_starship() {
    if ! command -v starship &> /dev/null; then
        curl -sS https://starship.rs/install.sh | sh
        mkdir -p $HOME/.config
        ln -s $(pwd)/starship.toml $HOME/.config/starship.toml
    fi
}

setup_tmux() {
    declare plugins_destination=$HOME/.tmux/plugins/tpm
    declare tmux_conf=$HOME/.tmux.conf
    if [ ! -d "$plugins_destination" ] ; then
        git clone https://github.com/tmux-plugins/tpm $plugins_destination
    fi
    if [ ! -e $tmux_conf ] ; then
        ln -s $(pwd)/tmux.conf $tmux_conf
    fi
}

setup_csh() {
    cp $cp_options cshrc $HOME/.cshrc
}

setup_git() {
    gitignore=$HOME/.gitignore
    if [ ! -e $gitignore ] || [ $overwrite = true ] ; then
        touch $gitignore
        append_if_new gitignore $gitignore
    fi
    gitconfig=$HOME/.gitconfig
    if [ ! -e $gitconfig ] || [ $overwrite = true ] ; then
        touch $gitconfig
        append_if_new gitconfig $gitconfig
    fi

    declare git_bash_completions="/usr/share/bash-completion/completions/git"
    declare git_bash_target="${BASHRC_D}"/git.bash
    if [ ! -e "${git_bash_target}" ] ; then
        ln -s "${git_bash_completions}" "${git_bash_target}"
    fi
}

setup_bash() {
    cp $cp_options inputrc $HOME/.inputrc
    declare bashrc="$HOME/.bashrc"
    if [ ! -e $bashrc ] || [ $overwrite = true ] ; then
        mkdir -p ${BASHRC_D}
        ln -s $(pwd)/bashrc.d/* ${BASHRC_D}
        append_if_new bashrc $bashrc
    fi
}

setup_zsh() {
    declare zshrc="$HOME/.zshrc"
    declare zshrc_d="$HOME/.shellrc/zshrc.d"
    if [[ ! -e $zshrc || $overwrite = true ]] ; then
        mkdir -p ${zshrc_d}
        ln -s $(pwd)/zshrc.d/* ${zshrc_d}
        append_if_new zshrc $zshrc
    fi
}

setup_repos() {
    mkdir -p $HOME/repos/github
}

setup_fzf() {
    destination=$HOME/.fzf
    if [ ! -d "$destination" ] ; then
        git clone --depth 1 https://github.com/junegunn/fzf.git "${destination}"
        cd "${destination}" && git pull && ./install --key-bindings --completion --no-update-rc && cd -
    fi
}

setup_zoxide() {
    if ! command -v zoxide &> /dev/null; then
        curl -sS https://raw.githubusercontent.com/ajeetdsouza/zoxide/main/install.sh | bash
    fi
}

append_with_newline() {
    # Append file $1 to file $2 with a newline in between
    if [ $# -ne 2 ]; then
        echo "Not enough arguments provided to $FUNCNAME"
        exit
    fi
    echo "$(awk 'FNR==1{print ""}1' $1)" >> $2
}

append_if_new() {
    # Append file $1 to file $2 if the first line of $1 is not in $2
    first_line=$(head -n 1 bashrc)
    if [ $# -ne 2 ]; then
        echo "Not enough arguments provided to $FUNCNAME"
        exit
    fi
    src=$1
    dst=$2
    first_line=$(head -n 1 $src)
    if ! grep -q -F "$first_line" $dst; then
        append_with_newline $src $dst
        return 0
    else
        return 1
    fi
}

setup_rvm() {
    if ! command -v rvm &> /dev/null; then
        gpg --keyserver keyserver.ubuntu.com --recv-keys 409B6B1796C275462A1703113804BB82D39DC0E3 7D2BAF1CF37B13E2069D6956105BD0E739499BDB
        curl -sSL https://get.rvm.io | bash
    fi
}

setup_alacritty() {
    setup_rust
    if ! command -v alacritty &> /dev/null; then
        cargo install alacritty
    fi
}

setup_ripgrep() {
    setup_rust
    if ! command -v rg &> /dev/null; then
        cargo install ripgrep
    fi
}

setup_rust() {
    if ! command -v cargo &> /dev/null; then
        echo "cargo could not be found, installing rust."
        # Install cargo
        curl https://sh.rustup.rs -sSf | sh
        # Install cargo binstall
        curl -L --proto '=https' --tlsv1.2 -sSf https://raw.githubusercontent.com/cargo-bins/cargo-binstall/main/install-from-binstall-release.sh | bash
    fi
}

# standard no-clobber option for cp
cp_options="-n"
overwrite=false
while getopts ":o" o; do
    case "${o}" in
        o)
            cp_options=""
            overwrite=true
            ;;
        *)
            usage
            ;;
    esac
done
shift $((OPTIND-1))

setup_vim
setup_starship
setup_tmux
setup_git
setup_bash
setup_csh
setup_zsh
setup_repos
setup_fzf
setup_pyenv
#setup_alacritty
setup_ripgrep
setup_zoxide
#do RVM last as it prefers that it's changes to PATH in ~/.bashrc are last
setup_rvm