#!/bin/bash
# Anvil Mesh Node — One-liner installer
# Usage: curl -fsSL https://anvil.sendbsv.com/install | sudo bash
#
# Requirements: Linux (amd64 or arm64), root/sudo, ~50MB disk

set -euo pipefail

ANVIL_VERSION="${ANVIL_VERSION:-latest}"
ANVIL_REPO="BSVanon/Anvil"
INSTALL_DIR="/opt/anvil"
SEED_PEER="${ANVIL_SEED:-wss://anvil.sendbsv.com/mesh}"
NODE_NAME="${ANVIL_NAME:-}"
API_PORT="9333"
CONFIG_FILE="/etc/anvil/node-a.toml"

# ── Colors & formatting ──
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
CYAN='\033[0;36m'
BOLD='\033[1m'
DIM='\033[2m'
NC='\033[0m'

pause_msg() {
  echo ""
  echo -e "  ${DIM}press enter to continue...${NC}"
  read -r < /dev/tty
}

# ── Detect architecture ──
ARCH=$(uname -m)
case "$ARCH" in
  x86_64)  BINARY="anvil-linux-amd64" ;;
  aarch64) BINARY="anvil-linux-arm64" ;;
  *)
    echo -e "${RED}Error: unsupported architecture $ARCH (need x86_64 or aarch64)${NC}"
    exit 1
    ;;
esac

# ── Check root ──
if [ "$(id -u)" -ne 0 ]; then
  echo -e "${RED}Error: run with sudo or as root${NC}"
  echo "  curl -fsSL https://anvil.sendbsv.com/install | sudo bash"
  exit 1
fi

# ══════════════════════════════════════════════════════════════
# SCREEN 1: Welcome
# ══════════════════════════════════════════════════════════════

clear
echo ""
echo ""
cat << 'BANNER'
       ╔═══════════════════════════════════════════════════════╗
       ║                                                       ║
       ║              ▄▀█ █▄░█ █░█ █ █░░                      ║
       ║              █▀█ █░▀█ ▀▄▀ █ █▄▄                      ║
       ║                                                       ║
       ║          BSV SPV Node · x402 Payments · Mesh          ║
       ║                                                       ║
       ╚═══════════════════════════════════════════════════════╝
BANNER
echo ""
echo ""
echo -e "  Welcome. This script will turn this machine into an"
echo -e "  Anvil mesh node in about 3 minutes."
echo ""
echo -e "  ${DIM}What happens next:${NC}"
echo ""
echo -e "    ${GREEN}▸${NC} Download the Anvil binary"
echo -e "    ${GREEN}▸${NC} Generate your node's unique identity"
echo -e "    ${GREEN}▸${NC} Sync BSV block headers"
echo -e "    ${GREEN}▸${NC} Show your wallet address to fund"
echo ""
echo -e "  ${DIM}You will need about 1,000,000 satoshis (~\$0.50 USD)${NC}"
echo -e "  ${DIM}to fund the node's wallet. Have a BSV wallet ready.${NC}"
echo ""

pause_msg

# ══════════════════════════════════════════════════════════════
# SCREEN 2: Download + Install
# ══════════════════════════════════════════════════════════════

clear
echo ""
echo ""
echo -e "  ${BOLD}━━━ STEP 1 of 4: Installing Anvil ━━━${NC}"
echo ""
echo ""
echo -e "  ${DIM}Downloading binary for ${ARCH}...${NC}"
echo ""

TMPBIN=$(mktemp /tmp/anvil-install.XXXXXX)
trap "rm -f $TMPBIN" EXIT

# Allow local binary for testing: ANVIL_LOCAL_BINARY=/path/to/anvil
if [ -n "${ANVIL_LOCAL_BINARY:-}" ] && [ -f "$ANVIL_LOCAL_BINARY" ]; then
  cp "$ANVIL_LOCAL_BINARY" "$TMPBIN"
  echo -e "    ${DIM}(using local binary: ${ANVIL_LOCAL_BINARY})${NC}"
else
  if [ "$ANVIL_VERSION" = "latest" ]; then
    DOWNLOAD_URL="https://github.com/${ANVIL_REPO}/releases/latest/download/${BINARY}"
  else
    DOWNLOAD_URL="https://github.com/${ANVIL_REPO}/releases/download/${ANVIL_VERSION}/${BINARY}"
  fi

  if command -v curl &>/dev/null; then
    curl -fsSL "$DOWNLOAD_URL" -o "$TMPBIN"
  elif command -v wget &>/dev/null; then
    wget -q "$DOWNLOAD_URL" -O "$TMPBIN"
  else
    echo -e "  ${RED}Error: curl or wget required${NC}"
    exit 1
  fi
fi
chmod 755 "$TMPBIN"

# Stop any running instance before overwriting
systemctl stop anvil-a 2>/dev/null || true
systemctl stop anvil-b 2>/dev/null || true
sleep 1

mkdir -p "$INSTALL_DIR"
cp "$TMPBIN" "${INSTALL_DIR}/anvil"
chmod 755 "${INSTALL_DIR}/anvil"

echo -e "    ${GREEN}✓${NC} Binary downloaded and installed"
echo ""

# ══════════════════════════════════════════════════════════════
# SCREEN 2b: Deploy (identity, config, systemd)
# ══════════════════════════════════════════════════════════════

echo -e "  ${DIM}Generating identity and configuring services...${NC}"
echo ""

DEPLOY_ARGS="--nodes a --skip-health"
if [ -n "$SEED_PEER" ]; then
  DEPLOY_ARGS="$DEPLOY_ARGS --seed $SEED_PEER"
fi
if [ -n "$NODE_NAME" ]; then
  DEPLOY_ARGS="$DEPLOY_ARGS --name $NODE_NAME"
fi

"$TMPBIN" deploy $DEPLOY_ARGS >/dev/null 2>&1

echo -e "    ${GREEN}✓${NC} Identity generated"
echo -e "    ${GREEN}✓${NC} Config written to /etc/anvil/"
echo -e "    ${GREEN}✓${NC} Systemd service created and started"
echo ""
echo ""
echo -e "  ${GREEN}${BOLD}  Installation complete.${NC}"
echo ""

sleep 2

# ══════════════════════════════════════════════════════════════
# SCREEN 3: Identity — SAVE THIS
# ══════════════════════════════════════════════════════════════

# Get node info
NODE_INFO=$("${INSTALL_DIR}/anvil" info -config "$CONFIG_FILE" -json 2>/dev/null || echo "")

IDENTITY=""
WALLET_ADDR=""
AUTH_TOKEN=""

if [ -n "$NODE_INFO" ]; then
  IDENTITY=$(echo "$NODE_INFO" | python3 -c "import sys,json; print(json.load(sys.stdin).get('identity_key',''))" 2>/dev/null || echo "")
  WALLET_ADDR=$(echo "$NODE_INFO" | python3 -c "import sys,json; print(json.load(sys.stdin).get('address',''))" 2>/dev/null || echo "")
  AUTH_TOKEN=$(echo "$NODE_INFO" | python3 -c "import sys,json; print(json.load(sys.stdin).get('auth_token',''))" 2>/dev/null || echo "")
fi

clear
echo ""
echo ""
echo -e "  ${BOLD}━━━ STEP 2 of 4: Your Node Identity ━━━${NC}"
echo ""
echo ""
echo -e "  ${RED}${BOLD}  ╔════════════════════════════════════════════════════╗${NC}"
echo -e "  ${RED}${BOLD}  ║                                                    ║${NC}"
echo -e "  ${RED}${BOLD}  ║   WRITE THIS DOWN. YOUR PRIVATE KEY CANNOT BE      ║${NC}"
echo -e "  ${RED}${BOLD}  ║   RECOVERED IF YOU LOSE IT.                        ║${NC}"
echo -e "  ${RED}${BOLD}  ║                                                    ║${NC}"
echo -e "  ${RED}${BOLD}  ╚════════════════════════════════════════════════════╝${NC}"
echo ""
echo ""
echo -e "  ${BOLD}Your private key (WIF) is stored in:${NC}"
echo ""
echo -e "    ${CYAN}/etc/anvil/node-a.env${NC}"
echo ""
echo -e "  ${BOLD}View it now and copy it somewhere safe:${NC}"
echo ""
echo -e "    ${CYAN}sudo cat /etc/anvil/node-a.env${NC}"
echo ""
echo ""
if [ -n "$IDENTITY" ]; then
  echo -e "  ${DIM}Your node's public identity:${NC}"
  echo -e "    ${IDENTITY}"
  echo ""
fi

pause_msg

# ══════════════════════════════════════════════════════════════
# SCREEN 4: Fund your node
# ══════════════════════════════════════════════════════════════

clear
echo ""
echo ""
echo -e "  ${BOLD}━━━ STEP 3 of 4: Fund Your Node ━━━${NC}"
echo ""
echo ""
echo -e "  Your node needs BSV to operate. Send ${BOLD}1,000,000 satoshis${NC}"
echo -e "  (about \$0.50 USD) to this address:"
echo ""
echo ""

if [ -n "$WALLET_ADDR" ]; then
  echo -e "       ┌──────────────────────────────────────────┐"
  echo -e "       │                                          │"
  echo -e "       │   ${GREEN}${BOLD}${WALLET_ADDR}${NC}   │"
  echo -e "       │                                          │"
  echo -e "       └──────────────────────────────────────────┘"

  # Try to generate a QR code if qrencode is available
  if command -v qrencode &>/dev/null; then
    echo ""
    echo -e "  ${DIM}Scan with your wallet:${NC}"
    echo ""
    qrencode -t ANSIUTF8 -m 2 "bitcoin:${WALLET_ADDR}?sv&amount=0.01" 2>/dev/null | while IFS= read -r line; do
      echo "       $line"
    done
  else
    # Try to install qrencode silently
    apt-get install -y qrencode >/dev/null 2>&1 || true
    if command -v qrencode &>/dev/null; then
      echo ""
      echo -e "  ${DIM}Scan with your wallet:${NC}"
      echo ""
      qrencode -t ANSIUTF8 -m 2 "bitcoin:${WALLET_ADDR}?sv&amount=0.01" 2>/dev/null | while IFS= read -r line; do
        echo "       $line"
      done
    fi
  fi
else
  echo -e "  ${RED}Could not derive address. Run manually:${NC}"
  echo -e "    ${CYAN}anvil info -config ${CONFIG_FILE}${NC}"
fi

echo ""
echo ""
echo -e "  ${YELLOW}${BOLD}After sending, wait for 1 confirmation.${NC}"
echo ""
echo -e "  ${DIM}While you wait, your node is syncing block headers${NC}"
echo -e "  ${DIM}in the background. We'll show you the progress next.${NC}"
echo ""

pause_msg

# ══════════════════════════════════════════════════════════════
# SCREEN 5: Header sync progress
# ══════════════════════════════════════════════════════════════

clear
echo ""
echo ""
echo -e "  ${BOLD}━━━ STEP 4 of 4: Syncing Block Headers ━━━${NC}"
echo ""
echo ""
echo -e "  ${DIM}Your node is downloading BSV block headers (80 bytes each).${NC}"
echo -e "  ${DIM}This is NOT the full blockchain — just headers for SPV${NC}"
echo -e "  ${DIM}verification. Usually takes 1-3 minutes.${NC}"
echo ""
echo ""

LAST_HEIGHT=0
PREV_HEIGHT=0
STALL_COUNT=0
SYNCED=false
MAX_WAIT=300
WAITED=0

while [ "$WAITED" -lt "$MAX_WAIT" ]; do
  STATUS=$(curl -s "http://127.0.0.1:${API_PORT}/status" 2>/dev/null || echo "")

  if [ -n "$STATUS" ]; then
    HEIGHT=$(echo "$STATUS" | python3 -c "import sys,json; print(json.load(sys.stdin).get('headers',{}).get('height',0))" 2>/dev/null || echo "0")

    if [ "$HEIGHT" -gt 0 ]; then
      if [ "$HEIGHT" = "$PREV_HEIGHT" ] && [ "$HEIGHT" -gt 900000 ]; then
        STALL_COUNT=$((STALL_COUNT + 1))
        if [ "$STALL_COUNT" -ge 3 ]; then
          SYNCED=true
          LAST_HEIGHT=$HEIGHT
          break
        fi
      else
        STALL_COUNT=0
      fi
      PREV_HEIGHT=$HEIGHT
      LAST_HEIGHT=$HEIGHT

      APPROX_TIP=942000
      [ "$HEIGHT" -gt "$APPROX_TIP" ] && APPROX_TIP=$HEIGHT
      PCT=$((HEIGHT * 100 / APPROX_TIP))
      BAR_WIDTH=40
      FILLED=$((PCT * BAR_WIDTH / 100))
      EMPTY=$((BAR_WIDTH - FILLED))
      BAR=""
      SPACE=""
      for ((i=0; i<FILLED; i++)); do BAR="${BAR}█"; done
      for ((i=0; i<EMPTY; i++)); do SPACE="${SPACE}░"; done

      printf "\r    ${CYAN}${BAR}${SPACE}${NC}  ${BOLD}%3d%%${NC}  block %s  " "$PCT" "$HEIGHT"
    fi
  else
    printf "\r    ${DIM}⏳  Starting up...${NC}                                        "
  fi

  sleep 2
  WAITED=$((WAITED + 2))
done

echo ""

if [ "$SYNCED" = true ]; then
  echo ""
  echo -e "    ${GREEN}✓${NC} Synced to block ${BOLD}${LAST_HEIGHT}${NC} — your node is at the chain tip."
else
  echo ""
  echo -e "    ${YELLOW}⏳${NC} Still syncing (block ${LAST_HEIGHT}). Running in background."
  echo -e "    ${DIM}   Check: curl -s http://localhost:${API_PORT}/status${NC}"
fi

sleep 2

# ══════════════════════════════════════════════════════════════
# SCREEN 6: Final — claim funds + you're live
# ══════════════════════════════════════════════════════════════

clear
echo ""
echo ""
cat << 'DONE_ART'

       ╔═══════════════════════════════════════════════════════╗
       ║                                                       ║
       ║              ◆  YOUR NODE IS LIVE  ◆                  ║
       ║                                                       ║
       ╚═══════════════════════════════════════════════════════╝

DONE_ART

echo ""
echo -e "  Headers are synced. Your node is connected to the Anvil"
echo -e "  mesh and peering with other nodes automatically."
echo ""
echo ""

if [ -n "$WALLET_ADDR" ] && [ -n "$AUTH_TOKEN" ]; then
  echo -e "  ${BOLD}LAST STEP: Claim your funds${NC}"
  echo ""
  echo -e "  ${DIM}If you've already sent BSV to your address and it has${NC}"
  echo -e "  ${DIM}at least 1 confirmation, run this command to import${NC}"
  echo -e "  ${DIM}the funds into your node's wallet:${NC}"
  echo ""
  echo -e "    ${CYAN}curl -X POST http://localhost:${API_PORT}/wallet/scan \\${NC}"
  echo -e "    ${CYAN}  -H \"Authorization: Bearer ${AUTH_TOKEN}\"${NC}"
  echo ""
  echo -e "  ${DIM}Run this any time you send more funds to your node.${NC}"
  echo ""
  echo ""
fi

echo -e "  ${DIM}─────────────────────────────────────────────────────────${NC}"
echo ""
echo -e "  ${BOLD}YOUR NODE${NC}"
echo ""
if [ -n "$WALLET_ADDR" ]; then
  echo -e "    Address:     ${GREEN}${WALLET_ADDR}${NC}"
fi
if [ -n "$IDENTITY" ]; then
  echo -e "    Identity:    ${DIM}${IDENTITY}${NC}"
fi
echo -e "    API:         ${CYAN}http://localhost:${API_PORT}/status${NC}"
echo -e "    Config:      /etc/anvil/node-a.toml"
echo -e "    Private key: /etc/anvil/node-a.env"
echo ""
echo -e "  ${DIM}─────────────────────────────────────────────────────────${NC}"
echo ""
echo -e "  ${BOLD}USEFUL COMMANDS${NC}"
echo ""
echo -e "    ${YELLOW}${BOLD}anvil help${NC}                                 ${YELLOW}All commands${NC}"
echo -e "    ${CYAN}sudo anvil info${NC}                            Node info"
echo -e "    ${CYAN}sudo anvil doctor${NC}                          Diagnostics"
echo -e "    ${CYAN}sudo journalctl -u anvil-a -f${NC}              Live logs"
echo -e "    ${CYAN}sudo systemctl restart anvil-a${NC}             Restart"
echo ""
echo -e "  ${DIM}─────────────────────────────────────────────────────────${NC}"
echo ""
echo -e "  ${BOLD}RENAME YOUR NODE${NC}"
echo ""
echo -e "  ${DIM}Your node was auto-named from its identity key.${NC}"
echo -e "  ${DIM}To give it a custom name:${NC}"
echo ""
echo -e "    ${CYAN}sudo sed -i 's|name = \"anvil-.*\"|name = \"my-node-name\"|' ${CONFIG_FILE}${NC}"
echo -e "    ${CYAN}sudo systemctl restart anvil-a${NC}"
echo ""
echo -e "  ${DIM}─────────────────────────────────────────────────────────${NC}"
echo ""
echo -e "  ${RED}${BOLD}  OPEN YOUR FIREWALL${NC}"
echo ""
echo -e "  ${RED}  Ports 8333 (mesh) and 9333 (API) MUST be open for${NC}"
echo -e "  ${RED}  inbound connections or your node cannot join the mesh.${NC}"
echo ""
echo -e "    ${CYAN}sudo ufw allow 8333/tcp${NC}"
echo -e "    ${CYAN}sudo ufw allow 9333/tcp${NC}"
echo ""
echo -e "  ${DIM}─────────────────────────────────────────────────────────${NC}"
echo ""
echo -e "  ${BOLD}YOUR EXPLORER${NC}"
echo ""
echo -e "  ${DIM}Visit your node's explorer in a browser:${NC}"
echo ""
echo -e "    ${CYAN}http://$(curl -4 -s ifconfig.me 2>/dev/null || echo '<your-ip>'):${API_PORT}/explorer${NC}"
echo ""
echo -e "  ${DIM}To log in and manage your wallet from the Explorer:${NC}"
echo ""
echo -e "    ${CYAN}sudo anvil token${NC}    ${DIM}→ copy the token → paste in Explorer → Node Login${NC}"
echo ""
echo ""
echo -e "  ${GREEN}${BOLD}Welcome to the Anvil mesh. ◆${NC}"
echo ""
echo ""