init.sh

#!/usr/bin/env bash

export DEBIAN_FRONTEND=noninteractive

local_bin="${HOME}/.local/bin"
workspace_dir="${HOME}/workspace"
dotfiles_dir="${workspace_dir}/dotfiles"

symlink() {
  # A symlink function that makes sure the target symlink's parent path
  # exists before creating the link

  # Expand ~ to $HOME if present
  local source="${1/#\~/$HOME}"
  local target="${2/#\~/$HOME}"

  [[ -e ${target} ]] && echo "'${target}' exists, skipping" && return

  mkdir -p "$(dirname "${target}")"
  ln -s "${source}" "${target}"
}


install_prerequisites() {
  echo "Installing prerequisites"

  if [[ $(uname) == "Linux" ]]; then
    sudo -E apt-get -yqq update > /dev/null
    sudo -E apt-get -yqq install \
        curl \
        git \
        > /dev/null
  fi
  if [[ $(uname) == "Darwin" ]]; then
    # On MacOS, install and use brew package manager
    if ! which brew &>/dev/null; then
      echo 'Installing brew package manager: https://brew.sh/'
      echo 'Requires user password'
      /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
    fi
  fi
}


setup_dotfiles_repo() {
  # Set up the dotfiles repository
  mkdir -p "${workspace_dir}"

  # Attempt to clone the repo if it doesnt exist on disk yet
  if [[ -d "${dotfiles_dir}" ]]; then
    echo "dotfiles folder already exists, skipping"
  else
    {
      git clone --recursive git@github.com:philipforget/dotfiles.git "${dotfiles_dir}"
    } || {
      echo "Unable to clone via ssh, falling back to https"
      git clone --recursive https://github.com/philipforget/dotfiles.git "${dotfiles_dir}"
    }
  fi
}

setup_symlinks() {
  mkdir -p "${workspace_dir}"
  mkdir -p "${dotfiles_dir}"
  mkdir -p "${local_bin}"

  # Set up symlinks
  symlink "${dotfiles_dir}/aliases" ~/.aliases
  symlink "${dotfiles_dir}/aliases_linux" ~/.aliases_linux
  symlink "${dotfiles_dir}/aliases_mac" ~/.aliases_mac
  symlink "${dotfiles_dir}/bash_custom" ~/.bash_custom
  symlink "${dotfiles_dir}/bash_linux" ~/.bash_linux
  symlink "${dotfiles_dir}/bash_mac" ~/.bash_mac
  symlink "${dotfiles_dir}/distraction" ~/.distraction
  symlink "${dotfiles_dir}/tmux" ~/.config/tmux
  symlink "${dotfiles_dir}/vim" ~/.vim
  symlink "${dotfiles_dir}/nvim" ~/.config/nvim
  symlink "${dotfiles_dir}/xmodmap" ~/.xmodmap
  symlink "${dotfiles_dir}/mise.toml" ~/.config/mise/config.toml
  symlink "${dotfiles_dir}/starship.toml" ~/.config/starship.toml

  symlink "${dotfiles_dir}/bin/sync-authorized-keys" "${local_bin}/sync-authorized-keys"

  # Delete the system-installed .gitconfig first if it exists
  rm "${HOME}/.gitconfig" || true
  symlink "${dotfiles_dir}/gitconfig" ~/.gitconfig

  if [[ $(uname) == "Darwin" ]]; then
    # Mac-only symlinks
    symlink "{$dotfiles_dir}/RectangleConfig.json" "~/Library/Application Support/Rectangle/RectangleConfig.json"
    symlink "${dotfiles_dir}/docker_config.mac.json" "~/.docker/config.json"
  fi
  if [[ $(uname) == "Linux" ]]; then
    # Linux-only symlinks
    symlink "${dotfiles_dir}/docker_config.json" ~/.docker/config.json
  fi

  # Add sourcing of ~/.bash_custom to .bashrc and .profile if not present
  if ! grep -Fxq 'source ~/.bash_custom' "${HOME}/.bashrc"; then
    echo 'source ~/.bash_custom' >>"${HOME}/.bashrc"
  fi
  if ! grep -Fxq 'source ~/.bash_custom' "${HOME}/.profile"; then
    echo 'source ~/.bash_custom' >>"${HOME}/.profile"
  fi
}

setup_system() {
  # Here's a nice place to add binaries
  mkdir -p "${local_bin}"

  # Install starship shell prompt to $local_bin, https://starship.rs/
  curl -sS https://starship.rs/install.sh | sh -s -- --force --bin-dir "${local_bin}"

  if [[ $(uname) == "Darwin" ]]; then
    echo "Installing packages with brew"
    brew install \
      age \
      bash \
      bash-completion@2 \
      colima \
      docker \
      git \
      git-lfs \
      mise \
      mosh \
      ripgrep \
      shellcheck \
      sops \
      tmux

    git lfs install

    brew install --cask \
      1password \
      google-chrome \
      microsoft-remote-desktop \
      rectangle \
      signal \
      wezterm

    # Set the brew-installed bash as the default shell
    brew_bash="$(brew --prefix)/bin/bash"
    # Bail early if ${brew_bash} doesn't exist for any reason
    if [[ ! -f "${brew_bash}" ]]; then
      echo "Unable to set brew-installed bash as shell!"
    else
      echo "Adding ${brew_bash} to /etc/shells if not present"
      grep "${brew_bash}" /etc/shells &>/dev/null || echo "${brew_bash}" | sudo tee -a /etc/shells
      if [[ "${SHELL}" != "${brew_bash}" ]]; then
        chsh -s "${brew_bash}" "$(whoami | xargs echo -n)"
      fi
    fi

    # Add ssh agent to system keychain on first unlock
    ssh_agent_config="AddKeysToAgent yes"
    grep "${ssh_agent_config}" ~/.ssh/config &>/dev/null || echo "${ssh_agent_config}" >>~/.ssh/config

  else
    if grep -qE "Ubuntu|Debian|Raspbian" /etc/issue; then
      echo "Installing required packages"
      # Use sudo -E to inherit our current environment, including
      # DEBIAN_FRONTEND set above
      sudo -E apt-get -qq update > /dev/null
      sudo -E apt-get -qq install -y \
        --no-install-recommends \
        age \
        bash-completion \
        build-essential \
        curl \
        dnsutils \
        git \
        git-lfs \
        libbz2-dev \
        libffi-dev \
        liblzma-dev \
        libncursesw5-dev \
        libreadline-dev \
        libsqlite3-dev \
        libssl-dev \
        libxml2-dev \
        libxmlsec1-dev \
        llvm \
        ripgrep \
        shellcheck \
        tk-dev \
        tmux \
        unzip \
        vim-nox \
        xz-utils \
        zlib1g-dev \
        > /dev/null

      git lfs install

      # Install sops from github releases
      curl --no-progress-meter -L \
        https://github.com/getsops/sops/releases/download/v3.8.1/sops-v3.8.1.linux.amd64 \
        -o "${local_bin}/sops" && chmod +x "${local_bin}/sops"
    fi
  fi
}

setup_mise() {
  mkdir -p "${local_bin}"
  # Install mise from their install script
  curl --no-progress-meter https://mise.run | sh

  eval "$("${HOME}"/.local/bin/mise activate bash)"
  mise trust ~/.config/mise/config.toml || echo "No global mise config, not trusting"

  mise plugin add usage
  mise use -g python@3.12

  # Mise gives us a nice cross-platform way of installing neovim, might as well
  # use it
  mise install -y neovim
  mise use -g neovim

  if [[ -f ~/.config/mise/config.toml ]]; then
    mise exec python -- python3 -m pip install \
      ipython \
      requests
  fi
}

sync_public_keys() {
  mkdir -p "${HOME}/.ssh"
  curl --no-progress-meter -L "https://github.com/${1}" >>"${HOME}/.ssh/authorized_keys"
}

init() {
  install_prerequisites

  local github_username="${1:-philipforget}"
  sync_public_keys "${github_username}"

  setup_system
  setup_symlinks
  setup_dotfiles_repo
  setup_mise

  echo "All done, launching a new shell process"
  exec bash
}

# When piping in from stdin (eg curl), BASH_SOURCE[0] will be unset, and
# when executing via ./init.sh or bash init.sh, BASH_SOURCE[0] and ${0}
# will be equal
if [ "${BASH_SOURCE[0]}" == "${0}" ] || [ -z "${BASH_SOURCE[0]}" ]; then
  # Tell bash to fail immediately on any error as well as unset variables
  set -eu -o pipefail
  init "$@"
fi