import 'dart:io'; import 'package:cw_core/node.dart'; import 'package:cw_core/pathForWallet.dart'; import 'package:cw_core/wallet_base.dart'; import 'package:cw_core/wallet_credentials.dart'; import 'package:cw_core/wallet_info.dart'; import 'package:cw_core/wallet_service.dart'; import 'package:cw_core/wallet_type.dart'; import 'package:cw_nano/nano_balance.dart'; import 'package:cw_nano/nano_client.dart'; import 'package:cw_nano/nano_mnemonic.dart'; import 'package:cw_nano/nano_util.dart'; import 'package:cw_nano/nano_wallet.dart'; import 'package:cw_nano/nano_wallet_info.dart'; import 'package:hive/hive.dart'; import 'package:bip39/bip39.dart' as bip39; class NanoNewWalletCredentials extends WalletCredentials { NanoNewWalletCredentials({required String name, String? password}) : super(name: name, password: password); } class NanoRestoreWalletFromSeedCredentials extends WalletCredentials { NanoRestoreWalletFromSeedCredentials( {required String name, required this.mnemonic, this.derivationType, int height = 0, String? password}) : super(name: name, password: password, height: height); final String mnemonic; final DerivationType? derivationType; } class NanoWalletLoadingException implements Exception { @override String toString() => 'Failure to load the wallet.'; } class NanoRestoreWalletFromKeysCredentials extends WalletCredentials { NanoRestoreWalletFromKeysCredentials({ required String name, required String password, required this.seedKey, this.derivationType, }) : super(name: name, password: password); final String seedKey; final DerivationType? derivationType; } class NanoWalletService extends WalletService { NanoWalletService(this.walletInfoSource); final Box walletInfoSource; static bool walletFilesExist(String path) => !File(path).existsSync() && !File('$path.keys').existsSync(); @override WalletType getType() => WalletType.nano; @override Future create(NanoNewWalletCredentials credentials) async { final mnemonic = bip39.generateMnemonic(); final nanoWalletInfo = NanoWalletInfo( walletInfo: credentials.walletInfo!, derivationType: DerivationType.nano, ); final wallet = NanoWallet( walletInfo: nanoWalletInfo, mnemonic: mnemonic, password: credentials.password!, ); return wallet; } @override Future remove(String wallet) async { final path = await pathForWalletDir(name: wallet, type: getType()); final file = Directory(path); final isExist = file.existsSync(); if (isExist) { await file.delete(recursive: true); } final walletInfo = walletInfoSource.values .firstWhere((info) => info.id == WalletBase.idFor(wallet, getType())); await walletInfoSource.delete(walletInfo.key); } @override Future rename(String currentName, String password, String newName) async { // final currentWalletInfo = walletInfoSource.values // .firstWhere((info) => info.id == WalletBase.idFor(currentName, getType())); // final currentWallet = NanoWallet(walletInfo: currentWalletInfo); // await currentWallet.renameWalletFiles(newName); // final newWalletInfo = currentWalletInfo; // newWalletInfo.id = WalletBase.idFor(newName, getType()); // newWalletInfo.name = newName; // await walletInfoSource.put(currentWalletInfo.key, newWalletInfo); } Future compareDerivationMethods({String? mnemonic, String? seedKey}) async { if (seedKey?.length == 128) { return DerivationType.bip39; } if (mnemonic?.split(' ').length == 12) { return DerivationType.bip39; } try { NanoClient nanoClient = NanoClient(); // TODO: figure out how to load the current node uri in this context: nanoClient.connect(Node( uri: NanoClient.BACKUP_NODE_URI, type: WalletType.nano, )); late String publicAddressStandard; late String publicAddressBip39; if (seedKey == null) { seedKey = bip39.mnemonicToEntropy(mnemonic).toUpperCase(); } publicAddressBip39 = await NanoUtil.hdSeedToAddress(seedKey, 0); publicAddressStandard = await NanoUtil.seedToAddress(seedKey, 0); // check if either has a balance: NanoBalance bip39Balance = await nanoClient.getBalance(publicAddressBip39); NanoBalance standardBalance = await nanoClient.getBalance(publicAddressStandard); // TODO: this is a super basic implementation, and if both addresses have balances // it might not be the one that the user wants, though it is unlikely if (bip39Balance.currentBalance > standardBalance.currentBalance) { return DerivationType.bip39; } else { return DerivationType.nano; } } catch (e) { return DerivationType.nano; } } @override Future restoreFromKeys(NanoRestoreWalletFromKeysCredentials credentials) async { throw UnimplementedError("restoreFromKeys"); // TODO: mnemonic can't be derived from the seedKey in the nano standard derivation // which complicates things // DerivationType derivationType = credentials.derivationType ?? await compareDerivationMethods(seedKey: credentials.seedKey); // String? mnemonic; // final nanoWalletInfo = NanoWalletInfo( // walletInfo: credentials.walletInfo!, // derivationType: derivationType, // ); // final wallet = await NanoWallet( // password: credentials.password!, // mnemonic: mnemonic ?? "", // we can't derive the mnemonic from the key in all cases // walletInfo: nanoWalletInfo, // ); // await wallet.init(); // await wallet.save(); // return wallet; } @override Future restoreFromSeed(NanoRestoreWalletFromSeedCredentials credentials) async { if (!bip39.validateMnemonic(credentials.mnemonic)) { throw NanoMnemonicIsIncorrectException(); } if (!NanoMnemomics.validateMnemonic(credentials.mnemonic.split(' '))) { throw NanoMnemonicIsIncorrectException(); } DerivationType derivationType = credentials.derivationType ?? await compareDerivationMethods(mnemonic: credentials.mnemonic); final nanoWalletInfo = NanoWalletInfo( walletInfo: credentials.walletInfo!, derivationType: derivationType, ); final wallet = await NanoWallet( password: credentials.password!, mnemonic: credentials.mnemonic, walletInfo: nanoWalletInfo, ); await wallet.init(); await wallet.save(); return wallet; } @override Future isWalletExit(String name) async => File(await pathForWallet(name: name, type: getType())).existsSync(); @override Future openWallet(String name, String password) async { final walletInfo = walletInfoSource.values.firstWhere((info) => info.id == WalletBase.idFor(name, getType())); final wallet = await NanoWalletBase.open( name: name, password: password, walletInfo: walletInfo, ); await wallet.init(); await wallet.save(); return wallet; } }