#!/usr/bin/env bash
# Configure all bridge destinations for WETH9 and WETH10
# Usage: ./configure-all-bridge-destinations.sh [private_key]

set -euo pipefail

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

# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
CYAN='\033[0;36m'
NC='\033[0m'

log_info() { echo -e "${BLUE}[INFO]${NC} $1"; }
log_success() { echo -e "${GREEN}[✓]${NC} $1"; }
log_warn() { echo -e "${YELLOW}[WARN]${NC} $1"; }
log_error() { echo -e "${RED}[ERROR]${NC} $1"; }
log_fix() { echo -e "${CYAN}[FIX]${NC} $1"; }

# Load environment variables if .env exists
if [ -f "$PROJECT_ROOT/.env" ]; then
    source "$PROJECT_ROOT/.env"
elif [ -f "$PROJECT_ROOT/../.env" ]; then
    source "$PROJECT_ROOT/../.env"
fi

# Configuration
RPC_URL="${RPC_URL_138:-http://192.168.11.250:8545}"
WETH9_BRIDGE="0x89dd12025bfCD38A168455A44B400e913ED33BE2"
WETH10_BRIDGE="0xe0E93247376aa097dB308B92e6Ba36bA015535D0"

# Parse arguments
PRIVATE_KEY_ARG="${1:-}"
WETH9_ETH_MAINNET="${2:-}"
WETH10_ETH_MAINNET="${3:-}"

# Use provided private key or from environment
if [ -n "$PRIVATE_KEY_ARG" ]; then
    PRIVATE_KEY="$PRIVATE_KEY_ARG"
elif [ -z "${PRIVATE_KEY:-}" ]; then
    log_error "PRIVATE_KEY not provided. Usage: $0 [private_key] [weth9_eth_mainnet] [weth10_eth_mainnet]"
    log_info "Or set PRIVATE_KEY in .env file"
    exit 1
fi

# Update Ethereum Mainnet addresses if provided
if [ -n "$WETH9_ETH_MAINNET" ] && echo "$WETH9_ETH_MAINNET" | grep -qE "^0x[0-9a-fA-F]{40}$"; then
    WETH9_DESTINATIONS["5009297550715157269"]="$WETH9_ETH_MAINNET"
    log_info "Using provided WETH9 Ethereum Mainnet address: $WETH9_ETH_MAINNET"
fi

if [ -n "$WETH10_ETH_MAINNET" ] && echo "$WETH10_ETH_MAINNET" | grep -qE "^0x[0-9a-fA-F]{40}$"; then
    WETH10_DESTINATIONS["5009297550715157269"]="$WETH10_ETH_MAINNET"
    log_info "Using provided WETH10 Ethereum Mainnet address: $WETH10_ETH_MAINNET"
fi

# Destination chains with their bridge addresses
# Format: selector:bridge_address
declare -A WETH9_DESTINATIONS=(
    ["11344663589394136015"]="0x8078a09637e47fa5ed34f626046ea2094a5cde5e"  # BSC
    ["4051577828743386545"]="0xa780ef19a041745d353c9432f2a7f5a241335ffe"   # Polygon
    ["6433500567565415381"]="0x8078a09637e47fa5ed34f626046ea2094a5cde5e"  # Avalanche
    ["15971525489660198786"]="0x8078a09637e47fa5ed34f626046ea2094a5cde5e"  # Base
    ["4949039107694359620"]="0x8078a09637e47fa5ed34f626046ea2094a5cde5e"  # Arbitrum
    ["3734403246176062136"]="0x8078a09637e47fa5ed34f626046ea2094a5cde5e"  # Optimism
    ["5009297550715157269"]="TBD"  # Ethereum Mainnet - needs to be provided
)

declare -A WETH10_DESTINATIONS=(
    ["11344663589394136015"]="0x105f8a15b819948a89153505762444ee9f324684"  # BSC
    ["4051577828743386545"]="0xdab0591e5e89295ffad75a71dcfc30c5625c4fa2"   # Polygon
    ["6433500567565415381"]="0x105f8a15b819948a89153505762444ee9f324684"  # Avalanche
    ["15971525489660198786"]="0x105f8a15b819948a89153505762444ee9f324684"  # Base
    ["4949039107694359620"]="0x105f8a15b819948a89153505762444ee9f324684"  # Arbitrum
    ["3734403246176062136"]="0x105f8a15b819948a89153505762444ee9f324684"  # Optimism
    ["5009297550715157269"]="TBD"  # Ethereum Mainnet - needs to be provided
)

declare -A CHAIN_NAMES=(
    ["11344663589394136015"]="BSC"
    ["4051577828743386545"]="Polygon"
    ["6433500567565415381"]="Avalanche"
    ["15971525489660198786"]="Base"
    ["4949039107694359620"]="Arbitrum"
    ["3734403246176062136"]="Optimism"
    ["5009297550715157269"]="Ethereum Mainnet"
)

# Get deployer address
DEPLOYER=$(cast wallet address --private-key "$PRIVATE_KEY" 2>/dev/null || echo "")
if [ -z "$DEPLOYER" ]; then
    log_error "Failed to get address from private key"
    exit 1
fi

log_info "========================================="
log_info "Configure All Bridge Destinations"
log_info "========================================="
log_info ""
log_info "Deployer: $DEPLOYER"
log_info "WETH9 Bridge: $WETH9_BRIDGE"
log_info "WETH10 Bridge: $WETH10_BRIDGE"
log_info ""

# Function to configure destination
configure_destination() {
    local bridge="$1"
    local selector="$2"
    local dest_address="$3"
    local chain_name="$4"
    local token_name="$5"
    
    # Check if already configured
    local current=$(cast call "$bridge" "destinations(uint64)" "$selector" --rpc-url "$RPC_URL" 2>/dev/null || echo "")
    local current_clean=$(echo "$current" | grep -oE "^0x[0-9a-fA-F]{40}$" | head -1 || echo "")
    
    # Check if result is valid address (not zero or empty)
    if [ -n "$current_clean" ] && ! echo "$current_clean" | grep -qE "^0x0+$" && [ "$current_clean" != "0x0000000000000000000000000000000000000000" ]; then
        log_success "✓ $token_name $chain_name: Already configured ($current_clean)"
        return 0
    fi
    
    if [ "$dest_address" = "TBD" ]; then
        log_warn "⚠ $token_name $chain_name: Address not provided (TBD)"
        return 1
    fi
    
    log_info "Configuring $token_name $chain_name..."
    local nonce=$(cast nonce "$DEPLOYER" --rpc-url "$RPC_URL" 2>/dev/null || echo "0")
    
    local tx_output=$(cast send "$bridge" \
        "addDestination(uint64,address)" \
        "$selector" \
        "$dest_address" \
        --rpc-url "$RPC_URL" \
        --private-key "$PRIVATE_KEY" \
        --gas-price 5000000000 \
        --nonce "$nonce" \
        2>&1 || echo "FAILED")
    
    if echo "$tx_output" | grep -qE "transactionHash"; then
        local tx_hash=$(echo "$tx_output" | grep -oE "transactionHash[[:space:]]+0x[0-9a-fA-F]{64}" | awk '{print $2}')
        log_success "✓ $token_name $chain_name: Configured ($tx_hash)"
        sleep 10
        return 0
    else
        log_error "✗ $token_name $chain_name: Failed"
        log_info "  Output: $(echo "$tx_output" | head -3)"
        return 1
    fi
}

# Configure WETH9 destinations
log_fix "========================================="
log_fix "Configuring WETH9 Bridge Destinations"
log_fix "========================================="
log_info ""

WETH9_SUCCESS=0
WETH9_FAILED=0
WETH9_SKIPPED=0

for SELECTOR in "${!WETH9_DESTINATIONS[@]}"; do
    DEST_ADDRESS="${WETH9_DESTINATIONS[$SELECTOR]}"
    CHAIN_NAME="${CHAIN_NAMES[$SELECTOR]}"
    
    if configure_destination "$WETH9_BRIDGE" "$SELECTOR" "$DEST_ADDRESS" "$CHAIN_NAME" "WETH9"; then
        ((WETH9_SUCCESS++)) || true
    elif [ "$DEST_ADDRESS" = "TBD" ]; then
        ((WETH9_SKIPPED++)) || true
    else
        ((WETH9_FAILED++)) || true
    fi
done

log_info ""
log_info "WETH9 Results: $WETH9_SUCCESS configured, $WETH9_FAILED failed, $WETH9_SKIPPED skipped"
log_info ""

# Configure WETH10 destinations
log_fix "========================================="
log_fix "Configuring WETH10 Bridge Destinations"
log_fix "========================================="
log_info ""

WETH10_SUCCESS=0
WETH10_FAILED=0
WETH10_SKIPPED=0

for SELECTOR in "${!WETH10_DESTINATIONS[@]}"; do
    DEST_ADDRESS="${WETH10_DESTINATIONS[$SELECTOR]}"
    CHAIN_NAME="${CHAIN_NAMES[$SELECTOR]}"
    
    if configure_destination "$WETH10_BRIDGE" "$SELECTOR" "$DEST_ADDRESS" "$CHAIN_NAME" "WETH10"; then
        ((WETH10_SUCCESS++)) || true
    elif [ "$DEST_ADDRESS" = "TBD" ]; then
        ((WETH10_SKIPPED++)) || true
    else
        ((WETH10_FAILED++)) || true
    fi
done

log_info ""
log_info "WETH10 Results: $WETH10_SUCCESS configured, $WETH10_FAILED failed, $WETH10_SKIPPED skipped"
log_info ""

# Final summary
log_info "========================================="
log_info "Configuration Summary"
log_info "========================================="
log_info ""

log_info "WETH9 Bridge:"
log_info "  Configured: $WETH9_SUCCESS"
log_info "  Failed: $WETH9_FAILED"
log_info "  Skipped: $WETH9_SKIPPED (Ethereum Mainnet - address needed)"

log_info ""
log_info "WETH10 Bridge:"
log_info "  Configured: $WETH10_SUCCESS"
log_info "  Failed: $WETH10_FAILED"
log_info "  Skipped: $WETH10_SKIPPED (Ethereum Mainnet - address needed)"

log_info ""

if [ $WETH9_SKIPPED -gt 0 ] || [ $WETH10_SKIPPED -gt 0 ]; then
    log_warn "⚠ Ethereum Mainnet addresses not provided"
    log_info "  To configure Ethereum Mainnet, use:"
    log_info "    ./scripts/fix-bridge-errors.sh [private_key] [ethereum_mainnet_bridge_address]"
fi

if [ $WETH9_FAILED -eq 0 ] && [ $WETH10_FAILED -eq 0 ]; then
    log_success "✓ All provided destinations configured successfully!"
else
    log_warn "⚠ Some destinations failed to configure"
    log_info "  Check transaction outputs above for details"
fi

log_info ""