stack_wallet/lib/services/mixins/fusion_wallet_interface.dart

import 'dart:async';
import 'dart:convert';
import 'dart:io';

import 'package:decimal/decimal.dart';
import 'package:fusiondart/fusiondart.dart';
import 'package:fusiondart/src/models/address.dart' as fusion_address;
import 'package:fusiondart/src/models/input.dart' as fusion_input;
import 'package:fusiondart/src/models/output.dart' as fusion_output;
import 'package:fusiondart/src/models/transaction.dart' as fusion_tx;
import 'package:isar/isar.dart';
import 'package:stackwallet/db/isar/main_db.dart';
import 'package:stackwallet/electrumx_rpc/cached_electrumx.dart';
import 'package:stackwallet/models/isar/models/isar_models.dart';
import 'package:stackwallet/services/tor_service.dart';
import 'package:stackwallet/utilities/amount/amount.dart';
import 'package:stackwallet/utilities/enums/coin_enum.dart';
import 'package:stackwallet/utilities/enums/derive_path_type_enum.dart';
import 'package:stackwallet/utilities/stack_file_system.dart';

const String kReservedFusionAddress = "reserved_fusion_address";

/// A mixin for the BitcoinCashWallet class that adds CashFusion functionality.
mixin FusionWalletInterface {
  // Passed in wallet data.
  static late final String _walletId;
  late final Coin _coin;
  static late final MainDB _db;
  late final CachedElectrumX _cachedElectrumX;
  late final TorService _torService;

  // Passed in wallet functions.
  late final Future<Address> Function(
    int chain,
    int index,
    DerivePathType derivePathType,
  ) _generateAddressForChain;

  /// Initializes the FusionWalletInterface mixin.
  ///
  /// This function must be called before any other functions in this mixin.
  ///
  /// Returns a `Future<void>` that resolves when Tor has been started.
  Future<void> initFusionInterface({
    required String walletId,
    required Coin coin,
    required MainDB db,
    required Future<Address> Function(
      int,
      int,
      DerivePathType,
    ) generateAddressForChain,
    required CachedElectrumX cachedElectrumX,
  }) async {
    // Set passed in wallet data.
    _walletId = walletId;
    _coin = coin;
    _db = db;
    _generateAddressForChain = generateAddressForChain;
    _torService = TorService.sharedInstance;
    _cachedElectrumX = cachedElectrumX;

    // Try getting the proxy info.
    //
    // Start the Tor service if it's not already running.  Returns if Tor is already
    // connected or else after Tor returns from start().
    try {
      _torService.getProxyInfo();
      // Proxy info successfully retrieved, Tor is connected.
      return;
    } catch (e) {
      // Init the Tor service if it hasn't already been.
      final torDir = await StackFileSystem.applicationTorDirectory();
      _torService.init(torDataDirPath: torDir.path);

      // Start the Tor service.
      return await _torService.start();
    }
  }

  /// Returns a list of all addresses in the wallet.
  Future<List<fusion_address.Address>> getFusionAddresses() async {
    List<Address> _addresses = await _db.getAddresses(_walletId).findAll();
    return _addresses.map((address) => address.toFusionAddress()).toList();
  }

  /// Returns a list of all transactions in the wallet for the given address.
  Future<Set<fusion_tx.Transaction>> getTransactionsByAddress(
      String address) async {
    var _txs = await _db.getTransactions(_walletId).findAll();

    // Use Future.wait to await all the futures in the set and then convert it to a set.
    var resultSet = await Future.wait(
        _txs.map((tx) => tx.toFusionTransaction(_cachedElectrumX)));

    return resultSet.toSet();
  }

  /// Returns a list of all UTXOs in the wallet for the given address.
  Future<List<fusion_input.Input>> getInputsByAddress(String address) async {
    var _utxos = await _db.getUTXOsByAddress(_walletId, address).findAll();

    List<Future<fusion_input.Input>> futureInputs =
        _utxos.map((utxo) => utxo.toFusionInput()).toList();

    return await Future.wait(futureInputs);
  }

  /// Creates a new reserved change address.
  Future<fusion_address.Address> createNewReservedChangeAddress() async {
    int? highestChangeIndex = await _db
        .getAddresses(_walletId)
        .filter()
        .typeEqualTo(AddressType.p2pkh)
        .subTypeEqualTo(AddressSubType.change)
        .derivationPath((q) => q.not().valueStartsWith("m/44'/0'"))
        .sortByDerivationIndexDesc()
        .derivationIndexProperty()
        .findFirst();

    Address address = await _generateAddressForChain(
      1, // change chain
      highestChangeIndex ?? 0,
      DerivePathTypeExt.primaryFor(_coin),
    );
    address = address.copyWith(otherData: kReservedFusionAddress);

    // TODO if we really want to be sure it's not used, call electrumx and check it

    Address? _address = await _db.getAddress(_walletId, address.value);
    if (_address != null) {
      // throw Exception("Address already exists");
      await _db.updateAddress(_address, address);
    } else {
      await _db.putAddress(address);
    }

    return address.toFusionAddress();
  }

  /// Returns a list of unused reserved change addresses.
  ///
  /// If there are not enough unused reserved change addresses, new ones are created.
  Future<List<fusion_address.Address>> getUnusedReservedChangeAddresses(
    int numberOfAddresses,
  ) async {
    // Fetch all transactions that have been sent to a reserved change address.
    final txns = await _db
        .getTransactions(_walletId)
        .filter()
        .address((q) => q.otherDataEqualTo(kReservedFusionAddress))
        .findAll();

    // Fetch all addresses that have been used in a transaction.
    final List<String> usedAddresses = txns
        .where((e) => e.address.value != null)
        .map((e) => e.address.value!.value)
        .toList(growable: false);

    // Fetch all reserved change addresses.
    final List<Address> addresses = await _db
        .getAddresses(_walletId)
        .filter()
        .otherDataEqualTo(kReservedFusionAddress)
        .findAll();

    // Initialize a list of unused reserved change addresses.
    final List<fusion_address.Address> unusedAddresses = [];

    // Add any unused reserved change addresses to the list.
    for (final address in addresses) {
      if (!usedAddresses.contains(address.value)) {
        unusedAddresses.add(address.toFusionAddress());
      }
    }

    // If there are not enough unused reserved change addresses, create new ones.
    if (unusedAddresses.length < numberOfAddresses) {
      for (int i = unusedAddresses.length; i < numberOfAddresses; i++) {
        unusedAddresses.add(await createNewReservedChangeAddress());
      }
    }

    // Return the list of unused reserved change addresses.
    return unusedAddresses;
  }

  /// Get an address.
  static Future<Address> getAddress(String addr) async {
    Address? address = await _db.getAddress(_walletId, addr);

    if (address == null) {
      throw Exception("Address not found");
    }

    return address;
  }

  /// Returns the current Tor proxy address.
  Future<({InternetAddress host, int port})> getSocksProxyAddress() async {
    /*
    // Start the Tor service if it's not already running.
    if (_torService.proxyInfo.port == -1) { // -1 indicates that the proxy is not running.
      await _torService.start(); // We already unawaited this in initFusionInterface...
    }
     */

    // TODO make sure we've properly awaited the Tor service starting before
    // returning the proxy address.

    // Return the proxy address.
    return _torService.getProxyInfo();
  }

  // Initial attempt for CashFusion integration goes here.

  /// Fuse the wallet's UTXOs.
  ///
  /// This function is called when the user taps the "Fuse" button in the UI.
  ///
  /// Returns:
  ///   A `Future<void>` that resolves when the fusion operation is finished.
  Future<void> fuse() async {
    // Initial attempt for CashFusion integration goes here.
    Fusion mainFusionObject = Fusion(
      getAddresses: () => getFusionAddresses(),
      getTransactionsByAddress: (String address) =>
          getTransactionsByAddress(address),
      getInputsByAddress: (String address) => getInputsByAddress(address),
      // createNewReservedChangeAddress: () => createNewReservedChangeAddress(),
      getUnusedReservedChangeAddresses: (int numberOfAddresses) =>
          getUnusedReservedChangeAddresses(numberOfAddresses),
      getSocksProxyAddress: () => getSocksProxyAddress(),
    );

    // Pass wallet functions to the Fusion object
    mainFusionObject.initFusion(
        getAddresses: getFusionAddresses,
        getTransactionsByAddress: getTransactionsByAddress,
        getInputsByAddress: getInputsByAddress,
        /*createNewReservedChangeAddress: createNewReservedChangeAddress,*/
        getUnusedReservedChangeAddresses: getUnusedReservedChangeAddresses,
        getSocksProxyAddress: getSocksProxyAddress);

    // Add stack UTXOs.
    List<UTXO> utxos = await _db.getUTXOs(_walletId).findAll();
    List<(String, int, int, List<int>)> coinList = [];

    // Loop through UTXOs, checking and adding valid ones.
    for (var e in utxos) {
      // Check if address is available.
      if (e.address == null) {
        // TODO we could continue here (and below during scriptPubKey validation) instead of throwing.
        throw Exception("UTXO ${e.txid}:${e.vout} address is null");
      }

      // Find public key.
      print("1 getting tx ${e.txid}");
      Map<String, dynamic> tx = await _cachedElectrumX.getTransaction(
          coin: _coin,
          txHash: e.txid,
          verbose: true); // TODO is verbose needed?

      // Check if scriptPubKey is available.
      if (tx["vout"] == null) {
        throw Exception("Vout in transaction ${e.txid} is null");
      }
      if (tx["vout"][e.vout] == null) {
        throw Exception("Vout index ${e.vout} in transaction is null");
      }
      if (tx["vout"][e.vout]["scriptPubKey"] == null) {
        throw Exception("scriptPubKey at vout index ${e.vout} is null");
      }
      if (tx["vout"][e.vout]["scriptPubKey"]["hex"] == null) {
        throw Exception("hex in scriptPubKey at vout index ${e.vout} is null");
      }

      // Assign scriptPubKey to pubKey.  TODO verify this is correct.
      List<int> pubKey =
          utf8.encode("${tx["vout"][e.vout]["scriptPubKey"]["hex"]}");

      // Add UTXO to coinList.
      coinList.add((e.txid, e.vout, e.value, pubKey));
    }

    // Add Stack UTXOs.
    await mainFusionObject.addCoinsFromWallet(coinList);

    // Fuse UTXOs.
    return await mainFusionObject.fuse();
    //print ("DEBUG FUSION bitcoincash_wallet.dart 1202");

    // TODO remove or fix code below.

    /*
    print("DEBUG: Waiting for any potential incoming data...");
    try {
      await Future.delayed(Duration(seconds: 5)); // wait for 5 seconds
    }
    catch (e) {
      print (e);
    }
    print("DEBUG: Done waiting.");

    bool mydebug1 = false;
    if (mydebug1 == true) {
      var serverIp = '167.114.119.46';
      var serverPort = 8787;

      List<int> frame = [
        118,
        91,
        232,
        180,
        228,
        57,
        109,
        207,
        0,
        0,
        0,
        45,
        10,
        43,
        10,
        7,
        97,
        108,
        112,
        104,
        97,
        49,
        51,
        18,
        32,
        111,
        226,
        140,
        10,
        182,
        241,
        179,
        114,
        193,
        166,
        162,
        70,
        174,
        99,
        247,
        79,
        147,
        30,
        131,
        101,
        225,
        90,
        8,
        156,
        104,
        214,
        25,
        0,
        0,
        0,
        0,
        0
      ];
      print("lets try to connect to a socket again");
      var socket = await Socket.connect(serverIp, serverPort);

      print('Connected to the server.');
      socket.add(frame);
      print('Sent frame: $frame');

      socket.listen((data) {
        print('Received from server: $data');
      }, onDone: () {
        print('Server closed connection.');
        socket.destroy();
      }, onError: (error) {
        print('Error: $error');
        socket.destroy();
      });
    }

    // await _checkCurrentChangeAddressesForTransactions();
    // await _checkCurrentReceivingAddressesForTransactions();
    */
  }

  Future<void> refreshFusion() {
    // TODO
    throw UnimplementedError(
        "TODO refreshFusion eg look up number of fusion participants connected/coordinating");
  }
}

/// An extension of Stack Wallet's Address class that adds CashFusion functionality.
extension FusionAddress on Address {
  fusion_address.Address toFusionAddress() {
    return fusion_address.Address(
        addr: value,
        publicKey: publicKey,
        derivationPath:
            fusion_address.DerivationPath(derivationPath?.value ?? ""));
    // TODO fix default derivation path.
  }
}

/// An extension of Stack Wallet's UTXO class that adds CashFusion functionality.
///
/// This class is used to convert Stack Wallet's UTXO class to FusionDart's
/// Input and Output classes.
extension FusionUTXO on UTXO {
  /// Converts a Stack Wallet UTXO to a FusionDart Input.
  Future<fusion_input.Input> toFusionInput() async {
    if (address == null) {
      throw Exception("toFutionInput Address is null");
    }

    // Search isar for address to get pubKey.
    Address addr = await FusionWalletInterface.getAddress(address!);

    return fusion_input.Input(
      prevTxid: utf8.encode(txid),
      prevIndex: vout,
      pubKey: addr.publicKey,
      amount: value,
    );
  }

  /// Converts a Stack Wallet UTXO to a FusionDart Output... eventually.
  Future<fusion_output.Output> toFusionOutput() async {
    if (address == null) {
      throw Exception("toFutionOutput Address is null");
    }

    // Search isar for address to get pubKey.
    Address addr = await FusionWalletInterface.getAddress(address!);

    return fusion_output.Output(
      addr: fusion_address.Address(
        addr: address!,
        publicKey: addr.publicKey, // TODO fix public key.
        derivationPath: null, // TODO fix derivation path.
      ),
      value: value,
    );
  }
}

/// An extension of Stack Wallet's Transaction class that adds CashFusion functionality.
extension FusionTransaction on Transaction {
  // WIP.
  Future<fusion_tx.Transaction> toFusionTransaction(
      CachedElectrumX cachedElectrumX) async {
    // Initialize Fusion Dart's Transaction object.
    fusion_tx.Transaction fusionTransaction = fusion_tx.Transaction();

    // WIP.
    fusionTransaction.Inputs = await Future.wait(inputs.map((e) async {
      // Find input amount.
      Map<String, dynamic> _tx = await cachedElectrumX.getTransaction(
          coin: Coin.bitcoincash,
          txHash: e.txid,
          verbose: true); // TODO is verbose needed?

      // Check if output amount is available.
      if (_tx.isEmpty) {
        throw Exception("Transaction not found for input: ${e.txid}");
      }
      if (_tx["vout"] == null) {
        throw Exception("Vout in transaction ${e.txid} is null");
      }
      if (_tx["vout"][e.vout] == null) {
        throw Exception("Vout index ${e.vout} in transaction is null");
      }
      if (_tx["vout"][e.vout]["value"] == null) {
        throw Exception("Value of vout index ${e.vout} in transaction is null");
      }
      // TODO replace with conditional chaining?

      // Assign vout value to amount.
      final value = Amount.fromDecimal(
        Decimal.parse(_tx["vout"][e.vout]["value"].toString()),
        fractionDigits: Coin.bitcoincash.decimals,
      );

      return fusion_input.Input(
        prevTxid: utf8.encode(e.txid), // TODO verify this is what we want.
        prevIndex: e.vout, // TODO verify this is what we want.
        pubKey: utf8.encode('0000'), // TODO fix public key.
        amount: value.raw.toInt(),
      );
    }).toList());

    fusionTransaction.Outputs = outputs.map((e) {
      /*
      if (e.scriptPubKey == null) {
        // TODO calculate scriptPubKey if it is null.
      }
      */

      fusion_address.DerivationPath? derivationPath;
      List<int>? pubKey;

      // Validate that we have all the required data.
      if (address.value == null) {
        // TODO calculate address if it is null.
        throw Exception(
            "address value is null for input: ${e.scriptPubKeyAddress}");
      } else {
        if (address.value!.publicKey.isEmpty || e.scriptPubKey != null) {
          pubKey = utf8.encode(e.scriptPubKey!);
          // TODO is this valid?
        } else {
          pubKey = address.value!
              .publicKey; // TODO IMPORTANT: this address may not be *the* address in question :)
        }
        if (address.value!.derivationPath != null) {
          derivationPath = fusion_address.DerivationPath(
              address.value!.derivationPath!.toString());
        } else {
          // TODO calculate derivation path if it is null.
          /*
          throw Exception(
              "derivationPath is null for input: ${e.scriptPubKeyAddress}");
          */
        }
      }

      // TODO handle case where address.value.publicKey is empty and e.scriptPubKey is null

      return fusion_output.Output(
        addr: fusion_address.Address(
          addr: e.scriptPubKeyAddress,
          publicKey: pubKey,
          derivationPath: derivationPath,
        ),
        value: e.value,
      );
    }).toList();

    return fusionTransaction;
  }
}