import 'dart:isolate'; import 'package:cw_core/wallet_info.dart' as xmr; import 'package:hive/hive.dart'; import 'package:mutex/mutex.dart'; import 'package:stackwallet/models/exchange/change_now/exchange_transaction.dart'; import 'package:stackwallet/models/exchange/response_objects/trade.dart'; import 'package:stackwallet/models/node_model.dart'; import 'package:stackwallet/models/notification_model.dart'; import 'package:stackwallet/models/trade_wallet_lookup.dart'; import 'package:stackwallet/services/wallets_service.dart'; import 'package:stackwallet/utilities/enums/coin_enum.dart'; import 'package:stackwallet/utilities/logger.dart'; class DB { static const String boxNameAddressBook = "addressBook"; static const String boxNameDebugInfo = "debugInfoBox"; static const String boxNameNodeModels = "nodeModels"; static const String boxNamePrimaryNodes = "primaryNodes"; static const String boxNameAllWalletsData = "wallets"; static const String boxNameNotifications = "notificationModels"; static const String boxNameWatchedTransactions = "watchedTxNotificationModels"; static const String boxNameWatchedTrades = "watchedTradesNotificationModels"; static const String boxNameTrades = "exchangeTransactionsBox"; static const String boxNameTradesV2 = "exchangeTradesBox"; static const String boxNameTradeNotes = "tradeNotesBox"; static const String boxNameTradeLookup = "tradeToTxidLookUpBox"; static const String boxNameFavoriteWallets = "favoriteWallets"; static const String boxNamePrefs = "prefs"; static const String boxNameWalletsToDeleteOnStart = "walletsToDeleteOnStart"; static const String boxNamePriceCache = "priceAPIPrice24hCache"; static const String boxNameDBInfo = "dbInfo"; static const String boxNameTheme = "theme"; String boxNameTxCache({required Coin coin}) => "${coin.name}_txCache"; String boxNameSetCache({required Coin coin}) => "${coin.name}_anonymitySetCache"; String boxNameUsedSerialsCache({required Coin coin}) => "${coin.name}_usedSerialsCache"; static bool _initialized = false; late final Box _boxAddressBook; late final Box _boxDebugInfo; late final Box _boxNodeModels; late final Box _boxPrimaryNodes; late final Box _boxAllWalletsData; late final Box _boxNotifications; late final Box _boxWatchedTransactions; late final Box _boxWatchedTrades; late final Box _boxTrades; late final Box _boxTradesV2; late final Box _boxTradeNotes; late final Box _boxFavoriteWallets; late final Box _walletInfoSource; late final Box _boxPrefs; late final Box _boxTradeLookup; late final Box _boxDBInfo; final Map> _walletBoxes = {}; final Map> _txCacheBoxes = {}; final Map> _setCacheBoxes = {}; final Map> _usedSerialsCacheBoxes = {}; // exposed for monero Box get moneroWalletInfoBox => _walletInfoSource; // mutex for stack backup final mutex = Mutex(); DB._(); static final DB _instance = DB._(); static DB get instance { // "This name does not uniquely identify an isolate. Multiple isolates in the same process may have the same debugName." // if (Isolate.current.debugName != "main") { // TODO: make sure this works properly if (Isolate.current.debugName != "main") { throw Exception( "DB.instance should not be accessed outside the main isolate!"); } return _instance; } // open hive boxes Future init() async { if (!_initialized) { if (Hive.isBoxOpen(boxNameDBInfo)) { _boxDBInfo = Hive.box(boxNameDBInfo); } else { _boxDBInfo = await Hive.openBox(boxNameDBInfo); } await Hive.openBox(boxNameWalletsToDeleteOnStart); if (Hive.isBoxOpen(boxNamePrefs)) { _boxPrefs = Hive.box(boxNamePrefs); } else { _boxPrefs = await Hive.openBox(boxNamePrefs); } _boxAddressBook = await Hive.openBox(boxNameAddressBook); _boxDebugInfo = await Hive.openBox(boxNameDebugInfo); if (Hive.isBoxOpen(boxNameNodeModels)) { _boxNodeModels = Hive.box(boxNameNodeModels); } else { _boxNodeModels = await Hive.openBox(boxNameNodeModels); } if (Hive.isBoxOpen(boxNamePrimaryNodes)) { _boxPrimaryNodes = Hive.box(boxNamePrimaryNodes); } else { _boxPrimaryNodes = await Hive.openBox(boxNamePrimaryNodes); } if (Hive.isBoxOpen(boxNameAllWalletsData)) { _boxAllWalletsData = Hive.box(boxNameAllWalletsData); } else { _boxAllWalletsData = await Hive.openBox(boxNameAllWalletsData); } _boxNotifications = await Hive.openBox(boxNameNotifications); _boxWatchedTransactions = await Hive.openBox(boxNameWatchedTransactions); _boxWatchedTrades = await Hive.openBox(boxNameWatchedTrades); _boxTrades = await Hive.openBox(boxNameTrades); _boxTradesV2 = await Hive.openBox(boxNameTradesV2); _boxTradeNotes = await Hive.openBox(boxNameTradeNotes); _boxTradeLookup = await Hive.openBox(boxNameTradeLookup); _walletInfoSource = await Hive.openBox(xmr.WalletInfo.boxName); _boxFavoriteWallets = await Hive.openBox(boxNameFavoriteWallets); await Future.wait([ Hive.openBox(boxNamePriceCache), _loadWalletBoxes(), _loadSharedCoinCacheBoxes(), ]); _initialized = true; } } Future _loadWalletBoxes() async { final names = _boxAllWalletsData.get("names") as Map? ?? {}; names.removeWhere((name, dyn) { final jsonObject = Map.from(dyn as Map); try { Coin.values.byName(jsonObject["coin"] as String); return false; } catch (e, s) { Logging.instance.log( "Error, ${jsonObject["coin"]} does not exist, $name wallet cannot be loaded", level: LogLevel.Error); return true; } }); final mapped = Map.from(names).map((name, dyn) => MapEntry( name, WalletInfo.fromJson(Map.from(dyn as Map)))); for (final entry in mapped.entries) { if (Hive.isBoxOpen(entry.value.walletId)) { _walletBoxes[entry.value.walletId] = Hive.box(entry.value.walletId); } else { _walletBoxes[entry.value.walletId] = await Hive.openBox(entry.value.walletId); } } } Future _loadSharedCoinCacheBoxes() async { for (final coin in Coin.values) { _txCacheBoxes[coin] = await Hive.openBox(boxNameTxCache(coin: coin)); _setCacheBoxes[coin] = await Hive.openBox(boxNameSetCache(coin: coin)); _usedSerialsCacheBoxes[coin] = await Hive.openBox(boxNameUsedSerialsCache(coin: coin)); } } ///////////////////////////////////////// Future addWalletBox({required String walletId}) async { if (_walletBoxes[walletId] != null) { throw Exception("Attempted overwrite of existing wallet box!"); } _walletBoxes[walletId] = await Hive.openBox(walletId); } Future removeWalletBox({required String walletId}) async { _walletBoxes.remove(walletId); } /////////////////////////////////////////// // reads List keys({required String boxName}) => Hive.box(boxName).keys.toList(growable: false); List values({required String boxName}) => Hive.box(boxName).values.toList(growable: false); T? get({ required String boxName, required dynamic key, }) => Hive.box(boxName).get(key); bool containsKey({required String boxName, required dynamic key}) => Hive.box(boxName).containsKey(key); // writes Future put( {required String boxName, required dynamic key, required T value}) async => await mutex .protect(() async => await Hive.box(boxName).put(key, value)); Future add({required String boxName, required T value}) async => await mutex.protect(() async => await Hive.box(boxName).add(value)); Future addAll( {required String boxName, required Iterable values}) async => await mutex .protect(() async => await Hive.box(boxName).addAll(values)); Future delete( {required dynamic key, required String boxName}) async => await mutex.protect(() async => await Hive.box(boxName).delete(key)); Future deleteAll({required String boxName}) async => await mutex.protect(() async => await Hive.box(boxName).clear()); Future deleteBoxFromDisk({required String boxName}) async => await mutex.protect(() async => await Hive.deleteBoxFromDisk(boxName)); }