install.sh

#!/bin/bash

set -e

usage() {
cat << EOF
usage:  options

This script install an opinionated debian environment.

OPTIONS:
  --all                   Install everything
  --tools                 Install tools
  --tmux                  Install tmux
  --vim                   Install vim
  --zsh                   Install zsh    
  -co,  --configure-only  Configure only, don't install anything
  -h,   --help            Show this message
EOF
}

CONFIGURE_ONLY=false
INSTALL_ALL=false
INSTALL_VIM=false
INSTALL_TMUX=false
INSTALL_TOOLS=false
INSTALL_ZSH=false

base_packages=(
  automake
  autoconf
  apt-transport-https
  bison
  build-essential
  curl
  file
  git
  htop
  libbz2-dev
  libffi-dev
  liblzma-dev
  libncurses5
  libncursesw5
  libncurses5-dev
  libncursesw5-dev
  libpq-dev
  libsqlite3-dev
  libssl-dev
  libreadline8
  libreadline-dev
  libtool
  libyaml-dev
  libxml2-dev
  libxmlsec1-dev
  libxslt-dev
  linux-headers-generic
  llvm
  python3-openssl
  software-properties-common
  ssh
  tk-dev
  unixodbc-dev
  unzip
  xdg-utils
  xz-utils
  zlib1g
  zlib1g-dev
)

zsh_packages=(
  libuser
  zsh
)

brew_base_packages=(
  gcc
  htop
  jq
  libpq
  net-tools
  # openssh - https://github.com/Homebrew/linuxbrew-core/pull/19684
  openssl 
  readline 
  sqlite3 
  wget
  xz 
  zlib
)

# Enable or disable colored logs
COLOR_LOG=true

## Color coded logging - https://misc.flogisoft.com/bash/tip_colors_and_formatting
error() {
  if [ $COLOR_LOG = true ]; then
    echo "$1"  # Red FG
  else
    echo "$1"
  fi
}

warn() {
  if [ $COLOR_LOG = true ]; then
    echo "$1"  # Yellow FG
  else
    echo "$1"
  fi
}

info() {
  if [ $COLOR_LOG = true ]; then
    echo "$1"  # Green FG
  else
    echo "$1"
  fi
}

debug() {
  if [ $COLOR_LOG = true ]; then
    echo "$1"  # Cyan FG
  else
    echo "$1"
  fi
}

asdf_install() {
  # Performs asdf installs and set the installed version as the global version
  # asdf_install [plugin] (version)
  # If version is not defined, will install the latest version

  info "  - Installing $1"
  set +e  # ignore error
  asdf plugin-add "$1"
  status_code=$?
  if [ $status_code -eq 0 ] || [ $status_code -eq 2 ]; then
      echo "$1 is already added"
  fi
  set -e  # exit on errors

  # Get latest version so we can install and set it as global
  if [ "$2" ]; then
    latest="$(asdf latest $1 $2)"
  else
    latest="$(asdf latest $1)"
  fi
  # default to passed in version. (needed for java)
  if [ ! "$latest" ]; then
    latest="$2"
  fi

  asdf install "$1" "$latest"
  asdf global "$1" "$latest"
}

perform_system_update() {
  info "Updating packages"
  sudo apt update
  sudo apt upgrade -y
}

install_packages() {
  info "Installing system packages"

  packages="${base_packages[@]}"
  if [ $INSTALL_ALL = true ]; then
    packages="${packages} ${zsh_packages[@]}"
  else
    if [ $INSTALL_ZSH = true ]; then
      packages="${packages} ${zsh_packages[@]}"
    fi
  fi

  sudo apt install -y ${packages}

  info "Installed system packages"
}

install_homebrew() {
  if [ ! -f "/home/linuxbrew/.linuxbrew/bin/brew" ]; then
    info "Installing Homebrew"
    /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install.sh)" </dev/null
    echo 'eval $(/home/linuxbrew/.linuxbrew/bin/brew shellenv)' >> $HOME/.zprofile
    echo 'eval $(/home/linuxbrew/.linuxbrew/bin/brew shellenv)' >> $HOME/.profile
    info "Installed Homebrew"
  else
    debug "Homebrew is already installed, skipping..."
  fi
  
  if [ ! "$(command -v brew)" ]; then
    info "Sourcing brew"
    eval $(/home/linuxbrew/.linuxbrew/bin/brew shellenv)
  fi

  info "Installing Homebrew base packages"
  brew install "${brew_base_packages[@]}"
  info "Installed Homebrew base packages"
}

install_adsf() {
  install_homebrew
  if [ ! -f "$(brew --prefix asdf)/libexec/asdf.sh" ]; then
    info "Installing asdf"

    brew install asdf
    echo -e "\n. $(brew --prefix asdf)/libexec/asdf.sh" >> ~/.bashrc
    echo -e "\n. $(brew --prefix asdf)/etc/bash_completion.d/asdf.bash" >> ~/.bashrc
    info "Installed asdf"
  else
    debug "asdf is already installed, skipping..."
  fi
  
  if [ ! "$(command -v asdf)" ]; then
    info "Sourcing asdf"
    . $(brew --prefix asdf)/libexec/asdf.sh
  fi
}

install_zsh() {
  info "Installing ZSH"
  sudo apt install -y "${zsh_packages[@]}"
}

configure_zsh() {
  info "Configuring ZSH"
  # Install Antigen - https://github.com/zsh-users/antigen
  mkdir -p $HOME/.antigen
  curl -L git.io/antigen > $HOME/.antigen/antigen.zsh

  wget https://github.com/MichielVanderlee/system_config/raw/master/.zshrc -O $HOME/.zshrc
  wget https://github.com/MichielVanderlee/system_config/raw/master/.p10k.zsh -O $HOME/.p10k.zsh
  wget https://github.com/MichielVanderlee/system_config/raw/master/.powerlevel9k -O $HOME/.powerlevel9k

  info "Updating shell"
  sudo chsh -s $(which zsh) $(whoami)
}

install_vim() {
  info "Installing Vim and Neovim"
  install_homebrew

  brew install vim nvim

  info "Installed Vim and Neovim"
}

configure_vim() {
  info "Configuring Vim and NeoVim"
  # Install vim & neovim dependencies
  curl -fLo ~/.local/share/nvim/site/autoload/plug.vim --create-dirs \
      https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim

  curl -fLo ~/.vim/autoload/plug.vim --create-dirs \
      https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim

  # Need to install via the vim linked python, not pyenv's python
  python3 -m pip install --user \
    pynvim \
    black \
    jedi

  mkdir -p $HOME/.config/nvim
  wget https://github.com/MichielVanderlee/system_config/raw/master/.vimrc -O $HOME/.vimrc
  wget https://github.com/MichielVanderlee/system_config/raw/master/init.vim -O $HOME/.config/nvim/init.vim
  wget https://github.com/MichielVanderlee/system_config/raw/master/vim.zip 
  unzip vim.zip -d $HOME
  rm vim.zip

  info "Configured Vim and NeoVim"
}

install_tmux() {
  info "Installing Tmux"
  install_adsf

  asdf_install tmux

  info "Installed Tmux"
}

configure_tmux() {
  info "Configuring Tmux"

  # Configure tmux
  if [ ! -d "$HOME/.tmux/plugins/tpm" ]; then
    git clone https://github.com/tmux-plugins/tpm $HOME/.tmux/plugins/tpm
    wget https://github.com/MichielVanderlee/system_config/raw/master/.tmux.conf -O $HOME/.tmux.conf
    info "Configured Tmux"
  else
    info "TMUX already configured, skipping..."
  fi
}

install_tools() {
  info "Installing Tools"
  install_vscode
  install_adsf
  install_pyenv

  # Install plugins
  asdf_install golang
  asdf_install java openjdk
  asdf_install maven
  asdf_install ruby

  info "Installing nodejs"
  ## Install NVM because that's what I use on windows, so don't mix commands to switch node versions
  # asdf plugin-add nodejs
  # asdf install nodejs latest:12
  # asdf global nodejs "$(asdf latest nodejs 12)"
  wget -qO- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.3/install.sh | bash

  pip install \
    awscli \
    pgcli

  brew install pspg

  # ptpython
  if [[ ! -d "$HOME/.ptpython" ]]; then
    info "Installing ptpython"
    pip install ptpython
    mkdir $HOME/.ptpython
    wget -O $HOME/.ptpython/config.py https://github.com/MichielVanderlee/system_config/raw/master/.ptpython-config.py
  fi

  # colorls
  info "Installing colorls"
  $(asdf which gem) install colorls

  # ammonite-repl
  sudo curl -L https://github.com/lihaoyi/Ammonite/releases/download/1.6.9/2.13-1.6.9 -o /usr/local/bin/amm
  sudo chmod +x /usr/local/bin/amm
  
  install_docker

  sudo sysctl -w vm.max_map_count=262144

  info "Installed Tools"
}

install_pyenv() {
  info "Installing PyEnv"
  # PyEnv - because asdf does not support virtual environment, yet! - https://github.com/asdf-vm/asdf/issues/636

  export PYENV_ROOT="$HOME/.pyenv"
  if [[ ! -d "$PYENV_ROOT" ]]; then
    # https://github.com/pyenv/pyenv/issues/1479#issuecomment-610683526
    debug "Removing linuxbrew from path"
    OLD_PATH="$PATH"
    export PATH="$(echo $PATH | tr : '\n' | grep -v linuxbrew | paste -s -d:)"

    git clone https://github.com/pyenv/pyenv.git $PYENV_ROOT
    git clone https://github.com/pyenv/pyenv-virtualenv.git $PYENV_ROOT/plugins/pyenv-virtualenv
    git clone https://github.com/momo-lab/pyenv-install-latest.git $PYENV_ROOT/plugins/pyenv-install-latest
    git clone https://github.com/jawshooah/pyenv-default-packages.git $PYENV_ROOT/plugins/pyenv-default-packages
    export PATH="$PYENV_ROOT/bin:$PATH"
    eval "$(pyenv init -)"  
    eval "$(pyenv virtualenv-init -)"
    pyenv install-latest 3.11
    pyenv global "$(pyenv install-latest --print 3.11)"

    debug "Resetting path"
    export PATH=$OLD_PATH
  else
    # Ensure we initialize pyenv so we can continue in case of reruns
    export PATH="$PYENV_ROOT/bin:$PATH"
    eval "$(pyenv init -)"  
    eval "$(pyenv virtualenv-init -)"
    info "Already installed PyEnv, skipping..."
  fi
}

install_docker() {
  info "Installing Docker"
  
  install_homebrew

  sudo apt-get install -y \
    apt-transport-https \
    ca-certificates \
    curl \
    gnupg2 \
    software-properties-common
  curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
  sudo add-apt-repository -y \
    "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
    $(lsb_release -cs) \
    stable"
  sudo apt-get update
  sudo apt-get install -y docker-ce

  sudo curl -L "https://github.com/docker/compose/releases/download/v2.17.3/docker-compose-$(uname -s)-$(uname -m)" -o /usr/local/bin/docker-compose
  sudo chmod +x /usr/local/bin/docker-compose

  sudo usermod -aG docker $USER

  info "Installed Docker"
}

install_vscode() {
  info "Installing VSCode"

  sudo apt install software-properties-common apt-transport-https wget -y
  wget -O- https://packages.microsoft.com/keys/microsoft.asc | sudo gpg --dearmor | sudo tee /usr/share/keyrings/vscode.gpg
  echo deb [arch=amd64 signed-by=/usr/share/keyrings/vscode.gpg] https://packages.microsoft.com/repos/vscode stable main | sudo tee /etc/apt/sources.list.d/vscode.list
  sudo apt update
  sudo apt install code

  info "Installed VSCode"
}

configure_gnome_theme(){
  info "Configuring Theme"
  CURRENT_PIC=$(gsettings get org.gnome.desktop.screensaver picture-uri)
  if [ "$CURRENT_PIC" != "'file://$HOME/Pictures/RS_Siege1.jpg'" ]; then
      wget -qO "$HOME/Pictures/RS_Siege1.jpg" https://github.com/MichielVanderlee/system_config/raw/master/assets/wallpapers/RS_Siege1.jpg

      gsettings set org.gnome.desktop.interface gtk-theme "Adwaita-dark"
      gsettings set org.gnome.desktop.interface icon-theme "Humanity-Dark"
      gsettings set org.gnome.desktop.background picture-uri "file://$HOME/Pictures/RS_Siege1.jpg"
      gsettings set org.gnome.desktop.screensaver picture-uri "file://$HOME/Pictures/RS_Siege1.jpg"
  fi

  # Install fonts
  CURRENT_FONT=$(gsettings get org.gnome.desktop.interface monospace-font-name)
  if [ "$CURRENT_FONT" != "'FiraCode Nerd Font 11'" ]; then
    info "Install Fonts"
    wget https://github.com/ryanoasis/nerd-fonts/releases/download/v3.0.1/FiraCode.zip
    mkdir -p $HOME/.fonts/truetype/FiraCode
    unzip FiraCode.zip -d $HOME/.fonts/truetype/FiraCode

    sudo fc-cache -f -v
    gsettings set org.gnome.desktop.interface monospace-font-name "FiraCode Nerd Font 11"
  fi

  # Set Terminal Theme
  info "Setting Terminal Theme"
  wget -qO terminal.profile https://github.com/MichielVanderlee/system_config/raw/master/terminal.profile
  cat terminal.profile | dconf load /org/gnome/terminal/legacy/profiles:/:b1dcc9dd-5262-4d8d-a863-c897e6d979b9/

  info "Configured Theme"
}

install() {
  perform_system_update
  install_packages

  if [[ $INSTALL_ALL = true || $INSTALL_TMUX = true ]]; then
    install_tmux
  fi
  if [[ $INSTALL_ALL = true || $INSTALL_TOOLS = true ]]; then
    install_tools
  fi
  if [[ $INSTALL_ALL = true || $INSTALL_VIM = true ]]; then
    install_vim
  fi
  if [[ $INSTALL_ALL = true || $INSTALL_ZSH = true ]]; then
    install_zsh
  fi
}

configure() {
  if [[ $INSTALL_ALL = true || $INSTALL_TMUX = true ]]; then
    configure_tmux
  fi
  if [[ $INSTALL_ALL = true || $INSTALL_TOOLS = true ]]; then
    os_release="$(awk -F= '/^NAME/{print $2}' /etc/os-release | tr -d '"')"
    if [ "$os_release" = 'Ubuntu' ]; then
      configure_gnome_theme
    fi
  fi
  if [[ $INSTALL_ALL = true || $INSTALL_VIM = true ]]; then
    configure_vim
  fi
  if [[ $INSTALL_ALL = true || $INSTALL_ZSH = true ]]; then
    configure_zsh
  fi
}

show_manual_next_steps() {
  info "Open a new terminal for all changes to take effect."

  if [[ $INSTALL_ALL = true || $INSTALL_TMUX = true ]]; then
    info " - Then open TMUX and press 'ctrl+a shift+i'. Ensure you're not nested in screen!'"
  fi
  if [[ $INSTALL_ALL = true || $INSTALL_VIM = true ]]; then
    info " - Then open vim, and nvim and run :PlugInstall"
  fi
}

main() {
  # default to ALL
  if [[ $INSTALL_ALL = false && $INSTALL_TMUX = false && $INSTALL_TOOLS = false && $INSTALL_VIM = false && $INSTALL_ZSH = false ]]; then
    INSTALL_ALL=true
  fi

  if [ ! $CONFIGURE_ONLY = true ]; then
    install
  fi
  configure  
  show_manual_next_steps
}

#############################################
# Parse Arguments
#############################################
while (("$#")); do
  case "$1" in
    # Add your arguments here

    --all)
      INSTALL_ALL=true
      shift
      ;;
    --tmux)
      INSTALL_TMUX=true
      shift
      ;;
    --tools)
      INSTALL_TOOLS=true
      shift
      ;;
    --vim)
      INSTALL_VIM=true
      shift
      ;;
    --zsh)
      INSTALL_ZSH=true
      shift
      ;;

    -co|--configure-only)
      CONFIGURE_ONLY=true
      shift
      ;;

    -nc|--no-color)
      COLOR_LOG=false
      shift
      ;;
    -h|--help)
      usage
      exit 0
      ;;
    --) # end argument parsing
      shift
      break
      ;;
    *)
      usage
      exit 1
      ;;
  esac
done

# Execute main function
main