mirrors/btcpayserver
1
0
Fork 0
mirror of https://github.com/btcpayserver/btcpayserver.git synced 2025-03-03 17:36:59 +01:00
Code Issues Projects Releases Packages Wiki Activity

Show total balances for each currency on List Wallets Page

Browse source
This commit is contained in:
Max Dignan 2021-09-15 10:52:57 -04:00 • committed by Andrew Camilleri
parent 88c925017d
commit b7a081b9a4
3 changed files with 33 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
BTCPayServer/Controllers/WalletsController.cs
View file

@ -259,10 +259,20 @@ namespace BTCPayServer.Controllers
{
walletVm.Balance = "";
}
walletVm.Network = wallet.Network;
walletVm.CryptoCode = wallet.Network.CryptoCode;
walletVm.StoreId = wallet.Store.Id;
walletVm.Id = new WalletId(wallet.Store.Id, wallet.Network.CryptoCode);
walletVm.StoreName = wallet.Store.StoreName;
if (wallets.BalanceForCryptoCode.ContainsKey(wallet.Network))
{
wallets.BalanceForCryptoCode[wallet.Network] = wallets.BalanceForCryptoCode[wallet.Network].Add(await GetBalanceAsMoney(wallet.Wallet, wallet.DerivationStrategy));
}
else
{
wallets.BalanceForCryptoCode[wallet.Network] = await GetBalanceAsMoney(wallet.Wallet, wallet.DerivationStrategy);
}
}
return View(wallets);
@ -1061,13 +1071,25 @@ namespace BTCPayServer.Controllers
return CurrentStore.GetDerivationSchemeSettings(NetworkProvider, walletId.CryptoCode);
}
private static async Task<string> GetBalanceString(BTCPayWallet wallet, DerivationStrategyBase derivationStrategy)
private static async Task<IMoney> GetBalanceAsMoney(BTCPayWallet wallet, DerivationStrategyBase derivationStrategy)
{
using CancellationTokenSource cts = new CancellationTokenSource(TimeSpan.FromSeconds(10));
try
{
var b = await wallet.GetBalance(derivationStrategy, cts.Token);
return (b.Available ?? b.Total).ShowMoney(wallet.Network);
return (b.Available ?? b.Total);
}
catch
{
return NBitcoin.Money.Zero;
}
}
private static async Task<string> GetBalanceString(BTCPayWallet wallet, DerivationStrategyBase derivationStrategy)
{
try
{
return (await GetBalanceAsMoney(wallet, derivationStrategy)).ShowMoney(wallet.Network);
}
catch
{

4
BTCPayServer/Models/WalletViewModels/ListWalletsViewModel.cs
View file

@ -1,4 +1,6 @@
using System.Collections.Generic;
using NBitcoin;
using BTCPayServer;
namespace BTCPayServer.Models.WalletViewModels
{
@ -12,8 +14,10 @@ namespace BTCPayServer.Models.WalletViewModels
public string Balance { get; set; }
public bool IsOwner { get; set; }
public WalletId Id { get; set; }
public BTCPayNetwork Network { get; set; }
}
public Dictionary<BTCPayNetwork, IMoney> BalanceForCryptoCode { get; set; } = new Dictionary<BTCPayNetwork, IMoney>();
public List<WalletViewModel> Wallets { get; set; } = new List<WalletViewModel>();
}
}

5
BTCPayServer/Views/Wallets/ListWallets.cshtml
View file

@ -16,6 +16,11 @@
<span class="fa fa-question-circle-o text-secondary" title="More information..."></span>
</a>
</small>
Total Balance:
@foreach(var balanceForACrypto in @Model.BalanceForCryptoCode)
{
<div>@(balanceForACrypto.Value.ShowMoney(balanceForACrypto.Key) + " - " + balanceForACrypto.Key.CryptoCode)</div>
}
</h2>
</div>
</div>