#!/usr/bin/env bash
# Setup script for microsoft/TokenEconomizer on Linux, from https://gitmanual.com/microsoft/TokenEconomizer
# 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/microsoft/TokenEconomizer" >&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_nvm() { load_nvm >/dev/null 2>&1; type nvm >/dev/null 2>&1; }

step '1/3 Clone the repository'
[ -d 'TokenEconomizer' ] || git clone https://github.com/microsoft/TokenEconomizer.git
cd 'TokenEconomizer'

step '2/3 Install the required runtimes'
# Node.js 22, Python >=3.10
# needs nvm: https://github.com/nvm-sh/nvm
if have_nvm; then nvm install 22; elif have node; then echo "nvm not found: using Node $(node -v)"; else echo "Needs Node.js: install nvm from https://github.com/nvm-sh/nvm" >&2; exit 1; fi
# Debian/Ubuntu
have python3 || sudo apt install -y python3 python3.10 python3-venv

step '3/3 Install dependencies'
npm install
python -m venv .venv
source .venv/bin/activate
pip install -e .

echo 'Set up microsoft/TokenEconomizer.'
echo 'Next, from the project folder:'
echo '  pytest'
echo 'Manual: https://gitmanual.com/microsoft/TokenEconomizer'