#!/bin/bash

# MEV Bot Production Readiness Verification Script
# This script validates that all 4 critical blockers have been fixed
# and the system is ready for production deployment.

set -e

echo ""
echo "╔════════════════════════════════════════════════════════╗"
echo "║   MEV BOT - PRODUCTION READINESS VERIFICATION v1.0     ║"
echo "║   Date: $(date '+%Y-%m-%d %H:%M:%S')                    ║"
echo "╚════════════════════════════════════════════════════════╝"
echo ""

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

CHECKS_PASSED=0
CHECKS_FAILED=0
CHECKS_WARNING=0

# Helper functions
pass_check() {
    echo -e "${GREEN}✅ PASS${NC}: $1"
    ((CHECKS_PASSED++))
}

fail_check() {
    echo -e "${RED}❌ FAIL${NC}: $1"
    ((CHECKS_FAILED++))
}

warn_check() {
    echo -e "${YELLOW}⚠️  WARN${NC}: $1"
    ((CHECKS_WARNING++))
}

info_check() {
    echo -e "${BLUE}ℹ️  INFO${NC}: $1"
}

echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "SECTION 1: CODE & BUILD VERIFICATION"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

# Check 1.1: Binary exists
if [ -f "./mev-bot" ]; then
    pass_check "MEV Bot binary exists and is executable"
else
    fail_check "MEV Bot binary not found - need to run 'make build'"
fi

# Check 1.2: Build succeeds
if make build > /tmp/build_log.txt 2>&1; then
    pass_check "Build succeeds with zero compilation errors"
else
    fail_check "Build failed - check 'make build' output"
    tail -20 /tmp/build_log.txt
fi

# Check 1.3: Pool validator file exists
if [ -f "pkg/scanner/market/pool_validator.go" ]; then
    LINES=$(wc -l < pkg/scanner/market/pool_validator.go)
    pass_check "Pool Validator deployed ($LINES lines)"
else
    fail_check "Pool Validator not found at pkg/scanner/market/pool_validator.go"
fi

# Check 1.4: Multi-hop liquidity validation in place
if grep -q "pool.Liquidity.Cmp(uint256.NewInt(0))" pkg/arbitrage/multihop.go 2>/dev/null; then
    pass_check "Multi-hop liquidity validation implemented"
else
    fail_check "Multi-hop liquidity validation not found"
fi

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "SECTION 2: RUNTIME VERIFICATION"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

# Check 2.1: Bot is running
BOT_PID=$(pgrep -f "mev-bot start" 2>/dev/null || echo "")
if [ ! -z "$BOT_PID" ]; then
    pass_check "MEV Bot process running (PID: $BOT_PID)"

    # Get uptime
    UPTIME=$(ps -p "$BOT_PID" -o etime= 2>/dev/null | tr -d ' ' || echo "unknown")
    info_check "Bot uptime: $UPTIME"
else
    warn_check "MEV Bot not currently running (expected if just verifying build)"
fi

# Check 2.2: Anvil fork is running
if pgrep -f "anvil --fork-url" > /dev/null 2>&1; then
    pass_check "Anvil fork running on port 8545"

    # Verify chain ID
    CHAIN=$(curl -s http://127.0.0.1:8545 -X POST \
      -H "Content-Type: application/json" \
      -d '{"jsonrpc":"2.0","method":"eth_chainId","params":[],"id":1}' 2>/dev/null | \
      jq -r '.result' 2>/dev/null || echo "")

    if [ "$CHAIN" = "0xa4b1" ]; then
        pass_check "Anvil forked to Arbitrum mainnet (Chain ID 42161)"
    else
        warn_check "Anvil chain ID unexpected: $CHAIN (expected 0xa4b1)"
    fi
else
    info_check "Anvil fork not running (not required for build verification)"
fi

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "SECTION 3: BLOCKER #1 - POOL ADDRESS VALIDATION"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

# Check 3.1: Pool validator in scanner
if grep -q "poolValidator := NewPoolValidator" pkg/scanner/market/scanner.go 2>/dev/null; then
    pass_check "Pool validator integrated into market scanner"
else
    fail_check "Pool validator integration not found in scanner.go"
fi

# Check 3.2: Validation check in fetchPoolData
if grep -q "pv.poolValidator.IsValidPoolAddress" pkg/scanner/market/scanner.go 2>/dev/null; then
    pass_check "Pool validation check in fetchPoolData() function"
else
    fail_check "Pool validation check not found in fetchPoolData()"
fi

# Check 3.3: Zero address validation
if grep -q 'addr == (common.Address{})' pkg/scanner/market/pool_validator.go 2>/dev/null; then
    pass_check "Zero address validation implemented"
else
    fail_check "Zero address validation not found"
fi

# Check 3.4: Contract existence check
if grep -q "eth_getCode\|CodeAt" pkg/scanner/market/pool_validator.go 2>/dev/null; then
    pass_check "Contract existence check via RPC implemented"
else
    fail_check "Contract existence check not found"
fi

# Check 3.5: Blacklist file
if [ -f "logs/pool_blacklist.json" ]; then
    BLACKLIST_COUNT=$(jq 'keys | length' logs/pool_blacklist.json 2>/dev/null || echo "0")
    pass_check "Pool blacklist loaded ($BLACKLIST_COUNT entries)"
else
    warn_check "Pool blacklist not yet created (will be created on first run)"
fi

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "SECTION 4: BLOCKER #2 - REAL LIQUIDITY VALIDATION"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

# Check 4.1: Liquidity nil check
if grep -q "pool == nil" pkg/arbitrage/multihop.go 2>/dev/null; then
    pass_check "Pool nil pointer check implemented"
else
    fail_check "Pool nil pointer check not found"
fi

# Check 4.2: Liquidity value check
if grep -q "pool.Liquidity.*Cmp.*uint256" pkg/arbitrage/multihop.go 2>/dev/null; then
    pass_check "Real liquidity (>0) validation implemented"
else
    fail_check "Real liquidity validation not found"
fi

# Check 4.3: SqrtPrice validation
if grep -q "pool.SqrtPriceX96.*Cmp" pkg/arbitrage/multihop.go 2>/dev/null; then
    pass_check "SqrtPrice validation for V3 pools implemented"
else
    fail_check "SqrtPrice validation not found"
fi

# Check 4.4: Type safety (uint256)
if grep -q "uint256.NewInt(0)" pkg/arbitrage/multihop.go 2>/dev/null; then
    pass_check "Type-safe uint256 comparisons used"
else
    fail_check "Type-safe comparisons not found"
fi

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "SECTION 5: BLOCKER #3 - SECURITY MANAGER"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

# Check 5.1: Security manager in main
if grep -q "NewSecurityManager\|securityManager" cmd/mev-bot/main.go 2>/dev/null; then
    pass_check "Security manager implemented in main.go"
else
    fail_check "Security manager not found in main.go"
fi

# Check 5.2: Production gating
if grep -q 'GO_ENV.*production\|os.Getenv.*GO_ENV' cmd/mev-bot/main.go 2>/dev/null; then
    pass_check "Security manager gated for production environment"
else
    warn_check "Production gating for security manager not explicitly shown"
fi

# Check 5.3: Environment variable set
if [ "$GO_ENV" = "production" ]; then
    pass_check "GO_ENV set to production mode"
else
    info_check "GO_ENV is '${GO_ENV:-not set}' (set to 'production' for production run)"
fi

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "SECTION 6: BLOCKER #4 - ARBITRAGE EXECUTION"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

# Check 6.1: All prerequisites met
if [ $CHECKS_FAILED -eq 0 ]; then
    pass_check "All prerequisites for arbitrage execution are in place"
else
    fail_check "Prerequisites for arbitrage execution NOT met (fix $CHECKS_FAILED items above)"
fi

# Check 6.2: Arbitrage service
if grep -q "arbitrageService\|ArbitrageService" cmd/mev-bot/main.go 2>/dev/null; then
    pass_check "Arbitrage service implemented and integrated"
else
    fail_check "Arbitrage service not found"
fi

# Check 6.3: Multi-hop scanner
if grep -q "NewMultiHopScanner\|multihop" pkg/arbitrage/service.go 2>/dev/null || \
   grep -q "multihop" cmd/mev-bot/main.go 2>/dev/null; then
    pass_check "Multi-hop arbitrage scanner implemented"
else
    warn_check "Multi-hop scanner integration not explicitly found"
fi

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "SECTION 7: LOG ANALYSIS"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

if [ -f "logs/mev-bot.log" ]; then
    TOTAL_LINES=$(wc -l < logs/mev-bot.log)
    ERROR_COUNT=$(grep -c "Error\|ERROR" logs/mev-bot.log 2>/dev/null || echo "0")
    POOL_ERRORS=$(grep -c "Error getting pool data" logs/mev-bot.log 2>/dev/null || echo "0")
    DETECTIONS=$(grep -c "Detected\|detected\|arbitrage\|opportunity" logs/mev-bot.log 2>/dev/null || echo "0")

    info_check "Total log lines: $TOTAL_LINES"
    info_check "Error messages: $ERROR_COUNT"

    if [ "$POOL_ERRORS" -eq "0" ]; then
        pass_check "NO pool data errors (0 detected) - Validator working!"
    elif [ "$POOL_ERRORS" -lt 5 ]; then
        warn_check "Low pool data errors ($POOL_ERRORS) - Acceptable"
    else
        fail_check "High pool data errors ($POOL_ERRORS) - May indicate issues"
    fi

    info_check "Opportunity detections: $DETECTIONS"
else
    info_check "No log file yet (will be created on first run)"
fi

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "SECTION 8: SYSTEM INTEGRATION"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

# Check 8.1: Configuration file
if [ -f "config/arbitrum_production.yaml" ]; then
    pass_check "Production configuration file exists"
else
    warn_check "Production configuration file not found"
fi

# Check 8.2: Environment file
if [ -f ".env.production" ]; then
    pass_check "Production environment file configured"
else
    warn_check "Production environment file not found"
fi

# Check 8.3: Tests pass
if make test > /tmp/test_log.txt 2>&1; then
    PASS_COUNT=$(grep -c "^PASS" /tmp/test_log.txt 2>/dev/null || echo "0")
    pass_check "All tests pass (include integration tests)"
else
    warn_check "Some tests may not be passing"
fi

# Check 8.4: Dashboard
if [ "$BOT_PID" != "" ] && curl -s http://localhost:8080/ > /dev/null 2>&1; then
    pass_check "Dashboard running on http://localhost:8080"
else
    info_check "Dashboard not currently accessible (expected if bot not running)"
fi

echo ""
echo "╔════════════════════════════════════════════════════════╗"
echo "║                    FINAL SUMMARY                       ║"
echo "╚════════════════════════════════════════════════════════╝"
echo ""

TOTAL_CHECKS=$((CHECKS_PASSED + CHECKS_FAILED + CHECKS_WARNING))

echo -e "Total Checks: $TOTAL_CHECKS"
echo -e "${GREEN}✅ Passed:   $CHECKS_PASSED${NC}"
echo -e "${RED}❌ Failed:   $CHECKS_FAILED${NC}"
echo -e "${YELLOW}⚠️  Warnings: $CHECKS_WARNING${NC}"
echo ""

# Calculate score
SCORE=$((CHECKS_PASSED * 100 / TOTAL_CHECKS))

echo "Overall Score: $SCORE/100"
echo ""

if [ $CHECKS_FAILED -eq 0 ]; then
    echo -e "${GREEN}╔════════════════════════════════════════════════════════╗${NC}"
    echo -e "${GREEN}║     ✅ PRODUCTION-READY - ALL BLOCKERS FIXED         ║${NC}"
    echo -e "${GREEN}║     Ready to deploy with: export GO_ENV=production   ║${NC}"
    echo -e "${GREEN}╚════════════════════════════════════════════════════════╝${NC}"
    echo ""
    exit 0
elif [ $CHECKS_FAILED -lt 3 ]; then
    echo -e "${YELLOW}╔════════════════════════════════════════════════════════╗${NC}"
    echo -e "${YELLOW}║  ⚠️  NEARLY READY - Fix remaining issues before deploy ║${NC}"
    echo -e "${YELLOW}╚════════════════════════════════════════════════════════╝${NC}"
    echo ""
    exit 1
else
    echo -e "${RED}╔════════════════════════════════════════════════════════╗${NC}"
    echo -e "${RED}║  ❌ NOT READY - Multiple blockers need attention       ║${NC}"
    echo -e "${RED}╚════════════════════════════════════════════════════════╝${NC}"
    echo ""
    exit 1
fi