proxmox/scripts/deployment/sweep-mainnet-aave-quote-push-receiver-surplus.sh

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

# Sweep retained surplus from a deployed Mainnet Aave quote-push receiver.
# Default: simulation only. Use --apply to broadcast.
#
# Env:
#   PRIVATE_KEY, ETHEREUM_MAINNET_RPC, AAVE_QUOTE_PUSH_RECEIVER_MAINNET
#   QUOTE_PUSH_SURPLUS_TOKEN_MAINNET    optional; defaults to mainnet USDC
#   QUOTE_PUSH_SURPLUS_RECIPIENT        optional; defaults to deployer derived from PRIVATE_KEY
#   QUOTE_PUSH_SURPLUS_RESERVE_RAW      optional; keep this much on the receiver when sweeping surplus mode
#   QUOTE_PUSH_SURPLUS_EXACT_AMOUNT_RAW optional; if > 0, sweep this exact amount instead of balance - reserve
#
# Usage:
#   source scripts/lib/load-project-env.sh
#   bash scripts/deployment/sweep-mainnet-aave-quote-push-receiver-surplus.sh --dry-run
#   QUOTE_PUSH_SURPLUS_RESERVE_RAW=50000 bash scripts/deployment/sweep-mainnet-aave-quote-push-receiver-surplus.sh --apply

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROXMOX_ROOT="$(cd "${SCRIPT_DIR}/../.." && pwd)"
SMOM="${PROXMOX_ROOT}/smom-dbis-138"
DEFAULT_USDC_MAINNET="0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48"

_qp_private_key="${PRIVATE_KEY-}"
_qp_rpc="${ETHEREUM_MAINNET_RPC-}"
_qp_receiver="${AAVE_QUOTE_PUSH_RECEIVER_MAINNET-}"
_qp_token="${QUOTE_PUSH_SURPLUS_TOKEN_MAINNET-}"
_qp_recipient="${QUOTE_PUSH_SURPLUS_RECIPIENT-}"
_qp_reserve="${QUOTE_PUSH_SURPLUS_RESERVE_RAW-}"
_qp_exact="${QUOTE_PUSH_SURPLUS_EXACT_AMOUNT_RAW-}"

# shellcheck disable=SC1091
source "${PROXMOX_ROOT}/scripts/lib/load-project-env.sh" 2>/dev/null || true
# shellcheck disable=SC1091
source "${SMOM}/scripts/load-env.sh" >/dev/null 2>&1 || true

[[ -n "$_qp_private_key" ]] && export PRIVATE_KEY="$_qp_private_key"
[[ -n "$_qp_rpc" ]] && export ETHEREUM_MAINNET_RPC="$_qp_rpc"
[[ -n "$_qp_receiver" ]] && export AAVE_QUOTE_PUSH_RECEIVER_MAINNET="$_qp_receiver"
[[ -n "$_qp_token" ]] && export QUOTE_PUSH_SURPLUS_TOKEN_MAINNET="$_qp_token"
[[ -n "$_qp_recipient" ]] && export QUOTE_PUSH_SURPLUS_RECIPIENT="$_qp_recipient"
[[ -n "$_qp_reserve" ]] && export QUOTE_PUSH_SURPLUS_RESERVE_RAW="$_qp_reserve"
[[ -n "$_qp_exact" ]] && export QUOTE_PUSH_SURPLUS_EXACT_AMOUNT_RAW="$_qp_exact"

unset _qp_private_key _qp_rpc _qp_receiver _qp_token _qp_recipient _qp_reserve _qp_exact

require_cmd() {
  command -v "$1" >/dev/null 2>&1 || {
    echo "[fail] missing required command: $1" >&2
    exit 1
  }
}

require_cmd cast
require_cmd forge

require_env() {
  local name="$1"
  if [[ -z "${!name:-}" ]]; then
    echo "[fail] missing required env: $name" >&2
    exit 1
  fi
}

pick_latest_receiver() {
  local latest_json="${SMOM}/broadcast/DeployAaveQuotePushFlashReceiver.s.sol/1/run-latest.json"
  if [[ ! -f "$latest_json" ]] || ! command -v jq >/dev/null 2>&1; then
    return 1
  fi
  jq -r '.transactions[]? | select(.transactionType == "CREATE" and .contractName == "AaveQuotePushFlashReceiver") | .contractAddress' \
    "$latest_json" | tail -n1
}

MODE="dry-run"
BROADCAST=()
for arg in "$@"; do
  case "$arg" in
    --dry-run) MODE="dry-run"; BROADCAST=() ;;
    --apply) MODE="apply"; BROADCAST=(--broadcast) ;;
    *)
      echo "[fail] unknown arg: $arg (use --dry-run or --apply)" >&2
      exit 2
      ;;
  esac
done

require_env PRIVATE_KEY
require_env ETHEREUM_MAINNET_RPC
if [[ -z "${AAVE_QUOTE_PUSH_RECEIVER_MAINNET:-}" ]]; then
  inferred_receiver="$(pick_latest_receiver || true)"
  if [[ -n "$inferred_receiver" && "$inferred_receiver" != "null" ]]; then
    export AAVE_QUOTE_PUSH_RECEIVER_MAINNET="$inferred_receiver"
  fi
fi
require_env AAVE_QUOTE_PUSH_RECEIVER_MAINNET

TOKEN="${QUOTE_PUSH_SURPLUS_TOKEN_MAINNET:-$DEFAULT_USDC_MAINNET}"
DEPLOYER="$(cast wallet address --private-key "$PRIVATE_KEY")"
RECIPIENT="${QUOTE_PUSH_SURPLUS_RECIPIENT:-$DEPLOYER}"
RESERVE_RAW="${QUOTE_PUSH_SURPLUS_RESERVE_RAW:-0}"
EXACT_RAW="${QUOTE_PUSH_SURPLUS_EXACT_AMOUNT_RAW:-0}"
receiver_owner="$(cast call "$AAVE_QUOTE_PUSH_RECEIVER_MAINNET" 'owner()(address)' --rpc-url "$ETHEREUM_MAINNET_RPC" 2>/dev/null | awk '{print $1}' || true)"

if [[ -z "$receiver_owner" ]]; then
  cat >&2 <<EOF
[fail] receiver does not expose owner()/sweep support: $AAVE_QUOTE_PUSH_RECEIVER_MAINNET
[hint] Redeploy the updated receiver, then point AAVE_QUOTE_PUSH_RECEIVER_MAINNET at the new address:
  bash scripts/deployment/deploy-mainnet-aave-quote-push-receiver.sh --apply
EOF
  exit 1
fi

if [[ "${receiver_owner,,}" != "${DEPLOYER,,}" ]]; then
  echo "[fail] deployer $DEPLOYER is not the receiver owner ($receiver_owner)" >&2
  exit 1
fi

receiver_balance_raw="$(
  cast call "$TOKEN" 'balanceOf(address)(uint256)' "$AAVE_QUOTE_PUSH_RECEIVER_MAINNET" --rpc-url "$ETHEREUM_MAINNET_RPC" \
    | awk '{print $1}'
)"

if (( EXACT_RAW > 0 )); then
  if (( receiver_balance_raw < EXACT_RAW )); then
    echo "[fail] receiver balance too small for exact sweep: have=$receiver_balance_raw need=$EXACT_RAW" >&2
    exit 1
  fi
  sweep_raw="$EXACT_RAW"
else
  if (( receiver_balance_raw <= RESERVE_RAW )); then
    echo "[stop] nothing to sweep: receiver_balance_raw=$receiver_balance_raw reserve_raw=$RESERVE_RAW" >&2
    exit 3
  fi
  sweep_raw=$((receiver_balance_raw - RESERVE_RAW))
fi

echo "=== sweep-mainnet-aave-quote-push-receiver-surplus ($MODE) ==="
echo "receiver=$AAVE_QUOTE_PUSH_RECEIVER_MAINNET"
echo "receiver_owner=$receiver_owner"
echo "token=$TOKEN"
echo "recipient=$RECIPIENT"
echo "receiver_balance_raw=$receiver_balance_raw"
echo "reserve_raw=$RESERVE_RAW"
echo "exact_raw=$EXACT_RAW"
echo "sweep_raw=$sweep_raw"

(
  cd "$SMOM"
  forge script script/flash/SweepAaveQuotePushFlashReceiverSurplus.s.sol:SweepAaveQuotePushFlashReceiverSurplus \
    --rpc-url "$ETHEREUM_MAINNET_RPC" \
    "${BROADCAST[@]}" \
    -vvvv
)

if [[ "$MODE" == "apply" ]]; then
  receiver_after="$(
    cast call "$TOKEN" 'balanceOf(address)(uint256)' "$AAVE_QUOTE_PUSH_RECEIVER_MAINNET" --rpc-url "$ETHEREUM_MAINNET_RPC" \
      | awk '{print $1}'
  )"
  recipient_after="$(
    cast call "$TOKEN" 'balanceOf(address)(uint256)' "$RECIPIENT" --rpc-url "$ETHEREUM_MAINNET_RPC" \
      | awk '{print $1}'
  )"
  echo "receiver_balance_after_raw=$receiver_after"
  echo "recipient_balance_after_raw=$recipient_after"
fi