stack_wallet/lib/services/wallets.dart

221 lines
7.2 KiB
Dart
Raw Normal View History

2023-05-26 21:21:16 +00:00
/*
* This file is part of Stack Wallet.
*
* Copyright (c) 2023 Cypher Stack
* All Rights Reserved.
* The code is distributed under GPLv3 license, see LICENSE file for details.
* Generated by Cypher Stack on 2023-05-26
*
*/
2023-10-31 17:13:26 +00:00
import 'package:isar/isar.dart';
2023-03-01 21:52:13 +00:00
import 'package:stackwallet/db/hive/db.dart';
2023-10-31 17:13:26 +00:00
import 'package:stackwallet/db/isar/main_db.dart';
2022-08-26 08:11:35 +00:00
import 'package:stackwallet/services/node_service.dart';
import 'package:stackwallet/services/transaction_notification_tracker.dart';
import 'package:stackwallet/utilities/enums/coin_enum.dart';
import 'package:stackwallet/utilities/enums/sync_type_enum.dart';
import 'package:stackwallet/utilities/logger.dart';
import 'package:stackwallet/utilities/prefs.dart';
2023-10-31 17:13:26 +00:00
import 'package:stackwallet/wallets/isar_models/wallet_info.dart';
import 'package:stackwallet/wallets/wallet/wallet.dart';
2022-08-26 08:11:35 +00:00
2023-10-31 17:13:26 +00:00
class Wallets {
2022-08-26 08:11:35 +00:00
Wallets._private();
static final Wallets _sharedInstance = Wallets._private();
static Wallets get sharedInstance => _sharedInstance;
late NodeService nodeService;
2023-10-31 17:13:26 +00:00
late MainDB mainDB;
2022-08-26 08:11:35 +00:00
2023-10-31 17:13:26 +00:00
bool get hasWallets => _wallets.isNotEmpty;
2022-08-26 08:11:35 +00:00
static bool hasLoaded = false;
2023-10-31 17:13:26 +00:00
final Map<String, Wallet> _wallets = {};
2022-08-26 08:11:35 +00:00
2023-10-31 17:13:26 +00:00
Wallet getWallet(String walletId) => _wallets[walletId]!;
Future<void> load(Prefs prefs, MainDB mainDB) async {
2022-08-26 08:11:35 +00:00
if (hasLoaded) {
return;
}
hasLoaded = true;
// clear out any wallet hive boxes where the wallet was deleted in previous app run
for (final walletId in DB.instance
.values<String>(boxName: DB.boxNameWalletsToDeleteOnStart)) {
2023-10-31 17:13:26 +00:00
await mainDB.isar.writeTxn(() async => await mainDB.isar.walletInfo
.where()
.walletIdEqualTo(walletId)
.deleteAll());
2022-08-26 08:11:35 +00:00
}
// clear list
await DB.instance
.deleteAll<String>(boxName: DB.boxNameWalletsToDeleteOnStart);
2023-10-31 17:13:26 +00:00
final walletInfoList = await mainDB.isar.walletInfo.where().findAll();
if (walletInfoList.isEmpty) {
return;
}
2022-08-26 08:11:35 +00:00
2023-10-31 17:13:26 +00:00
List<Future<void>> walletInitFutures = [];
List<({Wallet wallet, bool shouldAutoSync})> walletsToInitLinearly = [];
2022-08-26 08:11:35 +00:00
List<String> walletIdsToEnableAutoSync = [];
bool shouldAutoSyncAll = false;
switch (prefs.syncType) {
case SyncingType.currentWalletOnly:
// do nothing as this will be set when going into a wallet from the main screen
break;
case SyncingType.selectedWalletsAtStartup:
walletIdsToEnableAutoSync.addAll(prefs.walletIdsSyncOnStartup);
break;
case SyncingType.allWalletsOnStartup:
shouldAutoSyncAll = true;
break;
}
2023-10-31 17:13:26 +00:00
for (final walletInfo in walletInfoList) {
2022-08-26 08:11:35 +00:00
try {
Logging.instance.log(
2023-10-31 17:13:26 +00:00
"LOADING WALLET: ${walletInfo.name}:${walletInfo.walletId} "
"IS VERIFIED: ${walletInfo.isMnemonicVerified}",
level: LogLevel.Info,
);
if (walletInfo.isMnemonicVerified) {
// TODO: integrate this into the new wallets somehow?
// requires some thinking
final txTracker =
TransactionNotificationTracker(walletId: walletInfo.walletId);
final wallet = await Wallet.load(
walletId: walletInfo.walletId,
mainDB: mainDB,
secureStorageInterface: nodeService.secureStorageInterface,
nodeService: nodeService,
prefs: prefs,
);
2022-08-26 08:11:35 +00:00
2023-10-31 17:13:26 +00:00
final shouldSetAutoSync = shouldAutoSyncAll ||
walletIdsToEnableAutoSync.contains(walletInfo.walletId);
2022-08-26 08:11:35 +00:00
2023-10-31 17:13:26 +00:00
if (walletInfo.coin == Coin.monero ||
walletInfo.coin == Coin.wownero) {
// walletsToInitLinearly.add(Tuple2(manager, shouldSetAutoSync));
} else {
walletInitFutures.add(wallet.init().then((_) {
if (shouldSetAutoSync) {
wallet.shouldAutoSync = true;
2022-08-26 08:11:35 +00:00
}
2023-10-31 17:13:26 +00:00
}));
2022-08-26 08:11:35 +00:00
}
2023-10-31 17:13:26 +00:00
_wallets[wallet.walletId] = wallet;
2022-08-26 08:11:35 +00:00
} else {
// wallet creation was not completed by user so we remove it completely
2023-10-31 17:13:26 +00:00
await _deleteWallet(walletInfo.walletId);
// await walletsService.deleteWallet(walletInfo.name, false);
2022-08-26 08:11:35 +00:00
}
} catch (e, s) {
Logging.instance.log("$e $s", level: LogLevel.Fatal);
continue;
}
}
if (walletInitFutures.isNotEmpty && walletsToInitLinearly.isNotEmpty) {
await Future.wait([
_initLinearly(walletsToInitLinearly),
...walletInitFutures,
]);
} else if (walletInitFutures.isNotEmpty) {
await Future.wait(walletInitFutures);
} else if (walletsToInitLinearly.isNotEmpty) {
await _initLinearly(walletsToInitLinearly);
}
}
Future<void> loadAfterStackRestore(
2023-10-31 17:13:26 +00:00
Prefs prefs,
List<Wallet> wallets,
) async {
List<Future<void>> walletInitFutures = [];
List<({Wallet wallet, bool shouldAutoSync})> walletsToInitLinearly = [];
2022-08-26 08:11:35 +00:00
List<String> walletIdsToEnableAutoSync = [];
bool shouldAutoSyncAll = false;
switch (prefs.syncType) {
case SyncingType.currentWalletOnly:
// do nothing as this will be set when going into a wallet from the main screen
break;
case SyncingType.selectedWalletsAtStartup:
walletIdsToEnableAutoSync.addAll(prefs.walletIdsSyncOnStartup);
break;
case SyncingType.allWalletsOnStartup:
shouldAutoSyncAll = true;
break;
}
2023-10-31 17:13:26 +00:00
for (final wallet in wallets) {
Logging.instance.log(
"LOADING WALLET: ${wallet.walletInfo.name}:${wallet.walletId} IS VERIFIED: ${wallet.walletInfo.isMnemonicVerified}",
level: LogLevel.Info,
);
2022-08-26 08:11:35 +00:00
2023-10-31 17:13:26 +00:00
if (wallet.walletInfo.isMnemonicVerified) {
final shouldSetAutoSync = shouldAutoSyncAll ||
walletIdsToEnableAutoSync.contains(wallet.walletId);
2022-08-26 08:11:35 +00:00
2023-10-31 17:13:26 +00:00
if (wallet.walletInfo.coin == Coin.monero ||
wallet.walletInfo.coin == Coin.wownero) {
// walletsToInitLinearly.add(Tuple2(manager, shouldSetAutoSync));
} else {
walletInitFutures.add(wallet.init().then((value) {
if (shouldSetAutoSync) {
wallet.shouldAutoSync = true;
2022-08-26 08:11:35 +00:00
}
2023-10-31 17:13:26 +00:00
}));
2022-08-26 08:11:35 +00:00
}
2023-10-31 17:13:26 +00:00
_wallets[wallet.walletId] = wallet;
2022-08-26 08:11:35 +00:00
} else {
// wallet creation was not completed by user so we remove it completely
2023-10-31 17:13:26 +00:00
await _deleteWallet(wallet.walletId);
// await walletsService.deleteWallet(walletInfo.name, false);
2022-08-26 08:11:35 +00:00
}
}
if (walletInitFutures.isNotEmpty && walletsToInitLinearly.isNotEmpty) {
await Future.wait([
_initLinearly(walletsToInitLinearly),
...walletInitFutures,
]);
} else if (walletInitFutures.isNotEmpty) {
await Future.wait(walletInitFutures);
} else if (walletsToInitLinearly.isNotEmpty) {
await _initLinearly(walletsToInitLinearly);
}
}
2023-10-31 17:13:26 +00:00
Future<void> _initLinearly(
List<({Wallet wallet, bool shouldAutoSync})> dataList,
) async {
for (final data in dataList) {
await data.wallet.init();
if (data.shouldAutoSync && !data.wallet.shouldAutoSync) {
data.wallet.shouldAutoSync = true;
}
}
}
Future<void> _deleteWallet(String walletId) async {
// TODO proper clean up of other wallet data in addition to the following
await mainDB.isar.writeTxn(() async => await mainDB.isar.walletInfo
.where()
.walletIdEqualTo(walletId)
.deleteAll());
}
2022-08-26 08:11:35 +00:00
}