#!/bin/bash

# End-to-End Contract Testing Script
# This script tests all deployed contracts

set -e

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

# Load .env
if [ -f .env ]; then
    set -a
    source .env
    set +a
fi

RPC_URL=${RPC_URL:-${RPC_URL_138:-"http://192.168.11.250:8545"}}

echo -e "${BLUE}╔══════════════════════════════════════════════════════════════╗${NC}"
echo -e "${BLUE}║     End-to-End Contract Testing                            ║${NC}"
echo -e "${BLUE}╚══════════════════════════════════════════════════════════════╝${NC}"
echo ""

if [ -z "$PRIVATE_KEY" ]; then
    echo -e "${RED}Error: PRIVATE_KEY not set${NC}"
    exit 1
fi

TEST_RECIPIENT=${TEST_RECIPIENT:-"0x1111111111111111111111111111111111111111"}

# Test CompliantUSDT
if [ ! -z "$COMPLIANT_USDT_ADDRESS" ]; then
    echo -e "${YELLOW}Testing CompliantUSDT...${NC}"
    
    # Get balance
    BALANCE=$(cast call "$COMPLIANT_USDT_ADDRESS" "balanceOf(address)" "$(cast wallet address $PRIVATE_KEY)" --rpc-url "$RPC_URL" 2>/dev/null || echo "0")
    echo "  Deployer Balance: $BALANCE"
    
    # Test transfer (if balance > 0)
    if [ "$BALANCE" != "0" ] && [ "$BALANCE" != "" ]; then
        echo "  Testing transfer..."
        cast send "$COMPLIANT_USDT_ADDRESS" \
            "transfer(address,uint256)" \
            "$TEST_RECIPIENT" 1000000 \
            --rpc-url "$RPC_URL" \
            --private-key "$PRIVATE_KEY" \
            --legacy \
            --gas-price 20000000000 2>&1 | grep -E "(Success|Error)" || echo "  Transfer executed"
        
        # Verify transfer
        NEW_BALANCE=$(cast call "$COMPLIANT_USDT_ADDRESS" "balanceOf(address)" "$TEST_RECIPIENT" --rpc-url "$RPC_URL" 2>/dev/null || echo "0")
        if [ "$NEW_BALANCE" != "0" ]; then
            echo -e "  ${GREEN}✅ Transfer successful${NC}"
        fi
    fi
    
    # Test pause status
    PAUSED=$(cast call "$COMPLIANT_USDT_ADDRESS" "paused()" --rpc-url "$RPC_URL" 2>/dev/null || echo "false")
    echo "  Paused: $PAUSED"
    echo ""
fi

# Test TokenRegistry
if [ ! -z "$TOKEN_REGISTRY_ADDRESS" ]; then
    echo -e "${YELLOW}Testing TokenRegistry...${NC}"
    
    # Get token count
    COUNT=$(cast call "$TOKEN_REGISTRY_ADDRESS" "getTokenCount()" --rpc-url "$RPC_URL" 2>/dev/null || echo "0")
    echo "  Registered Tokens: $COUNT"
    
    # Get token by symbol
    if [ ! -z "$COMPLIANT_USDT_ADDRESS" ]; then
        TOKEN_ADDR=$(cast call "$TOKEN_REGISTRY_ADDRESS" "getTokenBySymbol(string)" "cUSDT" --rpc-url "$RPC_URL" 2>/dev/null || echo "")
        if [ "$TOKEN_ADDR" = "$COMPLIANT_USDT_ADDRESS" ]; then
            echo -e "  ${GREEN}✅ Token lookup by symbol works${NC}"
        else
            echo -e "  ${YELLOW}⚠️  Token lookup returned: $TOKEN_ADDR${NC}"
        fi
    fi
    echo ""
fi

# Test ComplianceRegistry
if [ ! -z "$COMPLIANCE_REGISTRY_ADDRESS" ]; then
    echo -e "${YELLOW}Testing ComplianceRegistry...${NC}"
    
    if [ ! -z "$COMPLIANT_USDT_ADDRESS" ]; then
        REGISTERED=$(cast call "$COMPLIANCE_REGISTRY_ADDRESS" "isContractRegistered(address)" "$COMPLIANT_USDT_ADDRESS" --rpc-url "$RPC_URL" 2>/dev/null || echo "false")
        echo "  CompliantUSDT Registered: $REGISTERED"
        
        if [ "$REGISTERED" = "true" ]; then
            STATUS=$(cast call "$COMPLIANCE_REGISTRY_ADDRESS" "getContractComplianceStatus(address)" "$COMPLIANT_USDT_ADDRESS" --rpc-url "$RPC_URL" 2>/dev/null || echo "")
            echo "  Compliance Status: Retrieved"
        fi
    fi
    echo ""
fi

# Test FeeCollector
if [ ! -z "$FEE_COLLECTOR_ADDRESS" ]; then
    echo -e "${YELLOW}Testing FeeCollector...${NC}"
    
    # Get balance
    BALANCE=$(cast call "$FEE_COLLECTOR_ADDRESS" "getBalance(address)" "0x0000000000000000000000000000000000000000" --rpc-url "$RPC_URL" 2>/dev/null || echo "0")
    echo "  ETH Balance: $BALANCE"
    
    # Get total collected
    TOTAL=$(cast call "$FEE_COLLECTOR_ADDRESS" "getTotalCollected(address)" "0x0000000000000000000000000000000000000000" --rpc-url "$RPC_URL" 2>/dev/null || echo "0")
    echo "  Total Collected: $TOTAL"
    echo ""
fi

echo -e "${BLUE}Testing Complete!${NC}"
echo ""