setup.sh

#!/bin/bash
set -e

WORKDIR="${PWD}"
shell_config=".zshrc"
antigen_config=".antigenrc"
str="alias vi='nvim'"
NEOVIM_PYENV_VER="3.9.6"


install_pyenv () {
    if [ ! -d ~/.pyenv ]; then
        git clone https://github.com/pyenv/pyenv.git ~/.pyenv
    fi

    export PYENV_ROOT="$HOME/.pyenv"
    export PATH="$PYENV_ROOT/bin:$PYENV_ROOT/shims:$PATH"
    pyenv_venv_dir=$(pyenv root)/plugins/pyenv-virtualenv

    if [ ! -d $pyenv_venv_dir ]; then
        curl https://pyenv.run | bash
    fi

    eval "$(pyenv init -)"
    eval "$(pyenv virtualenv-init -)"
}

install_packages () {
    if [ "$(uname)" = "Linux" ]; then
        yes '' | sudo apt-get install software-properties-common
        yes '' | sudo add-apt-repository ppa:neovim-ppa/stable

        sudo apt-get install apt-transport-https --yes


        sudo apt-get update
        yes | sudo apt-get install \
                ca-certificates \
                curl \
                openssl \
                libreadline-dev \
                sqlite3 \
                curl \
                libffi-dev \
                libbz2-dev \
                libssl-dev \
                libsqlite3-dev \
                liblzma-dev \
                zsh \
                universal-ctags \
                neovim \
                zsh-antigen \
                cmake \
                libfreetype6-dev \
                libfontconfig1-dev \
                xclip \
                python3-pip \
                python3-venv \
                direnv \
                ripgrep \
                tmux \
                clangd

        # install node
        curl -fsSL https://deb.nodesource.com/setup_current.x | sudo -E bash -
        sudo apt-get install -y nodejs

    	# setup for kubectl
    	curl -fsSL https://pkgs.k8s.io/core:/stable:/v1.30/deb/Release.key | sudo gpg --dearmor -o /etc/apt/keyrings/kubernetes-apt-keyring.gpg
    	sudo chmod 644 /etc/apt/keyrings/kubernetes-apt-keyring.gpg
	echo 'deb [signed-by=/etc/apt/keyrings/kubernetes-apt-keyring.gpg] https://pkgs.k8s.io/core:/stable:/v1.30/deb/ /' | sudo tee /etc/apt/sources.list.d/kubernetes.list
	sudo chmod 644 /etc/apt/sources.list.d/kubernetes.list
	sudo apt-get update
	sudo apt install -y kubectl

    	# setup for helm
	curl https://baltocdn.com/helm/signing.asc | gpg --dearmor | sudo tee /usr/share/keyrings/helm.gpg > /dev/null
	sudo apt-get install apt-transport-https --yes
	echo "deb [arch=$(dpkg --print-architecture) signed-by=/usr/share/keyrings/helm.gpg] https://baltocdn.com/helm/stable/debian/ all main" | sudo tee /etc/apt/sources.list.d/helm-stable-debian.list
	sudo apt-get update
	sudo apt-get install helm


    elif [ "$(uname)" = "Darwin" ]; then
        brew install \
            zsh \
            universal-ctags \
            neovim \
            antigen \
            direnv \
            ripgrep \
            helm \
            openssl \
            readline \
            sqlite3 \
            xz \
            zlib \
            kubectl \
            tmux \
            node

    fi


}

construct_shell_config () {
    # add lines to shell RC file if it doesn't already contain them
    if [ ! -f ~/$shell_config ]; then
        cp .zshrc ~/$shell_config
    elif grep "$str" ~/$shell_config > /dev/null; then
        echo "FOUND"
    else
        cat .zshrc >> ~/$shell_config
    fi
}

install_antigen () {
    # zsh package manager
    curl -L git.io/antigen > ~/antigen.zsh
    if [ ! -f ~/$antigen_config ]; then
        cp .antigenrc ~/$antigen_config
    fi
}

set_xdg_config_var () {
    # set XDG_CONFIG_HOME if it's not already set
    if [ ! $XDG_CONFIG_HOME ]; then
        mkdir -p ${XDG_CONFIG_HOME:=$HOME/.config}
    fi
}

install_nvm () {
    curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.38.0/install.sh | bash
}

install_python_version (){
    if [[ ! $(pyenv versions | grep $NEOVIM_PYENV_VER) ]]; then
        pyenv install $NEOVIM_PYENV_VER
    fi
}

install_pipx () {
    if [[ "$(uname)" == "Darwin" ]]; then
        brew install pipx
        pipx ensurepath
    else
        sudo apt install pipx
        pipx ensurepath

        export PATH="$HOME/.local/bin:$PATH"
    fi
}

install_poetry () {
    if [[ ! $(pipx list | grep "poetry") ]]; then
        pipx install poetry
    fi
}

setup_tmux () {
    cp $WORKDIR/.tmux.conf $HOME
    if [[ ! -d "$HOME/.tmux/plugins/tpm" ]]; then
        git clone https://github.com/tmux-plugins/tpm $HOME/.tmux/plugins/tpm
    fi
}

setup_neovim () {
    ln -s ${PWD}/nvim $XDG_CONFIG_HOME/nvim

    pri_dir=$(pwd)
    cd $HOME
    curl -fLo ~/.local/share/nvim/site/autoload/plug.vim --create-dirs \
        https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim
    cd $pri_dir

    install_python_version
    if [[ ! $(pyenv versions | grep neovim3) ]]; then
        pyenv virtualenv $NEOVIM_PYENV_VER neovim3
    fi
    pyenv activate neovim3

    pip install jedi
    pip install neovim
    pip install pylint

    nvim --headless +PlugInstall +qa
    nvim --headless +UpdateRemotePlugins +qa

    pyenv deactivate
}

setup_fonts () {
    # set up powerline and its fonts
    cd $HOME
    git clone --depth 1 https://github.com/ryanoasis/nerd-fonts
    cd nerd-fonts
    ./install.sh
    cd ..
    rm -rf nerd-fonts
    cd $WORKDIR
}

setup_fzf () {
    if [[ ! -d ~/.fzf ]]; then
        git clone --depth 1 https://github.com/junegunn/fzf ~/.fzf
        # install fzf and answer yes to all its prompts
        yes | ~/.fzf/install
    fi
}

setup_kube_for_wsl () {
    if [[ "$(uname)" == "Linux" ]]; then
        if [[ -z "${WSL_DISTRO_NAME}" && ! -f /etc/wsl.conf ]]; then
            sudo sh -c 'echo "[automount]\ncrossDistro = true" > /etc/wsl.conf'
        else
            echo "/etc/wsl.conf exists"
        fi
    fi
}

install_vue_cli () {
    if [[ ! -d ~/.npm-global ]]; then
        mkdir ~/.npm-global
        npm config set prefix '~/.npm-global'
        export PATH=~/.npm-global/bin:$PATH
    fi
    npm install -g @vue/cli
}


install_packages
construct_shell_config
install_antigen
set_xdg_config_var
install_pyenv
install_pipx
install_poetry
setup_neovim
setup_fonts
setup_fzf
setup_tmux
setup_kube_for_wsl
install_vue_cli

# Change user's default shell to zsh
sudo chsh -s $(which zsh) $USER
export SHELL=$(which zsh)
zsh