import { NextResponse } from 'next/server'
import { Connection, Keypair } from '@solana/web3.js'

export async function GET() {
  try {
    console.log('💰 Fetching real Solana wallet balance...')

    // Check if wallet is configured
    if (!process.env.SOLANA_PRIVATE_KEY) {
      return NextResponse.json({
        success: false,
        error: 'Wallet not configured',
        message: 'SOLANA_PRIVATE_KEY not found in environment'
      }, { status: 503 })
    }

    // Initialize connection and keypair
    const rpcUrl = process.env.SOLANA_RPC_URL || 'https://api.mainnet-beta.solana.com'
    const connection = new Connection(rpcUrl, 'confirmed')
    
    const privateKeyArray = JSON.parse(process.env.SOLANA_PRIVATE_KEY)
    const keypair = Keypair.fromSecretKey(new Uint8Array(privateKeyArray))

    // Get SOL balance
    const balance = await connection.getBalance(keypair.publicKey)
    const solBalance = balance / 1000000000 // Convert lamports to SOL

    // Get current SOL price with fallback
    let solPrice = 168.11 // Fallback price from our current market data
    let change24h = 0
    
    try {
      const priceResponse = await fetch(
        'https://api.coingecko.com/api/v3/simple/price?ids=solana&vs_currencies=usd&include_24hr_change=true'
      )
      
      if (priceResponse.ok) {
        const priceData = await priceResponse.json()
        if (priceData.solana?.usd) {
          solPrice = priceData.solana.usd
          change24h = priceData.solana.usd_24h_change || 0
          console.log(`💰 Using live SOL price: $${solPrice}`)
        } else {
          console.log(`⚠️ Using fallback SOL price: $${solPrice} (CoinGecko data invalid)`)
        }
      } else {
        console.log(`⚠️ Using fallback SOL price: $${solPrice} (CoinGecko rate limited)`)
      }
    } catch (priceError) {
      console.log(`⚠️ Using fallback SOL price: $${solPrice} (CoinGecko error: ${priceError.message})`)
    }

    const usdValue = solBalance * solPrice

    console.log(`💎 Real wallet: ${solBalance.toFixed(4)} SOL ($${usdValue.toFixed(2)})`)

    return NextResponse.json({
      success: true,
      balance: {
        totalValue: usdValue,
        availableBalance: usdValue,
        positions: [
          {
            symbol: 'SOL',
            price: solPrice,
            change24h: change24h,
            volume24h: 0,
            amount: solBalance,
            usdValue: usdValue
          }
        ]
      },
      wallet: {
        publicKey: keypair.publicKey.toString(),
        solBalance: solBalance,
        usdValue: usdValue
      },
      timestamp: Date.now()
    })

  } catch (error) {
    console.error('❌ Wallet balance API error:', error)
    return NextResponse.json({
      success: false,
      error: 'Failed to fetch wallet balance',
      message: error.message
    }, { status: 500 })
  }
}