import { PowerSun } from '@powersun/mcp-client';

// Public access (market data, estimates)
const ps = new PowerSun();
const prices = await ps.market.getPrices();
const estimate = await ps.market.estimateCost({
  txCount: 10,
  txType: 'trc20_transfer',
  durationMinutes: 60,
});

// Authenticated access (buy energy, check balance)
const client = new PowerSun({ apiKey: 'ps_your_key_here' });
const balance = await client.buyer.getBalance();
const order = await client.buyer.buyEnergy({
  targetAddress: 'TYourAddressHere...',
  txCount: 5,
  durationMinutes: 60,
});

import { PowerSun, PaymentRequiredError } from '@powersun/mcp-client';

const ps = new PowerSun(); // no API key

try {
  await ps.buyer.buyEnergy({
    targetAddress: 'TYourAddressHere...',
    txCount: 1,
    durationMinutes: 5,
  });
} catch (err) {
  if (err instanceof PaymentRequiredError) {
    // Send TRX to the deposit address
    console.log('Deposit address:', err.payment.depositAddress);
    console.log('Amount (TRX):', err.payment.requiredAmountTrx);

    // Poll for payment confirmation
    const status = await ps.paymentStatus(err.payment.intentId);
    console.log('Payment status:', status.status);
  }
}

const ps = new PowerSun();

// Step 1: Get challenge
const { challengeId, challenge } = await ps.register('TYourAddress...');

// Step 2: Sign challenge with TronLink / tronWeb
const signature = await tronWeb.trx.signMessageV2(challenge);

// Step 3: Verify and get API key
const { apiKey } = await ps.verify(challengeId, 'TYourAddress...', signature);

// Step 4: Use authenticated client
const client = new PowerSun({ apiKey });

const client = new PowerSun({ apiKey: 'ps_your_key_here' });

// Get swap quote (TRX → USDT)
const quote = await client.swap.getQuote({
  fromToken: 'TRX',
  toToken: 'USDT',
  amountIn: '10000000', // 10 TRX in SUN
  slippageBps: 100,     // 1% slippage
});

console.log('Expected output:', quote.quote.expectedAmountOut);
console.log('Energy cost:', quote.energy.costTrx, 'TRX');

// Sign the unsigned TX with tronWeb
const signedTx = await tronWeb.trx.sign(quote.unsignedTx);

// Broadcast with automatic energy delegation
const result = await client.buyer.broadcast({ txData: signedTx });