#!/usr/bin/env bash
# Direct deployment script - runs all deployment steps

set -e

cd /home/intlc/projects/proxmox/smom-dbis-138

# Load environment
if [[ -f ./scripts/load-env.sh ]]; then
    # shellcheck disable=SC1091
    source ./scripts/load-env.sh >/dev/null
fi
export RPC_URL="${RPC_URL:-${RPC_URL_138:-http://192.168.11.211:8545}}"
export GAS_PRICE="${GAS_PRICE:-20000000000}"

if [[ -z "${PRIVATE_KEY:-}" ]]; then
    echo "PRIVATE_KEY not set. Source smom-dbis-138/.env or ~/.secure-secrets/private-keys.env first." >&2
    exit 1
fi

echo "=========================================="
echo "Starting Deployment"
echo "=========================================="
echo "RPC URL: $RPC_URL"
echo "Deployer: $(cast wallet address "$PRIVATE_KEY")"
echo ""

# Step 1: Deploy ComplianceRegistry
echo "Step 1: Deploying ComplianceRegistry..."
forge script script/DeployComplianceRegistry.s.sol:DeployComplianceRegistry \
    --rpc-url "$RPC_URL" \
    --broadcast \
    --legacy \
    --gas-price "$GAS_PRICE" \
    --via-ir \
    -vv 2>&1 | tee /tmp/compliance-registry-deploy.log

COMPLIANCE_REGISTRY=$(grep -oP "ComplianceRegistry deployed at: \K0x[a-fA-F0-9]+" /tmp/compliance-registry-deploy.log | tail -1 || echo "")
if [ -z "$COMPLIANCE_REGISTRY" ]; then
    echo "⚠️  Could not extract ComplianceRegistry address from log"
    COMPLIANCE_REGISTRY=$(grep -oP "0x[a-fA-F0-9]{40}" /tmp/compliance-registry-deploy.log | tail -1 || echo "")
fi

echo "ComplianceRegistry: $COMPLIANCE_REGISTRY"
echo "COMPLIANCE_REGISTRY_ADDRESS=$COMPLIANCE_REGISTRY" >> .env
echo ""

# Step 2: Deploy CompliantUSDT
echo "Step 2: Deploying CompliantUSDT..."
forge script script/DeployCompliantUSDT.s.sol:DeployCompliantUSDT \
    --rpc-url "$RPC_URL" \
    --broadcast \
    --legacy \
    --gas-price "$GAS_PRICE" \
    --via-ir \
    -vv 2>&1 | tee /tmp/compliant-usdt-deploy.log

COMPLIANT_USDT=$(grep -oP "CompliantUSDT deployed at: \K0x[a-fA-F0-9]+" /tmp/compliant-usdt-deploy.log | tail -1 || echo "")
if [ -z "$COMPLIANT_USDT" ]; then
    COMPLIANT_USDT=$(grep -oP "0x[a-fA-F0-9]{40}" /tmp/compliant-usdt-deploy.log | tail -1 || echo "")
fi

echo "CompliantUSDT: $COMPLIANT_USDT"
echo "COMPLIANT_USDT_ADDRESS=$COMPLIANT_USDT" >> .env
echo ""

# Step 3: Deploy CompliantUSDC
echo "Step 3: Deploying CompliantUSDC..."
forge script script/DeployCompliantUSDC.s.sol:DeployCompliantUSDC \
    --rpc-url "$RPC_URL" \
    --broadcast \
    --legacy \
    --gas-price "$GAS_PRICE" \
    --via-ir \
    -vv 2>&1 | tee /tmp/compliant-usdc-deploy.log

COMPLIANT_USDC=$(grep -oP "CompliantUSDC deployed at: \K0x[a-fA-F0-9]+" /tmp/compliant-usdc-deploy.log | tail -1 || echo "")
if [ -z "$COMPLIANT_USDC" ]; then
    COMPLIANT_USDC=$(grep -oP "0x[a-fA-F0-9]{40}" /tmp/compliant-usdc-deploy.log | tail -1 || echo "")
fi

echo "CompliantUSDC: $COMPLIANT_USDC"
echo "COMPLIANT_USDC_ADDRESS=$COMPLIANT_USDC" >> .env
echo ""

# Step 4: Deploy TokenRegistry
echo "Step 4: Deploying TokenRegistry..."
forge script script/DeployTokenRegistry.s.sol:DeployTokenRegistry \
    --rpc-url "$RPC_URL" \
    --broadcast \
    --legacy \
    --gas-price "$GAS_PRICE" \
    -vv 2>&1 | tee /tmp/token-registry-deploy.log

TOKEN_REGISTRY=$(grep -oP "TokenRegistry deployed at: \K0x[a-fA-F0-9]+" /tmp/token-registry-deploy.log | tail -1 || echo "")
if [ -z "$TOKEN_REGISTRY" ]; then
    TOKEN_REGISTRY=$(grep -oP "0x[a-fA-F0-9]{40}" /tmp/token-registry-deploy.log | tail -1 || echo "")
fi

echo "TokenRegistry: $TOKEN_REGISTRY"
echo "TOKEN_REGISTRY_ADDRESS=$TOKEN_REGISTRY" >> .env
echo ""

# Step 5: Deploy FeeCollector
echo "Step 5: Deploying FeeCollector..."
forge script script/DeployFeeCollector.s.sol:DeployFeeCollector \
    --rpc-url "$RPC_URL" \
    --broadcast \
    --legacy \
    --gas-price "$GAS_PRICE" \
    -vv 2>&1 | tee /tmp/fee-collector-deploy.log

FEE_COLLECTOR=$(grep -oP "FeeCollector deployed at: \K0x[a-fA-F0-9]+" /tmp/fee-collector-deploy.log | tail -1 || echo "")
if [ -z "$FEE_COLLECTOR" ]; then
    FEE_COLLECTOR=$(grep -oP "0x[a-fA-F0-9]{40}" /tmp/fee-collector-deploy.log | tail -1 || echo "")
fi

echo "FeeCollector: $FEE_COLLECTOR"
echo "FEE_COLLECTOR_ADDRESS=$FEE_COLLECTOR" >> .env
echo ""

# Integration steps
if [ ! -z "$COMPLIANCE_REGISTRY" ] && [ ! -z "$COMPLIANT_USDT" ]; then
    echo "Registering CompliantUSDT in ComplianceRegistry..."
    cast send "$COMPLIANCE_REGISTRY" \
        "registerContract(address)" \
        "$COMPLIANT_USDT" \
        --rpc-url "$RPC_URL" \
        --private-key "$PRIVATE_KEY" \
        --legacy \
        --gas-price "$GAS_PRICE" 2>&1 | tail -5 || echo "Registration may have failed"
    echo ""
fi

if [ ! -z "$COMPLIANCE_REGISTRY" ] && [ ! -z "$COMPLIANT_USDC" ]; then
    echo "Registering CompliantUSDC in ComplianceRegistry..."
    cast send "$COMPLIANCE_REGISTRY" \
        "registerContract(address)" \
        "$COMPLIANT_USDC" \
        --rpc-url "$RPC_URL" \
        --private-key "$PRIVATE_KEY" \
        --legacy \
        --gas-price "$GAS_PRICE" 2>&1 | tail -5 || echo "Registration may have failed"
    echo ""
fi

if [ ! -z "$TOKEN_REGISTRY" ] && [ ! -z "$COMPLIANT_USDT" ]; then
    echo "Registering CompliantUSDT in TokenRegistry..."
    cast send "$TOKEN_REGISTRY" \
        "registerToken(address,string,string,uint8,bool,address)" \
        "$COMPLIANT_USDT" \
        "Tether USD (Compliant)" \
        "cUSDT" \
        6 \
        false \
        0x0000000000000000000000000000000000000000 \
        --rpc-url "$RPC_URL" \
        --private-key "$PRIVATE_KEY" \
        --legacy \
        --gas-price "$GAS_PRICE" 2>&1 | tail -5 || echo "Registration may have failed"
    echo ""
fi

if [ ! -z "$TOKEN_REGISTRY" ] && [ ! -z "$COMPLIANT_USDC" ]; then
    echo "Registering CompliantUSDC in TokenRegistry..."
    cast send "$TOKEN_REGISTRY" \
        "registerToken(address,string,string,uint8,bool,address)" \
        "$COMPLIANT_USDC" \
        "USD Coin (Compliant)" \
        "cUSDC" \
        6 \
        false \
        0x0000000000000000000000000000000000000000 \
        --rpc-url "$RPC_URL" \
        --private-key "$PRIVATE_KEY" \
        --legacy \
        --gas-price "$GAS_PRICE" 2>&1 | tail -5 || echo "Registration may have failed"
    echo ""
fi

echo "=========================================="
echo "Deployment Complete!"
echo "=========================================="
echo "ComplianceRegistry: $COMPLIANCE_REGISTRY"
echo "CompliantUSDT:      $COMPLIANT_USDT"
echo "CompliantUSDC:      $COMPLIANT_USDC"
echo "TokenRegistry:      $TOKEN_REGISTRY"
echo "FeeCollector:       $FEE_COLLECTOR"
echo ""
echo "All addresses saved to .env file"
