stack_wallet/lib/wallets/isar/providers/eth/token_balance_provider.dart
2024-01-11 11:19:23 -06:00

59 lines
1.8 KiB
Dart

import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:isar/isar.dart';
import 'package:stackwallet/models/balance.dart';
import 'package:stackwallet/models/isar/models/isar_models.dart';
import 'package:stackwallet/providers/db/main_db_provider.dart';
import 'package:stackwallet/wallets/isar/models/token_wallet_info.dart';
import 'package:stackwallet/wallets/isar/providers/util/watcher.dart';
final _twiProvider = ChangeNotifierProvider.family<Watcher,
({String walletId, String contractAddress})>(
(ref, data) {
final isar = ref.watch(mainDBProvider).isar;
final collection = isar.tokenWalletInfo;
TokenWalletInfo? initial = collection
.where()
.walletIdTokenAddressEqualTo(data.walletId, data.contractAddress)
.findFirstSync();
if (initial == null) {
initial = TokenWalletInfo(
walletId: data.walletId,
tokenAddress: data.contractAddress,
tokenFractionDigits: isar.ethContracts
.getByAddressSync(data.contractAddress)
?.decimals ??
2,
);
isar.writeTxnSync(() => isar.tokenWalletInfo.putSync(initial!));
}
final watcher = Watcher(
initial,
collection: collection,
);
ref.onDispose(() => watcher.dispose());
return watcher;
},
);
final pTokenWalletInfo = Provider.family<TokenWalletInfo,
({String walletId, String contractAddress})>(
(ref, data) {
return ref.watch(_twiProvider(data).select((value) => value.value))
as TokenWalletInfo;
},
);
final pTokenBalance =
Provider.family<Balance, ({String walletId, String contractAddress})>(
(ref, data) {
return ref.watch(_twiProvider(data).select(
(value) => (value.value as TokenWalletInfo).getCachedBalance()));
},
);