#!/usr/bin/env bash
# Setup script for rapid7/metasploit-framework on macOS, from https://gitmanual.com/rapid7/metasploit-framework
# Read it before you run it: the commands come from the repo's own README and files.
# Run: bash setup.sh
set -eo pipefail
trap 'echo "Setup stopped at the step above. Steps and fixes: https://gitmanual.com/rapid7/metasploit-framework" >&2' ERR
step() { printf '\n\033[1m==> %s\033[0m\n' "$1"; }
have() { command -v "$1" >/dev/null 2>&1; }
# Installers put programs here; make them usable right away.
export PATH="$HOME/.local/share/mise/shims:$HOME/.cargo/bin:$HOME/.local/bin:$HOME/.deno/bin:$HOME/.bun/bin:$PATH"
# nvm is a shell function, so load it into this script when it's installed.
export NVM_DIR="${NVM_DIR:-$HOME/.nvm}"
load_nvm() { [ -s "$NVM_DIR/nvm.sh" ] && . "$NVM_DIR/nvm.sh" || { have brew && [ -s "$(brew --prefix nvm 2>/dev/null)/nvm.sh" ] && mkdir -p "$NVM_DIR" && . "$(brew --prefix nvm)/nvm.sh"; }; }
have_nvm() { load_nvm >/dev/null 2>&1; type nvm >/dev/null 2>&1; }

step '1/4 Clone the repository'
[ -d 'metasploit-framework' ] || git clone --recurse-submodules https://github.com/rapid7/metasploit-framework.git
cd 'metasploit-framework'

step '2/4 Install the required runtimes'
# Ruby 3.3.13
have rbenv || brew install rbenv ruby-build
if have rbenv; then rbenv install --skip-existing; elif have ruby; then echo "rbenv not found: using $(ruby --version)"; else echo "Needs Ruby: install rbenv from https://github.com/rbenv/rbenv" >&2; exit 1; fi

step '3/4 Start backing services'
# Needs: PostgreSQL
# Docker Desktop
have docker || brew install --cask docker
docker compose -f docker-compose.yml up -d

step '4/4 Install dependencies'
bundle install

echo 'Set up rapid7/metasploit-framework.'
echo 'Manual: https://gitmanual.com/rapid7/metasploit-framework'