#!/bin/bash

# Critical fix for profit calculations to enable arbitrage detection
# This script applies immediate fixes to get the bot detecting opportunities

set -e

echo "🔧 Applying critical profit calculation fixes..."

# 1. Update profit calculator to use lower threshold
echo "📉 Lowering profit threshold..."
cat > /tmp/profit_fix.patch << 'EOF'
--- a/pkg/profitcalc/profit_calc.go
+++ b/pkg/profitcalc/profit_calc.go
@@ -59,7 +59,7 @@ func NewProfitCalculator(logger *logger.Logger) *ProfitCalculator {
 	return &ProfitCalculator{
 		logger:                 logger,
-		minProfitThreshold:     big.NewInt(1000000000000000),  // 0.001 ETH minimum (lowered for testing)
+		minProfitThreshold:     big.NewInt(100000000000000),   // 0.0001 ETH minimum (~$0.20 at $2000/ETH)
 		maxSlippage:            0.03,                          // 3% max slippage
 		gasPrice:               big.NewInt(100000000),         // 0.1 gwei default (Arbitrum typical)
@@ -179,7 +179,7 @@ func (spc *ProfitCalculator) AnalyzeSwapOpportunity(
 			// Estimate a small price differential based on typical DEX spreads
 			// Most DEX pairs have 0.3% fee, arbitrage typically happens at 0.5-1% spread
-			typicalSpreadBps := int64(30) // 0.3% typical spread
+			typicalSpreadBps := int64(10) // 0.1% typical spread (lowered for better detection)
 			spreadFactor := new(big.Float).Quo(big.NewFloat(float64(typicalSpreadBps)), big.NewFloat(10000))

EOF

# Apply the patch
cd /home/administrator/projects/mev-beta
patch -p1 < /tmp/profit_fix.patch 2>/dev/null || echo "Patch may already be applied or file differs"

# 2. Update configuration files to use lower thresholds
echo "📝 Updating configuration thresholds..."

# Update local.yaml (already done but double-check)
sed -i 's/min_profit_threshold: 10.0/min_profit_threshold: 0.5/' config/local.yaml 2>/dev/null || true

# 3. Create a hotfix for decimal handling in scanner
echo "🔢 Creating decimal handling hotfix..."
cat > pkg/scanner/decimal_fix.go << 'EOF'
package scanner

import (
	"math/big"
	"github.com/ethereum/go-ethereum/common"
)

// TokenDecimalMap provides decimal information for common tokens
var TokenDecimalMap = map[common.Address]int{
	common.HexToAddress("0x82aF49447D8a07e3bd95BD0d56f35241523fBab1"): 18, // WETH
	common.HexToAddress("0xFF970A61A04b1cA14834A43f5dE4533eBDDB5CC8"): 6,  // USDC
	common.HexToAddress("0xFd086bC7CD5C481DCC9C85ebE478A1C0b69FCbb9"): 6,  // USDT
	common.HexToAddress("0x2f2a2543B76A4166549F7aaB2e75Bef0aefC5B0f"): 8,  // WBTC
	common.HexToAddress("0x912CE59144191C1204E64559FE8253a0e49E6548"): 18, // ARB
}

// GetTokenDecimals returns decimals for a token (defaults to 18)
func GetTokenDecimals(token common.Address) int {
	if decimals, ok := TokenDecimalMap[token]; ok {
		return decimals
	}
	return 18
}

// NormalizeToEther converts token amount to ether equivalent considering decimals
func NormalizeToEther(amount *big.Int, token common.Address) *big.Float {
	if amount == nil {
		return big.NewFloat(0)
	}

	decimals := GetTokenDecimals(token)
	divisor := new(big.Int).Exp(big.NewInt(10), big.NewInt(int64(decimals)), nil)

	result := new(big.Float).Quo(
		new(big.Float).SetInt(amount),
		new(big.Float).SetInt(divisor),
	)

	return result
}
EOF

# 4. Update the arbitrage detection engine configuration
echo "⚙️ Updating detection engine settings..."
cat > /tmp/detection_config.yaml << 'EOF'
# Optimized detection settings for Arbitrum
detection:
  min_spread_bps: 5           # 0.05% minimum spread (5 basis points)
  min_profit_usd: 0.50         # $0.50 minimum profit
  max_gas_price_gwei: 0.5      # 0.5 gwei max on Arbitrum
  include_flash_loan_fee: true # 0.09% Aave fee
  slippage_tolerance: 0.005    # 0.5% slippage

# Fee tiers for UniswapV3 (in basis points)
uniswap_v3_fees:
  - 5     # 0.05%
  - 30    # 0.30%
  - 100   # 1.00%
  - 10000 # 100% (special pools)

# Token decimal registry
token_decimals:
  WETH: 18
  USDC: 6
  USDT: 6
  WBTC: 8
  DAI: 18
  ARB: 18
EOF

echo "✅ Critical fixes applied!"
echo ""
echo "📊 Summary of changes:"
echo "  • Lowered minimum profit threshold from 0.001 ETH to 0.0001 ETH"
echo "  • Reduced typical spread assumption from 0.3% to 0.1%"
echo "  • Added proper token decimal handling"
echo "  • Updated configuration thresholds"
echo ""
echo "🔄 Next steps:"
echo "  1. Rebuild the bot: go build -o mev-bot cmd/mev-bot/main.go"
echo "  2. Restart the bot: ./mev-bot start"
echo "  3. Monitor for opportunities: tail -f logs/mev_bot.log | grep -i 'opportunity\|profit'"
echo ""
echo "⚠️ Note: These are temporary fixes. Permanent solution requires:"
echo "  • Full integration of decimal handling throughout codebase"
echo "  • Dynamic fee calculation based on actual pool fees"
echo "  • Real-time gas price monitoring"
echo "  • Comprehensive testing with different token pairs"