| | | 1 | | using Microsoft.EntityFrameworkCore; |
| | | 2 | | |
| | | 3 | | namespace NLightning.Infrastructure.Repositories.Database.Bitcoin; |
| | | 4 | | |
| | | 5 | | using Domain.Bitcoin.Enums; |
| | | 6 | | using Domain.Bitcoin.Interfaces; |
| | | 7 | | using Domain.Bitcoin.Wallet.Models; |
| | | 8 | | using Persistence.Contexts; |
| | | 9 | | using Persistence.Entities.Bitcoin; |
| | | 10 | | |
| | | 11 | | public class WalletAddressesDbRepository(NLightningDbContext context) |
| | 0 | 12 | | : BaseDbRepository<WalletAddressEntity>(context), IWalletAddressesDbRepository |
| | | 13 | | { |
| | | 14 | | public async Task<WalletAddressModel?> GetUnusedAddressAsync(AddressType type, bool isChange) |
| | 0 | 15 | | { |
| | 0 | 16 | | var walletAddressEntity = await DbSet.AsNoTracking() |
| | 0 | 17 | | .Include(x => x.Utxos) |
| | 0 | 18 | | .Where(x => x.AddressType.Equals(type) |
| | 0 | 19 | | && x.IsChange.Equals(isChange)) |
| | 0 | 20 | | .Where(x => x.Utxos != null |
| | 0 | 21 | | && x.Utxos.Count().Equals(0)) |
| | 0 | 22 | | .OrderBy(x => x.Index) |
| | 0 | 23 | | .FirstOrDefaultAsync(); |
| | | 24 | | |
| | 0 | 25 | | return walletAddressEntity is null ? null : MapEntityToModel(walletAddressEntity); |
| | 0 | 26 | | } |
| | | 27 | | |
| | | 28 | | public async Task<uint> GetLastUsedAddressIndex(AddressType addressType, bool isChange) |
| | 0 | 29 | | { |
| | 0 | 30 | | var walletAddressEntity = await DbSet.AsNoTracking() |
| | 0 | 31 | | .Where(x => x.AddressType.Equals(addressType) |
| | 0 | 32 | | && x.IsChange.Equals(isChange)) |
| | 0 | 33 | | .OrderByDescending(x => x.Index) |
| | 0 | 34 | | .FirstOrDefaultAsync(); |
| | | 35 | | |
| | 0 | 36 | | return walletAddressEntity?.Index ?? 0; |
| | 0 | 37 | | } |
| | | 38 | | |
| | | 39 | | public void AddRange(List<WalletAddressModel> addresses) |
| | 0 | 40 | | { |
| | 0 | 41 | | var walletAddressEntities = addresses.Select(MapDomainToEntity); |
| | 0 | 42 | | DbSet.AddRange(walletAddressEntities); |
| | 0 | 43 | | } |
| | | 44 | | |
| | | 45 | | public void UpdateAsync(WalletAddressModel address) |
| | 0 | 46 | | { |
| | 0 | 47 | | var walletAddressEntity = MapDomainToEntity(address); |
| | 0 | 48 | | Update(walletAddressEntity); |
| | 0 | 49 | | } |
| | | 50 | | |
| | | 51 | | public IEnumerable<WalletAddressModel> GetAllAddresses() |
| | 0 | 52 | | { |
| | 0 | 53 | | return DbSet.AsNoTracking().AsEnumerable().Select(MapEntityToModel); |
| | 0 | 54 | | } |
| | | 55 | | |
| | | 56 | | private static WalletAddressEntity MapDomainToEntity(WalletAddressModel model) |
| | 0 | 57 | | { |
| | 0 | 58 | | return new WalletAddressEntity |
| | 0 | 59 | | { |
| | 0 | 60 | | Index = model.Index, |
| | 0 | 61 | | IsChange = model.IsChange, |
| | 0 | 62 | | AddressType = model.AddressType, |
| | 0 | 63 | | Address = model.Address |
| | 0 | 64 | | }; |
| | 0 | 65 | | } |
| | | 66 | | |
| | | 67 | | internal static WalletAddressModel MapEntityToModel(WalletAddressEntity entity) |
| | 0 | 68 | | { |
| | 0 | 69 | | return new WalletAddressModel(entity.AddressType, entity.Index, entity.IsChange, entity.Address); |
| | 0 | 70 | | } |
| | | 71 | | } |