install.sh

#!/bin/bash
set -e
echo '-- Welcome to ArchVim installation script :)'
echo '-- Make sure you have fast GitHub connection for best experience'

# ./install.sh nvim, for using NeoVim
# ./install.sh vim y, for not asking from terminal

VIMEXE=${1}
FORCE=${2}

cd "$(dirname $0)"/..
test -d ./.vim || (echo "ERROR: .vim not found, please download from github.com/archibate/vimrc/releases" && exit 1)
test -f ./.vim/install.sh || (echo "ERROR: .vim/install.sh not found, please download from github.com/archibate/vimrc/releases" && exit 1)

if [ "x$UID" == "x0" ] && [ "x$FORCE" != "xy" ]; then
    echo -n "-- This script doesn't need manually add sudo, if you run as root it will install for root user instead, sure to continue (Y/n)? "; read -n1 x; echo
    if [ "x$x" == "xn" ]; then exit 1; fi
fi

if [ "x$VIMEXE" == "x" ]; then
    if which vim 2> /dev/null; then VIMEXE=vim
    elif which nvim 2> /dev/null; then VIMEXE=nvim
    else
        echo 'ERROR: cannot find vim or nvim executable!'
        echo 'ERROR: Please install vim first: apt install vim'
        exit 1
    fi
fi
echo "-- Installing for vim executable: $VIMEXE"
echo "-- Installing for user: $USER ($UID)"


get_linux_distro() {
    if grep -Eq "Ubuntu" /etc/*-release 2> /dev/null; then
        echo "Ubuntu"
    elif grep -Eq "Deepin" /etc/*-release 2> /dev/null; then
        echo "Deepin"
    elif grep -Eq "Raspbian" /etc/*-release 2> /dev/null; then
        echo "Raspbian"
    elif grep -Eq "uos" /etc/*-release 2> /dev/null; then
        echo "UOS"
    elif grep -Eq "LinuxMint" /etc/*-release 2> /dev/null; then
        echo "LinuxMint"
    elif grep -Eq "elementary" /etc/*-release 2> /dev/null; then
        echo "elementaryOS"
    elif grep -Eq "Debian" /etc/*-release 2> /dev/null; then
        echo "Debian"
    elif grep -Eq "Kali" /etc/*-release 2> /dev/null; then
        echo "Kali"
    elif grep -Eq "Parrot" /etc/*-release 2> /dev/null; then
        echo "Parrot"
    elif grep -Eq "CentOS" /etc/*-release 2> /dev/null; then
        echo "CentOS"
    elif grep -Eq "fedora" /etc/*-release 2> /dev/null; then
        echo "fedora"
    elif grep -Eq "openSUSE" /etc/*-release 2> /dev/null; then
        echo "openSUSE"
    elif grep -Eq "Arch Linux" /etc/*-release 2> /dev/null; then
        echo "ArchLinux"
    elif grep -Eq "ManjaroLinux" /etc/*-release 2> /dev/null; then
        echo "ManjaroLinux"
    elif grep -Eq "Gentoo" /etc/*-release 2> /dev/null; then
        echo "Gentoo"
    elif grep -Eq "alpine" /etc/*-release 2> /dev/null; then
        echo "Alpine"
    elif [ "x$(uname -s)" == "xDarwin" ]; then
        echo "MacOS"
    else
        echo "Unknown"
    fi
}


detect_platform() {
  local platform="$(uname -s | tr '[:upper:]' '[:lower:]')"

  # check for MUSL
  if [ "${platform}" = "linux" ]; then
    if ldd /bin/sh | grep -i musl >/dev/null; then
      platform=linux_musl
    fi
  fi

  # mingw is Git-Bash
  if echo "${platform}" | grep -i mingw >/dev/null; then
    platform=win
  fi

  echo "${platform}"
}


detect_arch() {
  local arch="$(uname -m | tr '[:upper:]' '[:lower:]')"

  if echo "${arch}" | grep -i arm >/dev/null; then
    # ARM is fine
    echo "${arch}"
  else
    if [ "${arch}" = "i386" ]; then
      arch=x86
    elif [ "${arch}" = "x86_64" ]; then
      arch=x64
    elif [ "${arch}" = "aarch64" ]; then
      arch=arm64
    fi

    # `uname -m` in some cases mis-reports 32-bit OS as 64-bit, so double check
    if [ "${arch}" = "x64" ] && [ "$(getconf LONG_BIT)" -eq 32 ]; then
      arch=x86
    fi

    echo "${arch}"
  fi
}


backup_vimrc() {
    if [ "x$PWD" != "x$HOME" ]; then
        if [ -f ~/.vimrc ]; then
            echo "-- Backup existing .vimrc to ~/.vimrc.backup.$$"
            mv ~/.vimrc ~/.vimrc.backup.$$
        fi
        if [ -d ~/.vim ]; then
            echo "-- Backup existing .vim to ~/.vim.backup.$$"
            mv ~/.vim ~/.vim.backup.$$
        fi
    fi
}


install_vimrc() {
    if [ "x$PWD" != "x$HOME" ]; then
        echo "-- Installing .vimrc and .vim"
        cp -r .vimrc .vim ~/
    else
        echo "-- Already in home directory, skipping..."
    fi
}


install_coc_plugins() {
    bash <<EOF
set -e
pushd ~/

echo -e '\\n\\nZZZZ\\n\\n' | "$VIMEXE" -c "set mouse= | echo 'installing all Vim plugins, please wait...' | PlugInstall | echo 'done' | quit" 2>&1 | grep -v Warning
for x in coc-ccls coc-pyright coc-git coc-json; do
    echo "-- Installing coc plugin '\$x', please wait..."
    echo -e '\\n\\nZZZZ\\n\\n' | "$VIMEXE" -c "set mouse= | echo 'installing \$x, please wait...' | CocInstall -sync \$x | echo 'done' | quit" 2>&1 | grep -v Warning
done

mkdir -p ~/.config/coc/extensions/node_modules/coc-ccls
ln -sf node_modules/ws/lib ~/.config/coc/extensions/node_modules/coc-ccls/lib
echo '-- coc plugins installed successfully'

popd
EOF
}


install_pacman() {
    sudo pacman -S --noconfirm ripgrep
    sudo pacman -S --noconfirm nodejs
    sudo pacman -S --noconfirm ccls

    install_vimrc
    install_coc_plugins
}


install_ccls_from_source() {
    if which ccls 2> /dev/null && ccls --version; then
        echo '-- ccls already installed, skipping...'
    else
        if ! [ -d .vim/ccls ]; then
            echo '-- Cloning ccls source code from GitHub (please wait)...'
            mkdir -p /tmp/ccls-work.$$
            pushd /tmp/ccls-work.$$
            git clone https://github.com/MaskRay/ccls.git --depth=1 --recursive
            popd
            mv /tmp/ccls-work.$$/ccls .vim/
        fi
        cd .vim/ccls
        rm -rf /tmp/ccls-build.$$
        echo '-- Building ccls from source...'
        cmake -B /tmp/ccls-build.$$ -DCMAKE_BUILD_TYPE=Release
        cmake --build /tmp/ccls-build.$$ --config Release --parallel `grep -c ^processor /proc/cpuinfo || echo 1`
        sudo cmake --build /tmp/ccls-build.$$ --config Release --target install
        echo '-- Installed ccls successfully'
        rm -rf /tmp/ccls-build.$$
        cd ../..
    fi
}


install_nodejs_lts() {
    if ! which node 2> /dev/null || [ `node --version | sed s/v// | cut -f1 -d.` -lt 16 ]; then
        echo '-- Upgrading Node.js version to 16.x (it takes some time, please wait)...'
        NODEPLAT=$(detect_platform)
        NODEARCH=$(detect_arch)
        NODEVER=16.17.0
        NODEFILE=node-v$NODEVER-$NODEPLAT-$NODEARCH
        NODEURL=https://nodejs.org/dist/v$NODEVER/$NODEFILE.tar.xz
        echo "-- Downloading file from $NODEURL"
        sudo rm -rf /opt/archvim-nodejs
        sudo mkdir -p /opt/archvim-nodejs
        sudo bash -c "cd /opt/archvim-nodejs && (curl -Lf $NODEURL | tar -Jx)"
        NODEEXEC=/opt/archvim-nodejs/$NODEFILE/bin/node
        "$NODEEXEC" --version
        printf "let g:coc_node_path = '$NODEEXEC'\nlet g:coc_disable_startup_warning = 1\nsource ~/.vim/init.vim\n" > .vimrc
    fi
}


install_ripgrep_deb() {
    echo '-- Downloading ripgrep.deb from GitHub (please wait)...'
    RIPGREP_VERSION=$(curl -s "https://api.github.com/repos/BurntSushi/ripgrep/releases/latest" | grep -Po '"tag_name": "\K[0-9.]+')
    curl -Lo /tmp/vimrc-$$-ripgrep.deb "https://github.com/BurntSushi/ripgrep/releases/latest/download/ripgrep_${RIPGREP_VERSION}_amd64.deb"
    sudo apt install -y /tmp/vimrc-$$-ripgrep.deb
    rm -rf /tmp/vimrc-$$-ripgrep.deb
}


install_apt() {
    sudo apt-get install -y curl
    sudo apt-get install -y ripgrep || install_ripgrep_deb   # Ubuntu 18.04 doesn't have the ripgrep package...

    sudo apt-get install -y ccls || (sudo apt-get install -y clang libclang-dev cmake make g++ && install_ccls_from_source)
    install_nodejs_lts   # Ubuntu 20.04 only have Node.js version 11.x, the coc.nvim plugin requires 12.x and above however..
    install_vimrc
    install_coc_plugins
}


install_yum() {
    sudo yum-config-manager --add-repo=https://copr.fedorainfracloud.org/coprs/carlwgeorge/ripgrep/repo/epel-7/carlwgeorge-ripgrep-epel-7.repo
    sudo yum install -y ripgrep
    sudo yum install -y cmake make

    # "ClangConfig.cmake not found" error.
    # LLVM version below 7 is not supported for ccls.
    # Usually, building LLVM takes two or more hours.
    # You can also use Ninja to build LLVM spending less time.

    # install_llvm_from_source
    install_ccls_from_source
    install_nodejs_lts
    install_vimrc
    install_coc_plugins
}


install_brew() {
    brew install ripgrep 
    brew install cmake make gcc curl
    brew install node ccls

    install_vimrc
    install_coc_plugins
}


install_dnf() {
    sudo dnf install -y ripgrep
    sudo dnf install -y clang clang-devel
    sudo dnf install -y cmake make g++
    sudo dnf install -y curl

    install_ccls_from_source
    install_nodejs_lts
    install_vimrc
    install_coc_plugins
}


install_zypper() {
    sudo zypper in --no-confirm ripgrep ccls nodejs

    install_vimrc
    install_coc_plugins
}


install_ripgrep_from_source() {
    cd .vim
    echo '-- Cloning ripgrep source code from GitHub (please wait)...'
    git clone https://github.com/BurntSushi/ripgrep.git --depth=1
    cd ripgrep
    cargo build --release
    ./target/release/rg --version
    sudo cp ./target/release/rg /usr/local/bin
    cd ../..
}


install_llvm_from_source() {
    cd .vim
    echo '-- Cloning llvm-project source code from GitHub (please wait)...'
    git clone https://github.com/llvm/llvm-project.git --depth=1
    cd llvm-project
    cmake -B /tmp/llvm-build.$$ -DCMAKE_BUILD_TYPE=Release --enable-optimized --enable-targets=host-only -DLLVM_ENABLE_PROJECTS="clang" -G "Unix Makefiles" ./llvm
    cmake --build /tmp/llvm-build.$$ --config Release --parallel `grep -c ^processor /proc/cpuinfo || echo 1`
    sudo cmake --build /tmp/llvm-build.$$ --config Release --target install
    echo '-- Installed llvm successfully'
    rm -rf /tmp/llvm-build.$$
    cd ../..
}


install_any() {
    install_ripgrep_from_source
    install_ccls_from_source
    install_nodejs_lts
    install_vimrc
    install_coc_plugins
}


do_install() {
    distro=`get_linux_distro`
    echo "-- Linux distro detected: $distro"

    backup_vimrc

    if [ $distro == "Ubuntu" ]; then
        install_apt
    elif [ $distro == "Deepin" ]; then
        install_apt
    elif [ $distro == "Debian" ]; then
        install_apt
    elif [ $distro == "Kali" ]; then
        install_apt
    elif [ $distro == "Raspbian" ]; then
        install_apt
    elif [ $distro == "ArchLinux" ]; then
        install_pacman
    elif [ $distro == "ManjaroLinux" ]; then
        install_pacman
    elif [ $distro == "MacOS" ]; then
        install_brew
    elif [ $distro == "fedora" ]; then
        install_dnf
    elif [ $distro == "openSUSE" ]; then
        install_zypper
    elif [ $distro == "CentOS" ]; then
        install_yum
    else
        # TODO: add more Linux distros here..
        echo "-- WARNING: Unsupported Linux distro: $distro"
        echo "-- The script will try to install these packages from source: ripgrep ccls"
        echo "-- Note that ripgrep requires Rust, ccls requires Clang and LLVM to build, make sure you have them.."
        echo "-- If you know how to install them, feel free to contribute to this GitHub repository: github.com/archibate/vimrc"
        echo "-- Also, Node.js 12.x or above is required. Try the following patch script if you meet issues about coc.nvim:"
        cat <<EOF
mkdir -p ~/.config/coc/extensions/node_modules/coc-ccls
ln -sf node_modules/ws/lib ~/.config/coc/extensions/node_modules/coc-ccls/lib
EOF
        if [ "x$FORCE" != "xy" ]; then
            echo -n "-- Continue installing from source anyway (y/N)? "; read -n1 x; echo
            if [ "x$x" != "xy" ]; then exit 1; fi
        fi
        install_any
    fi

    echo "-- Installation complete, thank you for choosing ArchVim"
}

do_install