Return a vector of stakable shield notes

src/sapling/saplingscriptpubkeyman.cpp

@@ -6,6 +6,7 @@
 #include "sapling/saplingscriptpubkeyman.h"
 
 #include "chain.h" // for CBlockIndex
+#include "saplingscriptpubkeyman.h"
 #include "validation.h" // for ReadBlockFromDisk()
 
 void SaplingScriptPubKeyMan::AddToSaplingSpends(const uint256& nullifier, const uint256& wtxid)
@@ -540,6 +541,49 @@ void SaplingScriptPubKeyMan::GetFilteredNotes(
  }
  }
 }
+void SaplingScriptPubKeyMan::GetStakableNotes(std::vector<CShieldStake>& shieldInputs, int minDepth)
+{
+ LOCK(wallet->cs_wallet);
+ for (auto& p : wallet->mapWallet) {
+ const CWalletTx& wtx = p.second;
+ // Filter coinbase/coinstakes transactions that don't have Sapling outputs
+ if ((wtx.IsCoinBase() || wtx.IsCoinStake()) && wtx.mapSaplingNoteData.empty()) {
+ continue;
+ }
+ // Filter the transactions before checking for notes
+ const int depth = wtx.GetDepthInMainChain();
+ if (!IsFinalTx(wtx.tx, wallet->GetLastBlockHeight() + 1, GetAdjustedTime()) ||
+ depth < minDepth) {
+ continue;
+ }
+ for (const auto& it : wtx.mapSaplingNoteData) {
+ const SaplingOutPoint& op = it.first;
+ const SaplingNoteData& nd = it.second;
+
+ // skip sent notes
+ if (!nd.IsMyNote()) continue;
+
+ // recover plaintext and address
+ auto optNotePtAndAddress = wtx.DecryptSaplingNote(op);
+ assert(static_cast<bool>(optNotePtAndAddress));
+
+ const libzcash::SaplingIncomingViewingKey& ivk = *(nd.ivk);
+ const libzcash::SaplingNotePlaintext& notePt = optNotePtAndAddress->first;
+ const libzcash::SaplingPaymentAddress& pa = optNotePtAndAddress->second;
+ auto note = notePt.note(ivk).get();
+
+ // skip notes which cannot be spent
+ if (!HaveSpendingKeyForPaymentAddress(pa)) {
+ continue;
+ }
+
+ if (nd.nullifier && IsSaplingSpent(*nd.nullifier)) {
+ continue;
+ }
+ shieldInputs.emplace_back(CShieldStake(note.value(), *nd.nullifier));
+ }
+ }
+}
 
 /* Return list of available notes grouped by sapling address. */
 std::map<libzcash::SaplingPaymentAddress, std::vector<SaplingNoteEntry>> SaplingScriptPubKeyMan::ListNotes() const

src/sapling/saplingscriptpubkeyman.h

@@ -294,6 +294,9 @@ class SaplingScriptPubKeyMan {
  bool requireSpendingKey=true,
  bool ignoreLocked=true) const;
 
+ /* Return a list of notes that are stakable */
+ void GetStakableNotes(std::vector<CShieldStake>& shieldInputs, int minDepth);
+
  /* Return list of available notes grouped by sapling address. */
  std::map<libzcash::SaplingPaymentAddress, std::vector<SaplingNoteEntry>> ListNotes() const;