test.sh

#!/bin/bash
# local_test.sh — Spins up a complete local testing environment for Seville.
#
# Steps:
#   1. Start Anvil (local Ethereum node)
#   2. Deploy contracts and generate NodeConfig.toml
#   3. Start dstack simulator
#   4. cargo run lit-api-server
#   5. cargo run lit-actions
#   6. Serve lit-static via static-web-server
#
# Press Ctrl+C to tear down all background processes.

set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$SCRIPT_DIR"

# Simulator path (override with SIMULATOR_DIR env var if cloned elsewhere)
SIMULATOR_DIR="${SIMULATOR_DIR:-$HOME/GitHub/dstack/sdk/simulator}"
SIMULATOR_BIN="$SIMULATOR_DIR/dstack-simulator"

# Track background PIDs for cleanup
PIDS=()
SIM_TMP_OWNED=false

cleanup() {
    echo ""
    echo "==> Shutting down local environment..."
    for pid in "${PIDS[@]}"; do
        kill "$pid" 2>/dev/null || true
    done
    # Only clean up simulator temp dir if we created it
    if [ "$SIM_TMP_OWNED" = true ] && [ -n "${SIM_TMP:-}" ] && [ -d "${SIM_TMP:-}" ]; then
        rm -rf "$SIM_TMP"
    fi
    echo "==> All processes stopped."
}
trap cleanup EXIT INT TERM

# --------------------------------------------------------------------------
# 1. Start Anvil
# --------------------------------------------------------------------------
echo "==> Step 1: Starting Anvil..."

if ! command -v anvil &>/dev/null; then
    echo "ERROR: anvil is not installed."
    echo ""
    echo "Install Foundry (which includes anvil) by running:"
    echo "  curl -L https://foundry.paradigm.xyz | bash"
    echo "  foundryup"ƒ
    echo ""
    echo "Then re-run this script."
    exit 1
fi

anvil --silent &
ANVIL_PID=$!
PIDS+=("$ANVIL_PID")

# Wait for Anvil to be ready
for i in $(seq 1 10); do
    if curl -sf http://127.0.0.1:8545 -X POST \
        -H "Content-Type: application/json" \
        -d '{"jsonrpc":"2.0","method":"eth_blockNumber","params":[],"id":1}' \
        >/dev/null 2>&1; then
        echo "    Anvil is ready (PID $ANVIL_PID)."
        break
    fi
    if [ "$i" -eq 10 ]; then
        echo "ERROR: Anvil failed to start."
        exit 1
    fi
    sleep 1
done

# --------------------------------------------------------------------------
# 3. Start dstack simulator
# --------------------------------------------------------------------------
echo "==> Step 3: Starting dstack simulator..."

if [ ! -f "$SIMULATOR_BIN" ]; then
    echo "ERROR: dstack-simulator not found at $SIMULATOR_BIN"
    echo ""
    echo "Clone and build the simulator:"
    echo "  git clone https://github.com/Dstack-TEE/dstack.git ~/GitHub/dstack"
    echo "  cd ~/GitHub/dstack/sdk/simulator && bash build.sh"
    echo ""
    echo "Then re-run this script."
    exit 1
fi

# Check if simulator is already running
if pgrep -x dstack-simulator >/dev/null 2>&1; then
    echo "    dstack-simulator is already running."
    # Find the existing socket
    EXISTING_STATE="/tmp/dstack-sim.state"
    if [ -f "$EXISTING_STATE" ]; then
        _EXISTING_DIR=$(sed -n '1p' "$EXISTING_STATE")
        DSTACK_SOCKET="$_EXISTING_DIR/dstack.sock"
    else
        DSTACK_SOCKET="/tmp/dstack.sock"
    fi
    if [ ! -S "$DSTACK_SOCKET" ]; then
        echo "ERROR: dstack-simulator process found but socket $DSTACK_SOCKET does not exist."
        echo "       Kill the stale process (pkill dstack-simulator) and re-run."
        exit 1
    fi
else
    # Create isolated temp dir (same approach as justfile.sim)
    SIM_TMP=$(mktemp -d /tmp/dstack-sim-XXXXXX)
    SIM_TMP_OWNED=true
    DSTACK_SOCKET="$SIM_TMP/dstack.sock"

    cp "$SIMULATOR_DIR/appkeys.json" "$SIMULATOR_DIR/app-compose.json" \
       "$SIMULATOR_DIR/sys-config.json" "$SIMULATOR_DIR/attestation.bin" \
       "$SIMULATOR_DIR/dstack.toml" "$SIM_TMP/"

    (cd "$SIM_TMP" && exec "$SIMULATOR_BIN") >> "$SIM_TMP/dstack-simulator.log" 2>&1 &
    SIM_PID=$!
    PIDS+=("$SIM_PID")

    for i in $(seq 1 15); do
        if [ -S "$DSTACK_SOCKET" ]; then
            echo "    Simulator ready at $DSTACK_SOCKET (PID $SIM_PID)."
            break
        fi
        if [ "$i" -eq 15 ]; then
            echo "ERROR: dstack.sock never appeared."
            echo ""
            echo "Check log: $SIM_TMP/dstack-simulator.log"
            echo "You may need to set execute permissions:"
            echo "  chmod +x $SIMULATOR_BIN"
            exit 1
        fi
        printf "    waiting for dstack.sock (%d/15)...\n" "$i"
        sleep 1
    done
fi

export DSTACK_SOCKET


# --------------------------------------------------------------------------
# 2. Deploy contracts and create NodeConfig.toml
# --------------------------------------------------------------------------
echo "==> Step 2: Deploying contracts to Anvil..."

CONTRACTS_DIR="$SCRIPT_DIR/lit-api-server/blockchain/lit_node_express"

# Run the deploy and capture output to extract the AccountConfig address
DEPLOY_OUTPUT=$(make -C "$CONTRACTS_DIR" deploy_anvil 2>&1) || {
    echo "ERROR: Contract deployment failed."
    echo "$DEPLOY_OUTPUT"
    exit 1
}
echo "$DEPLOY_OUTPUT"

# Extract the last "deployed to 0x..." address — this is the AccountConfig (diamond proxy)
CONTRACT_ADDRESS=$(echo "$DEPLOY_OUTPUT" | grep -oE 'deployed to 0x[0-9a-fA-F]+' | tail -1 | sed 's/deployed to //')

if [ -z "$CONTRACT_ADDRESS" ] || [ ${#CONTRACT_ADDRESS} -ne 42 ]; then
    echo "ERROR: Could not extract a valid contract address from deploy output."
    echo "       Got: '${CONTRACT_ADDRESS:-<empty>}' (expected 0x + 40 hex chars)"
    exit 1
fi

echo "    Diamond proxy deployed at: $CONTRACT_ADDRESS"

# Write NodeConfig.toml for lit-api-server
cat > "$SCRIPT_DIR/lit-api-server/NodeConfig.toml" <<EOF
[chain]
name = "anvil"
contract_address = "$CONTRACT_ADDRESS"
EOF
echo "    NodeConfig.toml written."

# --------------------------------------------------------------------------
# 4. cargo run lit-api-server
# --------------------------------------------------------------------------
echo "==> Step 4: Starting lit-api-server..."

(cd "$SCRIPT_DIR/lit-api-server" && \
    DSTACK_SOCKET="$DSTACK_SOCKET" \
    cargo run --bin lit-api-server) &
API_PID=$!
PIDS+=("$API_PID")

# Wait for lit-api-server to respond
for i in $(seq 1 60); do
    if curl -sf http://localhost:8000/core/v1/health >/dev/null 2>&1 || \
       curl -sf http://localhost:8000/ >/dev/null 2>&1; then
        echo "    lit-api-server is ready (PID $API_PID)."
        break
    fi
    if ! kill -0 "$API_PID" 2>/dev/null; then
        echo "ERROR: lit-api-server process exited unexpectedly."
        exit 1
    fi
    if [ "$i" -eq 60 ]; then
        echo "    WARNING: lit-api-server may still be compiling/starting. Continuing..."
    fi
    sleep 2
done

# --------------------------------------------------------------------------
# 5. cargo run lit-actions
# --------------------------------------------------------------------------
echo "==> Step 5: Starting lit-actions..."

(cd "$SCRIPT_DIR" && \
    cargo run --manifest-path=lit-actions/Cargo.toml --bin lit_actions) &
ACTIONS_PID=$!
PIDS+=("$ACTIONS_PID")

# Wait for the unix socket to appear
for i in $(seq 1 60); do
    if [ -S /tmp/lit_actions.sock ]; then
        echo "    lit-actions is ready (PID $ACTIONS_PID)."
        break
    fi
    if ! kill -0 "$ACTIONS_PID" 2>/dev/null; then
        echo "ERROR: lit-actions process exited unexpectedly."
        exit 1
    fi
    if [ "$i" -eq 60 ]; then
        echo "    WARNING: lit-actions may still be compiling/starting. Continuing..."
    fi
    sleep 2
done

# --------------------------------------------------------------------------
# 6. Serve lit-static via static-web-server
# --------------------------------------------------------------------------
echo "==> Step 6: Starting static-web-server for lit-static..."

if ! command -v static-web-server &>/dev/null; then
    echo "ERROR: static-web-server is not installed."
    echo ""
    echo "Install via Homebrew:"
    echo "  brew install static-web-server"
    echo ""
    echo "Then re-run this script."
    exit 1
fi

static-web-server -p 8080 -d "$SCRIPT_DIR/lit-static" -g info &
STATIC_PID=$!
PIDS+=("$STATIC_PID")
echo "    static-web-server serving lit-static on http://localhost:8080 (PID $STATIC_PID)."

# --------------------------------------------------------------------------
# Ready
# --------------------------------------------------------------------------
echo ""
echo "============================================"
echo "  Local environment is running!"
echo "============================================"
echo "  Anvil (chain):        http://127.0.0.1:8545"
echo "  lit-api-server:       http://localhost:8000"
echo "  lit-actions (gRPC):   /tmp/lit_actions.sock"
echo "  lit-static:           http://localhost:8080"
echo "  dstack simulator:     $DSTACK_SOCKET"
echo "  Contract address:     $CONTRACT_ADDRESS"
echo "============================================"
echo ""
echo "Press Ctrl+C to stop all services."
echo ""

# Wait for any background process to exit
wait