hash-wallet/lib/exchange/provider/jupiter_exchange_provider.dart

import 'dart:convert';

import 'package:cake_wallet/.secrets.g.dart' as secrets;
import 'package:cake_wallet/exchange/exchange_provider_description.dart';
import 'package:cake_wallet/exchange/limits.dart';
import 'package:cake_wallet/exchange/provider/exchange_provider.dart';
import 'package:cake_wallet/exchange/trade.dart';
import 'package:cake_wallet/exchange/trade_not_created_exception.dart';
import 'package:cake_wallet/exchange/trade_request.dart';
import 'package:cake_wallet/exchange/trade_state.dart';
import 'package:cake_wallet/solana/solana.dart';
import 'package:cake_wallet/utils/exchange_provider_logger.dart';
import 'package:cw_core/amount_converter.dart';
import 'package:cw_core/crypto_currency.dart';
import 'package:cw_core/utils/print_verbose.dart';
import 'package:cw_core/utils/proxy_wrapper.dart';

class JupiterExchangeProvider extends ExchangeProvider {
  JupiterExchangeProvider();

  // Jupiter only supports Solana native SOL and Solana tokens
  bool _isSolanaCurrency(CryptoCurrency currency) =>
      currency == CryptoCurrency.sol || currency.tag == 'SOL';

  static const _baseUrl = 'api.jup.ag';
  static const _orderPath = '/ultra/v1/order';
  static const _executePath = '/ultra/v1/execute';

  // Wrapped SOL address (native SOL)
  static const _nativeSolMint = 'So11111111111111111111111111111111111111112';

  @override
  String get title => 'Jupiter';

  @override
  bool get isAvailable => true;

  @override
  bool get isEnabled => true;

  @override
  bool get supportsFixedRate => false; // Jupiter doesn't support fixed rate

  @override
  bool get supportsMemoOrDestinationTag => false;

  @override
  ExchangeProviderDescription get description => ExchangeProviderDescription.jupiter;

  @override
  Future<bool> checkIsAvailable() async => true;

  String _getTokenMint(CryptoCurrency currency) {
    // Handle native SOL
    if (currency == CryptoCurrency.sol) return _nativeSolMint;

    // Check if currency tag is SOL (indicating it's a Solana token)
    if (currency.tag != 'SOL') {
      throw Exception('Unsupported currency: ${currency.title} (not a Solana token)');
    }

    // Use solana proxy to get token address
    // The proxy will handle both SPLToken instances and CryptoCurrency
    // by searching through default tokens
    if (solana != null) {
      try {
        return solana!.getTokenAddress(currency);
      } catch (e) {
        printV('Error getting token address: $e');
        throw Exception('Unsupported currency: ${currency.title} (mint address not found: $e)');
      }
    }

    throw Exception('Unsupported currency: ${currency.title} (Solana proxy not available)');
  }

  @override
  Future<Limits?> fetchLimits({
    required CryptoCurrency from,
    required CryptoCurrency to,
    required bool isFixedRateMode,
  }) async {
    // The Ultra Swap API doesn't have a dedicated limits endpoint
    // The /order endpoint validates amounts and returns error codes:
    // - errorCode 1: Insufficient funds
    // - errorCode 2: Top up SOL for gas
    // - errorCode 3: Minimum amount for gasless

    // only return null for supported currencies
    if (_isSolanaCurrency(from) && _isSolanaCurrency(to)) {
      return Limits(min: null, max: null);
    } else {
      throw Exception('not supported');
    }
  }

  Map<String, String> _getHeaders() {
    final headers = <String, String>{};
    final apiKey = secrets.jupiterApiKey;
    if (apiKey.isNotEmpty) {
      headers['x-api-key'] = apiKey;
    }

    return headers;
  }

  Map<String, String>? _getReferralFeeConfig() {
    try {
      final referralFeeBpsStr = secrets.jupiterReferralFeeBps;
      final referralFeeBps = int.tryParse(referralFeeBpsStr) ?? 0;

      final referralAccount = secrets.jupiterReferralAccount;

      // Only enable if both are configured and valid
      if (referralFeeBps <= 0 || referralFeeBps > 10000 || referralAccount.isEmpty) {
        return null;
      }

      return {
        'referralFee': referralFeeBps.toString(),
        'referralAccount': referralAccount,
      };
    } catch (e) {
      return null;
    }
  }

  @override
  Future<double> fetchRate({
    required CryptoCurrency from,
    required CryptoCurrency to,
    required double amount,
    required bool isFixedRateMode,
    required bool isReceiveAmount,
  }) async {
    try {
      // must support both
      if (!_isSolanaCurrency(from) || !_isSolanaCurrency(to)) {
        return 0.0;
      }
      final inputMint = _getTokenMint(from);
      final outputMint = _getTokenMint(to);

      final amountInBaseUnits = AmountConverter.toBaseUnits(amount.toString(), from.decimals);

      final params = {
        'inputMint': inputMint,
        'outputMint': outputMint,
        'amount': amountInBaseUnits,
        // Note: taker is optional for quote-only requests
      };

      final uri = Uri.https(_baseUrl, _orderPath, params);
      final headers = _getHeaders();

      final response = await ProxyWrapper().get(
        clearnetUri: uri,
        headers: headers,
      );

      if (response.statusCode != 200) {
        ExchangeProviderLogger.logError(
          provider: description,
          function: 'fetchRate',
          error: Exception('Failed to fetch quote: ${response.statusCode}'),
          stackTrace: StackTrace.current,
          requestData: {
            'from': from.title,
            'to': to.title,
            'amount': amount,
            'isFixedRateMode': isFixedRateMode,
            'isReceiveAmount': isReceiveAmount,
          },
        );
        return 0.0;
      }

      final orderData = json.decode(response.body) as Map<String, dynamic>;
      final outAmount = BigInt.parse(orderData['outAmount'] as String);

      final outputAmount = AmountConverter.fromBaseUnits(outAmount.toString(), to.decimals);

      final rate = double.parse(outputAmount) / amount;

      ExchangeProviderLogger.logSuccess(
        provider: description,
        function: 'fetchRate',
        requestData: {
          'from': from.title,
          'to': to.title,
          'amount': amount,
          'isFixedRateMode': isFixedRateMode,
          'isReceiveAmount': isReceiveAmount,
        },
        responseData: {
          'rate': rate,
          'outputAmount': outputAmount,
        },
      );

      return rate;
    } catch (e, s) {
      ExchangeProviderLogger.logError(
        provider: description,
        function: 'fetchRate',
        error: e,
        stackTrace: s,
        requestData: {
          'from': from.title,
          'to': to.title,
          'amount': amount,
          'isFixedRateMode': isFixedRateMode,
          'isReceiveAmount': isReceiveAmount,
        },
      );
      printV('fetchRate error: $e');
      return 0.0;
    }
  }

  @override
  Future<Trade> createTrade({
    required TradeRequest request,
    required bool isFixedRateMode,
    required bool isSendAll,
  }) async {
    try {
      // must support both
      if (!_isSolanaCurrency(request.fromCurrency) || !_isSolanaCurrency(request.toCurrency)) {
        throw 'not supported currencies';
      }

      final inputMint = _getTokenMint(request.fromCurrency);
      final outputMint = _getTokenMint(request.toCurrency);

      final amountInBaseUnits =
          AmountConverter.toBaseUnits(request.fromAmount, request.fromCurrency.decimals);

      final isInternalTransfer = request.refundAddress == request.toAddress;

      final orderParams = <String, String>{
        'inputMint': inputMint,
        'outputMint': outputMint,
        'amount': amountInBaseUnits,
        'taker': request.refundAddress,
        if (!isInternalTransfer) 'receiver': request.toAddress,
      };

      final referralFeeConfig = _getReferralFeeConfig();
      if (referralFeeConfig != null) {
        orderParams['referralFee'] = referralFeeConfig['referralFee']!;
        orderParams['referralAccount'] = referralFeeConfig['referralAccount']!;
      }

      final orderUri = Uri.https(_baseUrl, _orderPath, orderParams);
      final headers = _getHeaders();

      final orderResponse = await ProxyWrapper().get(clearnetUri: orderUri, headers: headers);

      if (orderResponse.statusCode != 200) {
        final errorBody = orderResponse.body;
        ExchangeProviderLogger.logError(
          provider: description,
          function: 'createTrade',
          error: Exception('Failed to get order: ${orderResponse.statusCode} $errorBody'),
          stackTrace: StackTrace.current,
          requestData: {
            'from': request.fromCurrency.title,
            'to': request.toCurrency.title,
            'fromAmount': request.fromAmount,
            'toAmount': request.toAmount,
            'toAddress': request.toAddress,
            'refundAddress': request.refundAddress,
            'isFixedRateMode': isFixedRateMode,
            'isSendAll': isSendAll,
          },
        );
        throw TradeNotCreatedException(description);
      }

      final orderData = json.decode(orderResponse.body) as Map<String, dynamic>;

      // Check for errors in response
      if (orderData.containsKey('errorCode') || orderData.containsKey('errorMessage')) {
        final errorCode = orderData['errorCode'];
        final errorMessage = orderData['errorMessage'] ?? 'Unknown error';
        ExchangeProviderLogger.logError(
          provider: description,
          function: 'createTrade',
          error: Exception('Order error: $errorCode - $errorMessage'),
          stackTrace: StackTrace.current,
          requestData: {
            'from': request.fromCurrency.title,
            'to': request.toCurrency.title,
            'fromAmount': request.fromAmount,
            'toAmount': request.toAmount,
            'toAddress': request.toAddress,
            'refundAddress': request.refundAddress,
            'isFixedRateMode': isFixedRateMode,
            'isSendAll': isSendAll,
          },
        );
        throw TradeNotCreatedException(description);
      }

      // Extract response data
      final transaction = orderData['transaction'] as String?;
      final requestId = orderData['requestId'] as String?;
      final outAmount = orderData['outAmount'] as String? ?? '0.0';

      // Extract network fees from order response (in lamports)
      final signatureFeeLamports = (orderData['signatureFeeLamports'] as num?)?.toInt() ?? 0;

      final prioritizationFeeLamports =
          (orderData['prioritizationFeeLamports'] as num?)?.toInt() ?? 0;

      final rentFeeLamports = (orderData['rentFeeLamports'] as num?)?.toInt() ?? 0;

      final integratorFeeLamports = (orderData['integratorFeeLamports'] as num?)?.toInt() ?? 0;

      final networkFeeLamports = signatureFeeLamports + prioritizationFeeLamports + rentFeeLamports;
      final networkFeeInSol = networkFeeLamports / 1000000000.0;

      final integratorFeeInSol = integratorFeeLamports / 1000000000.0;

      final totalFeeInSol = networkFeeInSol + integratorFeeInSol;

      if (transaction == null || transaction.isEmpty) {
        throw Exception('No transaction returned from Jupiter order endpoint');
      }

      if (requestId == null || requestId.isEmpty) {
        throw Exception('No requestId returned from Jupiter order endpoint');
      }

      final receiveAmount = AmountConverter.fromBaseUnits(outAmount, request.toCurrency.decimals);

      ExchangeProviderLogger.logSuccess(
        provider: description,
        function: 'createTrade',
        requestData: {
          'from': request.fromCurrency.title,
          'to': request.toCurrency.title,
          'fromAmount': request.fromAmount,
          'toAmount': request.toAmount,
          'toAddress': request.toAddress,
          'refundAddress': request.refundAddress,
          'isFixedRateMode': isFixedRateMode,
          'isSendAll': isSendAll,
        },
        responseData: {
          'tradeId': requestId,
          'receiveAmount': receiveAmount,
          'hasTransaction': transaction.isNotEmpty,
          'requestId': requestId,
        },
      );

      return Trade(
        id: requestId,
        from: request.fromCurrency,
        to: request.toCurrency,
        provider: description,
        inputAddress: request.toAddress,
        refundAddress: request.refundAddress,
        state: TradeState.created,
        createdAt: DateTime.now(),
        amount: request.fromAmount,
        receiveAmount: receiveAmount,
        payoutAddress: request.toAddress,
        isSendAll: isSendAll,
        routerData: transaction,
        routerValue: requestId,
        fee: totalFeeInSol,
      );
    } catch (e, s) {
      ExchangeProviderLogger.logError(
        provider: description,
        function: 'createTrade',
        error: e,
        stackTrace: s,
        requestData: {
          'from': request.fromCurrency.title,
          'to': request.toCurrency.title,
          'fromAmount': request.fromAmount,
          'toAmount': request.toAmount,
          'toAddress': request.toAddress,
          'refundAddress': request.refundAddress,
          'isFixedRateMode': isFixedRateMode,
          'isSendAll': isSendAll,
        },
      );
      printV('createTrade error: $e');
      throw TradeNotCreatedException(description);
    }
  }

  /// Executes a signed Jupiter swap transaction via Jupiter's /execute endpoint
  Future<Map<String, dynamic>> executeSwap({
    required String signedTransaction,
    required String requestId,
  }) async {
    try {
      final executeUri = Uri.https(_baseUrl, _executePath);
      final headers = _getHeaders();
      headers['Content-Type'] = 'application/json';

      final body = json.encode({
        'signedTransaction': signedTransaction,
        'requestId': requestId,
      });

      final response = await ProxyWrapper().post(
        clearnetUri: executeUri,
        headers: headers,
        body: body,
      );

      if (response.statusCode != 200) {
        final errorBody = response.body;
        ExchangeProviderLogger.logError(
          provider: description,
          function: 'executeSwap',
          error: Exception('Failed to execute swap: ${response.statusCode} $errorBody'),
          stackTrace: StackTrace.current,
          requestData: {
            'requestId': requestId,
            'hasSignedTransaction': signedTransaction.isNotEmpty,
          },
        );
        throw Exception('Failed to execute swap: ${response.statusCode} $errorBody');
      }

      final executeData = json.decode(response.body) as Map<String, dynamic>;

      ExchangeProviderLogger.logSuccess(
        provider: description,
        function: 'executeSwap',
        requestData: {
          'requestId': requestId,
          'hasSignedTransaction': signedTransaction.isNotEmpty,
        },
        responseData: executeData,
      );

      return executeData;
    } catch (e, s) {
      ExchangeProviderLogger.logError(
        provider: description,
        function: 'executeSwap',
        error: e,
        stackTrace: s,
        requestData: {
          'requestId': requestId,
          'hasSignedTransaction': signedTransaction.isNotEmpty,
        },
      );
      rethrow;
    }
  }

  @override
  Future<Trade> findTradeById({required String id}) async {
    // Jupiter Ultra Swap API doesn't track trades by our trade ID
    //
    // Status tracking options:
    // 1. Use /execute endpoint with requestId + signedTransaction (requires storing signed tx)
    // 2. Check on-chain via transaction signature (txId) after transaction is sent
    //
    // Current implementation: We track status on-chain via transaction signature
    // The txId field in Trade is set after the transaction is sent and can be
    // used to check transaction status via Solana RPC.
    //
    // Note: To use /execute endpoint for status polling, we would need to:
    // - Store the signed transaction (not currently stored)
    // - Use requestId from routerValue
    // - Poll /ultra/v1/execute with both signedTransaction and requestId
    //
    // For now, throw exception to indicate status must be checked on-chain
    throw Exception(
        'Jupiter trade status must be checked on-chain using transaction signature (txId). '
        'After transaction is sent, txId will contain the signature for status checking.');
  }
}