aztec-install

#!/usr/bin/env bash
set -euo pipefail

# Colors
g="\033[32m" # Green
y="\033[33m" # Yellow
b="\033[34m" # Blue
p="\033[35m" # Purple
r="\033[0m"  # Reset
bold="\033[1m"

# Function to replace characters and add color
function print_colored() {
  local b=$'\033[34m' # Blue
  local y=$'\033[33m' # Yellow
  local r=$'\033[0m'  # Reset
  echo "$1" | sed -E "s/(█+)/${b}\1${y}/g"
}

function title() {
  echo
  print_colored " █████╗ ███████╗████████╗███████╗ ██████╗"
  print_colored "██╔══██╗╚══███╔╝╚══██╔══╝██╔════╝██╔════╝"
  print_colored "███████║  ███╔╝    ██║   █████╗  ██║"
  print_colored "██╔══██║ ███╔╝     ██║   ██╔══╝  ██║"
  print_colored "██║  ██║███████╗   ██║   ███████╗╚██████╗"
  print_colored "╚═╝  ╚═╝╚══════╝   ╚═╝   ╚══════╝ ╚═════╝"
  echo -e "${r}"
  echo -e "Welcome to the ${bold}${b}Aztec${r} installer! Your journey into blockchain privacy begins... ${bold}${p}now${r}."
  echo -e "We provide a docker image that works on any modern system (specifically: x64 or ARM64)."
  echo -e "Please ensure docker is installed for your platform: https://docs.docker.com/engine/install"
  echo
  if [ "$(uname -s)" == "Darwin" ]; then
    echo -e "${y}WARNING: For best performance we recommend adjusting your default docker settings:"
    echo -e "         - Under general, enable VirtioFS."
    echo -e "         - Under resources, set CPUs to ~80% your maximum."
    echo -e "         - Under resources, set Memory to ~80% your maximum."
    echo -e "You may receive a warning about your home directory being mounted into a container."
    echo -e "This is requested so we can read and write project files, that is all."
    echo -e "${r}"
  fi
  echo -e "This will install the following scripts and update your PATH if necessary:"
  echo -e "  ${bold}${g}aztec${r}         - a collection of tools to launch subsystems and interact with the aztec network."
  echo -e "  ${bold}${g}aztec-nargo${r}   - aztec's build of nargo, the noir compiler toolchain."
  echo -e "  ${bold}${g}aztec-up${r}      - a tool to upgrade the aztec toolchain to the latest, or specific versions."
  echo -e "  ${bold}${g}aztec-wallet${r}  - our minimalistic CLI wallet"
  echo
  read -p "Do you wish to continue? (y/n)" -n 1 -r
  echo
  if [[ ! $REPLY =~ ^[Yy]$ ]]; then
    exit 0
  fi
}

function info {
  echo -e "${g}$1${r}"
}

function warn {
  echo -e "${y}$1${r}"
}

AZTEC_PATH=$HOME/.aztec
BIN_PATH=${BIN_PATH:-$AZTEC_PATH/bin}

# Define version if specified, otherwise set to "latest".
VERSION=${VERSION:-"latest"}
INSTALL_HOST=install.aztec.network.s3-website.eu-west-2.amazonaws.com

[ -z "${NON_INTERACTIVE:-}" ] && title

# Check if Docker is available.
if ! command -v docker &>/dev/null; then
  warn "Docker is not installed. Please install Docker and try again."
  exit 1
fi

# Check if Docker is running.
if ! docker info &>/dev/null; then
  warn "Docker is not running. Please start Docker and try again."
  exit 1
fi

if ! docker compose &>/dev/null && ! command -v docker-compose &>/dev/null; then
  warn "WARNING: 'docker compose' not supported and docker-compose not found."
  warn "Continuing installation, but aztec-sandbox will not work."
fi

# Create a "hidden" `$HOME/.aztec` dir, so as not to clutter the user's cwd.
mkdir -p $AZTEC_PATH
mkdir -p $BIN_PATH

# Download containers from dockerhub. Tag them as latest.
function pull_container {
  docker pull aztecprotocol/$1:$VERSION

  # If not latest, retag to be latest so it runs from scripts.
  if [ $VERSION != "latest" ]; then
    docker tag aztecprotocol/$1:$VERSION aztecprotocol/$1:latest
  fi
}

export DOCKER_CLI_HINTS=false

if [ -z "${SKIP_PULL:-}" ]; then
  info "Pulling aztec version $VERSION..."
  pull_container aztec-nargo
  pull_container aztec
  pull_container cli-wallet
fi

# Download the Docker Compose file. Used by aztec.
curl -fsSL http://$INSTALL_HOST/docker-compose.sandbox.yml -o $AZTEC_PATH/docker-compose.sandbox.yml
curl -fsSL http://$INSTALL_HOST/docker-compose.test.yml -o $AZTEC_PATH/docker-compose.test.yml

function install_bin {
  local install_url
  if [ "$VERSION" != "latest" ]; then
    install_url="http://$INSTALL_HOST/$VERSION/$1"
  else
    install_url="http://$INSTALL_HOST/$1"
  fi
  curl -fsSL "$install_url" -o "$BIN_PATH/$1"
  chmod +x "$BIN_PATH/$1"
  echo "Installed: $BIN_PATH/$1"
}

info "Installing scripts in $BIN_PATH..."
rm -f $BIN_PATH/aztec*
install_bin .aztec-run
install_bin aztec
install_bin aztec-up
install_bin aztec-nargo
install_bin aztec-wallet

function update_path_env_var {
  TARGET_DIR="${1}"
  # Check if the target directory is in the user's PATH.
  if [[ ":$PATH:" != *":$TARGET_DIR:"* ]]; then
    # Determine the user's shell.
    SHELL_PROFILE=""
    case $SHELL in
    */bash)
      SHELL_PROFILE="$HOME/.bashrc"
      ;;
    */zsh)
      SHELL_PROFILE="$HOME/.zshrc"
      ;;
    # Add other shells as needed
    *)
      echo "Unsupported shell: $SHELL"
      return
      ;;
    esac

    if [ -z "${NON_INTERACTIVE:-}" ]; then
      # Inform the user about the change and ask for confirmation
      warn "The directory $TARGET_DIR is not in your PATH."
      read -p "Add it to $SHELL_PROFILE to make the aztec binaries accessible? (y/n)" -n 1 -r
      echo
      if [[ ! $REPLY =~ ^[Yy]$ ]]; then
        warn "Skipped updating PATH. You might need to add $TARGET_DIR to your PATH manually to use the binary."
        return
      fi
    fi

    # Add the target directory to the user's PATH in their profile.
    echo "export PATH=\"\$PATH:$TARGET_DIR\"" >>"$SHELL_PROFILE"

    if [ -z "${NON_INTERACTIVE:-}" ]; then
      info "Done! Starting fresh shell..."
      $SHELL
    fi
  fi
}

update_path_env_var $BIN_PATH

info "Done!"