#!/usr/bin/env bash

set -e

# Deploy CCIP Relay Infrastructure to Ethereum Mainnet
# This script deploys the relay router and bridge contracts

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

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

# Configuration
RPC_URL_MAINNET="${RPC_URL_MAINNET:-https://eth.llamarpc.com}"
WETH9_MAINNET="${WETH9_MAINNET:-0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2}"
PRIVATE_KEY="${PRIVATE_KEY:-}"
RELAYER_ADDRESS="${RELAYER_ADDRESS:-}"

if [ -z "$PRIVATE_KEY" ]; then
    echo "Error: PRIVATE_KEY environment variable not set"
    exit 1
fi

# Derive relayer address if not provided
if [ -z "$RELAYER_ADDRESS" ]; then
    RELAYER_ADDRESS=$(cast wallet address "$PRIVATE_KEY" 2>/dev/null || echo "")
    if [ -z "$RELAYER_ADDRESS" ]; then
        echo "Error: Could not derive relayer address from private key"
        exit 1
    fi
fi

echo "=== Deploying CCIP Relay Infrastructure to Ethereum Mainnet ==="
echo ""
echo "Configuration:"
echo "  RPC URL: $RPC_URL_MAINNET"
echo "  WETH9: $WETH9_MAINNET"
echo "  Relayer: $RELAYER_ADDRESS"
echo ""

# Check relayer has ETH
BALANCE=$(cast balance "$RELAYER_ADDRESS" --rpc-url "$RPC_URL_MAINNET" 2>/dev/null || echo "0")
echo "Relayer ETH balance: $(cast --to-unit "$BALANCE" 18 2>&1 | head -1 || echo "$BALANCE") ETH"
echo ""

if [ "$(cast --to-unit "$BALANCE" 18 2>&1 | head -1 || echo "0")" = "0" ]; then
    echo "⚠️  Warning: Relayer has no ETH. Deployment will fail."
    echo "   Please fund the relayer address: $RELAYER_ADDRESS"
    read -p "Continue anyway? (y/N) " -n 1 -r
    echo
    if [[ ! $REPLY =~ ^[Yy]$ ]]; then
        exit 1
    fi
fi

cd "$PROJECT_ROOT"

echo "Deploying contracts..."
echo ""

# Deploy using Foundry
forge script script/DeployCCIPRelay.s.sol:DeployCCIPRelay \
    --rpc-url "$RPC_URL_MAINNET" \
    --broadcast \
    --legacy \
    --gas-price $(cast gas-price --rpc-url "$RPC_URL_MAINNET" 2>/dev/null || echo "20000000000") \
    -vvv 2>&1 | tee /tmp/relay-deployment.log

# Extract deployed addresses
RELAY_ROUTER=$(grep -oE "CCIPRelayRouter deployed at: 0x[a-fA-F0-9]{40}" /tmp/relay-deployment.log | grep -oE "0x[a-fA-F0-9]{40}" | tail -1)
RELAY_BRIDGE=$(grep -oE "CCIPRelayBridge deployed at: 0x[a-fA-F0-9]{40}" /tmp/relay-deployment.log | grep -oE "0x[a-fA-F0-9]{40}" | tail -1)

if [ -n "$RELAY_ROUTER" ] && [ -n "$RELAY_BRIDGE" ]; then
    echo ""
    echo "=== Deployment Successful ==="
    echo "CCIPRelayRouter: $RELAY_ROUTER"
    echo "CCIPRelayBridge: $RELAY_BRIDGE"
    echo ""
    echo "Updating .env files..."
    
    # Update main .env
    if [ -f "$PROJECT_ROOT/.env" ]; then
        sed -i "s|CCIP_RELAY_ROUTER_MAINNET=.*|CCIP_RELAY_ROUTER_MAINNET=$RELAY_ROUTER|" "$PROJECT_ROOT/.env" 2>/dev/null || \
        echo "CCIP_RELAY_ROUTER_MAINNET=$RELAY_ROUTER" >> "$PROJECT_ROOT/.env"
        
        sed -i "s|CCIP_RELAY_BRIDGE_MAINNET=.*|CCIP_RELAY_BRIDGE_MAINNET=$RELAY_BRIDGE|" "$PROJECT_ROOT/.env" 2>/dev/null || \
        echo "CCIP_RELAY_BRIDGE_MAINNET=$RELAY_BRIDGE" >> "$PROJECT_ROOT/.env"
    fi
    
    # Update relay service .env
    if [ -f "$PROJECT_ROOT/services/relay/.env" ]; then
        sed -i "s|CCIP_RELAY_ROUTER_MAINNET=.*|CCIP_RELAY_ROUTER_MAINNET=$RELAY_ROUTER|" "$PROJECT_ROOT/services/relay/.env"
        sed -i "s|CCIP_RELAY_BRIDGE_MAINNET=.*|CCIP_RELAY_BRIDGE_MAINNET=$RELAY_BRIDGE|" "$PROJECT_ROOT/services/relay/.env"
        sed -i "s|RELAYER_ADDRESS=.*|RELAYER_ADDRESS=$RELAYER_ADDRESS|" "$PROJECT_ROOT/services/relay/.env"
    fi
    
    echo "Configuration updated!"
    echo ""
    echo "Next steps:"
    echo "1. Fund relay bridge with WETH9 tokens"
    echo "2. Start relay service: cd services/relay && npm start"
else
    echo ""
    echo "⚠️  Could not extract deployed addresses from logs"
    echo "   Please check deployment logs manually"
fi