Add GetBalance to CoinView

src/Stratis.Bitcoin.Features.BlockStore/AddressIndexing/AddressIndexerCV.cs

@@ -0,0 +1,103 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using NBitcoin;
+using Stratis.Bitcoin.Builder.Feature;
+using Stratis.Bitcoin.Controllers.Models;
+using Stratis.Bitcoin.Features.BlockStore.Models;
+using Stratis.Bitcoin.Features.Consensus.CoinViews;
+using Stratis.Bitcoin.Interfaces;
+
+namespace Stratis.Bitcoin.Features.BlockStore.AddressIndexing
+{
+    public class AddressIndexerCV : IAddressIndexer
+    {
+        private readonly Network network;
+        private readonly ICoinView coinView;
+        private readonly ChainIndexer chainIndexer;
+        private readonly IScriptAddressReader scriptAddressReader;
+
+        public ChainedHeader IndexerTip => GetTip();
+
+        public IFullNodeFeature InitializingFeature { set; private get; }
+
+        public AddressIndexerCV(Network network, ChainIndexer chainIndexer, IScriptAddressReader scriptAddressReader, ICoinView coinView)
+        {
+            this.network = network;
+            this.coinView = coinView;
+            this.chainIndexer = chainIndexer;
+            this.scriptAddressReader = scriptAddressReader;
+        }
+
+        private ChainedHeader GetTip()
+        {
+            this.coinView.Sync(this.chainIndexer);
+
+            return this.chainIndexer[this.coinView.GetTipHash().Hash];
+        }
+
+        public void Initialize()
+        {
+        }
+
+        private TxDestination AddressToDestination(string address)
+        {
+            var bitcoinAddress = BitcoinAddress.Create(address, this.network);
+            return this.scriptAddressReader.GetDestinationFromScriptPubKey(this.network, bitcoinAddress.ScriptPubKey).Single();
+        }
+
+        public AddressBalancesResult GetAddressBalances(string[] addresses, int minConfirmations = 0)
+        {
+            return new AddressBalancesResult()
+            {
+                Balances = addresses.Select(address => new AddressBalanceResult()
+                {
+                    Address = address,
+                    Balance = new Money(this.coinView.GetBalance(AddressToDestination(address)).First().satoshis)
+                }).ToList()
+            };
+        }
+
+        public LastBalanceDecreaseTransactionModel GetLastBalanceDecreaseTransaction(string address)
+        {
+            throw new NotImplementedException();
+        }
+
+        private IEnumerable<AddressBalanceChange> ToDiff(List<AddressBalanceChange> addressBalanceChanges)
+        {
+            for (int i = addressBalanceChanges.Count - 1; i > 0; i--)
+            {
+                yield return new AddressBalanceChange() { 
+                    BalanceChangedHeight = addressBalanceChanges[i].BalanceChangedHeight,
+                    Deposited = addressBalanceChanges[i].Satoshi < addressBalanceChanges[i - 1].Satoshi,
+                    Satoshi = Math.Abs(addressBalanceChanges[i].Satoshi - addressBalanceChanges[i - 1].Satoshi)
+                };
+            }
+        }
+
+        /// <inheritdoc/>
+        public VerboseAddressBalancesResult GetAddressIndexerState(string[] addresses)
+        {
+            // If the containing feature is not initialized then wait a bit.
+            this.InitializingFeature?.WaitInitialized();
+
+            return new VerboseAddressBalancesResult(this.IndexerTip.Height)
+            {
+                BalancesData = addresses.Select(address => new AddressIndexerData()
+                {
+                    Address = address,                    
+                    BalanceChanges = ToDiff(this.coinView.GetBalance(AddressToDestination(address)).Select(b => new AddressBalanceChange()
+                    {
+                         BalanceChangedHeight = (int)b.height,
+                         Deposited = b.satoshis >= 0,
+                         Satoshi = Math.Abs(b.satoshis)
+                    }).ToList()).ToList()
+                }).ToList()
+            };
+        }
+
+        public void Dispose()
+        {
+        }
+    }
+}

src/Stratis.Bitcoin.Features.BlockStore/BlockStoreFeature.cs

@@ -199,7 +199,7 @@ public static IFullNodeBuilder UseBlockStore(this IFullNodeBuilder fullNodeBuild
 
                         services.AddSingleton<StoreSettings>();
                         services.AddSingleton<IBlockStoreQueueFlushCondition, BlockStoreQueueFlushCondition>();
-                        services.AddSingleton<IAddressIndexer, AddressIndexer>();
+                        services.AddSingleton<IAddressIndexer, AddressIndexerCV>();
                         services.AddSingleton<IUtxoIndexer, UtxoIndexer>();
                     });
             });

src/Stratis.Bitcoin.Features.ColdStaking/ColdStakingDestinationReader.cs

@@ -2,6 +2,7 @@
 using NBitcoin;
 using Stratis.Bitcoin.Consensus;
 using Stratis.Bitcoin.Features.Wallet.Interfaces;
+using Stratis.Bitcoin.Interfaces;
 
 namespace Stratis.Bitcoin.Features.ColdStaking
 {

src/Stratis.Bitcoin.Features.Consensus.Tests/CoinViews/CoinviewTests.cs

@@ -43,7 +43,7 @@ public CoinviewTests()
             this.nodeStats = new NodeStats(this.dateTimeProvider, NodeSettings.Default(this.network), new Mock<IVersionProvider>().Object);
 
             this.coindb = new DBreezeCoindb(this.network, this.dataFolder, this.dateTimeProvider, this.loggerFactory, this.nodeStats, new DBreezeSerializer(this.network.Consensus.ConsensusFactory));
-            this.coindb.Initialize(new ChainedHeader(this.network.GetGenesis().Header, this.network.GenesisHash, 0));
+            this.coindb.Initialize(false);
 
             this.chainIndexer = new ChainIndexer(this.network);
             this.stakeChainStore = new StakeChainStore(this.network, this.chainIndexer, (IStakedb)this.coindb, this.loggerFactory);